<?php

/*
 * This file is part of the Sylius package.
 *
 * (c) Sylius Sp. z o.o.
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

declare(strict_types=1);

namespace Sylius\Component\Core\Model;

use Doctrine\Common\Collections\ArrayCollection;
use Doctrine\Common\Collections\Collection;
use Sylius\Component\Order\Model\AdjustmentInterface as BaseAdjustmentInterface;
use Sylius\Component\Order\Model\OrderInterface as BaseOrderInterface;
use Sylius\Component\Shipping\Model\Shipment as BaseShipment;
use Sylius\Component\Shipping\Model\ShipmentUnitInterface;
use Webmozart\Assert\Assert;

class Shipment extends BaseShipment implements ShipmentInterface
{
    /** @var BaseOrderInterface|null */
    protected $order;

    /** @var Collection<array-key, BaseAdjustmentInterface> */
    protected $adjustments;

    /** @var int */
    protected $adjustmentsTotal = 0;

    public function __construct()
    {
        parent::__construct();

        /** @var ArrayCollection<array-key, BaseAdjustmentInterface> $this->adjustments */
        $this->adjustments = new ArrayCollection();
    }

    public function getOrder(): ?BaseOrderInterface
    {
        Assert::nullOrIsInstanceOf($this->order, OrderInterface::class);

        return $this->order;
    }

    public function setOrder(?BaseOrderInterface $order): void
    {
        $this->order = $order;
    }

    public function getAdjustments(?string $type = null): Collection
    {
        if (null === $type) {
            return $this->adjustments;
        }

        return $this->adjustments->filter(function (BaseAdjustmentInterface $adjustment) use ($type): bool {
            return $type === $adjustment->getType();
        });
    }

    public function addAdjustment(BaseAdjustmentInterface $adjustment): void
    {
        /** @var AdjustmentInterface $adjustment */
        if ($this->hasAdjustment($adjustment)) {
            return;
        }

        $this->adjustments->add($adjustment);
        $adjustment->setShipment($this);
        $this->recalculateAdjustmentsTotal();
        $this->order->recalculateAdjustmentsTotal();
    }

    public function removeAdjustment(BaseAdjustmentInterface $adjustment): void
    {
        /** @var AdjustmentInterface $adjustment */
        if ($adjustment->isLocked() || !$this->hasAdjustment($adjustment)) {
            return;
        }

        $this->adjustments->removeElement($adjustment);
        $adjustment->setShipment(null);
        $this->recalculateAdjustmentsTotal();
        $this->order->recalculateAdjustmentsTotal();
    }

    public function hasAdjustment(BaseAdjustmentInterface $adjustment): bool
    {
        return $this->adjustments->contains($adjustment);
    }

    public function getAdjustmentsTotal(?string $type = null): int
    {
        $total = 0;
        foreach ($this->getAdjustments($type) as $adjustment) {
            if (!$adjustment->isNeutral()) {
                $total += $adjustment->getAmount();
            }
        }

        return $total;
    }

    public function removeAdjustments(?string $type = null): void
    {
        foreach ($this->getAdjustments($type) as $adjustment) {
            $this->removeAdjustment($adjustment);
        }
    }

    public function recalculateAdjustmentsTotal(): void
    {
        $this->adjustmentsTotal = $this->getAdjustmentsTotal();
    }

    public function getShippingUnitTotal(): int
    {
        return array_sum(array_map(function (ShipmentUnitInterface $shipmentUnit) {
            Assert::isInstanceOf($shipmentUnit, OrderItemUnitInterface::class);

            return $shipmentUnit->getTotal();
        }, $this->units->toArray()));
    }
}
