pmpro-sponsored-members.php

<?php
/*
Plugin Name: Paid Memberships Pro - Sponsored Members Add On
Plugin URI: https://www.paidmembershipspro.com/add-ons/pmpro-sponsored-members/
Description: Generate discount code for a main account holder to distribute to sponsored members.
Version: 0.10.2
Author: Paid Memberships Pro
Author URI: https://www.paidmembershipspro.com
Text Domain: pmpro-sponsored-members
Domain Path: /languages
*/

/*
	Set these to the ids of your main and sponsored levels.

	Now using a global array so you can have multiple main and sponsored levels.
	Array keys should be the main account level.

	global $pmprosm_sponsored_account_levels;
	$pmprosm_sponsored_account_levels = array(
		//set 5 seats at checkout
		1 => array(
			'main_level_id' => 1,		//redundant but useful
			'sponsored_level_id' => array(2,3),	//array or single id
			'seats' => 5
		),
		//seats based on field at checkout
		3 => array(
			'main_level_id' => 3,		//redundant but useful
			'sponsored_level_id' => 4,
			'seat_cost' => 250,
			'max_seats' => 10
		)
	);
*/

/*
	Set $pmprosm_sponsored_account_levels above here or in a custom plugin.
*/

//old constant values for reference. not used anymore
//define('PMPROSM_MAIN_ACCOUNT_LEVEL', 1);
//define('PMPROSM_SPONSORED_ACCOUNT_LEVEL', 2);
//define('PMPROSM_NUM_SEATS', 5);
//define('PMPROSM_SEAT_COST', 250);
//define('PMPROSM_MAX_SEATS', 10);

/**
 * Localize text for translations.
 * @since 0.8
 */
function pmprosm_load_textdomain(){
	load_plugin_textdomain( 'pmpro-sponsored-members', false, basename( dirname( __FILE__ ) ) . '/languages' );
}
add_action( 'init', 'pmprosm_load_textdomain' );

/**
 * Mark the plugin as MMPU-incompatible.
 */
function pmprosm_mmpu_incompatible_add_ons( $incompatible ) {
    $incompatible[] = 'PMPro Sponsored Members Add On';
    return $incompatible;
}
add_filter( 'pmpro_mmpu_incompatible_add_ons', 'pmprosm_mmpu_incompatible_add_ons' );

// Includes.
if( is_admin() ) {
	require_once(dirname(__FILE__) . '/includes/import-users-from-csv.php');
}

/**
 * Check if a level id is a "main account" level.
 */
function pmprosm_isMainLevel( $level_id ) {
	global $pmprosm_sponsored_account_levels;

	if( empty( $pmprosm_sponsored_account_levels ) ) {
		return false;
	}

	foreach( $pmprosm_sponsored_account_levels as $key => $values ) {
		if( $level_id == $key ) {
			return true;
		}
	}

	return false;
}

/**
 * Check if a level id is a "sponsored level".
 */
function pmprosm_isSponsoredLevel( $level_id ) {
	global $pmprosm_sponsored_account_levels;

	if( empty( $pmprosm_sponsored_account_levels ) ) {
		return false;
	}

	foreach( $pmprosm_sponsored_account_levels as $key => $values ) {
		if(is_array($values['sponsored_level_id']))
		{
			if(in_array($level_id, $values['sponsored_level_id']))
				return true;
		}
		else
		{
			if($values['sponsored_level_id'] == $level_id)
				return true;
		}
	}

	return false;
}

/**
 * Get values by main account level.
 */
function pmprosm_getValuesByMainLevel( $level_id ) {
	global $pmprosm_sponsored_account_levels;
	if( isset( $pmprosm_sponsored_account_levels[$level_id] ) ) {
		return $pmprosm_sponsored_account_levels[$level_id];
	} else {
		return false;
	}
}

/**
 * Get values by sponsored account level.
 */
function pmprosm_getValuesBySponsoredLevel( $level_id, $first = true ) {
	global $pmprosm_sponsored_account_levels;

	$pmprosm_sponsored_account_values = array();

	foreach( $pmprosm_sponsored_account_levels as $key => $values ) {
		if( is_array( $values['sponsored_level_id'] ) ) {
			if( in_array($level_id, $values['sponsored_level_id']) && $first ) {
				return $pmprosm_sponsored_account_levels[$key];
			} else {
				$pmprosm_sponsored_account_values[] = $pmprosm_sponsored_account_levels[$key];
			}
		} else {
			if( $values['sponsored_level_id'] == $level_id && $first ) {
				return $pmprosm_sponsored_account_levels[$key];
			} else {
				$pmprosm_sponsored_account_values[] = $pmprosm_sponsored_account_levels[$key];
			}
		}
	}

	return $pmprosm_sponsored_account_values;
}

/**
 * Code to run when a membership level is changed.
 * Cancel sub members when a main account cancels.
 * Activate sub members when changed to main account.
 * Generate a discount code when changing to main account level.
 */
function pmprosm_pmpro_after_change_membership_level( $level_id, $user_id ) {
	global $wpdb;

	if ( pmprosm_isMainLevel( $level_id ) ) { // Changing to a sponsoring level.
		//get values for this sponsorship
		$pmprosm_values = pmprosm_getValuesByMainLevel( $level_id );

		//check if this user already has a discount code
		$code_id = pmprosm_getCodeByUserID( $user_id );

		//make sure the code is still around
		if( $code_id ) {
			$code_exists = $wpdb->get_var("SELECT id FROM $wpdb->pmpro_discount_codes WHERE id = '" . esc_sql( $code_id ) . "' LIMIT 1");
			if( ! $code_exists ) {
				$code_id = false;
			}
		}

		//no code, make one
		if( empty( $code_id ) ) {
			//if seats cost money and there are no seats, just return
			if( ! empty( $pmprosm_values['seat_cost'] ) && empty( $_REQUEST['seats'] ) ) {
				return;
			}

			//check for seats
			if(isset($_REQUEST['seats']))
				$uses = intval($_REQUEST['seats']);
			elseif(isset($_SESSION['seats']))
				$uses = intval($_SESSION['seats']);
			elseif(!empty($pmprosm_values['seats']))
				$uses = $pmprosm_values['seats'];
			elseif(!empty($pmprosm_values['min_seats']))
				$uses = $pmprosm_values['min_seats'];
			else
				$uses = "";

			//create a new code
			pmprosm_createSponsorCode( $user_id, $level_id, $uses );

			//make sure seats is correct in user meta
			update_user_meta( $user_id, 'pmprosm_seats', $uses );
		} elseif( ! empty( $pmprosm_values['sponsored_level_id'] ) ) {
			//update sponsor code and sub accounts
			pmprosm_sponsored_account_change( $level_id, $user_id );

			//make sure we only do it once
			remove_action( 'pmpro_after_checkout', 'pmprosm_pmpro_after_checkout_sponsor_account_change', 10, 2 );
		}
	} else { // They are cancelling or changing to a non-sponsor level.
		//is there a discount code attached to this user?
		$code_id = pmprosm_getCodeByUserID( $user_id );

		//if so find all users who signed up with that and cancel them as well
		if( ! empty( $code_id ) ) {
			$sqlQuery = "SELECT user_id FROM $wpdb->pmpro_discount_codes_uses WHERE code_id = '" . esc_sql( $code_id ) . "'";
			$sub_user_ids = $wpdb->get_col($sqlQuery);

			if( ! empty( $sub_user_ids ) ) {
				foreach( $sub_user_ids as $sub_user_id ) {
					//cancel their membership
					pmpro_changeMembershipLevel( 0, $sub_user_id );
				}
			}
		}

		//remove seats from meta
		update_user_meta( $user_id, 'pmprosm_seats', '' );
	} 
}
add_action( 'pmpro_after_change_membership_level', 'pmprosm_pmpro_after_change_membership_level', 10, 2 );

/**
 * Create a new sponsor discount code.
 */
function pmprosm_createSponsorCode( $user_id, $level_id, $uses = '' ) {
	global $wpdb;

	// Get values for this sponsorship.
	$pmprosm_values = pmprosm_getValuesByMainLevel( $level_id );

	// Generate a new code. change these values if you want.
	if( version_compare(PMPRO_VERSION, '1.7.5' ) > 0 ) {
		//seed parameter added in version 1.7.6
		$code = "S" . pmpro_getDiscountCode( $user_id );
	} else {
		$code = "S" . pmpro_getDiscountCode();
	}
	
	$starts = date( 'Y-m-d', current_time( 'timestamp' ) );
	$expires = date( 'Y-m-d', strtotime( '+1 year', current_time( 'timestamp' ) ) );

	$sponsored_code_settings = apply_filters( 'pmprosm_sponsored_code_settings', array( 'code' => $code, 'starts' => $starts, 'expires' => $expires, 'uses' => $uses ) );

	$sponsored_code = $sponsored_code_settings['code'];
	$code_starts = $sponsored_code_settings['starts'];
	$code_expires = $sponsored_code_settings['expires'];
	$code_uses = ! empty( $sponsored_code_settings['uses'] ) ? intval( $sponsored_code_settings['uses'] ) : '';

	$sqlQuery = "INSERT INTO $wpdb->pmpro_discount_codes (code, starts, expires, uses) VALUES('" . esc_sql( $sponsored_code ) . "', '" . esc_sql( $code_starts ) . "', '" . esc_sql( $code_expires ) . "', '$code_uses')";

	if( $wpdb->query( $sqlQuery ) !== false ) {
		//set code in user meta
		$code_id = $wpdb->insert_id;
		pmprosm_setCodeUserID( $code_id, $user_id );

		//okay update levels for code
		if( ! is_array( $pmprosm_values['sponsored_level_id'] ) ) {
			$sponsored_levels = array($pmprosm_values['sponsored_level_id']);
		} else {
			$sponsored_levels = $pmprosm_values['sponsored_level_id'];
		}

		foreach( $sponsored_levels as $sponsored_level ) {
			//default values for discount code; everything free
			$discount_code = array(
				'code_id'=>esc_sql($code_id),
				'level_id'=>esc_sql($sponsored_level),
				'initial_payment'=>'0',
				'billing_amount'=>'0',
				'cycle_number'=>'0',
				'cycle_period'=>"'Month'",
				'billing_limit'=>'0',
				'trial_amount'=>'0',
				'trial_limit'=>'0',
				'expiration_number'=>'0',
				'expiration_period' => "'Month'"
			);

			// Allow override of the discount code values.
			if( ! empty( $pmprosm_values['discount_code'] ) ) {
				$discount_code = array_merge( $discount_code, $pmprosm_values['discount_code'] );
			}

			$sqlQuery = "INSERT INTO $wpdb->pmpro_discount_codes_levels (code_id,
																		 level_id,
																		 initial_payment,
																		 billing_amount,
																		 cycle_number,
																		 cycle_period,
																		 billing_limit,
																		 trial_amount,
																		 trial_limit,
																		 expiration_number,
																		 expiration_period)
														VALUES(" . implode(",", $discount_code) . ")";
			$wpdb->query( $sqlQuery );
		}

		//code created
		return $code_id;
	}

	//something went wrong
	return false;
}

/**
 * This is the function that handles updating discount codes
 * and sub accounts when a sponsor renews or changes levels.
 */
