1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
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;
/**
* Provide category id. Find category in default store group by specified website and category name or create new one
*/
class CategoryResolver
{
/**
* @var StoreManager
*/
private $storeManager;
/**
* @var CategoryFactory
*/
private $categoryFactory;
/**
* @var CollectionFactory
*/
private $collectionFactory;
/**
* @var CategoryRepositoryInterface
*/
private $categoryRepository;
/**
* @var array
*/
private $categories = [];
/**
* @param StoreManager $storeManager
* @param CategoryFactory $categoryFactory
* @param CategoryRepositoryInterface $categoryRepository
* @param CollectionFactory $collectionFactory
* @internal param Category $category
*/
public function __construct(
StoreManager $storeManager,
CategoryFactory $categoryFactory,
CategoryRepositoryInterface $categoryRepository,
CollectionFactory $collectionFactory
) {
$this->storeManager = $storeManager;
$this->categoryFactory = $categoryFactory;
$this->collectionFactory = $collectionFactory;
$this->categoryRepository = $categoryRepository;
}
/**
* Get category id
*
* @param int $websiteId
* @param string $categoryName
* @return int
*/
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];
}
}