TaxClassProcessor.php

Go to the documentation of this file.

<?php
namespace Magento\CatalogImportExport\Model\Import\Product;

use Magento\CatalogImportExport\Model\Import\Product\Type\AbstractType;
use Magento\Tax\Model\ClassModel;

class TaxClassProcessor
{
    const ATRR_CODE = 'tax_class_id';

    protected $taxClasses;

    protected $collectionFactory;

    protected $classModelFactory;

    public function __construct(
        \Magento\Tax\Model\ResourceModel\TaxClass\CollectionFactory $collectionFactory,
        \Magento\Tax\Model\ClassModelFactory $classModelFactory
    ) {
        $this->collectionFactory = $collectionFactory;
        $this->classModelFactory = $classModelFactory;
        $this->initTaxClasses();
    }

    protected function initTaxClasses()
    {
        if (empty($this->taxClasses)) {
            $collection = $this->collectionFactory->create();
            $collection->addFieldToFilter('class_type', ClassModel::TAX_CLASS_TYPE_PRODUCT);
            /* @var $collection \Magento\Tax\Model\ResourceModel\TaxClass\Collection */
            foreach ($collection as $taxClass) {
                $this->taxClasses[$taxClass->getClassName()] = $taxClass->getId();
            }
        }
        return $this;
    }

    protected function createTaxClass($taxClassName, AbstractType $productTypeModel)
    {
        $taxClass = $this->classModelFactory->create();
        $taxClass->setClassType(ClassModel::TAX_CLASS_TYPE_PRODUCT);
        $taxClass->setClassName($taxClassName);
        $taxClass->save();

        $id = $taxClass->getId();

        $productTypeModel->addAttributeOption(self::ATRR_CODE, $id, $id);

        return $id;
    }

    public function upsertTaxClass($taxClassName, AbstractType $productTypeModel)
    {
        if (!isset($this->taxClasses[$taxClassName])) {
            $this->taxClasses[$taxClassName] = $this->createTaxClass($taxClassName, $productTypeModel);
        }

        return $this->taxClasses[$taxClassName];
    }
}