newsupernova/vendor/vertex/module-tax/Observer/VertexCalculationExtensionLoader.php

<?php
/**
 * @copyright  Vertex. All rights reserved.  https://www.vertexinc.com/
 * @author     Mediotype                     https://www.mediotype.com/
 */

namespace Vertex\Tax\Observer;

use Magento\Sales\Api\Data\InvoiceExtensionFactory;
use Magento\Sales\Api\OrderAddressRepositoryInterface;
use Magento\Sales\Api\OrderRepositoryInterface;
use Magento\Sales\Model\Order;
use Magento\Sales\Model\Order\Invoice;

/**
 * Handles loading and assignment of necessary data for Tax Calculation purposes
 */
class VertexCalculationExtensionLoader
{
    /** @var OrderAddressRepositoryInterface */
    private $addressRepository;

    /** @var InvoiceExtensionFactory */
    private $invoiceExtensionFactory;

    /** @var OrderRepositoryInterface */
    private $orderRepository;

    /**
     * @param InvoiceExtensionFactory $invoiceExtensionFactory
     * @param OrderAddressRepositoryInterface $addressRepository
     * @param OrderRepositoryInterface $orderRepository
     */
    public function __construct(
        InvoiceExtensionFactory $invoiceExtensionFactory,
        OrderAddressRepositoryInterface $addressRepository,
        OrderRepositoryInterface $orderRepository
    ) {
        $this->invoiceExtensionFactory = $invoiceExtensionFactory;
        $this->addressRepository = $addressRepository;
        $this->orderRepository = $orderRepository;
    }

    /**
     * Load the addresses and Order for an Invoice onto the Vertex Tax Calculation extension attributes
     *
     * This is necessary as there are two different ways an Invoice comes across the wire with respect to the addresses
     * attached to it:
     *
     * The first way is when it is an existing Order that is invoiced.  In this scenario, the Billing Address ID and the
     * Shipping Address ID are set on the Invoice.
     *
     * The second way is when it is a newly placed Order on the frontend, with an online payment method that captures
     * payment and triggers an invoice.  In this scenario, the Order's addresses are not yet in the database and are not
     * yet tied to the Invoice.
     *
     * Additionally, during the second scenario we do not want to load up the Order through the repository, as it can
     * cause additional issues.  In this scenario, we want to use the Order that is already attached to the Invoice.
     *
     * @param Invoice $originalInvoice
     * @return Invoice
     */
    public function loadOnInvoice(Invoice $originalInvoice)
    {
        $invoice = clone $originalInvoice;

        if (!$invoice->getExtensionAttributes()) {
            $invoice->setExtensionAttributes($this->invoiceExtensionFactory->create());
        } else {
            $invoice->setExtensionAttributes(clone $originalInvoice->getExtensionAttributes());
        }

        try {
            $order = $invoice->getOrder() ?: $this->orderRepository->get($invoice->getOrderId());
        } catch (\Exception $e) {
            $order = null;
        }

        $invoice->getExtensionAttributes()
            ->setVertexTaxCalculationOrder($order);

        if ($order && $order->getBillingAddress()) {
            $invoice->getExtensionAttributes()
                ->setVertexTaxCalculationBillingAddress($order->getBillingAddress());
        } elseif ($invoice->getBillingAddressId()) {
            $invoice->getExtensionAttributes()
                ->setVertexTaxCalculationBillingAddress(
                    $this->addressRepository->get($invoice->getBillingAddressId())
                );
        }

        if ($order instanceof Order && $order->getShippingAddress()) {
            $invoice->getExtensionAttributes()
                ->setVertexTaxCalculationShippingAddress($order->getShippingAddress());
        } elseif ($invoice->getShippingAddressId()) {
            $invoice->getExtensionAttributes()
                ->setVertexTaxCalculationShippingAddress(
                    $this->addressRepository->get($invoice->getShippingAddressId())
                );
        }

        return $invoice;
    }
}