function pmprosm_sponsored_account_change( $level_id, $user_id ) {
	global $wpdb;

	$pmprosm_values = pmprosm_getValuesByMainLevel( $level_id );
	$code_id = pmprosm_getCodeByUserID( $user_id );

	//update seats for code
	if(isset($_REQUEST['seats']))
		$seats = intval($_REQUEST['seats']);
	elseif(isset($_SESSION['seats']))
		$seats = intval($_SESSION['seats']);
	elseif(!empty($pmprosm_values['seats']))
		$seats = $pmprosm_values['seats'];
	else
		$seats = "";

	// Extend the expiration date of the discount code just in case.
	$expires = date( 'Y-m-d', strtotime( '+1 year', current_time( 'timestamp' ) ) );

	// Support for customizing the discount code expires/uses.
	$sponsored_code_settings = apply_filters( 'pmprosm_sponsored_code_settings', array( 'code' => '', 'starts' => '', 'expires' => $expires, 'uses' => $seats ) );

	$sqlQuery = "UPDATE $wpdb->pmpro_discount_codes SET uses = '" . $sponsored_code_settings['uses'] . "', expires = '" . $sponsored_code_settings['expires'] . "' WHERE id = '" . $code_id . "' LIMIT 1";
	$wpdb->query( $sqlQuery );

	//activate/deactivate old accounts
	if( ! empty( $pmprosm_values['sponsored_accounts_at_checkout'] ) ) {
		// cannot rely on old_sub_accounts_active to be set.  May have old accounts, but may only create new ones
		// so we need to deactivate old accounts.
		$children = pmprosm_getChildren( $user_id );
		if( $children ) {
			$old_sub_accounts_active = $_REQUEST['old_sub_accounts_active'];

			for( $i = 0; $i < count( $children ); $i++ ) {
				if( is_array( $old_sub_accounts_active ) && in_array( $children[$i], $old_sub_accounts_active ) ) {
					//they should have their level/etc from before
				} else {
					//remove their level
					pmpro_changeMembershipLevel( 0, $children[$i] );

					//remove discount code use
					pmprosm_removeDiscountCodeUse( $children[$i], $code_id );
				}
			}
		}
	}

	// if old sub accounts cancelled above from user request and "sponsored_accounts_at_checkout"
	//   then old accounts to deactivate should be gone so ok to drop into this code.
	// if not sponsored_accounts_at_checkout then
	//see if we should enable some accounts
	$sqlQuery = "SELECT user_id FROM $wpdb->pmpro_discount_codes_uses WHERE code_id = '" . esc_sql( $code_id ) . "'";
	$sub_user_ids = $wpdb->get_col( $sqlQuery );

	if( ! empty( $sub_user_ids ) ) {
		//check if they have enough seats
		if( $seats >= count( $sub_user_ids ) ) {
			$count = 0;
			foreach( $sub_user_ids as $sub_user_id ) {
				$count++;

				//change their membership
				if( is_array( $pmprosm_values['sponsored_level_id'] ) ) {
					//what level did this user have last that is a sponsored level?
					$last_level_id = $wpdb->get_var("SELECT membership_id FROM $wpdb->pmpro_memberships_users WHERE user_id = '" . esc_sql( $sub_user_id ) . "' AND status = 'inactive' ORDER BY id DESC");

					//okay give them that level back
					if( in_array($last_level_id, $pmprosm_values['sponsored_level_id'] ) ) {
						if( pmprosm_changeMembershipLevelWithCode( $last_level_id, $sub_user_id, $code_id ) ) {

							// Update code use with new order.
							pmprosm_removeDiscountCodeUse( $sub_user_id, $code_id );
							pmprosm_addDiscountCodeUse( $sub_user_id, $last_level_id, $code_id );
						}
					}
				} else {
					if( pmprosm_changeMembershipLevelWithCode( $pmprosm_values['sponsored_level_id'], $sub_user_id, $code_id ) ) {
						// update code Use with new order
						pmprosm_removeDiscountCodeUse( $sub_user_id, $code_id );
						pmprosm_addDiscountCodeUse( $sub_user_id, $pmprosm_values['sponsored_level_id'], $code_id );
					}
				}
			}
		} else {
			// Get code.
			$code = $wpdb->get_var("SELECT code FROM $wpdb->pmpro_discount_codes WHERE id = '" . esc_sql( $code_id ) . "' LIMIT 1");

			// Cancel sponsnored accounts.
			foreach( $sub_user_ids as $sub_user_id ) {
				//cancel their membership
				pmpro_changeMembershipLevel( 0, $sub_user_id );
			}

			//detach sponsored accounts
			$sqlQuery = "DELETE FROM $wpdb->pmpro_discount_codes_uses WHERE code_id = '" . esc_sql( $code_id ) . "'";
			$wpdb->query( $sqlQuery );

			//we better warn them
			if( is_admin() ) {
				// Assuming an admin update.
				set_transient( "pmprosm_error", sprintf(__("This user has fewer seats than they had sponsored accounts. The sponsored accounts have been deactivated. The user must have his sponsored accounts checkout again using the code: %s.", "pmpro-sponsored-members"), $code ) );
			} else {
				// Assuming a checkout.
				add_filter( 'pmpro_confirmation_url', 'pmprosm_pmpro_confirmation_url_lowseats' );
			}
		}
	}
}

/**
 * Make sure we trigger an update when a sponsor renews the same level.
 */
function pmprosm_pmpro_after_checkout_sponsor_account_change( $user_id ) {
	global $pmpro_level;

	// Get checkout level.
	$level = pmpro_getLevelAtCheckout();
	$level_id = empty( $level->id ) ? false : $level->id;

    // Handle sponsored accounts.
    if ( pmprosm_isMainLevel( $level_id ) ) {
		pmprosm_sponsored_account_change($level_id, $user_id);		
	}
}
add_action( 'pmpro_after_checkout', 'pmprosm_pmpro_after_checkout_sponsor_account_change', 10, 2 );

/**
 * Low seats message for confirmation message.
 */
// Add param to checkout URL (queued up in pmprosm_sponsored_account_change()).
function pmprosm_pmpro_confirmation_url_lowseats($url)
{
	$url .= "&lowseats=1";

	return $url;
}
// Add message to confirmation page.
function pmprosm_pmpro_confirmation_message_lowseats( $message ) {
	global $wpdb, $current_user;
	$code_id = pmprosm_getCodeByUserID($current_user->ID);
	$code = $wpdb->get_var("SELECT code FROM $wpdb->pmpro_discount_codes WHERE id = '" . esc_sql( $code_id ) . "' LIMIT 1");

	$message .= sprintf(__( "<p><strong>Notice:</strong>Your current membership has fewer seats than you had sponsored accounts. The accounts have been deactivated. You must have your sponsored accounts checkout again using your code: %s.</p>", 'pmpro-sponsored-members' ), $code );

	return $message;
}
if( ! empty( $_REQUEST['lowseats'] ) ) {
	add_filter( 'pmpro_confirmation_message', 'pmprosm_pmpro_confirmation_message_lowseats' );
}
	
//low seats message for edit user page
function pmprosm_admin_head_errors() {
	$error = get_transient( 'pmprosm_error' );
	if( ! empty( $error ) ) {
	?>
		<script>
		jQuery(document).ready(function() {
			jQuery('div.wrap h2').after('<div id="message" class="updated"><p><?php echo $error;?></p></div>');
		});
		</script>
	<?php

		delete_transient("pmprosm_error");
	}
}
add_action( 'admin_head', 'pmprosm_admin_head_errors' );

/**
 * Function to get children of a sponsor
 */
function pmprosm_getChildren( $user_id = NULL ) {

    global $wpdb, $current_user;

    $children = array();

    if( empty( $user_id ) ) {
        if( is_user_logged_in() ) {
			$user_id = $current_user->ID;
		} else {
			return false;
		}
    }

    $code_id = pmprosm_getCodeByUserID( $user_id );

	if( empty( $code_id ) ) {
		return false;
	}

	$children = $wpdb->get_col("SELECT user_id FROM $wpdb->pmpro_memberships_users WHERE code_id = '" . esc_sql( $code_id ) . "' AND status = 'active'");

	// If sponsor account is expired or cancelled,
	// then children accounts are no longer active.
	// So we can get a list of old children accounts
	// by getting all the uses of the discount code.

	if ( empty( $children ) ) {
		$sqlQuery = "SELECT user_id FROM $wpdb->pmpro_discount_codes_uses WHERE code_id = '" . esc_sql( $code_id ) . "'";
		$children = $wpdb->get_col( $sqlQuery );
	}
	
    return $children;
}

/**
 * Get the sponsoring user's ID from a given code ID.
 */
function pmprosm_getCodeUserID( $code_id ) {
	$code_user_ids = get_option( 'pmpro_code_user_ids' );

	if( ! empty( $code_user_ids[$code_id] ) ) {
		return $code_user_ids[$code_id];
	} else {
		return false;
	}
}

/**
 * Link a discount code to the sponsoring user.
 */
function pmprosm_setCodeUserID( $code_id, $user_id ) {
	$code_user_ids = get_option( 'pmpro_code_user_ids' );
	$code_user_ids[$code_id] = $user_id;

	return update_option( 'pmpro_code_user_ids', $code_user_ids );
}

/**
 * Unlink a discount code from the sponsoring user.
 */
function pmprosm_deleteCodeUserID( $code_id ) {
	$code_user_ids = get_option( 'pmpro_code_user_ids' );
	unset($code_user_ids[$code_id]);

	return update_option( 'pmpro_code_user_ids', $code_user_ids );
}

/**
 * Get discount code by user.
 */
function pmprosm_getCodeByUserID( $user_id ) {
	$code_user_ids = get_option( 'pmpro_code_user_ids' );

	if( is_array($code_user_ids) ) {
		foreach( $code_user_ids as $code_id => $code_user_id ) {
			if( $code_user_id == $user_id ) {
				return $code_id;
			}
		}
	}

	return false;
}

/**
 * Get user by discount code.
 */
function pmprosm_getUserByCodeID( $needle ) {
	$code_user_ids = get_option( 'pmpro_code_user_ids' );

	if( is_array($code_user_ids ) ) {
		foreach( $code_user_ids as $code_id => $code_user_id ) {
			if( $code_id == $needle ) {
				return $code_user_id;
			}
		}
	}

	return false;
}

/**
 * Get a discount code object from a code_id.
 */
function pmprosm_getDiscountCodeByCodeID( $code_id ) {
	static $discount_codes;

	if( !isset( $discount_codes[$code_id] ) ) {
		global $wpdb;
		$discount_codes[$code_id] = $wpdb->get_row( "SELECT * FROM $wpdb->pmpro_discount_codes WHERE id = '" . esc_sql( $code_id ) . "' LIMIT 1");
	}

	return $discount_codes[$code_id];
}

/**
 * Show a user's discount code on the confirmation page.
 */
function pmprosm_pmpro_confirmation_message( $message ) {
	global $current_user, $wpdb;

	$code_id = pmprosm_getCodeByUserID( $current_user->ID );

	if( ! empty( $code_id ) ) {
		$pmprosm_values = pmprosm_getValuesByMainLevel( $current_user->membership_level->ID );
		$code = pmprosm_getDiscountCodeByCodeID( $code_id );

		if( empty( $pmprosm_values['sponsored_level_id'] ) ) {
			return $message;
		}

		if( ! is_array($pmprosm_values['sponsored_level_id'] ) ) {
			$sponsored_level_ids = array($pmprosm_values['sponsored_level_id']);
		} else {
			$sponsored_level_ids = $pmprosm_values['sponsored_level_id'];
		}

		// No sponsored levels to use codes for.
		if( empty( $sponsored_level_ids ) || empty( $sponsored_level_ids[0] ) ) {
			return $message;
		}

		$pmpro_levels = pmpro_getAllLevels( false, true );

		$code_urls = array();
		foreach( $sponsored_level_ids as $sponsored_level_id ) {
			$level_name = $pmpro_levels[$sponsored_level_id]->name;
			$code_urls[] = array( "name"=>$level_name, "url"=>pmpro_url( "checkout", "?level=" . $sponsored_level_id . "&discount_code=" . $code->code ) );
		}
	}

	if( ! empty( $code ) ) {
		if ( empty( $pmprosm_values['hide_display_discount_code']) || $pmprosm_values['hide_display_discount_code'] === false ) {

			if ( count( $code_urls ) > 1 ) {
				$message .= "<div class=\"pmpro_content_message\"><p>" . __( "Give this code to your sponsored members to use at checkout:", "pmpro-sponsored-members" ) . " <strong>" . $code->code . "</strong></p><p>" . __( "Or provide one of these direct links to register:", "pmpro-sponsored-members" ) . "<br /></p>";
			} else {
				$message .= "<div class=\"pmpro_content_message\"><p>" . __( "Give this code to your sponsored members to use at checkout:", "pmpro-sponsored-members" ) . " <strong>" . $code->code . "</strong></p><p>" . __( "Or provide this direct link to register:", "pmpro-sponsored-members" ) . "<br /></p>";
			}

			$message .= "<ul>";
			foreach ( $code_urls as $code_url ) {
				$message .= "<li>" . $code_url['name'] . ":<strong> " . esc_url( $code_url['url'] ) . "</strong></li>";
			}
			$message .= "</ul>";

			if ( empty( $code->uses ) ) {
				$message .= __( "This code has unlimited uses.", "pmpro-sponsored-members" );
			} else {
				$message .= sprintf( _n( 'This code has %d use.', 'This code has %d uses.', $code->uses, 'pmpro-sponsored-members' ), $code->uses );
			}

			$message .= "</div>";
		}
        $member_ids = pmprosm_getChildren( $current_user->ID );
		if (isset($pmprosm_values['list_sponsored_accounts']) && $pmprosm_values['list_sponsored_accounts'] === true) {
			if ( ! empty( $member_ids ) ) {
				$message .= "<hr />";
				$message .= pmprosm_display_sponsored_accounts( $member_ids );
			}
		}
	}
	return $message;
}
add_filter( 'pmpro_confirmation_message', 'pmprosm_pmpro_confirmation_message' );

