1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
<?php
namespace AutomateWoo;
if ( ! defined( 'ABSPATH' ) ) {
exit;
}
/**
* @class Action_Memberships_Change_Plan
* @since 2.8
*/
class Action_Memberships_Change_Plan extends Action_Memberships_Abstract {
/**
* The data items required by the action.
*
* @var array
*/
public $required_data_items = [ 'customer' ];
/**
* Method to set the action's admin props.
*/
public function load_admin_details() {
parent::load_admin_details();
$this->title = __( 'Create / Change Membership Plan For User', 'automatewoo' );
$this->description = __( "Changes the plan of a user's active membership. If no active membership exists a new membership can optionally be created.", 'automatewoo' );
}
/**
* Method to load the action's fields.
**/
public function load_fields() {
$plans = Memberships_Helper::get_membership_plans();
$from_plan = new Fields\Select();
$from_plan->set_options( $plans );
$from_plan->set_name( 'from_plan' );
$from_plan->set_title( __( 'Existing plan', 'automatewoo' ) );
$from_plan->set_placeholder( __( '[None]', 'automatewoo' ) );
$from_plan->set_description( __( 'Leave this blank to only create new memberships.', 'automatewoo' ) );
$to_plan = new Fields\Select();
$to_plan->set_options( $plans );
$to_plan->set_name( 'to_plan' );
$to_plan->set_title( __( 'New plan', 'automatewoo' ) );
$to_plan->set_required();
$allow_create = new Fields\Checkbox();
$allow_create->set_name( 'allow_create' );
$allow_create->set_title( __( 'Allow membership creation?', 'automatewoo' ) );
$allow_create->set_description( __( 'If checked a new membership will be created for the user if they are not a member of the existing plan.', 'automatewoo' ) );
$this->add_field( $from_plan );
$this->add_field( $to_plan );
$this->add_field( $allow_create );
}
/**
* Runs the action
*/
public function run() {
$customer = $this->workflow->data_layer()->get_customer();
$existing_plan_id = absint( $this->get_option( 'from_plan' ) );
$new_plan_id = absint( $this->get_option( 'to_plan' ) );
$allow_create = $this->get_option( 'allow_create' );
$new_plan = $new_plan_id ? wc_memberships_get_membership_plan( $new_plan_id ) : false;
if ( ! $customer->is_registered() || ! $new_plan ) {
return;
}
// check if user already has a membership with the new plan
$check_for_membership = wc_memberships_get_user_membership( $customer->get_user_id(), $new_plan );
if ( $check_for_membership ) {
$this->workflow->log_action_note( $this, __( 'Plan could not be changed or created because the user already has a membership using the new plan.', 'automatewoo' ) );
return;
}
$membership = $existing_plan_id ? wc_memberships_get_user_membership( $customer->get_user_id(), $existing_plan_id ) : false;
if ( $membership ) {
$this->change_membership_plan( $membership, $new_plan );
} elseif ( $allow_create ) {
// if no existing plan and allow create is checked, create a new plan for the user
$this->create_membership( $customer, $new_plan );
}
}
/**
* @param Customer $customer
* @param \WC_Memberships_Membership_Plan $new_plan
*
* @throws \Exception If an error happens creating the membership.
*/
public function create_membership( $customer, $new_plan ) {
$membership = wc_memberships_create_user_membership(
[
'user_id' => $customer->get_user_id(),
'plan_id' => $new_plan->get_id(),
]
);
if ( is_wp_error( $membership ) ) {
throw new \Exception( esc_attr( $membership->get_error_message() ) );
}
$membership->add_note(
esc_textarea(
sprintf(
// translators: The Workflow ID
__( 'Membership created by AutomateWoo workflow #%s', 'automatewoo' ),
$this->workflow->get_id()
)
)
);
$this->workflow->log_action_note(
$this,
sprintf(
// translators: The Membership ID
__( 'Membership #%s successfully created.', 'automatewoo' ),
$membership->get_id()
)
);
}
/**
* @param \WC_Memberships_User_Membership $membership
* @param \WC_Memberships_Membership_Plan $new_plan
*/
public function change_membership_plan( $membership, $new_plan ) {
$membership->add_note(
sprintf(
// translators: %1$s The Membership plan name, %2$s the new plan name, %3$s The Workflow ID
__( 'Membership plan changed from %1$s to %2$s by AutomateWoo workflow:#%3$s', 'automatewoo' ),
$membership->get_plan()->get_name(),
$new_plan->get_name(),
$this->workflow->get_id()
)
);
wp_update_post(
[
'ID' => $membership->get_id(),
'post_parent' => $new_plan->get_id(),
]
);
}
}
|