RESTRICT_CONTENT_ADD_MEMBERSHIP_LEVEL
Class RESTRICT_CONTENT_ADD_MEMBERSHIP_LEVEL
Source Source
File: src/integrations/restrict-content/actions/restrict-content-add-membership-level.php
class RESTRICT_CONTENT_ADD_MEMBERSHIP_LEVEL { /** * Integration code * @var string */ public static $integration = 'RC'; private $action_code; private $action_meta; /** * Set up Automator action constructor. */ public function __construct() { $this->action_code = 'RCADDMEMBERSHIPLEVEL'; $this->action_meta = 'RCMEMBERSHIPLEVEL'; $this->define_action(); } /** * Define and register the action by pushing it into the Automator object */ public function define_action() { $action = [ 'author' => 'Uncanny Automator', 'support_link' => Automator()->get_author_support_link( $this->action_code, 'integration/restrict-content/' ), 'integration' => self::$integration, 'code' => $this->action_code, /* translators: Logged-in trigger - Popup Maker */ 'sentence' => sprintf( esc_attr__( 'Add the user to {{a membership level:%1$s}}', 'uncanny-automator' ), $this->action_meta ), /* translators: Logged-in trigger - Popup Maker */ 'select_option_name' => esc_attr__( 'Add the user to {{a membership level}}', 'uncanny-automator' ), 'priority' => 11, 'accepted_args' => 3, 'execution_function' => [ $this, 'add_rcp_membership' ], 'options_group' => [ $this->action_meta => [ Automator()->helpers->recipe->restrict_content->options->get_membership_levels( null, $this->action_meta, [ 'any' => false ] ), Automator()->helpers->recipe->field->text_field( 'RCMEMBERSHIPEXPIRY', esc_attr__( 'Expiry date', 'uncanny-automator' ), true, 'text', '', false, esc_attr__( 'Leave empty to use expiry settings from the membership level, or type a specific date in the format YYYY-MM-DD', 'uncanny-automator' ) ), ], ], ]; Automator()->register->action( $action ); } /** * Validation function when the trigger action is hit * * @param $user_id * @param $action_data * @param $recipe_id */ public function add_rcp_membership( $user_id, $action_data, $recipe_id, $args ) { $level_id = absint( $action_data['meta'][ $this->action_meta ] ); $expiry_date = Automator()->parse->text( $action_data['meta']['RCMEMBERSHIPEXPIRY'], $recipe_id, $user_id, $args ); // Get all the active membership level IDs. $level_ids = rcp_get_membership_levels( [ 'status' => 'active', 'fields' => 'id' ] ); if ( empty( $level_ids ) ) { Automator()->complete_action( $user_id, $action_data, $recipe_id, __( 'You must have at least one active membership level.', 'rcp' ) ); return; } $customer = rcp_get_customer_by_user_id( $user_id ); $newest_time = current_time( 'timestamp' ); $created_date = date( 'Y-m-d H:i:s', $newest_time ); // Create a new customer record if one does not exist. if ( empty( $customer ) ) { $customer_id = rcp_add_customer( [ 'user_id' => absint( $user_id ), 'date_registered' => $created_date, ] ); } else { $customer_id = $customer->get_id(); } // Now add the membership. /* * For the time always active status. */ $status = 'active'; $membership_args = [ 'customer_id' => absint( $customer_id ), 'user_id' => $user_id, 'object_id' => ! empty( $level_id ) ? $level_id : $level_ids[ array_rand( $level_ids ) ], // specified or random membership level ID 'status' => $status, 'created_date' => $created_date, 'gateway' => 'manual', 'subscription_key' => rcp_generate_subscription_key(), ]; if ( ! empty( $expiry_date ) ) { $membership_args['expiration_date'] = date( 'Y-m-d H:i:s', strtotime( $expiry_date ) ); } $membership_id = rcp_add_membership( $membership_args ); // Add membership meta to designate this as a generated record so we can deleted it later. rcp_add_membership_meta( $membership_id, 'rcp_generated_via_UA', $recipe_id ); $membership = rcp_get_membership( $membership_id ); // Generate a transaction ID. $auth_key = defined( 'AUTH_KEY' ) ? AUTH_KEY : ''; $transaction_id = strtolower( md5( $membership_args['subscription_key'] . date( 'Y-m-d H:i:s' ) . $auth_key . uniqid( 'rcp', true ) ) ); // Create a corresponding payment record. $payment_args = array( 'subscription' => rcp_get_subscription_name( $membership_args['object_id'] ), 'object_id' => $membership_args['object_id'], 'date' => $membership_args['created_date'], 'amount' => $membership->get_initial_amount(), 'subtotal' => $membership->get_initial_amount(), 'user_id' => $user_id, 'subscription_key' => $membership_args['subscription_key'], 'transaction_id' => $transaction_id, 'status' => 'pending' == $membership_args['status'] ? 'pending' : 'complete', 'gateway' => 'manual', 'customer_id' => $customer_id, 'membership_id' => $membership_id, ); $rcp_payments = new \RCP_Payments(); $payment_id = $rcp_payments->insert( $payment_args ); // Add payment meta to designate this as a generated record so we can delete it later. $rcp_payments->add_meta( $payment_id, 'rcp_generated_via_UA', $recipe_id ); Automator()->complete_action( $user_id, $action_data, $recipe_id ); } }
Expand full source code Collapse full source code View on Github
Methods Methods
- __construct — Set up Automator action constructor.
- add_rcp_membership — Validation function when the trigger action is hit
- define_action — Define and register the action by pushing it into the Automator object