/**
 * Delete code connection when a discount code is deleted.
 */
function pmprosm_pmpro_delete_discount_code( $code_id ) {
	pmprosm_deleteCodeUserID( $code_id );
}
add_action( 'pmpro_delete_discount_code', 'pmprosm_pmpro_delete_discount_code' );

/**
 * Only let members using a sponsored discount code sign up for the sponsored level
 */
function pmprosm_pmpro_registration_checks( $pmpro_continue_registration ) {
	//only bother if things are okay so far
	if( ! $pmpro_continue_registration ) {
		return $pmpro_continue_registration;
	}

	//level = PMPROSM_SPONSORED_ACCOUNT_LEVEL and there is no discount code, then show an error message
	global $pmpro_level, $discount_code, $wpdb;

	if( pmprosm_isSponsoredLevel( $pmpro_level->id ) && empty( $discount_code ) && !pmprosm_isMainLevel( $pmpro_level->id ) ) {
		$pmprosm_values = pmprosm_getValuesBySponsoredLevel( $pmpro_level->id, false );
		$continue_reg = true;

		foreach( $pmprosm_values as $pmprosm_value ) {
			$check_sponsored_level = false;

			// Check if array.
			if( is_array( $pmprosm_value['sponsored_level_id'] ) ) {
				$check_sponsored_level = in_array( $pmpro_level->id, $pmprosm_value );
			} elseif( $pmprosm_value['sponsored_level_id'] == $pmpro_level->id ) {
				$check_sponsored_level = true;
			}

			if( $continue_reg && isset( $pmprosm_value['discount_code_required'] ) && ! empty( $pmprosm_value['discount_code_required'] ) && $check_sponsored_level ) {
				$continue_reg = false;
				break;
			}
		}

		if( ! $continue_reg ) {
			pmpro_setMessage( __( "You must use a valid discount code to register for this level.", "pmpro-sponsored-members" ), "pmpro_error" );
			return false;
		}
	}

	// If a discount code is being used, check that the main account is active.
	if( pmprosm_isSponsoredLevel( $pmpro_level->id ) && !empty( $discount_code ) ) {
		$pmprosm_values = pmprosm_getValuesBySponsoredLevel( $pmpro_level->id, false );

		$code_id = $wpdb->get_var( "SELECT id FROM $wpdb->pmpro_discount_codes WHERE code = '" . esc_sql($discount_code) . "' LIMIT 1" );
		if( ! empty( $code_id ) ) {
			$code_user_id = pmprosm_getCodeUserID( $code_id );

			// If no user is found for the discount code,
			// assume it's a normal discount code.
			if ( ! $code_user_id ) {
				return $pmpro_continue_registration;
			}

			$continue_reg = false;

			foreach( $pmprosm_values as $pmprosm_value ) {
				if( ! $continue_reg && ! empty( $code_user_id ) && ! pmpro_hasMembershipLevel( $pmprosm_value['main_level_id'], $code_user_id ) ) {
					$continue_reg = false;
				} else {
					$continue_reg = true;
				}
			}

			if( ! $continue_reg ) {
				pmpro_setMessage( __( "The sponsor for this code is inactive. Ask them to renew their account.", "pmpro-sponsored-members" ), "pmpro_error" );
				return false;
			}

			if( ! empty( $code_user_id ) && $code_user_id == get_current_user_id() ) {
				pmpro_setMessage( __( "Sponsors are not permitted to sign up for sponsored levels. This is most likely a mistake.", "pmpro-sponsored-members" ), "pmpro_error" );
				return false;
			}
		}
	}

	// If the level has max or min seats, check them.
	if( pmprosm_isMainLevel( $pmpro_level->id ) ) {
		$pmprosm_values = pmprosm_getValuesBySponsoredLevel( $pmpro_level->id, false );
		if( isset( $pmprosm_values['max_seats']) && intval( $_REQUEST['seats']) > intval( $pmprosm_values['max_seats'] ) ) {
			pmpro_setMessage( __("The maximum number of seats allowed is " . intval( $pmprosm_values['max_seats'] ) . ".", "pmpro-sponsored-members" ), "pmpro_error" );
			return false;
		}
		elseif( isset( $pmprosm_values['min_seats'] ) && intval( $_REQUEST['seats'] ) < intval( $pmprosm_values['min_seats'] ) ) {
			pmpro_setMessage( __( "The minimum number of seats allowed is " . intval( $pmprosm_values['min_seats'] ) . ".", "pmpro-sponsored-members" ), "pmpro_error" );
			return false;
		}
	}

	return $pmpro_continue_registration;
}
add_filter( 'pmpro_registration_checks', 'pmprosm_pmpro_registration_checks' );

// add parent account column to the discount codes table view
function pmprosm_pmpro_discountcodes_extra_cols_header() {
	?>
	<th><?php esc_html_e( "Parent Account", "pmpro-sponsored-members" );?></th>
	<?php
}
add_action( "pmpro_discountcodes_extra_cols_header", "pmprosm_pmpro_discountcodes_extra_cols_header" );

function pmprosm_pmpro_discountcodes_extra_cols_body( $code ) {
	$code_user_id = pmprosm_getCodeUserID( $code->id );
	$code_user = get_userdata( $code_user_id );
	?>
	<td>
	<?php if( ! empty( $code_user_id ) && ! empty( $code_user ) ) { ?>
		<a href="<?php echo get_edit_user_link( $code_user_id ); ?>"><?php echo $code_user->user_login; ?></a>
	<?php } elseif( ! empty( $code_user_id ) && empty( $code_user ) ) { ?>
		<em>Missing User</em>
	<?php } ?>
	</td>
	<?php
}
add_action( "pmpro_discountcodes_extra_cols_body", "pmprosm_pmpro_discountcodes_extra_cols_body" );


/**
 * Add user id field to discount code page.
 */
function pmprosm_pmpro_discount_code_after_settings() {
	$code_id = intval( $_REQUEST['edit'] );

	if( ! empty( $_REQUEST['user_id'] ) ) {
		$code_user_id = intval( $_REQUEST['user_id'] );
	} elseif( $code_id > -1 ) {
		$code_user_id = pmprosm_getCodeUserID( $code_id );
	} else {
		$code_user_id = "";
	}
?>
<h2><?php esc_html_e( "For Sponsored Accounts", "pmpro-sponsored-members" );?></h2>
<table class="form-table">
<tbody>
<tr>
    <th scope="row" valign="top"><label for="user_id"><?php esc_html_e( "User ID:", "pmpro-sponsored-members" );?></label></th>
    <td>
		<input name="user_id" type="text" size="10" value="<?php if( ! empty( $code_user_id ) ) { echo esc_attr( $code_user_id ); }?>" />
		<small class="pmpro_lite"><?php esc_html_e( "The user ID of the main account holder.", "pmpro-sponsored-members" );?></small>
	</td>
</tr>
</tbody>
</table>
<?php
}
add_action( "pmpro_discount_code_after_settings", "pmprosm_pmpro_discount_code_after_settings" );

/**
 * Save the code user id when saving a discount code.
 */
function pmprosm_pmpro_save_discount_code( $code_id ) {
	//fix in case this is a new discount code (for PMPro versions < 1.7.1)
	if( $code_id < 0 ) {
		global $wpdb;
		$code_id = $wpdb->get_var( "SELECT id FROM $wpdb->pmpro_discount_codes ORDER BY id DESC LIMIT 1" );
	}

	if( ! empty( $code_id ) ) {
		$code_user_id = intval( $_REQUEST['user_id'] );
		pmprosm_setCodeUserID( $code_id, $code_user_id );
	}
}
add_action( "pmpro_save_discount_code", "pmprosm_pmpro_save_discount_code", 5 );

