<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\InventoryReservationCli\Model\SalableQuantityInconsistency;

use Magento\Framework\Serialize\SerializerInterface;
use Magento\InventoryReservationCli\Model\SalableQuantityInconsistency;
use Magento\InventoryReservationCli\Model\SalableQuantityInconsistencyFactory;
use Magento\InventoryReservationCli\Model\ResourceModel\GetOrderIncrementId;
use Magento\InventoryReservationsApi\Model\ReservationInterface;
use Magento\InventorySalesApi\Model\StockByWebsiteIdResolverInterface;
use Magento\Sales\Api\Data\OrderInterface;

/**
 * Collects all existing and missing reservations in order to calculate inconsistency
 */
class Collector
{
    /**
     * @var SalableQuantityInconsistency[]
     */
    private $items = [];

    /**
     * @var SalableQuantityInconsistencyFactory
     */
    private $salableQuantityInconsistencyFactory;

    /**
     * @var SerializerInterface
     */
    private $serializer;

    /**
     * @var StockByWebsiteIdResolverInterface
     */
    private $stockByWebsiteIdResolver;

    /**
     * @var GetOrderIncrementId
     */
    private $getOrderIncrementId;

    /**
     * @param SalableQuantityInconsistencyFactory $salableQuantityInconsistencyFactory
     * @param SerializerInterface $serializer
     * @param StockByWebsiteIdResolverInterface $stockByWebsiteIdResolver
     * @param GetOrderIncrementId $getOrderIncrementId
     */
    public function __construct(
        SalableQuantityInconsistencyFactory $salableQuantityInconsistencyFactory,
        SerializerInterface $serializer,
        StockByWebsiteIdResolverInterface $stockByWebsiteIdResolver,
        GetOrderIncrementId $getOrderIncrementId
    ) {
        $this->salableQuantityInconsistencyFactory = $salableQuantityInconsistencyFactory;
        $this->serializer = $serializer;
        $this->stockByWebsiteIdResolver = $stockByWebsiteIdResolver;
        $this->getOrderIncrementId = $getOrderIncrementId;
    }

    /**
     * Add reservation to collectors items
     *
     * @param ReservationInterface $reservation
     */
    public function addReservation(ReservationInterface $reservation): void
    {
        $metadata = $this->serializer->unserialize($reservation->getMetadata());
        $objectId = $metadata['object_id'];
        $objectIncrementId = !empty($metadata['object_increment_id'])
            ? $metadata['object_increment_id']
            : $this->getOrderIncrementId->execute((int)$objectId);
        $stockId = $reservation->getStockId();
        $key = $objectIncrementId . '-' . $stockId;

        if (!isset($this->items[$key])) {
            $this->items[$key] = $this->salableQuantityInconsistencyFactory->create();
        }

        $this->items[$key]->setObjectId((int)$objectId);
        $this->items[$key]->setOrderIncrementId((string)$objectIncrementId);
        $this->items[$key]->setHasAssignedOrder((int)$objectId || (string)$objectIncrementId);
        $this->items[$key]->setStockId((int)$stockId);
        $this->items[$key]->addItemQty($reservation->getSku(), $reservation->getQuantity());
    }

    /**
     * Add order to collectors items
     *
     * @param OrderInterface $order
     */
    public function addOrder(OrderInterface $order): void
    {
        $objectIncrementId = $order->getIncrementId();
        $websiteId = (int)$order->getStore()->getWebsiteId();
        $stockId = (int)$this->stockByWebsiteIdResolver->execute((int)$websiteId)->getStockId();
        $key = $objectIncrementId . '-' . $stockId;

        if (!isset($this->items[$key])) {
            $this->items[$key] = $this->salableQuantityInconsistencyFactory->create();
        }

        $this->items[$key]->setOrderIncrementId($objectIncrementId);
        $this->items[$key]->setHasAssignedOrder(true);
        $this->items[$key]->setOrderStatus($order->getStatus());
    }

    /**
     * Add order to collectors items
     *
     * @param array $orderData
     */
    public function addOrderData(array $orderData): void
    {
        $objectIncrementId = $orderData['increment_id'];
        $websiteId = (int)$orderData['website_id'];
        $stockId = (int)$this->stockByWebsiteIdResolver->execute((int)$websiteId)->getStockId();
        $key = $objectIncrementId . '-' . $stockId;

        if (!isset($this->items[$key])) {
            $this->items[$key] = $this->salableQuantityInconsistencyFactory->create();
        }

        $this->items[$key]->setOrderIncrementId($orderData['increment_id']);
        $this->items[$key]->setHasAssignedOrder(true);
        $this->items[$key]->setOrderStatus($orderData['status']);
    }

    /**
     * Getter for items
     *
     * @return SalableQuantityInconsistency[]
     */
    public function getItems(): array
    {
        return $this->items;
    }

    /**
     * Setter for items
     *
     * @param SalableQuantityInconsistency[] $items
     */
    public function setItems(array $items): void
    {
        $this->items = $items;
    }
}