context;
$posted_affiliate = $_POST[ $context . '_affiliate'];
$affiliate_selection = $this->get_affiliate_selection();
// Input field. Accepts either an affiliate ID or username
if ( 'input' === $affiliate_selection ) {
if ( isset( $posted_affiliate ) && $posted_affiliate ) {
if ( is_numeric( $posted_affiliate ) ) {
$affiliate_id = $posted_affiliate;
} elseif ( is_string( $posted_affiliate ) ) {
// get affiliate ID from username
$user = get_user_by( 'login', sanitize_text_field( urldecode( $posted_affiliate ) ) );
if ( $user ) {
$affiliate_id = affwp_get_affiliate_id( $user->ID );
}
}
}
} else {
// select menu
if ( isset( $posted_affiliate ) && $posted_affiliate ) {
$affiliate_id = $posted_affiliate;
}
}
// Return the affiliate ID.
return absint( $affiliate_id );
}
/**
* Get affiliate selection
* @since 1.0.3
*/
public function get_affiliate_selection() {
$affiliate_selection = affiliate_wp()->settings->get( 'checkout_referrals_affiliate_selection' );
return $affiliate_selection;
}
/**
* Validates an affiliate
*
* @since 1.0.3
* @param $affiliate $affiliate username or ID of affiliate
*/
public function is_valid_affiliate( $affiliate = '' ) {
// set flag to false
$valid_affiliate = false;
if ( is_numeric( $affiliate ) ) {
// affiliate ID provided
if ( affwp_is_active_affiliate( $affiliate ) ) {
$valid_affiliate = true;
}
} else {
// username provided. Uppercase or lowercase usernames are ok
if ( affwp_is_active_affiliate( affiliate_wp()->tracking->get_affiliate_id_from_login( $affiliate ) ) ) {
$valid_affiliate = true;
}
}
return $valid_affiliate;
}
/**
* Error messages
*
* @since 1.0.3
*/
public function get_error( $affiliate = '' ) {
// Whether an affiliate is required to be selected or entered
$require_affiliate = affiliate_wp()->settings->get( 'checkout_referrals_require_affiliate' );
// either input or select menu
$affiliate_selection = $this->get_affiliate_selection();
// the affiliate that was submitted
$affiliate_submitted = isset( $affiliate ) && $affiliate ? $affiliate : '';
$error = '';
/**
* Affiliate is required but not affiliate was selected/entered
*/
if ( $require_affiliate && ! $affiliate_submitted ) {
if ( 'input' === $affiliate_selection ) {
// input field
$error = __( 'Please enter an affiliate', 'affiliatewp-checkout-referrals' );
} else {
// select menu
$error = __( 'Please select an affiliate', 'affiliatewp-checkout-referrals' );
}
} else {
/**
* Validate the affiliate submitted
* Set error if affiliate was submitted but the affiliate is invalid
*/
if ( $affiliate_submitted && ! $this->is_valid_affiliate( $affiliate_submitted ) ) {
$error = __( 'Please enter a valid affiliate', 'affiliatewp-checkout-referrals' );
}
}
if ( $error ) {
return apply_filters( 'affwp_checkout_referrals_require_affiliate_error', $error );
} else {
return false;
}
}
}