//show existing sponsored accounts and add a dropdown to choose number of seats on checkout page
function pmprosm_pmpro_checkout_boxes() {
	global $current_user, $pmpro_level, $pmpro_currency_symbol;

	//only for PMPROSM_MAIN_ACCOUNT_LEVEL
	if( empty( $pmpro_level) || ! pmprosm_isMainLevel( $pmpro_level->id ) ) {
		return;
	}

	//make sure options are defined for this
	$pmprosm_values = pmprosm_getValuesByMainLevel( $pmpro_level->id );

	if ( empty( $pmprosm_values ) ) {
		return;
	}

	$can_edit_seats = ! empty( $pmprosm_values['max_seats'] ) || ! empty( $pmprosm_values['min_seats'] );
	$seat_cost = empty( $pmprosm_values['seat_cost'] ) ? null : $pmprosm_values['seat_cost'];
	$max_seats = empty( $pmprosm_values['max_seats'] ) ? null : $pmprosm_values['max_seats'];

	//get seats from submit
	if( isset( $_REQUEST['seats'] ) ) {
		$seats = intval( $_REQUEST['seats'] );
	} elseif( ! empty( $current_user->ID ) ) {
		$seats = get_user_meta( $current_user->ID, "pmprosm_seats", true );
	} else {
		$seats = "";
	}

    if( isset( $pmprosm_values['seats'] ) && $seats == "" ) {
		$seats = $pmprosm_values['seats'];
	}

	if( $seats == "" ) {
		// leaving blank ('') causes this to be unlimited.
		$seats = 0;
	}

	if( is_array( $pmprosm_values['sponsored_level_id'] ) ) {
		$sponsored_level = pmpro_getLevel( $pmprosm_values['sponsored_level_id'][0] );
	} else {
		$sponsored_level = pmpro_getLevel( $pmprosm_values['sponsored_level_id'] );
	}
	?>
	<div id="pmpro_extra_seats" class="pmpro_checkout">
		<?php
			if ( $can_edit_seats ) {
		?>
			<hr />
			<h2>
				<span class="pmpro_checkout-h2-name">
				<?php
					if ( $seat_cost >  0) {
						esc_html_e( "Would you like to purchase extra seat(s)?", "pmpro-sponsored-members" );
					} else {
						esc_html_e( "Would you like to create extra account(s)?", "pmpro-sponsored-members" );
					}
				?>
				</span>
			</h2>
		<?php } ?>
		<div class="pmpro_checkout-fields">
			<div class="pmpro_checkout-field pmpro_checkout-field-seats">

				<?php
					if ( $can_edit_seats ) {
				?>
						<label for="seats"><?php echo __( 'Number of Seats', 'pmpro-sponsored-members' );?></label>
						<input type="text" id="seats" name="seats" value="<?php echo esc_attr( $seats ); ?>" size="10" />
				<?php } else { ?>
						<input type="hidden" id="seats" name="seats" value="<?php echo esc_attr( $seats ); ?>" size="10" />
				<?php } ?>

				<?php
					if ( $can_edit_seats ) {
				?>
						<p class="pmpro_small">
							<?php
								//min seats defaults to 1
								if( isset( $pmprosm_values['min_seats'] ) ) {
									$min_seats = $pmprosm_values['min_seats'];
								} else {
									$min_seats = 1;
								}

								if ( $max_seats > 1 ) {
									if ( isset( $pmprosm_values['seat_cost_text'] ) ) {
										printf( esc_html__( "Enter a number from %d to %d. %s", "pmpro-sponsored-members" ), $min_seats, $pmprosm_values['max_seats'], $pmprosm_values['seat_cost_text'] );
									} else {
										$seat_cost = empty( $pmprosm_values['seat_cost'] ) ? 0 : $pmprosm_values['seat_cost'];
										printf( esc_html__( "Enter a number from %d to %d. +%s per extra seat.", "pmpro-sponsored-members" ), $min_seats, $pmprosm_values['max_seats'], $pmpro_currency_symbol . $seat_cost );									
									}
								}
							?>
						</p>
				<?php } ?>
				<?php
				//adding sub accounts at checkout?
				if(!empty($pmprosm_values['sponsored_accounts_at_checkout']))
				{
					// add extra_seat_prompt_text
					if(!empty($pmprosm_values['extra_seat_prompt_text'])) {
						echo '<div id="pmpro_extra_seat_prompt">';
						echo  $pmprosm_values['extra_seat_prompt_text'];
						echo '</div>';
					}
					//look for existing sponsored accounts
					$children = pmprosm_getChildren($current_user->ID);
					if(!empty($children))
					{
						echo "<hr />";

						//get checkbox values if there
						if( isset( $_REQUEST['old_sub_accounts_active'] ) ) {
							$old_sub_accounts_active = $_REQUEST['old_sub_accounts_active'];
						} else {
							$old_sub_accounts_active = array();
						}

						$i = 0;
						foreach( $children as $child_id ) {
						?>
						<div>
							<?php
								//get user
								$child = get_userdata( $child_id );
                                if ( $child === false ) {
									continue;  //user does not exist, so skip it.
								}

								//acive?
								if( pmpro_hasMembershipLevel( NULL, $child_id ) ) {
									$active = true;
								} else {
									$active = false;
								}

								//checked?
                            	// can't assume order of old_sub_accounts_active
								if( in_array( $child_id, $old_sub_accounts_active ) ) {
									$checked = true;
								} else {
									$checked = $active;
								}
							?>
							<label><?php echo esc_html( $child->display_name );?></label>
							<input type="checkbox" id="old_sub_accounts_active_<?php echo intval( $i );?>" class="old_sub_accounts_active" name="old_sub_accounts_active[]" value="<?php echo intval( $child_id );?>" <?php checked($checked, true);?> />
							<label class="pmpro_normal pmpro_clickable" for="old_sub_accounts_active_<?php echo intval( $i );?>">
							<?php if( ! empty( $active ) ) { ?>
								<?php esc_html_e( 'Keep checked to keep this account active.', 'pmpro-sponsored-members' ); ?>
							<?php } else { ?>
								<?php esc_html_e( 'Check to reactivate this account.', 'pmpro-sponsored-members' ); ?>
							<?php } ?>
							</label>
						</div>
						<?php
							$i++;
						}
					}	//end existing sponsored accounts

					echo "<div id = 'sponsored_accounts'>";

					if( ! empty( $_REQUEST['add_sub_accounts_username'] ) ) {
						$child_usernames = $_REQUEST['add_sub_accounts_username'];
					} elseif( $seats ) {
						$child_usernames = array_fill( 0, $seats, '' );
					} else {
						$child_usernames = array();
					}

					if( ! empty( $_REQUEST['add_sub_accounts_first_name'] ) ) {
						$child_first_names = $_REQUEST['add_sub_accounts_first_name'];
					} elseif( $seats ) {
						$child_first_names = array_fill(0, $seats, '');
					} else {
						$child_first_names = array();
					}

					if( ! empty( $_REQUEST['add_sub_accounts_last_name'] ) ) {
						$child_last_names = $_REQUEST['add_sub_accounts_last_name'];
					} elseif($seats) {
						$child_last_names = array_fill(0, $seats, '');
					} else {
						$child_last_names = array();
					}

					if( !empty( $_REQUEST['add_sub_accounts_email'] ) ) {
						$child_emails = $_REQUEST['add_sub_accounts_email'];
					} elseif($seats) {
						$child_emails = array_fill(0, $seats, '');
					} else {
						$child_emails = array();
					}

					for( $i = 0; $i < count( $child_usernames ); $i++ ) {
						if( is_array( $child_usernames ) ) {
							$child_username = $child_usernames[$i];
						} else {
							$child_username = "";
						}

						if( is_array( $child_usernames ) ) {
							$child_first_name = $child_first_names[$i];
						} else {
							$child_first_name = "";
						}

						if( is_array( $child_usernames ) ) {
							$child_last_name = $child_last_names[$i];
						} else {
							$child_last_name = "";
						}

						if( is_array( $child_usernames ) ) {
							$child_email = $child_emails[$i];
						} else {
							$child_email = "";
						}
					?>
					<div id="sponsored_account_<?php echo intval( $i );?>">
						<hr />
                        <div><h2><?php echo esc_html( $sponsored_level->name ); esc_html_e(' account information.', 'pmpro-sponsored-members'); ?> </h2>
                            <h4><?php if (isset($pmprosm_values['sponsored_header_text']))
									echo $pmprosm_values['sponsored_header_text'];
								else
									esc_html_e('Please fill in following information and account(s) will be created.', 'pmpro-sponsored-members');
								?></h4>
                        </div>
                        <?php if(!empty($pmprosm_values['children_get_name'])) { ?>
							<div class="pmpro_checkout-field">
								<label><?php echo esc_html__("First Name", "pmpro-sponsored-members");?></label>
								<input type="text" name="add_sub_accounts_first_name[]" class="pmpro_required" value="<?php echo esc_attr($child_first_name);?>" size="20" />
							</div>
							<div class="pmpro_checkout-field">
								<label><?php echo esc_html__("Last Name", "pmpro-sponsored-members");?></label>
								<input type="text" name="add_sub_accounts_last_name[]" class="pmpro_required" value="<?php echo esc_attr($child_last_name);?>" size="20" />
							</div>
						<?php } ?>
						<?php if(empty($pmprosm_values['children_hide_username'])) { ?>
							<div class="pmpro_checkout-field">
								<label><?php echo esc_html__("Username", "pmpro-sponsored-members");?></label>
								<input type="text" name="add_sub_accounts_username[]" class="pmpro_required" value="<?php echo esc_attr($child_username);?>" size="20" />
							</div>
						<?php } ?>
						<?php
						// The `children_hide_email` option is deprecated, but keeping it in the code in case someone is using it.
						if(empty($pmprosm_values['children_hide_email'])) { ?>
							<div class="pmpro_checkout-field">
								<label><?php echo esc_html__("Email", "pmpro-sponsored-members");?></label>
								<input type="text" name="add_sub_accounts_email[]" class="pmpro_required" value="<?php echo esc_attr($child_email);?>" size="20" />
							</div>
						<?php } ?>
						<?php if(empty($pmprosm_values['children_hide_password'])) { ?>
							<div class="pmpro_checkout-field">
								<label><?php echo esc_html__("Password", "pmpro-sponsored-members");?></label>
								<input type="password" name="add_sub_accounts_password[]" class="pmpro_required" value="" size="20" />
							</div>
						<?php } ?>
						<?php do_action('pmprosm_children_fields', $i, $seats);?>
					</div>
					<?php
					}

					echo "</div>";

					/*
						Get the HTML for the empty extra fields and save it to a variable.
					*/
					ob_start();
					do_action("pmprosm_children_fields", false, $seats);
					$empty_child_fields = ob_get_contents();
					ob_end_clean();
					//also clean it up a bit
					$empty_child_fields = str_replace("\n", "", $empty_child_fields);
				}	//if(!empty($pmprosm_values['sponsored_accounts_at_checkout']))
				?>

				<script>
				jQuery(document).ready(function() {
					var pmpro_base_level_is_free = <?php if(pmpro_isLevelFree($pmpro_level)) echo "true"; else echo "false";?>;
					var seat_cost = <?php echo ( empty( $pmprosm_values['seat_cost'] ) ? 0 : intval( $pmprosm_values['seat_cost'] ) )?>;
					var min_seats = <?php if(!empty($pmprosm_values['min_seats'])) echo intval($pmprosm_values['min_seats']); elseif (!empty($pmprosm_values['seats'])) echo intval($pmprosm_values['seats']); else echo "0";?>;
					var max_seats = <?php if(!empty($pmprosm_values['max_seats'])) echo intval($pmprosm_values['max_seats']); elseif (!empty($pmprosm_values['seats'])) echo intval($pmprosm_values['seats']); else echo "false";?>;

					//update things when the # of seats changes
					jQuery('#seats, input.old_sub_accounts_active').bind("change", function() {
						seatsChanged();
					});

					//run it once on load too
					seatsChanged();

					function seatsChanged() {
						//num seats entered
						seats = parseInt(jQuery('#seats').val());

						//num of old seats checked
						old_sub_accounts_active = 0;
						jQuery("input.old_sub_accounts_active:checked").each(function(){
							old_sub_accounts_active += 1;
						});

						//max sure not over max
						if( max_seats && seats > max_seats ) {
							seats = max_seats;
							jQuery('#seats').val(seats);
						}

						//and not under min
						if( min_seats && seats < min_seats ) {
							seats = min_seats;
							jQuery('#seats').val(seats);
						}

						<?php
							//how many child seats are shown now (if sponsored_accounts_at_checkout is set)
							if(!empty($pmprosm_values['sponsored_accounts_at_checkout']))
							{
							?>
                            if( seats > 0) {
                                jQuery('#pmpro_extra_seat_prompt').show();
                            } else {
                                jQuery('#pmpro_extra_seat_prompt').hide();
                            }
							if(jQuery('#sponsored_accounts')) {
								children = jQuery('#sponsored_accounts').children();
								i = children.length-1;

								//how many should we show
								newseats = seats - old_sub_accounts_active;

								if(newseats < children.length) {
									while(i >= newseats) {
										jQuery(children[i]).remove();
										i--;
									}
								} else if(newseats > children.length) {
									i = children.length;

									while (i < newseats) {
                                        var div = '<div id="sponsored_account_'+i+'"><hr /><div><h2><?php echo esc_html( $sponsored_level->name ); esc_html_e(" account information # XXXX", 'pmpro-sponsored-members'); ?> </h2><h4><?php if (isset($pmprosm_values["sponsored_header_text"]))echo $pmprosm_values["sponsored_header_text"];else esc_html_e("Please fill in following information and account(s) will be created.", 'pmpro-sponsored-members');?></h4></div><?php if(!empty($pmprosm_values["children_get_name"])) { ?><label>First Name</label><input type="text" name="add_sub_accounts_first_name[]" value="" class="input" size="20" /><br><label>Last Name</label><input type="text" name="add_sub_accounts_last_name[]" value="" class="input" size="20" /><br><?php } ?><?php if(empty($pmprosm_values["children_hide_username"])){ ?><label>Username</label><input type="text" name="add_sub_accounts_username[]" value="" class="input pmpro_required" size="20" /> <span class="pmpro_asterisk"><abbr title="Required Field">*</abbr></span><br><?php } ?><?php if(empty($pmprosm_values["children_hide_email"])){ ?><label>Email</label><input type="text" name="add_sub_accounts_email[]" value"" class="input pmpro_required" size="20" /> <span class="pmpro_asterisk"><abbr title="Required Field">*</abbr></span><br><?php } ?><?php if(empty($pmprosm_values["children_hide_password"])){ ?><label>Password</label><input type="password" name="add_sub_accounts_password[]" value="" class="input pmpro_required" size="20" /> <span class="pmpro_asterisk"><abbr title="Required Field">*</abbr></span><?php } ?><?php echo $empty_child_fields;?></div>';
                                        newdiv = div.replace(/XXXX/g,i+1);
                                        jQuery('#sponsored_accounts').append(newdiv); i++;
									}
								}
							}
							<?php
							}
						?>

						if(pmpro_base_level_is_free && seat_cost && seats) {
							//need to show billing fields
							jQuery('#pmpro_payment_method').show();
							jQuery('#pmpro_billing_address_fields').show();
							jQuery('#pmpro_payment_information_fields').show();
						} else if(pmpro_base_level_is_free) {
							//need to hide billing fields
							jQuery('#pmpro_payment_method').hide();
							jQuery('#pmpro_billing_address_fields').hide();
							jQuery('#pmpro_payment_information_fields').hide();
						}

						<?php do_action('pmprosm_seats_changed_js'); ?>
					}
				});
				</script>
			</div> <!-- end pmpro_checkout-field-seats -->
		</div> <!-- end pmpro_checkout-fields -->
	</div> <!-- end pmpro_extra_seats -->
<?php
}
add_action( "pmpro_checkout_boxes", "pmprosm_pmpro_checkout_boxes" );

