CategoryResolver.php

Go to the documentation of this file.

<?php
namespace Magento\Setup\Fixtures;

use Magento\Catalog\Api\CategoryRepositoryInterface;
use Magento\Catalog\Model\CategoryFactory;
use Magento\Catalog\Model\ResourceModel\Category\CollectionFactory;
use Magento\Store\Model\StoreManager;

class CategoryResolver
{
    private $storeManager;

    private $categoryFactory;

    private $collectionFactory;

    private $categoryRepository;

    private $categories = [];

    public function __construct(
        StoreManager $storeManager,
        CategoryFactory $categoryFactory,
        CategoryRepositoryInterface $categoryRepository,
        CollectionFactory $collectionFactory
    ) {
        $this->storeManager = $storeManager;
        $this->categoryFactory = $categoryFactory;
        $this->collectionFactory = $collectionFactory;
        $this->categoryRepository = $categoryRepository;
    }

    public function getCategory($websiteId, $categoryName)
    {
        $categoryKey = $websiteId . $categoryName;

        if (!isset($this->categories[$categoryKey])) {
            $website = $this->storeManager->getWebsite($websiteId);
            $rootCategoryId = $website->getDefaultGroup()->getRootCategoryId();
            $website->getDefaultGroup()->getStoreId();
            $category = $this->collectionFactory->create()
                ->addFieldToFilter('parent_id', $rootCategoryId)
                ->addFieldToFilter('name', $categoryName)
                ->fetchItem();
            if ($category && $category->getId()) {
                $this->categories[$categoryKey] = $category->getId();
            } else {
                $category = $this->categoryFactory->create(
                    [
                        'data' => [
                            'parent_id' => $rootCategoryId,
                            'name' => $categoryName,
                            'position' => 1,
                            'is_active' => true,
                            'available_sort_by' => ['position', 'name'],
                            'url_key' => $categoryName . '-' . $websiteId
                        ]
                    ]
                );
                $category = $this->categoryRepository->save($category);
                $this->categories[$categoryKey] = $category->getId();
            }
        }

        return $this->categories[$categoryKey];
    }
}