Add new methods to check if category flat is enabled

adampmoss · May 8, 2017 · c9eb855 · c9eb855
1 parent 9bbac94
commit c9eb855
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/app/code/community/Creare/CreareSeoSitemap/Block/Sitemap.php b/app/code/community/Creare/CreareSeoSitemap/Block/Sitemap.php
@@ -6,6 +6,7 @@
     protected $store;
     protected $sitemapHelper;
     protected $xmlSitemaps = false;
+    protected $_flatEnabled = array();
 
     public function __construct(array $args)
     {
@@ -20,6 +21,36 @@ public function __construct(array $args)
         parent::__construct($args);
     }
 
+    /**
+     * Retrieve Catalog Product Flat Helper object
+     *
+     * @return Mage_Catalog_Helper_Product_Flat
+     */
+    public function getFlatHelper()
+    {
+        return Mage::helper('catalog/category_flat');
+    }
+
+    /**
+     * Retrieve is flat enabled flag
+     * Return always false if magento run admin
+     *
+     * @return bool
+     */
+    public function isEnabledFlat()
+    {
+        // Flat Data can be used only on frontend
+        if (Mage::app()->getStore()->isAdmin()) {
+            return false;
+        }
+        $storeId = $this->getStoreId();
+        if (!isset($this->_flatEnabled[$storeId])) {
+            $flatHelper = $this->getFlatHelper();
+            $this->_flatEnabled[$storeId] = $flatHelper->isAvailable() && $flatHelper->isBuilt($storeId);
+        }
+        return $this->_flatEnabled[$storeId];
+    }
+
     /**
      * @return Creare_CreareSeoSitemap_Helper_Data
      */