Keep consistent Stripe customer in V2 and V3 Elements

solidusio · Aug 8, 2020 · 5d43fed · 5d43fed
1 parent 4f67582
commit 5d43fed
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/app/models/spree/payment_method/stripe_credit_card.rb b/app/models/spree/payment_method/stripe_credit_card.rb
@@ -119,6 +119,14 @@ def create_profile(payment)
           creditcard = source
         end
 
+        user_stripe_payment_sources = payment.source.user&.wallet&.wallet_payment_sources&.select do |wps|
+          wps.payment_source.payment_method.type == 'Spree::PaymentMethod::StripeCreditCard'
+        end
+        if user_stripe_payment_sources.present?
+          customer_id = user_stripe_payment_sources.map {|ps| ps.payment_source&.gateway_customer_profile_id }.compact.last
+          options[:customer] = customer_id
+        end
+
         response = gateway.store(creditcard, options)
         if response.success?
           if v3_intents?
@@ -130,8 +138,8 @@ def create_profile(payment)
           else
             payment.source.update!(
               cc_type: payment.source.cc_type,
-              gateway_customer_profile_id: response.params['id'],
-              gateway_payment_profile_id: response.params['default_source'] || response.params['default_card']
+              gateway_customer_profile_id: options[:customer] ? response.params['customer'] : response.params['id'],
+              gateway_payment_profile_id: options[:customer] ? response.params['id'] : response.params['default_source'] || response.params['default_card']
             )
           end
         else