add ca support (#115)

Affirm · Nov 2, 2023 · fc9122c · fc9122c
1 parent 01ed1bb
commit fc9122c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/Model/Config.php b/Model/Config.php
@@ -85,6 +85,7 @@ class Config implements ConfigInterface
     const SUFFIX_CANADA = '_ca';
     const KEY_ASLOWAS_DEVELOPER = 'affirm_aslowas_developer';
     const KEY_PIXEL = 'affirm_pixel';
+    const PARTIAL_CAPTURE_COUNTRIES = ['USA', 'CAN'];
     /**#@-*/
 
     /**
@@ -743,7 +744,7 @@ public function getCheckoutFlowType()
      */
     public function getPartialCapture($countryCode = self::COUNTRY_CODE_USA)
     {
-        return $this->getConfigData('partial_capture') && $countryCode == self::COUNTRY_CODE_USA;
+        return $this->getConfigData('partial_capture') && in_array($countryCode, self::PARTIAL_CAPTURE_COUNTRIES);
     }
 
     /**

diff --git a/Model/Plugin/Payment/CanCapturePartial.php b/Model/Plugin/Payment/CanCapturePartial.php
@@ -31,6 +31,7 @@ class CanCapturePartial
    * Define constants
    */
   const DEFAULT_COUNTRY_CODE = 'USA';
+  const PARTIAL_CAPTURE_COUNTRIES = ['USA', 'CAN'];
 
   /**
    * Constructor
@@ -58,7 +59,7 @@ public function afterCanCapturePartial(Payment $subject, $result)
       $countryCode = $subject->getData()['additional_information']['country_code'];
     }
 
-    if (!$this->affirmPaymentConfig->getPartialCapture() || $countryCode != self::DEFAULT_COUNTRY_CODE) {
+    if (!$this->affirmPaymentConfig->getPartialCapture() || !in_array( $countryCode, self::PARTIAL_CAPTURE_COUNTRIES ) ) {
       return false;
     }
     return $result;