vendor/sylius/sylius/src/Sylius/Component/Core/Model/Shipment.php line 22

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of the Sylius package.
  5.  *
  6.  * (c) Paweł Jędrzejewski
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. declare(strict_types=1);
  14. namespace Sylius\Component\Core\Model;
  16. use Doctrine\Common\Collections\ArrayCollection;
  17. use Doctrine\Common\Collections\Collection;
  18. use Sylius\Component\Order\Model\AdjustmentInterface as BaseAdjustmentInterface;
  19. use Sylius\Component\Order\Model\OrderInterface as BaseOrderInterface;
  20. use Sylius\Component\Shipping\Model\Shipment as BaseShipment;
  22. class Shipment extends BaseShipment implements ShipmentInterface
  23. {
  24.     /** @var BaseOrderInterface|null */
  25.     protected $order;
  27.     /**
  28.      * @var Collection|BaseAdjustmentInterface[]
  29.      *
  30.      * @psalm-var Collection<array-key, BaseAdjustmentInterface>
  31.      */
  32.     protected $adjustments;
  34.     /**
  35.      * @var int
  36.      */
  37.     protected $adjustmentsTotal = 0;
  39.     public function __construct()
  40.     {
  41.         parent::__construct();
  43.         /** @var ArrayCollection<array-key, BaseAdjustmentInterface> $this->adjustments */
  44.         $this->adjustments = new ArrayCollection();
  45.     }
  47.     public function getOrder(): ?BaseOrderInterface
  48.     {
  49.         return $this->order;
  50.     }
  52.     public function setOrder(?BaseOrderInterface $order): void
  53.     {
  54.         $this->order = $order;
  55.     }
  57.     public function getAdjustments(?string $type = null): Collection
  58.     {
  59.         if (null === $type) {
  60.             return $this->adjustments;
  61.         }
  63.         return $this->adjustments->filter(function (AdjustmentInterface $adjustment) use ($type): bool {
  64.             return $type === $adjustment->getType();
  65.         });
  66.     }
  68.     public function addAdjustment(BaseAdjustmentInterface $adjustment): void
  69.     {
  70.         /** @var AdjustmentInterface $adjustment */
  71.         if ($this->hasAdjustment($adjustment)) {
  72.             return;
  73.         }
  75.         $this->adjustments->add($adjustment);
  76.         $adjustment->setShipment($this);
  77.         $this->recalculateAdjustmentsTotal();
  78.         $this->order->recalculateAdjustmentsTotal();
  79.     }
  81.     public function removeAdjustment(BaseAdjustmentInterface $adjustment): void
  82.     {
  83.         /** @var AdjustmentInterface $adjustment */
  84.         if ($adjustment->isLocked() || !$this->hasAdjustment($adjustment)) {
  85.             return;
  86.         }
  88.         $this->adjustments->removeElement($adjustment);
  89.         $adjustment->setShipment(null);
  90.         $this->recalculateAdjustmentsTotal();
  91.         $this->order->recalculateAdjustmentsTotal();
  92.     }
  94.     public function hasAdjustment(BaseAdjustmentInterface $adjustment): bool
  95.     {
  96.         return $this->adjustments->contains($adjustment);
  97.     }
  99.     public function getAdjustmentsTotal(?string $type = null): int
  100.     {
  101.         $total = 0;
  102.         foreach ($this->getAdjustments($type) as $adjustment) {
  103.             if (!$adjustment->isNeutral()) {
  104.                 $total += $adjustment->getAmount();
  105.             }
  106.         }
  108.         return $total;
  109.     }
  111.     public function removeAdjustments(?string $type = null): void
  112.     {
  113.         foreach ($this->getAdjustments($type) as $adjustment) {
  114.             $this->removeAdjustment($adjustment);
  115.         }
  116.     }
  118.     public function recalculateAdjustmentsTotal(): void
  119.     {
  120.         $this->adjustmentsTotal = $this->getAdjustmentsTotal();
  121.     }
  122. }