Allows customers to select membership duration at checkout. Adjusts charge and expiration date accordingly.

pmpro-select-membership-duration.php

/*
	Allows customers to select membership duration at checkout. Adjusts charge and expiration date accordingly.
*/
function my_pmprorh_init()
{
	//don't break if Register Helper is not loaded
	if(!function_exists( 'pmprorh_add_registration_field' )) {
		return false;
	}
	
	//define the fields
	$fields = array();

	$fields[] = new PMProRH_Field(
		'membership_duration',
		'select',					
		array(
			'label'		=> 'Membership Duration',
			'options'	 => array(1 => '1 Year', 2 => '2 Years', 3 => '3 Years')
		)
	);

	//add the fields into a new checkout_boxes are of the checkout page
	foreach($fields as $field)
		pmprorh_add_registration_field(
			'checkout_boxes',				// location on checkout page
			$field						// PMProRH_Field object
		);
	//that's it. see the PMPro Register Helper readme for more information and examples.
}
add_action( 'init', 'my_pmprorh_init' );

function my_pmpro_checkout_level($level)
{	 
	 $duration = $_REQUEST['membership_duration'];
	 
	 
	 if($duration === '1 Year')
		  $duration = 1;
	 if($duration === '2 Years')
		  $duration = 2;
	 if($duration === '3 Years')		 
		  $duration = 3;
	 
	 
	 if(is_int($duration))
	 {
		  $level->expiration_number = $duration;
		  $level->expiration_period = 'Year';
		  $level->initial_payment = $duration*$level->initial_payment;
	 }
	 
	 return $level;
}
add_filter("pmpro_checkout_level", "my_pmpro_checkout_level");