add visibleOnly

thelia · Feb 21, 2017 · 34f3a35 · 34f3a35
1 parent 70b243e
commit 34f3a35
Showing 1 changed file with 9 additions and 17 deletions.
diff --git a/core/lib/Thelia/Model/Category.php b/core/lib/Thelia/Model/Category.php
@@ -44,17 +44,21 @@ public function getRewrittenUrlViewName()
      *
      * @return int
      */
-    public function countAllProducts()
+    public function countAllProducts($visibleOnly = false)
     {
         $children = CategoryQuery::findAllChild($this->getId());
         array_push($children, $this);
 
         $countProduct = 0;
 
         foreach ($children as $child) {
-            $countProduct += ProductQuery::create()
-                ->filterByCategory($child)
-                ->count();
+            $req = ProductQuery::create();
+            $req->filterByCategory($child);
+            if($visibleOnly) {
+                $req->filterByVisible(true);
+            }
+
+            $countProduct += $req->count();
         }
 
         return $countProduct;
@@ -70,19 +74,7 @@ public function countAllProducts()
      */
     public function countAllProductsVisibleOnly()
     {
-        $children = CategoryQuery::findAllChild($this->getId());
-        array_push($children, $this);
-
-        $countProduct = 0;
-
-        foreach ($children as $child) {
-            $countProduct += ProductQuery::create()
-                ->filterByCategory($child)
-                ->filterByVisible(true)
-                ->count();
-        }
-
-        return $countProduct;
+        return countAllProducts(true);
     }
 
     /**