<?php
/**
 * Refer to LICENSE.txt distributed with the Temando Shipping module for notice of license
 */
namespace Temando\Shipping\Plugin\Quote\Address;

use Magento\Framework\Exception\NoSuchEntityException;
use Magento\Quote\Model\Quote\Address;
use Magento\Quote\Model\Quote\Address\ToOrderAddress;
use Magento\Sales\Api\Data\OrderAddressExtensionInterface;
use Magento\Sales\Api\Data\OrderAddressExtensionInterfaceFactory;
use Magento\Sales\Api\Data\OrderAddressInterface;
use Temando\Shipping\Model\Config\ModuleConfigInterface;
use Temando\Shipping\Model\ResourceModel\Repository\AddressRepositoryInterface;

/**
 * @package Temando\Shipping\Plugin
 * @author  Christoph Aßmann <christoph.assmann@netresearch.de>
 * @license https://opensource.org/licenses/osl-3.0.php Open Software License (OSL 3.0)
 * @link    https://www.temando.com/
 */
class ToOrderAddressPlugin
{
    /**
     * @var ModuleConfigInterface
     */
    private $config;

    /**
     * @var AddressRepositoryInterface
     */
    private $addressRepository;

    /**
     * @var OrderAddressExtensionInterfaceFactory
     */
    private $addressExtensionFactory;

    /**
     * ToOrderAddressPlugin constructor.
     *
     * @param ModuleConfigInterface $config
     * @param AddressRepositoryInterface $addressRepository
     * @param OrderAddressExtensionInterfaceFactory $addressExtensionFactory
     */
    public function __construct(
        ModuleConfigInterface $config,
        AddressRepositoryInterface $addressRepository,
        OrderAddressExtensionInterfaceFactory $addressExtensionFactory
    ) {
        $this->config = $config;
        $this->addressRepository = $addressRepository;
        $this->addressExtensionFactory = $addressExtensionFactory;
    }

    /**
     * Copy extension attributes (dynamic fields selected in checkout) to order
     * shipping address. When the order gets placed in Magento, these checkout
     * fields must be transmitted to the API. There is no need to persist them
     * to data storage though.
     *
     * @see \Temando\Shipping\Plugin\Sales\OrderRepositoryPlugin::afterSave
     *
     * @param ToOrderAddress $subject
     * @param OrderAddressInterface $orderAddress
     * @param Address $quoteAddress
     * @return OrderAddressInterface
     */
    public function afterConvert(
        ToOrderAddress $subject,
        OrderAddressInterface $orderAddress,
        Address $quoteAddress
    ) {
        $storeId = $quoteAddress->getQuote()->getStoreId();
        if (!$this->config->isEnabled($storeId)) {
            return $orderAddress;
        }

        if ($quoteAddress->getAddressType() !== Address::ADDRESS_TYPE_SHIPPING) {
            // no need to handle billing addresses
            return $orderAddress;
        }

        try {
            $checkoutAddress = $this->addressRepository->getByQuoteAddressId($quoteAddress->getId());
        } catch (NoSuchEntityException $e) {
            // no additional fields selected during checkout
            return $orderAddress;
        }

        $extensionAttributes = $orderAddress->getExtensionAttributes();
        if (!$extensionAttributes instanceof OrderAddressExtensionInterface) {
            $extensionAttributes = $this->addressExtensionFactory->create();
        }

        $extensionAttributes->setCheckoutFields($checkoutAddress->getServiceSelection());
        $orderAddress->setExtensionAttributes($extensionAttributes);

        return $orderAddress;
    }
}