UpdateTaxRegionId.php

Go to the documentation of this file.

<?php
namespace Magento\Tax\Setup\Patch\Data;

use Magento\Directory\Model\RegionFactory;
use Magento\Framework\Api\Search\SearchCriteriaFactory;
use Magento\Tax\Api\TaxRateRepositoryInterface;
use Magento\Framework\App\ResourceConnection;
use Magento\Framework\Setup\Patch\DataPatchInterface;
use Magento\Framework\Setup\Patch\PatchVersionInterface;
use Magento\Tax\Setup\TaxSetupFactory;

class UpdateTaxRegionId implements DataPatchInterface, PatchVersionInterface
{
    private $moduleDataSetup;

    private $taxRateRepository;

    private $searchCriteriaFactory;

    private $regionFactory;

    public function __construct(
        \Magento\Framework\Setup\ModuleDataSetupInterface $moduleDataSetup,
        TaxRateRepositoryInterface $taxRateRepository,
        SearchCriteriaFactory $searchCriteriaFactory,
        \Magento\Directory\Model\RegionFactory $regionFactory
    ) {
        $this->moduleDataSetup = $moduleDataSetup;
        $this->taxRateRepository = $taxRateRepository;
        $this->searchCriteriaFactory = $searchCriteriaFactory;
        $this->regionFactory = $regionFactory;
    }

    public function apply()
    {
        $this->moduleDataSetup->getConnection()->startSetup();

        //Update the tax_region_id
        $taxRateList = $this->taxRateRepository->getList($this->searchCriteriaFactory->create());
        foreach ($taxRateList->getItems() as $taxRateData) {
            $regionCode = $this->parseRegionFromTaxCode($taxRateData->getCode());
            if ($regionCode) {
                $region = $this->regionFactory->create();
                $region->loadByCode($regionCode, $taxRateData->getTaxCountryId());
                if ($taxRateData->getTaxPostcode() === null) {
                    $taxRateData->setTaxPostcode('*');
                }
                $taxRateData->setTaxRegionId($region->getRegionId());
                $this->taxRateRepository->save($taxRateData);
            }
        }
        $this->moduleDataSetup->getConnection()->endSetup();
    }

    public static function getDependencies()
    {
        return [
            UpdateTaxClassAttributeVisibility::class
        ];
    }

    public static function getVersion()
    {
        return '2.0.3';
    }

    public function getAliases()
    {
        return [];
    }

    private function parseRegionFromTaxCode($taxCode)
    {
        $result = '';
        $parts = explode('-', $taxCode, 3);

        if (isset($parts[1])) {
            $result = $parts[1];
        }

        return $result;
    }
}