Custom price with shopping cart

Author: rez1dent3

src/Internal/Dto/ItemDto.php

@@ -11,7 +11,8 @@ final class ItemDto implements ItemDtoInterface
 {
     public function __construct(
         private ProductInterface $product,
-        private int $quantity
+        private int $quantity,
+        private int|string|null $price,
     ) {
     }
 
@@ -23,6 +24,11 @@ public function items(): array
         return array_fill(0, $this->quantity, $this->product);
     }
 
+    public function getPrice(): int|string|null
+    {
+        return $this->price;
+    }
+
     public function product(): ProductInterface
     {
         return $this->product;

src/Internal/Dto/ItemDtoInterface.php

@@ -16,5 +16,7 @@ public function items(): array;
 
     public function count(): int;
 
+    public function getPrice(): int|string|null;
+
     public function product(): ProductInterface;
 }

src/Objects/Cart.php

@@ -21,15 +21,10 @@
 final class Cart implements Countable, CartInterface
 {
     /**
-     * @var ProductInterface[]
+     * @var array<string, ItemDtoInterface[]>
      */
     private array $items = [];
 
-    /**
-     * @var array<string, int>
-     */
-    private array $quantity = [];
-
     private array $meta = [];
 
     public function __construct(
@@ -51,14 +46,16 @@ public function withMeta(array $meta): self
         return $self;
     }
 
-    public function withItem(ProductInterface $product, int $quantity = 1): self
+    public function withItem(ProductInterface $product, int $quantity = 1, int|string|null $price = null): self
     {
         $self = clone $this;
 
         $productId = $self->productId($product);
+        if (!isset($self->items[$productId])) {
+            $self->items[$productId] = [];
+        }
 
-        $self->quantity[$productId] = $self->getQuantity($product) + $quantity;
-        $self->items[$productId] = $product;
+        $self->items[$productId][] = new ItemDto($product, $quantity, $price);
 
         return $self;
     }
@@ -80,29 +77,28 @@ public function getItems(): array
     {
         $items = [];
         foreach ($this->items as $item) {
-            $count = $this->getQuantity($item);
-            for ($i = 0; $i < $count; ++$i) {
-                $items[] = $item;
+            foreach ($item as $datum) {
+                $items[] = $datum->items();
             }
         }
 
-        return $items;
-    }
-
-    /**
-     * @return ProductInterface[]
-     */
-    public function getUniqueItems(): array
-    {
-        return $this->items;
+        return array_merge(...$items);
     }
 
     public function getTotal(Customer $customer): string
     {
         $result = 0;
-        foreach ($this->items as $item) {
-            $price = $this->math->mul($this->getQuantity($item), $item->getAmountProduct($customer));
-            $result = $this->math->add($result, $price);
+        $prices = [];
+        foreach ($this->items as $productId => $_items) {
+            foreach ($_items as $item) {
+                $product = $item->product();
+                $prices[$productId] = $item->getPrice()
+                    ?? $prices[$productId]
+                    ?? $item->product()->getAmountProduct($customer);
+
+                $price = $this->math->mul($this->getQuantity($product), $prices[$productId]);
+                $result = $this->math->add($result, $price);
+            }
         }
 
         return (string) $result;
@@ -115,24 +111,26 @@ public function count(): int
 
     public function getQuantity(ProductInterface $product): int
     {
-        return $this->quantity[$this->productId($product)] ?? 0;
+        /** @var ItemDtoInterface[] $items */
+        $quantity = 0;
+        $items = $this->items[$this->productId($product)] ?? [];
+        foreach ($items as $item) {
+            $quantity += $item->count();
+        }
+
+        return $quantity;
     }
 
     /**
      * @throws CartEmptyException
      */
     public function getBasketDto(): BasketDtoInterface
     {
-        $items = array_map(
-            fn (ProductInterface $product): ItemDtoInterface => new ItemDto($product, $this->getQuantity($product)),
-            $this->getUniqueItems()
-        );
-
-        if ($items === []) {
+        if ($this->items === []) {
             throw new CartEmptyException('Cart is empty', ExceptionInterface::CART_EMPTY);
         }
 
-        return new BasketDto($items, $this->getMeta());
+        return new BasketDto(array_merge(...array_values($this->items)), $this->getMeta());
     }
 
     private function productId(ProductInterface $product): string

src/Traits/CartPay.php

@@ -120,14 +120,17 @@ public function payCart(CartInterface $cart, bool $force = false): array
             $basketDto = $cart->getBasketDto();
             $prepareService = app(PrepareServiceInterface::class);
             $assistantService = app(AssistantServiceInterface::class);
-            foreach ($cart->getBasketDto()->cursor() as $product) {
-                $transfers[] = $prepareService->transferLazy(
-                    $this,
-                    $product,
-                    Transfer::STATUS_PAID,
-                    $product->getAmountProduct($this),
-                    $assistantService->getMeta($basketDto, $product)
-                );
+            foreach ($cart->getBasketDto()->items() as $item) {
+                foreach ($item->items() as $_item) {
+                    $product = $item->product();
+                    $transfers[] = $prepareService->transferLazy(
+                        $this,
+                        $product,
+                        Transfer::STATUS_PAID,
+                        $item->getPrice() ?? $product->getAmountProduct($this),
+                        $assistantService->getMeta($basketDto, $product)
+                    );
+                }
             }
 
             if (!$force) {

tests/Units/Domain/BasketTest.php

@@ -17,8 +17,8 @@ final class BasketTest extends TestCase
     public function testCount(): void
     {
         $item = new Item();
-        $productDto1 = new ItemDto($item, 24);
-        $productDto2 = new ItemDto($item, 26);
+        $productDto1 = new ItemDto($item, 24, null);
+        $productDto2 = new ItemDto($item, 26, null);
         $basket = new BasketDto([$productDto1, $productDto2], []);
 
         self::assertEmpty($basket->meta());

tests/Units/Domain/ProductTest.php

@@ -10,6 +10,7 @@
 use Bavix\Wallet\Models\Transaction;
 use Bavix\Wallet\Models\Transfer;
 use Bavix\Wallet\Models\Wallet;
+use Bavix\Wallet\Objects\Cart;
 use Bavix\Wallet\Test\Infra\Factories\BuyerFactory;
 use Bavix\Wallet\Test\Infra\Factories\ItemFactory;
 use Bavix\Wallet\Test\Infra\Factories\ItemWalletFactory;
@@ -260,6 +261,28 @@ public function testPayFreeOutOfStock(): void
         $buyer->payFree($product);
     }
 
+    public function testPayCustomPrice(): void
+    {
+        /**
+         * @var Buyer $buyer
+         * @var Item  $product
+         */
+        $buyer = BuyerFactory::new()->create();
+        $product = ItemFactory::new()->create([
+            'quantity' => 1,
+            'price' => 5_000,
+        ]);
+
+        $buyer->deposit(1_000);
+        self::assertSame(1_000, $buyer->balanceInt);
+
+        $cart = app(Cart::class)
+            ->withItem($product, price: 1_000);
+
+        $transfers = $buyer->payCart($cart);
+        self::assertCount(1, $transfers);
+    }
+
     /**
      * @see https://github.com/bavix/laravel-wallet/issues/237
      *