automatorwp / integrations / restrict-content-pro / includes / actions / add-membership.php

add-membership.php in AutomatorWP – No-Code Workflow Automation, Integration & Webhooks Plugin, now with AI 6.0.1, at integrations/restrict-content-pro/includes/actions/add-membership.php

184 lines 8.5 KB

Raw Download Zip

1	<?php
2	/**
3	* Add Membership
4	*
5	* @package AutomatorWP\Integrations\Restrict_Content_Pro\Actions\Add_Membership
6	* @author AutomatorWP <[email protected]>, Ruben Garcia <[email protected]>
7	* @since 1.0.0
8	*/
9	// Exit if accessed directly
10	if( !defined( 'ABSPATH' ) ) exit;
11
12	class AutomatorWP_Restrict_Content_Pro_Add_Membership extends AutomatorWP_Integration_Action {
13
14	public $integration = 'restrict_content_pro';
15	public $action = 'restrict_content_pro_add_membership';
16
17	/**
18	* Register the trigger
19	*
20	* @since 1.0.0
21	*/
22	public function register() {
23
24	automatorwp_register_action( $this->action, array(
25	'integration' => $this->integration,
26	'label' => __( 'Add membership level to user', 'automatorwp' ),
27	'select_option' => __( 'Add <strong>membership level</strong> to user', 'automatorwp' ),
28	/* translators: %1$s: Membership. */
29	'edit_label' => sprintf( __( 'Add %1$s to user', 'automatorwp' ), '{membership}' ),
30	/* translators: %1$s: Membership. */
31	'log_label' => sprintf( __( 'Add %1$s to user', 'automatorwp' ), '{membership}' ),
32	'options' => array(
33	'membership' => array(
34	'from' => 'membership',
35	'fields' => array(
36	'membership' => array(
37	'name' => __( 'Membership Level:', 'automatorwp' ),
38	'type' => 'select',
39	'classes' => 'automatorwp-ajax-selector',
40	'option_none' => true,
41	'option_none_value' => 'any',
42	'option_none_label' => __( 'a membership level', 'automatorwp' ),
43	'option_custom' => true,
44	'option_custom_value' => 'custom',
45	'option_custom_label' => __( 'Use a custom value', 'automatorwp' ),
46	'option_custom_desc' => '',
47	'attributes' => array(
48	'data-action' => 'automatorwp_restrict_content_pro_get_memberships',
49	'data-option-none' => true,
50	'data-option-none-value' => 'any',
51	'data-option-none-label' => __( 'a membership level', 'automatorwp' ),
52	'data-option-custom' => true,
53	'data-option-custom-value' => 'custom',
54	'data-option-custom-label' => __( 'Use a custom value', 'automatorwp' ),
55	'data-placeholder' => '',
56	),
57	'options_cb' => 'automatorwp_restrict_content_pro_options_cb_membership',
58	'default' => 'any'
59	),
60	'membership_custom' => automatorwp_utilities_custom_field( array(
61	'option_custom_desc' => __( 'Membership ID', 'automatorwp' ),
62	) ),
63	'status' => array(
64	'name' => __( 'Status:', 'automatorwp' ),
65	'type' => 'select',
66	'options' => array(
67	'active' => __( 'Active', 'automatorwp' ),
68	'pending' => __( 'Pending', 'automatorwp' ),
69	'expired' => __( 'Expired', 'automatorwp' ),
70	'cancelled' => __( 'Cancelled', 'automatorwp' ),
71	),
72	'default' => 'active'
73	),
74	'expiration' => array(
75	'name' => __( 'Expiration Date:', 'automatorwp' ),
76	'desc' => __( 'Enter the membership expiration date in format YYYY-MM-DD. Leave empty for lifetime.', 'automatorwp' ),
77	'type' => 'text',
78	'default' => ''
79	),
80	)
81	)
82	),
83	) );
84
85	}
86
87	/**
88	* Action execution function
89	*
90	* @since 1.0.0
91	*
92	* @param stdClass $action The action object
93	* @param int $user_id The user ID
94	* @param array $action_options The action's stored options (with tags already passed)
95	* @param stdClass $automation The action's automation object
96	*/
97	public function execute( $action, $user_id, $action_options, $automation ) {
98
99	// Shorthand
100	$level_id = $action_options['membership'];
101	$status = $action_options['status'];
102	$expiration_date = $action_options['expiration'];
103
104	// Bail if not membership level has been configured
105	if( empty( $level_id ) \|\| $level_id === 'any' ) {
106	return;
107	}
108
109	$customer = rcp_get_customer_by_user_id( $user_id );
110
111	// Create a new customer record if one does not exist
112	if ( empty( $customer ) ) {
113	$customer_id = rcp_add_customer( array(
114	'user_id' => absint( $user_id ),
115	'date_registered' => date( 'Y-m-d H:i:s', current_time( 'timestamp' ) )
116	) );
117	} else {
118	$customer_id = $customer->get_id();
119	}
120
121	$membership_args = array(
122	'customer_id' => absint( $customer_id ),
123	'user_id' => $user_id,
124	'object_id' => $level_id,
125	'status' => $status,
126	'created_date' => date( 'Y-m-d H:i:s', current_time( 'timestamp' ) ),
127	'gateway' => 'manual',
128	'subscription_key' => rcp_generate_subscription_key()
129	);
130
131	switch ( $status ) {
132	case 'expired' :
133	// Set the expiration date if expired
134	$membership_args['expiration_date'] = date( 'Y-m-d H:i:s', current_time( 'timestamp' ) );
135	break;
136	case 'cancelled' :
137	// Set the cancellation date if cancelled
138	$membership_args['cancellation_date'] = date( 'Y-m-d H:i:s', current_time( 'timestamp' ) );
139	break;
140	case 'active' :
141	if ( ! empty( $expiration_date ) ) {
142	$membership_args['expiration_date'] = date( 'Y-m-d H:i:s', strtotime( $expiration_date ) );
143	}
144	break;
145	}
146
147	$membership_id = rcp_add_membership( $membership_args );
148
149	// Add membership meta to designate this as a generated record from this action
150	rcp_add_membership_meta( $membership_id, 'automatorwp_restrict_content_action_id', $action->id );
151
152	$membership = rcp_get_membership( $membership_id );
153
154	// Generate a transaction ID
155	$auth_key = defined( 'AUTH_KEY' ) ? AUTH_KEY : '';
156	$transaction_id = strtolower( md5( $membership_args['subscription_key'] . date( 'Y-m-d H:i:s' ) . $auth_key . uniqid( 'rcp', true ) ) );
157
158	// Create a corresponding payment record.
159	$payment_args = array(
160	'subscription' => rcp_get_subscription_name( $membership_args['object_id'] ),
161	'object_id' => $membership_args['object_id'],
162	'date' => $membership_args['created_date'],
163	'amount' => $membership->get_initial_amount(),
164	'subtotal' => $membership->get_initial_amount(),
165	'user_id' => $user_id,
166	'subscription_key' => $membership_args['subscription_key'],
167	'transaction_id' => $transaction_id,
168	'status' => 'pending' == $membership_args['status'] ? 'pending' : 'complete',
169	'gateway' => 'manual',
170	'customer_id' => $customer_id,
171	'membership_id' => $membership_id
172	);
173
174	$rcp_payments = new RCP_Payments();
175	$payment_id = $rcp_payments->insert( $payment_args );
176
177	// Add payment meta to designate this as a generated record from this action
178	$rcp_payments->add_meta( $payment_id, 'automatorwp_restrict_content_action_id', $action->id );
179
180	}
181
182	}
183
184	new AutomatorWP_Restrict_Content_Pro_Add_Membership();