//adjust price based on seats
function pmprosm_pmpro_checkout_levels( $level ) {
	//get seats from submit
	if( isset( $_REQUEST['seats'] ) ) {
		$seats = intval( $_REQUEST['seats'] );
	} else {
		$seats = "";
	}

	if( ! empty( $seats ) ) {
		$pmprosm_values = pmprosm_getValuesByMainLevel($level->id);
		if( ! empty( $pmprosm_values['seat_cost'] ) ) {
			if((!isset($pmprosm_values['apply_seat_cost_to_initial_payment']) && $level->initial_payment > 0) || !empty($pmprosm_values['apply_seat_cost_to_initial_payment']))
			{
				if(!empty($pmprosm_values['apply_seat_cost_to_initial_payment']) && $pmprosm_values['apply_seat_cost_to_initial_payment'] === "sponsored_level") {
					$sponsored_level = pmpro_getLevel($pmprosm_values['sponsored_level_id']);
					$level->initial_payment += $sponsored_level->initial_payment * $seats;
				} else {
					$level->initial_payment += $pmprosm_values['seat_cost'] * $seats;
				}
			}

			if((!isset($pmprosm_values['apply_seat_cost_to_billing_amount']) && $level->billing_amount > 0) || !empty($pmprosm_values['apply_seat_cost_to_billing_amount']))
			{
				if(!empty($pmprosm_values['apply_seat_cost_to_billing_amount']) && $pmprosm_values['apply_seat_cost_to_billing_amount'] === "sponsored_level") {
					$sponsored_level = pmpro_getLevel($pmprosm_values['sponsored_level_id']);
					$level->billing_amount += $sponsored_level->billing_amount * $seats;
					$level->cycle_number = $sponsored_level->cycle_number;
					$level->cycle_period = $sponsored_level->cycle_period;
				} else {
					$level->billing_amount += $pmprosm_values['seat_cost'] * $seats;

					if(!empty($pmprosm_values['seat_cost_cycle_number'])) {
						$level->cycle_number = $pmprosm_values['seat_cost_cycle_number'];
					}
					if(!empty($pmprosm_values['seat_cost_cycle_period'])) {
						$level->cycle_period = $pmprosm_values['seat_cost_cycle_period'];
					}
				}
			}
		}
	}

	return $level;
}
add_filter( "pmpro_checkout_level", "pmprosm_pmpro_checkout_levels" );

//save seats at checkout
function pmprosm_pmpro_after_checkout( $user_id ) {
	global $current_user, $pmprosm_sponsored_account_levels, $pmpro_level;

	$level = pmpro_getLevelAtCheckout();
	$level_id = empty( $level->id ) ? false : $level->id;

	if( empty( $level_id ) ) {
		return;
	}

	$parent_level = pmprosm_getValuesByMainLevel( $level_id );

	//get seats from submit
	if( ! empty( $parent_level['seats'] ) ) {
		$seats = $parent_level['seats'];
	} elseif( isset( $_REQUEST['seats'] ) ) {
		$seats = intval($_REQUEST['seats']);
	} else {
		$seats = "";
	}

	update_user_meta( $user_id, "pmprosm_seats", $seats );

	if( ! empty( $parent_level['sponsored_accounts_at_checkout'] ) ) {
		//Create additional child member here
		if( ! empty( $_REQUEST['add_sub_accounts_username'] ) ) {
			$child_username = $_REQUEST['add_sub_accounts_username'];
		} else {
			$child_username = array();
		}

		if( ! empty( $_REQUEST['add_sub_accounts_first_name'] ) ) {
			$child_first_name = $_REQUEST['add_sub_accounts_first_name'];
		} else {
			$child_first_name = array();
		}

		if( ! empty( $_REQUEST['add_sub_accounts_last_name'] ) ) {
			$child_last_name = $_REQUEST['add_sub_accounts_last_name'];
		} else {
			$child_last_name = array();
		}

		$child_password = isset( $_REQUEST['add_sub_accounts_password'] ) ? $_REQUEST['add_sub_accounts_password'] : '';
		$child_email = isset( $_REQUEST['add_sub_accounts_email'] ) ? $_REQUEST['add_sub_accounts_email'] : '';

		// Bail if the child accounts aren't set.
		if ( ! is_array( $child_email ) && empty( $child_email ) ) {
			return;
		}
		$sponsored_code = pmprosm_getCodeByUserID( $user_id );

		if( $parent_level ) {
			if( is_array( $parent_level['sponsored_level_id'] ) ) {
				$child_level_id = $parent_level['sponsored_level_id'][0];
			} else {
				$child_level_id = $parent_level['sponsored_level_id'];
			}

			//create new child accounts
			for( $i = 0; $i < count( $child_email ); $i++ ) {
				//if a blank entry is find, skip it
				if( empty( $child_email[$i] ) ) {
					continue;
				}

				// We may need to generate a username
				if ( ! empty( $child_username[$i] ) ) {
					$new_username = $child_username[$i];
				} else {
					$new_email = empty( $child_email[$i] ) ? null : $child_email[$i];
					$new_first_name = empty( $child_first_name[$i] ) ? null : $child_first_name[$i];
					$new_last_name = empty( $child_last_name[$i] ) ? null : $child_last_name[$i];
					$new_username = pmpro_generateUsername( $new_first_name, $new_last_name, $new_email );
				}

				// If we don't have a password, generate one.
				$new_password = isset( $child_password[$i] ) ? $child_password[$i] : wp_generate_password();

				// Create the user.
				$child_user_id = wp_create_user( $new_username, $new_password, $child_email[$i] );

				if( is_wp_error($child_user_id) ) {

					$error_code = $child_user_id->get_error_code();
					$existing_user = null;

					// check the error code & look up user if it's a duplicate email.
					if ( $error_code == 'existing_user_email' ) {
						$existing_user = get_user_by( 'email', $child_email[ $i ] );
					} else {
						// skip this user (quietly).
						// TODO: Should probably return message to admin?
						$existing_user = null;
						continue;
					}

					// Have an actual WP_User object & is user already a member?
					if( ! is_null($existing_user) && ! pmpro_getMembershipLevelForUser( $existing_user->ID ) ) {
						$child_user_id = $existing_user->ID;
					} else {
						continue;
					}
				}

				// Update first/last.
				if( ! empty( $child_first_name[$i] ) ) {
					update_user_meta($child_user_id, "first_name", $child_first_name[$i]);
				}
				if( ! empty( $child_last_name[$i] ) ) {
					update_user_meta($child_user_id, "last_name", $child_last_name[$i]);
				}

				if( pmprosm_changeMembershipLevelWithCode( $child_level_id, $child_user_id, $sponsored_code ) ) {
					pmprosm_addDiscountCodeUse( $child_user_id, $child_level_id, $sponsored_code );
				}

				// User_id here is the parent user id
				do_action( 'pmprosm_after_child_created', $child_user_id, $user_id, $i );
			}
		}
	}
}
add_action( "pmpro_after_checkout", "pmprosm_pmpro_after_checkout" );

function pmprosm_after_checkout_children_updated( $user_id ) {
	global $current_user, $pmprosm_sponsored_account_levels, $pmpro_level;

	$level = pmpro_getLevelAtCheckout();
	$level_id = empty( $level->id ) ? false : $level->id;

	if( empty( $level_id ) ) {
		return;
	}

	$parent_level = pmprosm_getValuesByMainLevel( $level_id );

	if( ! empty( $parent_level['sponsored_accounts_at_checkout'] ) ) {
		$children = pmprosm_getChildren( $user_id );
		if ( $children ) {
			//get last order
			$parentOrder = new MemberOrder();
			$parentOrder->getLastMemberOrder( $user_id, "success" );

			$sponsor_code_id = pmprosm_getCodeByUserID( $user_id );
			foreach ($children as $child_user_id) {

				$order_id = pmprosm_getOrderByCodeUser( $sponsor_code_id, $child_user_id );

				if ( ! empty( $order_id ) ) {
					$invoice = new MemberOrder( $order_id );
					//set some child order fields to match parents
					$invoice->billing->name    = isset( $parentOrder->billing->name ) ? $parentOrder->billing->name : '';
					$invoice->billing->street  = isset( $parentOrder->billing->street ) ? $parentOrder->billing->street : '';
					$invoice->billing->city    = isset( $parentOrder->billing->city ) ? $parentOrder->billing->city : '';
					$invoice->billing->state   = isset( $parentOrder->billing->state ) ? $parentOrder->billing->state : '';
					$invoice->billing->zip     = isset( $parentOrder->billing->zip ) ? $parentOrder->billing->zip : '';
					$invoice->billing->country = isset( $parentOrder->billing->country ) ? $parentOrder->billing->country : '';
					$invoice->billing->phone   = isset( $parentOrder->billing->phone ) ? $parentOrder->billing->phone : '';
					$invoice->status           = isset( $parentOrder->status ) ? $parentOrder->status : '';
					$invoice->saveOrder();
				} else {
					$invoice = null;
				}

				$sendemails = apply_filters( "pmpro_send_checkout_emails", true );

				if ( $sendemails ) { // Send the e-mails only if the flag is set to true
					$child_user                   = get_userdata( $child_user_id );
					$child_user->membership_level = pmpro_getMembershipLevelForUser( $child_user_id );

					//send email to member
					$pmproemail = new PMProEmail();
					$pmproemail->sendCheckoutEmail( $child_user, $invoice );

					//send email to admin
					$pmproemail = new PMProEmail();
					$pmproemail->sendCheckoutAdminEmail( $child_user, $invoice );
				}
			}
		}
	}
}
add_action("pmpro_after_checkout", "pmprosm_after_checkout_children_updated",30,1);

/**
 * Change a user's level and also set the code_id.
 */
function pmprosm_changeMembershipLevelWithCode($level_id, $user_id, $code_id) {
	$child_level = pmpro_getLevel($level_id);

	// Set the start date to NOW() but allow filters.
	$startdate = apply_filters( "pmprosm_checkout_start_date", date( 'Y-m-d H:i:s' ), $user_id, $child_level );

	$custom_level = array(
		'user_id' => $user_id,
		'membership_id' => $level_id,
		'code_id' => $code_id,
		'initial_payment' => $child_level->initial_payment,
		'billing_amount' => $child_level->billing_amount,
		'cycle_number' => $child_level->cycle_number,
		'cycle_period' => $child_level->cycle_period,
		'billing_limit' => $child_level->billing_limit,
		'trial_amount' => $child_level->trial_amount,
		'trial_limit' => $child_level->trial_limit,
		'startdate' => $startdate,
		'enddate' => ''
		);

	return pmpro_changeMembershipLevel( $custom_level, $user_id );
}
/**
 * Get order_id from code_id & user_id.
 */
function pmprosm_getOrderByCodeUser( $code_id, $user_id ) {
	global $wpdb;

	$sqlQuery = "SELECT order_id FROM $wpdb->pmpro_discount_codes_uses WHERE user_id = '" . esc_sql( $user_id ) . "' AND code_id = '". esc_sql( $code_id ) ."' ";
	$order_id = $wpdb->get_var( $sqlQuery );
	return $order_id;

}

