fix Fee.php

buckaroo-it · Aug 21, 2024 · 5829a6d · 5829a6d
1 parent 634cdaa
commit 5829a6d
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/Block/Totals/Fee.php b/Block/Totals/Fee.php
@@ -52,12 +52,27 @@ public function getTitle(): string
     public function getTotal(): float
     {
         $totalData = $this->getSegment();
-        $extensionAttributes = $totalData['extension_attributes'] ?? null;
+        if (false === is_array($totalData)) {
+            throw new \UnexpectedValueException('Expecting an array but getting '.gettype($totalData));
+        }
+
+        $extensionAttributes = $totalData['extension_attributes'];
+
         if (
-            $extensionAttributes
+            is_array($extensionAttributes) &&
+            isset($extensionAttributes['buckaroo_fee']) &&
+            is_scalar($extensionAttributes['buckaroo_fee'])
         ) {
-            return (floatval($extensionAttributes->getBuckarooFee()['buckaroo_fee'][0]));
+            return floatval($extensionAttributes['buckaroo_fee']);
         }
+
+        if ($extensionAttributes instanceof \Magento\Quote\Api\Data\TotalSegmentExtension) {
+            /** @var \Magento\Quote\Api\Data\TotalSegmentExtension $extensionAttributes */
+            if ($extensionAttributes->getBuckarooFee() !== null) {
+                return $extensionAttributes->getBuckarooFee();
+            }
+        }
+
         return 0;
     }
 }