ACP2E-2300: allow in sitemap only categories and products from public shared catalog.

Author: alzota

app/code/Magento/Sitemap/Model/ResourceModel/Catalog/Category.php

@@ -6,7 +6,8 @@
 namespace Magento\Sitemap\Model\ResourceModel\Catalog;
 
 use Magento\CatalogUrlRewrite\Model\CategoryUrlRewriteGenerator;
-use Magento\Framework\DB\Select;
+use Magento\Framework\App\ObjectManager;
+use Magento\Framework\Event\ManagerInterface;
 
 /**
  * Sitemap resource catalog collection model
@@ -19,7 +20,7 @@ class Category extends \Magento\Framework\Model\ResourceModel\Db\AbstractDb
     /**
      * Collection Zend Db select
      *
-     * @var Select
+     * @var \Magento\Framework\DB\Select
      */
     protected $_select;
 
@@ -46,24 +47,32 @@ class Category extends \Magento\Framework\Model\ResourceModel\Db\AbstractDb
      */
     protected $metadataPool;
 
+    /**
+     * @var ManagerInterface
+     */
+    private $eventManager;
+
     /**
      * @param \Magento\Framework\Model\ResourceModel\Db\Context $context
      * @param \Magento\Store\Model\StoreManagerInterface $storeManager
      * @param \Magento\Catalog\Model\ResourceModel\Category $categoryResource
      * @param \Magento\Framework\EntityManager\MetadataPool $metadataPool
      * @param string $connectionName
+     * @param ManagerInterface|null $eventManager
      */
     public function __construct(
         \Magento\Framework\Model\ResourceModel\Db\Context $context,
         \Magento\Store\Model\StoreManagerInterface $storeManager,
         \Magento\Catalog\Model\ResourceModel\Category $categoryResource,
         \Magento\Framework\EntityManager\MetadataPool $metadataPool,
-        $connectionName = null
+        $connectionName = null,
+        ManagerInterface $eventManager = null
     ) {
         $this->_storeManager = $storeManager;
         $this->_categoryResource = $categoryResource;
         parent::__construct($context, $connectionName);
         $this->metadataPool = $metadataPool;
+        $this->eventManager = $eventManager ?? ObjectManager::getInstance()->get(ManagerInterface::class);
     }
 
     /**
@@ -123,7 +132,12 @@ public function getCollection($storeId)
 
         $this->_addFilter($storeId, 'is_active', 1);
 
-        $query = $connection->query($this->prepareSelectStatement($this->_select));
+        $this->eventManager->dispatch(
+            'sitemap_category_select_init',
+            ['select' => $this->_select]
+        );
+
+        $query = $connection->query($this->_select);
         while ($row = $query->fetch()) {
             $category = $this->_prepareCategory($row);
             $categories[$category->getId()] = $category;
@@ -132,17 +146,6 @@ public function getCollection($storeId)
         return $categories;
     }
 
-    /**
-     * Allow to modify select statement with plugins
-     *
-     * @param Select $select
-     * @return Select
-     */
-    public function prepareSelectStatement(Select $select)
-    {
-        return $select;
-    }
-
     /**
      * Prepare category
      *
@@ -167,14 +170,14 @@ protected function _prepareCategory(array $categoryRow)
      * @param string $attributeCode
      * @param mixed $value
      * @param string $type
-     * @return Select|bool
+     * @return \Magento\Framework\DB\Select|bool
      */
     protected function _addFilter($storeId, $attributeCode, $value, $type = '=')
     {
         $meta = $this->metadataPool->getMetadata(\Magento\Catalog\Api\Data\CategoryInterface::class);
         $linkField = $meta->getLinkField();
 
-        if (!$this->_select instanceof Select) {
+        if (!$this->_select instanceof \Magento\Framework\DB\Select) {
             return false;
         }