/**
 * Add a row to pmpro_discount_codes_uses with a blank order
 */
function pmprosm_addDiscountCodeUse( $user_id, $level_id, $code_id ) {
	global $wpdb;

	$user = get_userdata( $user_id );

	//add blank order
	$morder = new MemberOrder();
	$morder->code = pmpro_getDiscountCode();
	$morder->InitialPayment = 0;
	$morder->Email = $user->user_email;
	$morder->gateway = "free";	//sponsored
	$morder->user_id = $user_id;
	$morder->membership_id = $level_id;
	$morder->status = 'success';
	$morder->notes = __( 'Child account created during checkout of sponsored member.', 'pmpro-sponsored-members' );
	$morder->saveOrder();

	if( ! empty( $morder->id ) ) {
		$code_order_id = $morder->id;
	} else {
		$code_order_id = "";
	}

	global $wpdb;
	$wpdb->query("INSERT INTO $wpdb->pmpro_discount_codes_uses (code_id, user_id, order_id, timestamp) VALUES('" . esc_sql($code_id) . "', '" . esc_sql($user_id) . "', '" . intval($code_order_id) . "', now())");
}

/**
 * Remove a row from pmpro_discount_code_uses.
 */
function pmprosm_removeDiscountCodeUse( $user_id, $code_id ) {
	global $wpdb;

	$wpdb->query("DELETE FROM $wpdb->pmpro_discount_codes_uses WHERE user_id = '" . esc_sql($user_id) . "' AND code_id = '" . esc_sql($code_id) . "'");
}

function pmprosm_pmpro_registration_checks_sponsored_accounts( $okay ) {
	global $pmpro_msg, $pmpro_msgt;

	$level = pmpro_getLevelAtCheckout();
	$level_id = empty( $level->id ) ? false : $level->id;

	// Only if we're adding accounts at checkout.
	$pmprosm_values = pmprosm_getValuesByMainLevel( $level_id );
	if( empty( $pmprosm_values['sponsored_accounts_at_checkout'] ) ) {
		return $okay;
	}

	// Get number of old accounts to test later.
	if( ! empty( $_REQUEST['old_sub_accounts_active'] ) ) {
		$num_old_accounts = count($_REQUEST['old_sub_accounts_active']);
	} else {
		$num_old_accounts = 0;
	}

	// Get seats.
	if( ! empty( $_REQUEST['seats'] ) ) {
		$seats = intval($_REQUEST['seats']);
	} else {
		$seats = 0;
	}

	// How many new accounts?
	$num_new_accounts = $seats - $num_old_accounts;

	// Get account values.
	if( ! empty( $_REQUEST['add_sub_accounts_username'] ) ) {
		$child_usernames = $_REQUEST['add_sub_accounts_username'];
	} else {
		$child_usernames = array();
	}

	if( ! empty( $_REQUEST['add_sub_accounts_first_name'] ) ) {
		$child_first_names = $_REQUEST['add_sub_accounts_first_name'];
	} else {
		$child_first_names = array();
	}

	if( ! empty( $_REQUEST['add_sub_accounts_last_name'] ) ) {
		$child_last_names = $_REQUEST['add_sub_accounts_last_name'];
	} else {
		$child_last_names = array();
	}

	if( ! empty( $_REQUEST['add_sub_accounts_email'] ) ) {
		$child_emails = $_REQUEST['add_sub_accounts_email'];
	} else {
		$child_emails = array();
	}

	if( ! empty( $_REQUEST['add_sub_accounts_password'] ) ) {
		$child_passwords = $_REQUEST['add_sub_accounts_password'];
	} else {
		$child_passwords = array();
	}

	//check that these emails and usernames are unique
	$unique_usernames = array_unique( array_filter( $child_usernames ) );
	$unique_emails = array_unique( array_filter( $child_emails ) );
	$passwords = array_filter( $child_passwords );

	// Expressions to simplify error logic below.
	$creating_accounts = $num_new_accounts > 0;
	$requiring_usernames = ( ! isset( $pmprosm_values['children_hide_username'] ) || $pmprosm_values['children_hide_username'] !== true );
	$usernames_provided = count( $unique_usernames ) >= $num_new_accounts;
	$requiring_passwords = ( ! isset( $pmprosm_values['children_hide_password'] ) || $pmprosm_values['children_hide_password'] !== true );
	$passwords_provided = count( $passwords ) >= $num_new_accounts;
	$emails_provided = count( $unique_emails ) >= $num_new_accounts;

	if( $creating_accounts && $requiring_usernames && ! $usernames_provided ) {
		pmpro_setMessage( esc_html__( "Please enter a username for each new sponsored account." ), "pmpro_error" );
		$okay = false;
	} elseif( $creating_accounts && $requiring_passwords && ! $passwords_provided ) {
		pmpro_setMessage( esc_html__( "Please enter a password for each new sponsored account." ), "pmpro_error" );
		$okay = false;
	} elseif( $creating_accounts && ! $emails_provided ) {
		pmpro_setMessage( esc_html__( "Please enter an email for each new sponsored account." ), "pmpro_error" );
		$okay = false;
	} elseif( count( $unique_usernames ) != count( $child_usernames ) || count( $unique_emails ) != count( $child_emails ) ) {
		pmpro_setMessage( esc_html__( "Each sponsored account must have a unique username and email address." ), "pmpro_error" );
		$okay = false;
	} elseif( count( $child_emails ) + $num_old_accounts > $seats ) {
		pmpro_setMessage( esc_html__( "You have more accounts checked than you are purchasing seats. Increase the number of seats or deactivate some accounts." ), "pmpro_error" );
		$okay = false;
	} elseif( isset( $_REQUEST['username'] ) && in_array( $_REQUEST['username'], $child_usernames ) ) {
		pmpro_setMessage( esc_html__( "A sponsored account must have a different username than the main account." ), "pmpro_error" );
		$okay = false;
	} elseif( isset( $_REQUEST['bemail']) && in_array( $_REQUEST['bemail'], $child_emails ) ) {
		pmpro_setMessage( esc_html__( "A sponsored account must have a different email than the main account." ), "pmpro_error" );
		$okay = false;
	} else {
		foreach( $child_usernames as $child_username ) {
			// Check if child usernames already in use.
			if( username_exists( $child_username ) ) {
					$pmpro_msg = "The username <b>".$child_username."</b> already exists. Please select a different username";
					$pmpro_msgt = "pmpro_error";
					pmpro_setMessage($pmpro_msg,"pmpro_error");
					$okay = false;
			}
		}

		foreach( $child_emails as $child_email ) {
			// Check if child emails already in use.
			if( email_exists( $child_email ) ) {
				$pmpro_msg = "That email <b>".$child_email."</b> already exists. Please select a different email";
				$pmpro_msgt = "pmpro_error";
				pmpro_setMessage( $pmpro_msg, "pmpro_error" );

				$okay = false;
			} elseif( ! is_email( $child_email ) ) {
				$pmpro_msg = "<b>".$child_email."</b> is not a valid email address. Please select a different email";
				$pmpro_msgt = "pmpro_error";
				pmpro_setMessage( $pmpro_msg, "pmpro_error" );

				$okay = false;
			}
		}
	}

	return $okay;
}
add_action( 'pmpro_registration_checks', 'pmprosm_pmpro_registration_checks_sponsored_accounts' );

// Save fields in session for PayPal Express/etc.
function pmprosm_pmpro_paypalexpress_session_vars() {
	if(!empty($_REQUEST['seats']))
		$_SESSION['seats'] = $_REQUEST['seats'];
	else
		$_SESSION['seats'] = "";
	if(!empty($_REQUEST['add_sub_accounts_username']))
		$_SESSION['add_sub_accounts_username'] = $_REQUEST['add_sub_accounts_username'];
	else
		$_SESSION['add_sub_accounts_username'] = "";
	if(!empty($_REQUEST['add_sub_accounts_password']))
		$_SESSION['add_sub_accounts_password'] = $_REQUEST['add_sub_accounts_password'];
	else
		$_SESSION['add_sub_accounts_password'] = "";
	if(!empty($_REQUEST['add_sub_accounts_email']))
		$_SESSION['add_sub_accounts_email'] = $_REQUEST['add_sub_accounts_email'];
	else
		$_SESSION['add_sub_accounts_password'] = "";
	if(!empty($_REQUEST['add_sub_accounts_first_name']))
		$_SESSION['add_sub_accounts_first_name'] = $_REQUEST['add_sub_accounts_first_name'];
	else
		$_SESSION['add_sub_accounts_first_name'] = "";
	if(!empty($_REQUEST['add_sub_accounts_last_name']))
		$_SESSION['add_sub_accounts_last_name'] = $_REQUEST['add_sub_accounts_last_name'];
	else
		$_SESSION['add_sub_accounts_last_name'] = "";
	if(!empty($_REQUEST['old_sub_accounts_active']))
		$_SESSION['old_sub_accounts_active'] = $_REQUEST['old_sub_accounts_active'];
	else
		$_SESSION['old_sub_accounts_active'] = "";
}
add_action( "pmpro_paypalexpress_session_vars", "pmprosm_pmpro_paypalexpress_session_vars" );
add_action( "pmpro_before_send_to_twocheckout", "pmprosm_pmpro_paypalexpress_session_vars", 10, 2 );

// Load fields from session if available.
function pmprosm_init_load_session_vars( $param ) {
	if( empty( $_REQUEST['seats'] ) && ! empty( $_SESSION['seats'] ) ) {
		$_REQUEST['seats'] = $_SESSION['seats'];
		unset( $_SESSION['seats'] );
	}
	if( empty( $_REQUEST['add_sub_accounts_username'] ) && ! empty( $_SESSION['add_sub_accounts_username'] ) ) {
		$_REQUEST['add_sub_accounts_username'] = $_SESSION['add_sub_accounts_username'];
		unset( $_SESSION['add_sub_accounts_username'] );
	}
	if( empty( $_REQUEST['add_sub_accounts_password'] ) && ! empty( $_SESSION['add_sub_accounts_password'] ) ) {
		$_REQUEST['add_sub_accounts_password'] = $_SESSION['add_sub_accounts_password'];
		unset( $_SESSION['add_sub_accounts_password'] );
	}
	if( empty( $_REQUEST['add_sub_accounts_email'] ) && ! empty( $_SESSION['add_sub_accounts_email'] ) ) {
		$_REQUEST['add_sub_accounts_email'] = $_SESSION['add_sub_accounts_email'];
		unset( $_SESSION['add_sub_accounts_email'] );
	}
	if( empty( $_REQUEST['add_sub_accounts_first_name'] ) && ! empty( $_SESSION['add_sub_accounts_first_name'] ) ) {
		$_REQUEST['add_sub_accounts_first_name'] = $_SESSION['add_sub_accounts_first_name'];
		unset( $_SESSION['add_sub_accounts_first_name'] );
	}
	if( empty( $_REQUEST['add_sub_accounts_last_name'] ) && ! empty( $_SESSION['add_sub_accounts_last_name'] ) ) {
		$_REQUEST['add_sub_accounts_last_name'] = $_SESSION['add_sub_accounts_last_name'];
		unset( $_SESSION['add_sub_accounts_last_name'] );
	}
	if( empty( $_REQUEST['old_sub_accounts_active'] ) && ! empty( $_SESSION['old_sub_accounts_active'] ) ) {
		$_REQUEST['old_sub_accounts_active'] = $_SESSION['old_sub_accounts_active'];
		unset( $_SESSION['old_sub_accounts_active'] );
	}

	return $param;
}
add_action( 'pmpro_checkout_preheader', 'pmprosm_init_load_session_vars', 5 );

// Add the 'seats' parameter to the Paypal Express return url so we charge the correct amount
function pmprosm_paypal_express_return_url_parameters( $params ) {
	if( isset( $_REQUEST['seats'] ) ) {
		$params['seats'] = intval( $_REQUEST['seats'] );
	}
	return $params;
}
add_filter( "pmpro_paypal_express_return_url_parameters", "pmprosm_paypal_express_return_url_parameters" );

