patch gift

Author: rez1dent3

src/Traits/CanBePaid.php

@@ -161,4 +161,24 @@ public function forceRefund(Product $product, bool $gifts = false): bool
         return $this->refund($product, true, $gifts);
     }
 
+    /**
+     * @param Product $product
+     * @param bool $force
+     * @return bool
+     */
+    public function refundGift(Product $product, bool $force = false): bool
+    {
+        return $this->refund($product, $force, true);
+    }
+
+    /**
+     * @param Product $product
+     * @return bool
+     * @throws
+     */
+    public function forceRefundGift(Product $product): bool
+    {
+        return $this->refundGift($product, true);
+    }
+
 }

src/Traits/HasGift.php

@@ -24,38 +24,74 @@ trait HasGift
      *
      * @param Wallet $to
      * @param Product $product
+     * @param bool $force
      * @return Transfer
      */
-    public function gift(Wallet $to, Product $product): Transfer
+    public function gift(Wallet $to, Product $product, bool $force = false): Transfer
     {
         /**
-         * this comment is needed for syntax highlighting )
-         * @var \Bavix\Wallet\Models\Wallet $to
+         * Who's giving? Let's call him Santa Claus
          */
-        return DB::transaction(function () use ($to, $product) {
+        $santa = $this;
+
+        /**
+         * @return Transfer
+         */
+        $callback = function () use ($santa, $product, $force) {
             $amount = $product->getAmountProduct();
             $meta = $product->getMetaProduct();
             $fee = Tax::fee($product, $amount);
-            $withdraw = $this->withdraw($amount + $fee, $meta);
+
+            /**
+             * Santa pays taxes
+             */
+            if ($force) {
+                $withdraw = $santa->withdraw($amount + $fee, $meta);
+            } else {
+                $withdraw = $santa->forceWithdraw($amount + $fee, $meta);
+            }
+
             $deposit = $product->deposit($amount, $meta);
-            return $to->assemble($product, $withdraw, $deposit);
-        });
+            return $this->assemble($product, $withdraw, $deposit);
+        };
+
+        /**
+         * Unfortunately,
+         * I think it is wrong to make the "assemble" method public.
+         * That's why I address him like this!
+         */
+        return DB::transaction(
+            $callback->bindTo($to, \get_class($to))
+        );
     }
 
     /**
      * Give the goods safely.
      *
      * @param Wallet $to
      * @param Product $product
+     * @param bool $force
      * @return Transfer|null
      */
-    public function safeGift(Wallet $to, Product $product): ?Transfer
+    public function safeGift(Wallet $to, Product $product, bool $force = false): ?Transfer
     {
         try {
-            return $this->gift($to, $product);
+            return $this->gift($to, $product, $force);
         } catch (\Throwable $throwable) {
             return null;
         }
     }
 
+    /**
+     * to give force)
+     *
+     * @param Wallet $to
+     * @param Product $product
+     * @return Transfer
+     */
+    public function forceGift(Wallet $to, Product $product): Transfer
+    {
+        return $this->gift($to, $product, true);
+    }
+
 }

src/Traits/HasWallet.php

@@ -164,6 +164,7 @@ public function forceTransfer(Wallet $wallet, int $amount, ?array $meta = null):
     /**
      * this method adds a new transfer to the transfer table
      *
+     * @param string $status
      * @param Wallet $wallet
      * @param Transaction $withdraw
      * @param Transaction $deposit
@@ -172,10 +173,17 @@ public function forceTransfer(Wallet $wallet, int $amount, ?array $meta = null):
      */
     protected function assemble(Wallet $wallet, Transaction $withdraw, Transaction $deposit): Transfer
     {
+        $status = Transfer::STATUS_PAID;
+        if ($this->getMorphClass() !== $withdraw->payable_type ||
+            $this->getKey() !== $withdraw->payable_id) {
+            $status = Transfer::STATUS_GIFT;
+        }
+
         /**
          * @var Model $wallet
          */
         return \app('bavix.wallet::transfer')->create([
+            'status' => $status,
             'deposit_id' => $deposit->getKey(),
             'withdraw_id' => $withdraw->getKey(),
             'from_type' => $this->getMorphClass(),

tests/GiftTest.php

@@ -0,0 +1,68 @@
+<?php
+
+namespace Bavix\Wallet\Test;
+
+use Bavix\Wallet\Test\Models\Buyer;
+use Bavix\Wallet\Test\Models\Item;
+
+class GiftTest extends TestCase
+{
+
+    /**
+     * @return void
+     */
+    public function testGift(): void
+    {
+        /**
+         * @var Buyer $first
+         * @var Buyer $second
+         * @var Item $product
+         */
+        [$first, $second] = factory(Buyer::class, 2)->create();
+        $product = factory(Item::class)->create([
+            'quantity' => 1,
+        ]);
+
+        $this->assertEquals($first->balance, 0);
+        $this->assertEquals($second->balance, 0);
+
+        $first->deposit($product->getAmountProduct());
+        $this->assertEquals($first->balance, $product->getAmountProduct());
+
+        $first->wallet->gift($second, $product);
+        $this->assertEquals($first->balance, 0);
+        $this->assertEquals($second->balance, 0);
+        $this->assertNull($first->paid($product, true));
+        $this->assertNotNull($second->paid($product, true));
+    }
+
+    /**
+     * @return void
+     */
+    public function testRefund(): void
+    {
+        /**
+         * @var Buyer $first
+         * @var Buyer $second
+         * @var Item $product
+         */
+        [$first, $second] = factory(Buyer::class, 2)->create();
+        $product = factory(Item::class)->create([
+            'quantity' => 1,
+        ]);
+
+        $this->assertEquals($first->balance, 0);
+        $this->assertEquals($second->balance, 0);
+
+        $first->deposit($product->getAmountProduct());
+        $this->assertEquals($first->balance, $product->getAmountProduct());
+
+        $first->wallet->gift($second, $product);
+        $this->assertEquals($first->balance, 0);
+        $this->assertEquals($second->balance, 0);
+
+        $this->assertFalse($second->wallet->safeRefund($product));
+        $this->assertTrue($second->wallet->refundGift($product));
+    }
+
+}