action_name = __( 'Assign Membership Plan', 'wp-marketing-automations-pro' ); $this->action_desc = __( 'This action assigns/ update the plan of a user\'s active membership. If no active membership exists a new membership will be created', 'wp-marketing-automations-pro' ); $this->required_fields = array( 'user_id', 'to_plan', 'from_plan' ); $this->support_v2 = true; } public static function get_instance() { if ( null === self::$ins ) { self::$ins = new self(); } return self::$ins; } /** * Localize data for html fields for the current action. */ public function admin_enqueue_assets() { if ( BWFAN_Common::is_load_admin_assets( 'automation' ) ) { $data = $this->get_view_data(); BWFAN_Core()->admin->set_actions_js_data( $this->get_class_slug(), 'membership_plans', $data ); } } public function get_view_data() { $plans = wc_memberships_get_membership_plans(); $plans_array = []; foreach ( $plans as $plan ) { $plans_array[ $plan->get_id() ] = $plan->get_name(); } return $plans_array; } /** * Show the html fields for the current action. */ public function get_view() { ?> ID : 0; } $data_to_set['user_id'] = $user_id; return $data_to_set; } /** * Execute the current action. * Return 3 for successful execution , 4 for permanent failure. * * @param $action_data * * @return array */ public function execute_action( $action_data ) { $this->set_data( $action_data['processed_data'] ); $result = $this->process(); if ( true === $result ) { return array( 'status' => 3, 'message' => 'Membership plan is assigned / updated to user\'s membership', ); } return $result; } /** * Process and do the actual processing for the current action. * This function is present in every action class. */ public function process() { $is_required_fields_present = $this->check_fields( $this->data, $this->required_fields ); if ( false === $is_required_fields_present ) { return $this->show_fields_error(); } $new_plan_id = absint( $this->data['to_plan'] ); $existing_plan_id = absint( $this->data['from_plan'] ); $user = get_userdata( $this->data['user_id'] ); if ( false === $user ) { return array( 'status' => 4, 'message' => __( 'User does not exists', 'wp-marketing-automations-pro' ), ); } $selected_plan = $new_plan_id ? wc_memberships_get_membership_plan( $this->data['to_plan'] ) : false; // Selected plan must exists in order to execute this action if ( false === $selected_plan ) { return array( 'status' => 4, 'message' => __( 'Membership Plan does not exists', 'wp-marketing-automations-pro' ), ); } $new_membership_by_plan = wc_memberships_get_user_membership( $user->ID, $selected_plan ); // Plan is already included in one of the user's membership if ( $new_membership_by_plan ) { return array( 'status' => 4, 'message' => __( 'Membership plan is already included in user\'s membership', 'wp-marketing-automations-pro' ), ); } $existing_membership = $existing_plan_id ? wc_memberships_get_user_membership( $user->ID, $existing_plan_id ) : false; if ( $existing_membership ) { $membership_updated = $this->change_membership_plan( $existing_membership, $selected_plan ); if ( ! $membership_updated ) { return array( 'status' => 4, 'message' => __( 'Unable to update plan to user\'s membership', 'wp-marketing-automations-pro' ), ); } return true; } else { $membership_created = $this->create_new_membership( $user, $selected_plan ); if ( ! $membership_created ) { return array( 'status' => 4, 'message' => __( 'Unable to assign plan to user', 'wp-marketing-automations-pro' ), ); } return true; } } public function process_v2() { $new_plan_id = absint( $this->data['to_plan'] ); $user = get_userdata( $this->data['user_id'] ); if ( false === $user ) { return $this->skipped_response( __( 'User does not exists', 'wp-marketing-automations-pro' ) ); } $selected_plan = $new_plan_id ? wc_memberships_get_membership_plan( $this->data['to_plan'] ) : false; // Selected plan must exists in order to execute this action if ( false === $selected_plan ) { return $this->skipped_response( __( 'Membership Plan does not exists', 'wp-marketing-automations-pro' ) ); } $new_membership_by_plan = wc_memberships_get_user_membership( $user->ID, $selected_plan ); // Plan is already included in one of the user's membership if ( $new_membership_by_plan ) { return $this->success_message( __( 'Membership plan is already included in user\'s membership.', 'wp-marketing-automations-pro' ) ); } $membership_created = $this->create_new_membership( $user, $selected_plan ); if ( ! $membership_created ) { return $this->skipped_response( __( 'Unable to assign plan to user', 'wp-marketing-automations-pro' ) ); } return $this->success_message( __( 'Plan Assigned.', 'wp-marketing-automations-pro' ) ); } /** * Change membership plan of User Membership * * @param WC_Memberships_User_Membership $membership * @param WC_Memberships_Membership_Plan $plan * * @return bool */ private function change_membership_plan( $membership, $plan ) { $updated = wp_update_post( array( 'ID' => $membership->get_id(), 'post_parent' => $plan->get_id(), ) ); if ( $updated ) { return true; } return false; } /** * Create new User Membership * * @param WP_User $user * @param WC_Memberships_Membership_Plan $plan * * @return bool */ private function create_new_membership( $user, $plan ) { try { wc_memberships_create_user_membership( [ 'user_id' => $user->ID, 'plan_id' => $plan->get_id(), ] ); } catch ( Exception $e ) { return false; } return true; } public function get_fields_schema() { return [ [ "id" => 'to_plan', "label" => __( 'New Membership Plan', 'wp-marketing-automations-pro' ), "type" => 'custom_search', 'autocompleterOption' => [ 'path' => 'wcm_plans', 'slug' => 'wcm_plans', 'labelText' => 'plan' ], "allowFreeTextSearch" => false, "required" => true, "errorMsg" => __( "New Membership plan is required", 'wp-marketing-automations-pro' ), "multiple" => false ], ]; } public function get_desc_text( $data ) { $data = json_decode( wp_json_encode( $data ), true ); if ( ! isset( $data['to_plan'] ) || empty( $data['to_plan'] ) ) { return ''; } $plans = []; foreach ( $data['to_plan'] as $plan ) { if ( ! isset( $plan['name'] ) || empty( $plan['name'] ) ) { continue; } $plans[] = $plan['name']; } return $plans; } } /** * Register this action. Registering the action will make it eligible to see it on single automation screen in select actions dropdown. */ return 'BWFAN_WCM_Update_Plan';