Completed
Push — master ( 6d8164...bf7249 )
by Roy
03:26
created

WC_Stripe_Payment_Tokens::get_instance()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 3
Code Lines 2

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
dl 0
loc 3
rs 10
c 0
b 0
f 0
cc 1
eloc 2
nc 1
nop 0
1
<?php
2
if ( ! defined( 'ABSPATH' ) ) {
3
	exit;
4
}
5
6
/**
7
 * Handles and process WC payment tokens API.
8
 * Seen in checkout page and my account->add payment method page.
9
 *
10
 * @since 4.0.0
11
 */
12
class WC_Stripe_Payment_Tokens {
13
	private static $_this;
14
15
	/**
16
	 * Constructor.
17
	 *
18
	 * @since 4.0.0
19
	 * @version 4.0.0
20
	 */
21
	public function __construct() {
22
		self::$_this = $this;
23
24
		add_filter( 'woocommerce_get_customer_payment_tokens', array( $this, 'woocommerce_get_customer_payment_tokens' ), 10, 3 );
25
		add_filter( 'woocommerce_payment_methods_list_item', array( $this, 'get_account_saved_payment_methods_list_item_sepa' ), 10, 2 );
26
		add_filter( 'woocommerce_get_credit_card_type_label', array( $this, 'normalize_sepa_label' ) );
27
		add_action( 'woocommerce_payment_token_deleted', array( $this, 'woocommerce_payment_token_deleted' ), 10, 2 );
28
		add_action( 'woocommerce_payment_token_set_default', array( $this, 'woocommerce_payment_token_set_default' ) );
29
	}
30
31
	/**
32
	 * Public access to instance object.
33
	 *
34
	 * @since 4.0.0
35
	 * @version 4.0.0
36
	 */
37
	public static function get_instance() {
38
		return self::$_this;
39
	}
40
41
	/**
42
	 * Normalizes the SEPA IBAN label on My Account page.
43
	 *
44
	 * @since 4.0.0
45
	 * @version 4.0.0
46
	 * @param string $label
47
	 * @return string $label
48
	 */
49
	public function normalize_sepa_label( $label ) {
50
		if ( 'sepa iban' === strtolower( $label ) ) {
51
			return 'SEPA IBAN';
52
		}
53
54
		return $label;
55
	}
56
57
	/**
58
	 * Gets saved tokens from API if they don't already exist in WooCommerce.
59
	 *
60
	 * @since 3.1.0
61
	 * @version 4.0.0
62
	 * @param array $tokens
63
	 * @return array
64
	 */
65
	public function woocommerce_get_customer_payment_tokens( $tokens = array(), $customer_id, $gateway_id ) {
66
		if ( is_user_logged_in() && class_exists( 'WC_Payment_Token_CC' ) ) {
67
			$stored_tokens = array();
68
69
			foreach ( $tokens as $token ) {
70
				$stored_tokens[] = $token->get_token();
71
			}
72
73
			if ( 'stripe' === $gateway_id ) {
74
				$stripe_customer = new WC_Stripe_Customer( $customer_id );
75
				$stripe_sources  = $stripe_customer->get_sources();
76
77
				foreach ( $stripe_sources as $source ) {
78
					if ( isset( $source->type ) && 'card' === $source->type ) {
79
						if ( ! in_array( $source->id, $stored_tokens ) ) {
80
							$token = new WC_Payment_Token_CC();
81
							$token->set_token( $source->id );
82
							$token->set_gateway_id( 'stripe' );
83
84
							if ( 'source' === $source->object && 'card' === $source->type ) {
85
								$token->set_card_type( strtolower( $source->card->brand ) );
86
								$token->set_last4( $source->card->last4 );
87
								$token->set_expiry_month( $source->card->exp_month );
88
								$token->set_expiry_year( $source->card->exp_year );
89
							}
90
91
							$token->set_user_id( $customer_id );
92
							$token->save();
93
							$tokens[ $token->get_id() ] = $token;
94
						}
95
					} else {
96
						if ( ! in_array( $source->id, $stored_tokens ) && 'card' === $source->object ) {
97
							$token = new WC_Payment_Token_CC();
98
							$token->set_token( $source->id );
99
							$token->set_gateway_id( 'stripe' );
100
							$token->set_card_type( strtolower( $source->brand ) );
101
							$token->set_last4( $source->last4 );
102
							$token->set_expiry_month( $source->exp_month );
103
							$token->set_expiry_year( $source->exp_year );
104
							$token->set_user_id( $customer_id );
105
							$token->save();
106
							$tokens[ $token->get_id() ] = $token;
107
						}
108
					}
109
				}
110
			}
111
112
			if ( 'stripe_sepa' === $gateway_id ) {
113
				$stripe_customer = new WC_Stripe_Customer( $customer_id );
114
				$stripe_sources  = $stripe_customer->get_sources();
115
116
				foreach ( $stripe_sources as $source ) {
117
					if ( isset( $source->type ) && 'sepa_debit' === $source->type ) {
118
						if ( ! in_array( $source->id, $stored_tokens ) ) {
119
							$token = new WC_Payment_Token_SEPA();
120
							$token->set_token( $source->id );
121
							$token->set_gateway_id( 'stripe_sepa' );
122
							$token->set_last4( $source->sepa_debit->last4 );
123
							$token->set_user_id( $customer_id );
124
							$token->save();
125
							$tokens[ $token->get_id() ] = $token;
126
						}
127
					}
128
				}
129
			}
130
		}
131
132
		return $tokens;
133
	}
134
135
	/**
136
	 * Controls the output for SEPA on the my account page.
137
	 *
138
	 * @since 4.0.0
139
	 * @version 4.0.0
140
	 * @param  array             $item         Individual list item from woocommerce_saved_payment_methods_list
141
	 * @param  WC_Payment_Token $payment_token The payment token associated with this method entry
142
	 * @return array                           Filtered item
143
	 */
144
	public function get_account_saved_payment_methods_list_item_sepa( $item, $payment_token ) {
145
		if ( 'sepa' !== strtolower( $payment_token->get_type() ) ) {
146
			return $item;
147
		}
148
149
		$item['method']['last4'] = $payment_token->get_last4();
150
		$item['method']['brand'] = esc_html__( 'SEPA IBAN', 'woocommerce-gateway-stripe' );
151
152
		return $item;
153
	}
154
155
	/**
156
	 * Delete token from Stripe.
157
	 *
158
	 * @since 3.1.0
159
	 * @version 4.0.0
160
	 */
161
	public function woocommerce_payment_token_deleted( $token_id, $token ) {
162 View Code Duplication
		if ( 'stripe' === $token->get_gateway_id() || 'stripe_sepa' === $token->get_gateway_id() ) {
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated across your project.

Duplicated code is one of the most pungent code smells. If you need to duplicate the same code in three or more different places, we strongly encourage you to look into extracting the code into a single class or operation.

You can also find more detailed suggestions in the “Code” section of your repository.

Loading history...
163
			$stripe_customer = new WC_Stripe_Customer( get_current_user_id() );
164
			$stripe_customer->delete_source( $token->get_token() );
165
		}
166
	}
167
168
	/**
169
	 * Set as default in Stripe.
170
	 *
171
	 * @since 3.1.0
172
	 * @version 4.0.0
173
	 */
174
	public function woocommerce_payment_token_set_default( $token_id ) {
175
		$token = WC_Payment_Tokens::get( $token_id );
176
177 View Code Duplication
		if ( 'stripe' === $token->get_gateway_id() || 'stripe_sepa' === $token->get_gateway_id() ) {
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated across your project.

Duplicated code is one of the most pungent code smells. If you need to duplicate the same code in three or more different places, we strongly encourage you to look into extracting the code into a single class or operation.

You can also find more detailed suggestions in the “Code” section of your repository.

Loading history...
178
			$stripe_customer = new WC_Stripe_Customer( get_current_user_id() );
179
			$stripe_customer->set_default_source( $token->get_token() );
180
		}
181
	}
182
}
183
184
new WC_Stripe_Payment_Tokens();
185