// Add code and seats fields to profile for admins.
function pmprosm_profile_fields_seats( $user ) {
	global $wpdb;
	$user->membership_level = pmpro_getMembershipLevelForUser( $user->ID );
	if( current_user_can( "edit_users" ) && ! empty( $user->membership_level ) ) {
		
		// Removing a sponsored member?
		if( isset( $_REQUEST['pmprosm_remove_member_id'] ) && isset( $_REQUEST['pmprosm_remove_member_level'] ) ) {
			// Check nonce.
			if( wp_verify_nonce( $_REQUEST['_wpnonce'], 'pmprosm_remove_member' ) ) {
				// Nonce is good. Remove the member.
				$removed = pmprosm_remove_member_from_seat( intval( $_REQUEST['pmprosm_remove_member_id'] ), intval( $_REQUEST['pmprosm_remove_member_level'] ), $user->ID );
				
				if ( $removed !== false ) {
					$member = get_userdata( intval( $_REQUEST['pmprosm_remove_member_id'] ) );
					?>
					<div class="notice notice-success"><p>
					<?php
					echo esc_html( sprintf( __( 'Sponsored User: %s was removed. (Membership Level: %s)', 'pmpro-sponsored-members' ), $member->display_name, intval( $_REQUEST['pmprosm_remove_member_level'] ) ) );
					?>
					</p></div>
					<?php
				}
			}
		}
		?>
		<hr />
		<h2><?php esc_html_e( 'Sponsored Seats', 'pmpro-sponsored-members' ); ?></h2>
		<table class="form-table">
			<?php
				//get the user's sponsor code
				$sponsor_code_id = pmprosm_getCodeByUserID( $user->ID );
				$code = pmprosm_getDiscountCodeByCodeID( $sponsor_code_id );
				if( ! empty( $code ) ) {
					?>
					<tr>
						<th><label for="sponsor_code"><?php esc_html_e( 'Sponsor Code', 'pmpro-sponsored-members' ); ?></label></th>
						<td>
							<?php echo $code->code; ?>
						</td>
					</tr>
					<tr>
						<th><label for="seats"><?php esc_html_e( 'Seats', 'pmpro-sponsored-members' ); ?></label></th>
						<td>
							<?php
								$seats = intval( get_user_meta( $user->ID, "pmprosm_seats", true ) );
							?>
							<input type="text" id="seats" name="seats" size="5" value="<?php echo esc_attr( $seats );?>" />
						</td>
					</tr>
					<?php
					} else { ?>
					<tr>
						<th><label for="sponsor_code"><?php esc_html_e( 'Sponsor Code', 'pmpro-sponsored-members' ); ?></label></th>
						<td><em class="muted"><?php esc_html_e( 'This membership level does not include a sponsor code.', 'pmpro-sponsored-members' ); ?></em></td>
					</tr>
					<?php } ?>

				<?php
					//get the user's parent account
					$parent = pmprosm_getSponsor( $user->ID );
					if( !empty( $parent ) ) {
						?>
						<tr>
							<th><label for="parent"><?php esc_html_e( 'Parent', 'pmpro-sponsored-members' ); ?></label></th>
							<td><a href="<?php echo get_edit_user_link( $parent->ID ); ?>"><?php echo esc_html( $parent->display_name ); ?></a></td>
						</tr>
						<?php
					}
				?>
		</table>
		<?php
			//get members
			$member_ids = pmprosm_getChildren( $user->ID );
			if ( !empty( $member_ids) ) {
            // this was already in profile so don't restrict by 'list_sponsored_accounts' - Keep backward compatability
                echo "<hr />";
                echo pmprosm_display_sponsored_accounts( $member_ids );
			}
	}
}
add_action( 'show_user_profile', 'pmprosm_profile_fields_seats' );
add_action( 'edit_user_profile', 'pmprosm_profile_fields_seats' );

/**
 * Output HTML for list of sponsored members.
 */
function pmprosm_display_sponsored_accounts( $member_ids ) {
    global $current_user;
	
	// Make sure we have something to display.
	if ( empty( $member_ids) ) {
		return '';
	}
	
	// Display sponsored members.
	$count = 0;
	ob_start();
    ?>

    <h2><?php esc_html_e( "Sponsored Members", "pmpro-sponsored-members" );?></h2>
    <div class="pmpro-sponsored-members_children" <?php if( count( $member_ids ) > 4 ) { ?>style="height: 150px; overflow: auto;"<?php } ?>>
        <table class="wp-list-table widefat striped" width="100%" cellpadding="0" cellspacing="0" border="0">
            <thead>
            <tr>
                <th><?php esc_html_e( 'Date', 'pmpro-sponsored-members' ); ?></th>
                <th><?php esc_html_e( 'Name', 'pmpro-sponsored-members' ); ?></th>
                <th><?php esc_html_e( 'Email', 'pmpro-sponsored-members' ); ?></th>
                <th><?php esc_html_e( 'Level', 'pmpro-sponsored-members' ); ?></th>
				<th></th>
            </tr>
            </thead>
            <tbody>
			<?php
			foreach($member_ids as $member_id)
			{
				$member = get_userdata($member_id);
				if(empty($member)) {
					continue;
				}
				$member->membership_level = pmpro_getMembershipLevelForUser($member_id);
				
				// Figure out URL based on where we are.
				if ( $_SERVER['SCRIPT_NAME'] == '/wp-admin/user-edit.php' ) {
					// Editing another user in the admin.
					$qargs = array(
						'user_id' => intval($_REQUEST['user_id']),
						'pmprosm_remove_member_id' => $member->ID,
						'pmprosm_remove_member_level' => $member->membership_level->id
					);
					$remove_url = add_query_arg( $qargs, admin_url('user-edit.php') );
				} elseif ( $_SERVER['SCRIPT_NAME'] == '/wp-admin/profile.php' ) {
					// Editing yourself on profile.php
					$qargs = array(
						'pmprosm_remove_member_id' => $member->ID,
						'pmprosm_remove_member_level' => $member->membership_level->id
					);
					$remove_url = add_query_arg( $qargs, admin_url('profile.php') );
				} else {
					// Assume frontend account page.
					$qargs = array(
						'user_id' => $current_user->ID,
						'pmprosm_remove_member_id' => $member->ID,
						'pmprosm_remove_member_level' => $member->membership_level->id
					);
					$remove_url = add_query_arg( $qargs, pmpro_url('account') );
				}
				$remove_url = wp_nonce_url( $remove_url, 'pmprosm_remove_member' );
				?>
                <tr<?php if($count++ % 2 == 1) { ?> class="alternate"<?php } ?>>
                    <td><?php echo date(get_option("date_format"), $member->membership_level->startdate); ?></td>
                    <td><?php echo esc_html( $member->display_name ); ?></td>
                    <td>
						<?php if ( current_user_can( 'edit_users' ) ) { ?>
							<a href="<?php echo get_edit_user_link($member_id); ?>"><?php echo esc_html( $member->user_email ); ?></a>
						<?php } else { ?>
							<?php echo esc_html( $member->user_email ); ?>
						<?php } ?>
                    </td>
                    <td><?php echo esc_html( $member->membership_level->name ); ?></td>
					<td>
						<?php
						$delete_text = esc_attr__('Are you sure you want to remove this member?', 'pmpro-sponsored-members' );
						?>
						<a href="<?php echo $remove_url; ?>" onclick="<?php echo esc_js('return confirm("' . $delete_text . '");'); ?>"><?php _e( 'Remove', 'pmpro-sponsored-members'); ?></a>
					</td>
                </tr>
				<?php
			}
			?>
            </tbody>
        </table>
    </div>
    <?php
	$content = ob_get_contents();
	ob_end_clean();
	return $content;
}

/**
 * Cancel a member and free up the discount code use.
 * @since 0.9.1
 */
function pmprosm_remove_member_from_seat( $child_id, $level_id, $parent_id ) {
	if( pmprosm_isSponsoredLevel( $level_id ) ) {
		// Check if the parent is really the parent of this child.
		$real_parent = pmprosm_getSponsor( $child_id );
		if ( empty( $real_parent) || $real_parent->ID !== $parent_id ) {
			return false;
		}
		
		// We're good. Remove.
		$removed = pmpro_cancelMembershipLevel( $level_id, $child_id );
	    if( $removed !== false ) {
	        $code_id = pmprosm_getCodeByUserID( $parent_id );
	        pmprosm_removeDiscountCodeUse( $child_id, $code_id );
	        return true;
	    }
    }
	
	return false;
}

/**
 * Save seats on profile save.
 */
function pmprosm_profile_update_seats( $user_id ) {
	//make sure they can edit
	if ( !current_user_can( 'edit_user', $user_id ) ) {
		return false;
	}

	//only let admin's edit the seats
	if( current_user_can( "manage_options" ) && isset( $_POST['seats'] ) ) {
		//update user meta
		update_user_meta( $user_id, "pmprosm_seats", intval($_POST['seats']) );

		//update code
		global $wpdb;
		$code_id = pmprosm_getCodeByUserID( $user_id );
		$sqlQuery = "UPDATE $wpdb->pmpro_discount_codes SET uses = '" . intval($_POST['seats']) . "' WHERE id = '" . esc_sql( $code_id ) . "' LIMIT 1";
		$wpdb->query($sqlQuery);
	}
}
add_action( 'profile_update', 'pmprosm_profile_update_seats' );

/**
 * Show seats on the account page and show if they have been claimed.
 */
function pmprosm_the_content_account_page( $content ) {
	global $post, $pmpro_pages, $current_user, $wpdb;

	if( ! is_admin() && isset( $post->ID ) && $post->ID == $pmpro_pages['account'] ) {
		//what's their code?
		$code_id = pmprosm_getCodeByUserID( $current_user->ID );

		if ( isset( $current_user->membership_level->ID ) ) {
			$pmprosm_values = pmprosm_getValuesByMainLevel( $current_user->membership_level->ID );
		} else {
			$pmprosm_values = '';
		}

		if(! empty( $code_id ) && ! empty( $pmprosm_values ) ) {
			$code = pmprosm_getDiscountCodeByCodeID( $code_id );

			if( ! is_array( $pmprosm_values['sponsored_level_id'] ) ) {
				$sponsored_level_ids = array($pmprosm_values['sponsored_level_id']);
			} else {
				$sponsored_level_ids = $pmprosm_values['sponsored_level_id'];
			}

			// No sponsored levels to use codes for.
			if( empty( $sponsored_level_ids ) || empty( $sponsored_level_ids[0] ) ) {
				return $content;
			}

			$code_urls = pmprosm_get_checkout_urls( $code );

			ob_start();

			$limit = 1;
			if( isset( $pmprosm_values['max_seats'] ) ) {
				$limit = $pmprosm_values['max_seats'];
			}

			// Removing a sponsored member?
			if( isset( $_REQUEST['pmprosm_remove_member_id'] ) && isset( $_REQUEST['pmprosm_remove_member_level'] ) ) {
				// Check nonce.
				if( wp_verify_nonce( $_REQUEST['_wpnonce'], 'pmprosm_remove_member' ) ) {
					// Nonce is good. Remove the member.
					$removed = pmprosm_remove_member_from_seat( intval( $_REQUEST['pmprosm_remove_member_id'] ), intval( $_REQUEST['pmprosm_remove_member_level'] ), $current_user->ID );
					
					if ( $removed !== false ) {
						$member = get_userdata( intval( $_REQUEST['pmprosm_remove_member_id'] ) );
						$removed_message = sprintf( __( 'Sponsored User: %s was removed. (Membership Level: %s)', 'pmpro-sponsored-members' ), $member->display_name, intval( $_REQUEST['pmprosm_remove_member_level'] ) );
					}
				}
			}

			// Get members.
			$member_ids = pmprosm_getChildren($current_user->ID);
			?>
			<div id="pmpro_account-sponsored" class="pmpro_box">

				<h2><?php esc_html_e( "Sponsored Seats", "pmpro-sponsored-members" );?></h2>
                <?php if (empty($pmprosm_values['hide_display_discount_code']) || $pmprosm_values['hide_display_discount_code'] === false ) { ?>
                    <p><?php printf(esc_html__("Give this code to your sponsored members to use at checkout: %s", "pmpro-sponsored-members"), '<strong>' . $code->code . '</strong>');?></p>
                    <?php if(count($code_urls) > 1) { ?>
                        <p><?php esc_html_e("Or provide one of these direct links to register:", "pmpro-sponsored-members");?></p>
                    <?php } else { ?>
                        <p><?php esc_html_e("Or provide this direct link to register:", "pmpro-sponsored-members");?></p>
                    <?php } ?>

                    <ul>
                        <?php foreach($code_urls as $code_url) { ?>
                            <li><?php echo esc_html( $code_url['name'] );?>: <strong><a target="_blank" href="<?php echo esc_url( $code_url['url'] );?>"><?php echo esc_url(  $code_url['url'] );?></a></strong></li>
                        <?php } ?>
                    </ul>
                <?php } // hide_display_discount_code ?>

                <div class="pmpro_message pmpro_default">
					<?php if(empty($code->uses)) { ?>
						<?php esc_html_e( "This code has unlimited uses.", "pmpro-sponsored-members" );?>
					<?php } else { ?>
						<?php printf( esc_html__("%s/%s uses.", "pmpro-sponsored-members" ), count( $member_ids ), $code->uses );?>
					<?php } ?>
				</div>
				<?php
                    // Did we remove a child earlier?
					if ( ! empty( $removed_message ) ) {
					?>
					<div class="pmpro_message pmpro_success"><?php echo esc_html( $removed_message ); ?></div>
					<?php
					}
					
					// use same account display as in admin
                    if ( ! empty( $member_ids ) ) {
                        echo "<hr />";
                        echo pmprosm_display_sponsored_accounts( $member_ids );
                    }
                ?>
			</div> <!-- end pmpro_account-sponsored -->
			<?php

			$temp_content = ob_get_contents();
			ob_end_clean();

			$content = str_replace( '<!-- end pmpro_account-profile -->', '<!-- end pmpro_account-profile -->' . $temp_content, $content );
		}
	}

	return $content;
}
add_filter( "the_content", "pmprosm_the_content_account_page", 30 );

