Remove Amazon Pay from payment gateways when changing subscription payment method if customer is logged out of Amazon account (woocommerce#52)

Change redirect url if request redirect is empty.

Author: jrodger

woocommerce-gateway-amazon-payments-advanced.php

@@ -186,6 +186,8 @@ public function __construct() {
 		add_action( 'wp_ajax_amazon_get_order_reference', array( $this, 'ajax_get_order_reference' ) );
 		add_action( 'wp_ajax_nopriv_amazon_get_order_reference', array( $this, 'ajax_get_order_reference' ) );
 
+		// WC Subscription Hook
+		add_filter( 'woocommerce_subscriptions_process_payment_for_change_method_via_pay_shortcode', array( $this, 'filter_payment_method_changed_result' ), 10, 2 );
 	}
 
 	/**
@@ -501,6 +503,7 @@ public function checkout_init( $checkout ) {
 		// Don't try to render the Amazon widgets if we don't have the prerequisites
 		// for each mode.
 		if ( ( ! $enable_login_app && empty( $this->reference_id ) ) || ( $enable_login_app && empty( $this->access_token ) ) ) {
+			add_filter( 'woocommerce_available_payment_gateways', array( $this, 'remove_amazon_gateway' ) );
 			return;
 		}
 
@@ -1104,6 +1107,24 @@ public function update_amazon_widgets_fragment( $fragments ) {
 
 	}
 
+	/**
+	 * Remove amazon gateway.
+	 *
+	 * @param $gateways
+	 *
+	 * @return array
+	 */
+	public function remove_amazon_gateway( $gateways ) {
+
+		foreach ( $gateways as $gateway_key => $gateway ) {
+			if ( 'amazon_payments_advanced' === $gateway_key ) {
+				unset( $gateways[ $gateway_key ] );
+			}
+		}
+
+		return $gateways;
+	}
+
 	/**
 	 * Remove all gateways except amazon
 	 *
@@ -1639,6 +1660,21 @@ protected function amazon_keys_already_set() {
 	private function get_settings_url() {
 		return admin_url( 'admin.php?page=wc-settings&tab=checkout&section=amazon_payments_advanced' );
 	}
+
+	/**
+	 * Set redirect URL if the result redirect URL is empty
+	 *
+	 * @param mixed $result
+	 * @param WC_Subscription $subscription
+	 *
+	 * @return mixed
+	 */
+	public function filter_payment_method_changed_result( $result, $subscription ) {
+		if ( empty( $result['redirect'] ) && ! empty( $subscription ) && method_exists( $subscription, 'get_view_order_url' ) ) {
+			$result['redirect'] = $subscription->get_view_order_url();
+		}
+		return $result;
+	}
 }
 
 /**