55 */
66namespace Magento \Catalog \Model \Category ;
77
8+ use Magento \Catalog \Api \Data \CategoryTreeInterface ;
9+ use Magento \Catalog \Api \Data \CategoryTreeInterfaceFactory ;
10+ use Magento \Catalog \Model \Category ;
11+ use Magento \Catalog \Model \ResourceModel \Category \Collection ;
12+ use Magento \Catalog \Model \ResourceModel \Category \TreeFactory ;
13+ use Magento \Framework \App \ObjectManager ;
814use Magento \Framework \Data \Tree \Node ;
15+ use Magento \Framework \Exception \LocalizedException ;
16+ use Magento \Framework \Exception \NoSuchEntityException ;
17+ use Magento \Store \Model \StoreManagerInterface ;
918
1019/**
1120 * Retrieve category data represented in tree structure
@@ -18,54 +27,54 @@ class Tree
1827 protected $ categoryTree ;
1928
2029 /**
21- * @var \Magento\Store\Model\ StoreManagerInterface
30+ * @var StoreManagerInterface
2231 */
2332 protected $ storeManager ;
2433
2534 /**
26- * @var \Magento\Catalog\Model\ResourceModel\Category\ Collection
35+ * @var Collection
2736 */
2837 protected $ categoryCollection ;
2938
3039 /**
31- * @var \Magento\Catalog\Api\Data\ CategoryTreeInterfaceFactory
40+ * @var CategoryTreeInterfaceFactory
3241 */
3342 protected $ treeFactory ;
3443
3544 /**
36- * @var \Magento\Catalog\Model\ResourceModel\Category\ TreeFactory
45+ * @var TreeFactory
3746 */
3847 private $ treeResourceFactory ;
3948
4049 /**
4150 * @param \Magento\Catalog\Model\ResourceModel\Category\Tree $categoryTree
42- * @param \Magento\Store\Model\ StoreManagerInterface $storeManager
43- * @param \Magento\Catalog\Model\ResourceModel\Category\ Collection $categoryCollection
44- * @param \Magento\Catalog\Api\Data\ CategoryTreeInterfaceFactory $treeFactory
45- * @param \Magento\Catalog\Model\ResourceModel\Category\ TreeFactory|null $treeResourceFactory
51+ * @param StoreManagerInterface $storeManager
52+ * @param Collection $categoryCollection
53+ * @param CategoryTreeInterfaceFactory $treeFactory
54+ * @param TreeFactory|null $treeResourceFactory
4655 */
4756 public function __construct (
4857 \Magento \Catalog \Model \ResourceModel \Category \Tree $ categoryTree ,
49- \ Magento \ Store \ Model \ StoreManagerInterface $ storeManager ,
50- \ Magento \ Catalog \ Model \ ResourceModel \ Category \ Collection $ categoryCollection ,
51- \ Magento \ Catalog \ Api \ Data \ CategoryTreeInterfaceFactory $ treeFactory ,
52- \ Magento \ Catalog \ Model \ ResourceModel \ Category \ TreeFactory $ treeResourceFactory = null
58+ StoreManagerInterface $ storeManager ,
59+ Collection $ categoryCollection ,
60+ CategoryTreeInterfaceFactory $ treeFactory ,
61+ TreeFactory $ treeResourceFactory = null
5362 ) {
5463 $ this ->categoryTree = $ categoryTree ;
5564 $ this ->storeManager = $ storeManager ;
5665 $ this ->categoryCollection = $ categoryCollection ;
5766 $ this ->treeFactory = $ treeFactory ;
58- $ this ->treeResourceFactory = $ treeResourceFactory ?? \ Magento \ Framework \ App \ ObjectManager::getInstance ()
59- ->get (\ Magento \ Catalog \ Model \ ResourceModel \ Category \ TreeFactory::class);
67+ $ this ->treeResourceFactory = $ treeResourceFactory ?? ObjectManager::getInstance ()
68+ ->get (TreeFactory::class);
6069 }
6170
6271 /**
6372 * Get root node by category.
6473 *
65- * @param \Magento\Catalog\Model\ Category|null $category
74+ * @param Category|null $category
6675 * @return Node|null
67- * @throws \Magento\Framework\Exception\ LocalizedException
68- * @throws \Magento\Framework\Exception\ NoSuchEntityException
76+ * @throws LocalizedException
77+ * @throws NoSuchEntityException
6978 */
7079 public function getRootNode ($ category = null )
7180 {
@@ -86,28 +95,28 @@ public function getRootNode($category = null)
8695 /**
8796 * Get node by category.
8897 *
89- * @param \Magento\Catalog\Model\ Category $category
98+ * @param Category $category
9099 * @return Node
91- * @throws \Magento\Framework\Exception\ LocalizedException
92- * @throws \Magento\Framework\Exception\ NoSuchEntityException
100+ * @throws LocalizedException
101+ * @throws NoSuchEntityException
93102 */
94- protected function getNode (\ Magento \ Catalog \ Model \ Category $ category )
103+ protected function getNode (Category $ category )
95104 {
96105 $ nodeId = $ category ->getId ();
97106 $ categoryTree = $ this ->treeResourceFactory ->create ();
98107 $ node = $ categoryTree ->loadNode ($ nodeId );
99108 $ node ->loadChildren ();
100109 $ this ->prepareCollection ();
101- $ this -> categoryTree ->addCollectionData ($ this ->categoryCollection );
110+ $ categoryTree ->addCollectionData ($ this ->categoryCollection );
102111 return $ node ;
103112 }
104113
105114 /**
106115 * Prepare category collection.
107116 *
108117 * @return void
109- * @throws \Magento\Framework\Exception\ LocalizedException
110- * @throws \Magento\Framework\Exception\ NoSuchEntityException
118+ * @throws LocalizedException
119+ * @throws NoSuchEntityException
111120 */
112121 protected function prepareCollection ()
113122 {
@@ -128,16 +137,16 @@ protected function prepareCollection()
128137 /**
129138 * Get tree by node.
130139 *
131- * @param \Magento\Framework\Data\Tree\ Node $node
140+ * @param Node $node
132141 * @param int $depth
133142 * @param int $currentLevel
134- * @return \Magento\Catalog\Api\Data\ CategoryTreeInterface
143+ * @return CategoryTreeInterface
135144 */
136145 public function getTree ($ node , $ depth = null , $ currentLevel = 0 )
137146 {
138- /** @var \Magento\Catalog\Api\Data\ CategoryTreeInterface[] $children */
147+ /** @var CategoryTreeInterface[] $children */
139148 $ children = $ this ->getChildren ($ node , $ depth , $ currentLevel );
140- /** @var \Magento\Catalog\Api\Data\ CategoryTreeInterface $tree */
149+ /** @var CategoryTreeInterface $tree */
141150 $ tree = $ this ->treeFactory ->create ();
142151 $ tree ->setId ($ node ->getId ())
143152 ->setParentId ($ node ->getParentId ())
@@ -153,10 +162,10 @@ public function getTree($node, $depth = null, $currentLevel = 0)
153162 /**
154163 * Get node children.
155164 *
156- * @param \Magento\Framework\Data\Tree\ Node $node
165+ * @param Node $node
157166 * @param int $depth
158167 * @param int $currentLevel
159- * @return \Magento\Catalog\Api\Data\ CategoryTreeInterface[]|[]
168+ * @return CategoryTreeInterface[]|[]
160169 */
161170 protected function getChildren ($ node , $ depth , $ currentLevel )
162171 {
0 commit comments