// Make sure we declare this in global space.
global $pmprosm_user_sponsors;

/**
 * Get a user's sponsoring member.
 */
function pmprosm_getSponsor( $user_id, $force = false) {
	global $wpdb, $pmprosm_user_sponsors;

	if( ! empty( $pmprosm_user_sponsors[$user_id] ) && ! $force ) {
		return $pmprosm_user_sponsors[$user_id];
	}

	//make sure this user has one of the sponsored levels
	$user_level = pmpro_getMembershipLevelForUser( $user_id );
	if( empty( $user_level ) || ! pmprosm_isSponsoredLevel( $user_level->id ) ) {
		$pmprosm_user_sponsors[$user_id] = false;
		return $pmprosm_user_sponsors[$user_id];
	}

	//what code did this user_id sign up for?
	$sqlQuery = "SELECT code_id FROM $wpdb->pmpro_discount_codes_uses WHERE user_id = '" . esc_sql( $user_id ) . "' ORDER BY id DESC";
	$code_id = $wpdb->get_var( $sqlQuery );

	// Found a code?
	if( empty( $code_id ) ) {
		$pmprosm_user_sponsors[$user_id] = false;
		return $pmprosm_user_sponsors[$user_id];
	}

	//okay find sponsor
	$sponsor_user_id = pmprosm_getUserByCodeID( $code_id );

	$pmprosm_user_sponsors[$user_id] = get_userdata( $sponsor_user_id );
	return $pmprosm_user_sponsors[$user_id];
}

/**
 * Add code to confirmation email.
 */
function pmprosm_pmpro_email_body( $body, $pmpro_email ) {
	global $wpdb, $pmprosm_sponsored_account_levels;

	//only checkout emails, not admins
	if(strpos($pmpro_email->template, "checkout") !== false && strpos($pmpro_email->template, "admin") === false && strpos($pmpro_email->template, "debug") === false) {
		//get the user_id from the email
		$user_id = $wpdb->get_var("SELECT ID FROM $wpdb->users WHERE user_email = '" . esc_sql( $pmpro_email->data['user_email'] ) . "' LIMIT 1");
		$level_id = $pmpro_email->data['membership_id'];
		$code_id = pmprosm_getCodeByUserID( $user_id );

		if( ! empty( $user_id ) && ! empty( $code_id ) && pmprosm_isMainLevel( $level_id ) ) {
			// Get code.
			$code = pmprosm_getDiscountCodeByCodeID( $code_id );

			// Get sponsored levels.
			$pmprosm_values = pmprosm_getValuesByMainLevel( $level_id );
			if( ! is_array( $pmprosm_values['sponsored_level_id'] ) ) {
				$sponsored_level_ids = array($pmprosm_values['sponsored_level_id']);
			} else {
				$sponsored_level_ids = $pmprosm_values['sponsored_level_id'];
			}

			// No sponsored levels to use codes for.
			if( empty( $sponsored_level_ids ) || empty( $sponsored_level_ids[0] ) ) {
				return $body;
			}

			if(isset($pmprosm_values['add_created_accounts_to_confirmation_email']) && $pmprosm_values['add_created_accounts_to_confirmation_email'] === true) {
				$children = pmprosm_getChildren( $user_id );
				if( ! empty( $children ) ) {
					$message = "<p>" . esc_html__( "Accounts created at checkout:", "pmpro-sponsored-members" ) . "<br />";
					$message .= "<ul>";
					foreach ( $children as $child_id ) {
						$child = get_userdata($child_id);
						$message .= "<li>" . esc_html( $child->display_name ) . " ( " . $child->user_email . " ) </li>";
					}
					$message .= "</ul>";

					$body = $message . "<hr />" . $body;
				}
			}

			// Check if we should update confirmation email.
			if ( isset( $pmprosm_values['hide_display_discount_code'] ) && $pmprosm_values['hide_display_discount_code'] === true ) {
				return $body;
			}

			// Check if we should update confirmation email.
			if( isset( $pmprosm_values['add_code_to_confirmation_email'] ) && $pmprosm_values['add_code_to_confirmation_email'] === false ) {
				return $body;
			}

			// Figure out urls for code.
			$code_urls = array();
			$pmpro_levels = pmpro_getAllLevels(true, true);
			foreach( $sponsored_level_ids as $sponsored_level_id ) {
				$level_name = $pmpro_levels[$sponsored_level_id]->name;
				$code_urls[] = array("name"=>$level_name, "url"=>pmpro_url( "checkout", "?level=" . $sponsored_level_id . "&discount_code=" . $code->code ) );
			}

			// Build message.
			$message = "<p>" . sprintf( esc_html__("Give this code to your sponsored members to use at checkout: %s", "pmpro-sponsored-members" ), $code->code ) . "<br />";

			if( count( $code_urls ) > 1 ) {
				$message .= esc_html__("Or provide one of these direct links to register:", "pmpro-sponsored-members") . "</p>";
			} else {
				$message .= esc_html__("Or provide this direct link to register:", "pmpro-sponsored-members") . "</p>";
			}

			$message .= "<ul>";
			foreach( $code_urls as $code_url ) {
				$message .= "<li>" . esc_html( $code_url['name'] ) . ": <strong>" . esc_url( $code_url['url'] ) . "</strong></li>";
			}
			$message .= "</ul>";

			$body = $message . "<hr />" . $body;
		}
	}

	return $body;
}
add_filter( "pmpro_email_body", "pmprosm_pmpro_email_body", 10, 2 );

/**
 * Function to add links to the plugin row meta
 */
function pmprosm_plugin_row_meta( $links, $file ) {
	if( strpos( $file, 'pmpro-sponsored-members.php' ) !== false ) {
		$new_links = array(
			'<a href="' . esc_url('https://www.paidmembershipspro.com/add-ons/pmpro-sponsored-members/')  . '" title="' . esc_attr__( 'View Documentation', 'pmpro-sponsored-members' ) . '">' . esc_html__( 'Docs', 'pmpro-sponsored-members' ) . '</a>',
			'<a href="' . esc_url('https://paidmembershipspro.com/support/') . '" title="' . esc_attr__( 'Visit Customer Support Forum', 'pmpro-sponsored-members' ) . '">' . esc_html__( 'Support', 'pmpro-sponsored-members' ) . '</a>',
		);
		$links = array_merge( $links, $new_links );
	}
	return $links;
}
add_filter( 'plugin_row_meta', 'pmprosm_plugin_row_meta', 10, 2 );

/**
 * This function generates direct checkout links from a discount code for future uses.
 * @param object $code This is the discount code object.
 * @since
 * @return array.
 */
function pmprosm_get_checkout_urls( $code ) {
	global $wpdb;

	$checkout_urls = array();

	if ( ! empty( $code ) && is_object( $code ) ) {
		$code_id = $code->id;
	} else {
		return;
	}

	$sql_code = "SELECT level_id, name FROM $wpdb->pmpro_discount_codes_levels c INNER JOIN $wpdb->pmpro_membership_levels l ON c.level_id = l.id WHERE c.code_id =" . esc_sql( $code_id );

	$levels_id = $wpdb->get_results( $sql_code );

	foreach ( $levels_id as $value ) {
		$checkout_urls[] = array( "name" => $value->name, "url" => pmpro_url( "checkout", "?level=" . $value->level_id . "&discount_code=" . $code->code ) );
	}

	return $checkout_urls;
}

/**
 * Add sponsor's code to memberslist CSV export
 */
function pmprosm_pmpro_members_list_csv_extra_columns( $columns ) {
	$columns['sponsorcode'] = 'pmprosm_pmpro_members_list_csv_sponsorcode';

	return $columns;
}
add_filter( 'pmpro_members_list_csv_extra_columns', 'pmprosm_pmpro_members_list_csv_extra_columns' );

/**
 * Call back to get the sponsor code
 */
function pmprosm_pmpro_members_list_csv_sponsorcode( $user ) {
	$sponsor_code_id = pmprosm_getCodeByUserID( $user->ID );
	$sponsor_code = pmprosm_getDiscountCodeByCodeID( $sponsor_code_id );
	if( !empty( $sponsor_code ) ) {
		return $sponsor_code->code;
	} else {
		return '';
	}
}

/**
 * Add Sponsor or Sponsor Code to the Member's List Table
 * Add Header
 */
function pmprosm_pmpro_memberslist_extra_cols_header() {
?>
<th><?php esc_html_e( 'Sponsor/Code', 'pmpro-sponsored-members' ); ?></th>
<?php
}
add_action( 'pmpro_memberslist_extra_cols_header', 'pmprosm_pmpro_memberslist_extra_cols_header' );

/**
 * Add Sponsor or Sponsor Code to the Member's List Table
 * Add Column Content
 */
function pmprosm_pmpro_memberslist_extra_cols_body( $theuser ) {
	$sponsor_code_id = pmprosm_getCodeByUserID( $theuser->ID );
	$sponsor_code = pmprosm_getDiscountCodeByCodeID( $sponsor_code_id );
	$sponsor = pmprosm_getSponsor( $theuser->ID );
?>
<td>
	<?php
		if( !empty( $sponsor) ) {
			$user_link = '<a href="' . add_query_arg('user_id', $sponsor->ID, admin_url('user-edit.php') ) . '">' . esc_html( $sponsor->user_login ) . '</a>';
			printf( __( 'Sponsored by %s', 'pmpro-sponsored-members' ), $user_link );
		}
		if( !empty( $sponsor_code ) ) {
			echo '<a href="' . add_query_arg( array( 'page' => 'pmpro-discountcodes', 'edit' => $sponsor_code_id ), admin_url( 'admin.php' ) ) . '">' . esc_html( $sponsor_code->code ) . '</a>';
		}
	?>
</td>
<?php
}
add_action( 'pmpro_memberslist_extra_cols_body', 'pmprosm_pmpro_memberslist_extra_cols_body' );