folder and content fake

folder and content loops

core/lib/Thelia/Model/Folder.php

@@ -4,6 +4,37 @@ namespace Thelia\Model;
 
 use Thelia\Model\Base\Folder as BaseFolder;
 
-class Folder extends BaseFolder {
+class Folder extends BaseFolder
+{
+    /**
+     * @return int number of contents for the folder
+     */
+    public function countChild()
+    {
+        return FolderQuery::countChild($this->getId());
+    }
 
+    /**
+     *
+     * count all products for current category and sub categories
+     *
+     * @return int
+     */
+    public function countAllContents()
+    {
+        $children = FolderQuery::findAllChild($this->getId());
+        array_push($children, $this);
+
+        $contentsCount = 0;
+
+        foreach($children as $child)
+        {
+            $contentsCount += ProductQuery::create()
+                ->filterByCategory($child)
+                ->count();
+        }
+
+        return $contentsCount;
+
+    }
 }