FileMaster

use SureTriggers\Integrations\AutomateAction;
use SureTriggers\Traits\SingletonLoader;

/**
 * AffiliateLinkCustomer
 *
 * @category AffiliateLinkCustomer
 * @package  SureTriggers
 * @author   BSF <username@example.com>
 * @license  https://www.gnu.org/licenses/gpl-3.0.html GPLv3
 * @link     https://www.brainstormforce.com/
 * @since    1.0.0
 */
class AffiliateLinkCustomer extends AutomateAction {

/**
	 * Integration type.
	 *
	 * @var string
	 */
	public $integration = 'AffiliateWP';

/**
	 * Action name.
	 *
	 * @var string
	 */
	public $action = 'affiliate_link_customer';

use SingletonLoader;

/**
	 * Register a action.
	 *
	 * @param array $actions actions.
	 * @return array
	 */
	public function register( $actions ) {
		$actions[ $this->integration ][ $this->action ] = [
			'label'    => __( 'Link Customer to an Affiliate for Lifetime Commissions', 'suretriggers' ),
			'action'   => $this->action,
			'function' => [ $this, 'action_listener' ],
		];
		return $actions;
	}

/**
	 * Action listener.
	 *
	 * @param int   $user_id user_id.
	 * @param int   $automation_id automation_id.
	 * @param array $fields fields.
	 * @param array $selected_options selectedOptions.
	 * @psalm-suppress UndefinedMethod
	 * @throws Exception Exception.
	 * 
	 * @return array|bool|void
	 */
	public function _action_listener( $user_id, $automation_id, $fields, $selected_options ) {
		
		$lifetime_user_name = $selected_options['affiliate_id'];
		$affiliate_user     = get_user_by( 'ID', $lifetime_user_name );
		$user_email         = $selected_options['customer_email'];

if ( ! function_exists( 'affiliate_wp' ) || ! function_exists( 'affwp_get_customer' ) || ! function_exists( 'affwp_add_customer' ) || ! function_exists( 'affiliate_wp_lifetime_commissions' ) ) {
			return [
				'status'  => 'error',
				'message' => 'AffiliateWP functions not found.',
			];
		}

if ( is_email( $user_email ) ) {
			$user = get_user_by( 'email', $user_email );
			if ( $user ) {
				$user_ID = $user->ID;
				if ( $affiliate_user ) {
					// Prevent affiliates from setting themselves as a lifetime customer.
					if ( $affiliate_user->ID === $user_ID ) {
						return [
							'status'  => 'error',
							'message' => 'Can not add affiliate themselves as a lifetime customer',
						];
					}

$customer = affiliate_wp()->customers->get_by( 'user_id', $user_ID );

// If the customer does not exist, create it.
					if ( ! $customer ) {
						$customer = affwp_get_customer(
							affwp_add_customer(
								[
									'user_id' => $user_ID,
									'email'   => $user->user_email,
								] 
							) 
						);
					}

$lifetime_customer = affiliate_wp_lifetime_commissions()
					->lifetime_customers->get_by( 'affwp_customer_id', $customer->customer_id );
					$affiliate         = '';

$affiliate = affiliate_wp()->affiliates->get_by( 'affiliate_id', $affiliate_user->ID );

// If the affiliate was unset, and this user is currently a customer, delink the affiliate and bail.
					if ( $lifetime_customer && ! $affiliate ) {
						affiliate_wp_lifetime_commissions()
						->lifetime_customers->delete( $lifetime_customer->lifetime_customer_id );
						return [
							'status'  => 'error',
							'message' => 'Affiliate was not set.',
						];
					}

// Add a new lifetime customer if the provided lifetime customer does not exist.
					if ( ! $lifetime_customer ) {
						affiliate_wp_lifetime_commissions()->lifetime_customers->add(
							[
								'affwp_customer_id' => $customer->customer_id,
								'affiliate_id'      => $affiliate->affiliate_id,
							] 
						);
						return affiliate_wp_lifetime_commissions()
						->lifetime_customers->get_by( 'affwp_customer_id', $customer->customer_id );
						// Otherwise, confirm the affiliate is actually a new affiliate before changing.
					} elseif ( $affiliate->affiliate_id !== $lifetime_customer->affiliate_id ) {
						affiliate_wp_lifetime_commissions()->lifetime_customers->update( 
							$lifetime_customer->affwp_customer_id,
							[
								'affiliate_id' => $affiliate->affiliate_id,
							] 
						);
						return affiliate_wp_lifetime_commissions()
						->lifetime_customers->get_by( 'affwp_customer_id', $customer->customer_id );
					}
				} else {
					return [
						'status'  => 'error',
						'message' => 'Affiliate User not exists.',
					];
				}
			} else {
				return [
					'status'  => 'error',
					'message' => 'User not exists.',
				];
			}
		} else {
			return [
				'status'  => 'error',
				'message' => 'Please enter valid email address.',
			];
		}
	}
}

AffiliateLinkCustomer::get_instance();

Back

Edit File: affiliate-link-customer.php