Add allowed wallets param

HPWebdeveloper · Jan 11, 2024 · 11bf4a1 · 11bf4a1
1 parent 218bc2c
commit 11bf4a1
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 9 deletions.
diff --git a/src/Interfaces/WalletOperations.php b/src/Interfaces/WalletOperations.php
@@ -10,7 +10,7 @@ public function getWalletBalanceByType(string $walletType);
 
     public function hasSufficientBalance($value): bool;
 
-    public function pay(int|float $orderValue, ?string $notes = null);
+    public function pay(int|float $orderValue, array $allowedWallets = [], ?string $notes = null): void;
 
     public function deposit(string $type, int|float $amount, ?string $notes = null): bool;
-}
+}
diff --git a/src/Services/PocketServices.php b/src/Services/PocketServices.php
@@ -4,14 +4,14 @@
 
 class PocketServices
 {
-    public function deposit($user, $type, $amount, $notes = null)
+    public function deposit($user, $type, $amount, $notes = null): bool
     {
         return $user->deposit($type, $amount, $notes);
     }
 
-    public function pay($user, $orderValue, $notes = null)
+    public function pay($user, $orderValue, array $allowedWallets = [], ?string $notes = null): void
     {
-        return $user->pay($orderValue, $notes);
+        $user->pay($orderValue, $allowedWallets, $notes);
     }
 
     public function checkBalance($user)
@@ -23,4 +23,4 @@ public function walletBalanceByType($user, $type)
     {
         return $user->getWalletBalanceByType($type);
     }
-}
+}
diff --git a/src/Traits/HandlesPayment.php b/src/Traits/HandlesPayment.php
@@ -13,19 +13,28 @@ trait HandlesPayment
      *
      * @throws InsufficientBalanceException
      */
-    public function pay(int|float $orderValue, ?string $notes = null): void
+    public function pay(int|float $orderValue, array $allowedWallets = [], ?string $notes = null): void
     {
         if (! $this->hasSufficientBalance($orderValue)) {
             throw new InsufficientBalanceException('Insufficient balance to cover the order.');
         }
 
-        DB::transaction(function () use ($orderValue, $notes) {
+        DB::transaction(function () use ($orderValue, $notes, $allowedWallets) {
             $remainingOrderValue = $orderValue;
 
             $walletsInOrder = $this->wallets()->whereIn('type', $this->walletsInOrder())->get();
 
+            /**
+             * @param string $wallet
+             * @return bool $useWallet
+             * */
+            $useWallet = fn ($wallet) => count($allowedWallets) < 1 || in_array($wallet, $allowedWallets);
+
+            /**
+             * @var BalanceOperation $wallet
+             */
             foreach ($walletsInOrder as $wallet) {
-                if (! $wallet || ! $wallet->hasBalance()) {
+                if (! $wallet || ! $wallet->hasBalance() || !$useWallet($wallet->type->value)) {
                     continue;
                 }