Save payment method info in subscription

Author: annemirasol

includes/payment-methods/class-wc-stripe-upe-payment-gateway.php

@@ -1039,6 +1039,17 @@ private function process_payment_with_confirmation_token( int $order_id ) {
 				}
 			}
 
+			if ( $payment_information['has_subscription'] ) {
+				$this->maybe_update_source_on_subscription_order(
+					$order,
+					(object) [
+						'payment_method' => $payment_method_id,
+						'customer'       => $payment_information['customer'],
+					],
+					'stripe' // TODO: get the correct payment gateway ID
+				);
+			}
+
 			$return_url = $this->get_return_url( $order );
 
 			return [
@@ -2213,6 +2224,7 @@ protected function prepare_payment_information_from_request( WC_Order $order ) {
 			'payment_method_details'        => [],
 			'payment_type'                  => 'single', // single | recurring.
 			'capture_method'                => $capture_method,
+			'save_payment_method_to_store'  => $save_payment_method_to_store,
 		];
 
 		if ( 'us_bank_account' === $selected_payment_type ) {
@@ -2224,22 +2236,22 @@ protected function prepare_payment_information_from_request( WC_Order $order ) {
 			unset( $payment_information['payment_method_details'] );
 			unset( $payment_information['capture_method'] );
 
-			$confirmation_token_id                               = sanitize_text_field( wp_unslash( $_POST['wc-stripe-confirmation-token'] ?? '' ) );
-			$payment_information['confirmation_token']           = $confirmation_token_id;
-			$payment_information['save_payment_method_to_store'] = false; // TODO Is this needed for subscriptions?
-			$payment_information['payment_method_options']       = $this->get_payment_method_options_for_confirmation_tokens(
+			$confirmation_token_id                         = sanitize_text_field( wp_unslash( $_POST['wc-stripe-confirmation-token'] ?? '' ) );
+			$payment_information['confirmation_token']     = $confirmation_token_id;
+			$payment_information['payment_method_options'] = $this->get_payment_method_options_for_confirmation_tokens(
 				$selected_payment_type,
 				[
 					'capture_method'   => $capture_method,
 					'has_subscription' => $payment_information['has_subscription'],
 				]
 			);
+			// TODO: remove this once Amazon Pay supports saved payment methods
+			$payment_information['save_payment_method_to_store'] = false;
 		} else {
-			$payment_method_details                              = WC_Stripe_API::get_payment_method( $payment_method_id );
-			$payment_information['payment_method']               = $payment_method_id;
-			$payment_information['payment_method_details']       = $payment_method_details;
-			$payment_information['save_payment_method_to_store'] = $save_payment_method_to_store;
-			$payment_information['payment_method_options']       = $this->get_payment_method_options(
+			$payment_method_details                        = WC_Stripe_API::get_payment_method( $payment_method_id );
+			$payment_information['payment_method']         = $payment_method_id;
+			$payment_information['payment_method_details'] = $payment_method_details;
+			$payment_information['payment_method_options'] = $this->get_payment_method_options(
 				$selected_payment_type,
 				$order,
 				$payment_method_details