/*
	Add On:	PMPro Sponsored Members
	Desc:	Update the associated discount code's expiration date to match the expiration date of the parent level.
*/

function pmprosm_pmpro_after_checkout_update_code_expiration($user_id, $order)
{
	global $pmpro_level, $wpdb;

	//get level
	if(!empty($pmpro_level))
		$level_id = $pmpro_level->id;
	elseif(!empty($_REQUEST['level']))
		$level_id = intval($_REQUEST['level']);
	else
		$level_id = false;

	if(empty($level_id))
		return;

    if (pmprosm_isMainLevel($level_id))
    {
    		$code_id = pmprosm_getCodeByUserID($user_id);

		$level = pmpro_getMembershipLevelForUser($user_id);

		if(!empty($level->enddate))
		{
			$wpdb->update(
				$wpdb->pmpro_discount_codes,
				array(
					'id'=>max($code_id, 0),
					'expires' => date("Y-m-d", $level->enddate),
				),
				array('id' => $code_id)
			);
		}
    }

}
add_action("pmpro_after_checkout", "pmprosm_pmpro_after_checkout_update_code_expiration", 15, 2);