AttributeCollectionProvider.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryImportExport\Model\Export;

use Magento\Eav\Model\Entity\AttributeFactory;
use Magento\Framework\Data\Collection;
use Magento\ImportExport\Model\Export\Factory as CollectionFactory;
use Magento\InventoryApi\Api\Data\SourceItemInterface;
use Magento\InventoryImportExport\Model\Export\Source\StockStatus;

class AttributeCollectionProvider
{
    private $collection;

    private $attributeFactory;

    public function __construct(
        CollectionFactory $collectionFactory,
        AttributeFactory $attributeFactory
    ) {
        $this->collection = $collectionFactory->create(Collection::class);
        $this->attributeFactory = $attributeFactory;
    }

    public function get(): Collection
    {
        if (count($this->collection) === 0) {
            $sourceCodeAttribute = $this->attributeFactory->create();
            $sourceCodeAttribute->setId(SourceItemInterface::SOURCE_CODE);
            $sourceCodeAttribute->setDefaultFrontendLabel(SourceItemInterface::SOURCE_CODE);
            $sourceCodeAttribute->setAttributeCode(SourceItemInterface::SOURCE_CODE);
            $sourceCodeAttribute->setBackendType('varchar');
            $this->collection->addItem($sourceCodeAttribute);

            $skuAttribute = $this->attributeFactory->create();
            $skuAttribute->setId(SourceItemInterface::SKU);
            $skuAttribute->setBackendType('varchar');
            $skuAttribute->setDefaultFrontendLabel(SourceItemInterface::SKU);
            $skuAttribute->setAttributeCode(SourceItemInterface::SKU);
            $this->collection->addItem($skuAttribute);

            $statusIdAttribute = $this->attributeFactory->create();
            $statusIdAttribute->setId(SourceItemInterface::STATUS);
            $statusIdAttribute->setDefaultFrontendLabel(SourceItemInterface::STATUS);
            $statusIdAttribute->setAttributeCode(SourceItemInterface::STATUS);
            $statusIdAttribute->setBackendType('int');
            $statusIdAttribute->setFrontendInput('select');
            $statusIdAttribute->setSourceModel(StockStatus::class);
            $this->collection->addItem($statusIdAttribute);

            $quantityAttribute = $this->attributeFactory->create();
            $quantityAttribute->setId(SourceItemInterface::QUANTITY);
            $quantityAttribute->setBackendType('decimal');
            $quantityAttribute->setDefaultFrontendLabel(SourceItemInterface::QUANTITY);
            $quantityAttribute->setAttributeCode(SourceItemInterface::QUANTITY);
            $this->collection->addItem($quantityAttribute);
        }

        return $this->collection;
    }
}