From 4ec10c4faa8beaac15480093b9a3343a3e022fbd Mon Sep 17 00:00:00 2001
From: Diatrex <diahshalabi@hotmail.com>
Date: Tue, 11 Aug 2020 17:15:31 +0300
Subject: [PATCH] #1893 [dress-theme] GG Anasayfa Giydirme

---
 .../src/Category/CategoryCriteria.php         | 27 +++++++++++++++++++
 .../src/Category/CategoryRepository.php       |  9 ++++++-
 2 files changed, 35 insertions(+), 1 deletion(-)

diff --git a/addons/default/visiosoft/cats-module/src/Category/CategoryCriteria.php b/addons/default/visiosoft/cats-module/src/Category/CategoryCriteria.php
index 71865b8b6..b4712eb7b 100644
--- a/addons/default/visiosoft/cats-module/src/Category/CategoryCriteria.php
+++ b/addons/default/visiosoft/cats-module/src/Category/CategoryCriteria.php
@@ -1,8 +1,35 @@
 <?php namespace Visiosoft\CatsModule\Category;
 
 use Anomaly\Streams\Platform\Entry\EntryCriteria;
+use Anomaly\Streams\Platform\Stream\Contract\StreamInterface;
+use Illuminate\Database\Eloquent\Builder;
+use Visiosoft\CatsModule\Category\Contract\CategoryRepositoryInterface;
 
 class CategoryCriteria extends EntryCriteria
 {
 
+    private $categoryRepository;
+
+    public function __construct(
+        Builder $query, StreamInterface $stream, $method,
+        CategoryRepositoryInterface $categoryRepository
+    )
+    {
+        parent::__construct($query, $stream, $method);
+        $this->categoryRepository = $categoryRepository;
+    }
+
+    public function getMainCats() {
+        $mainCats = $this->categoryRepository->newQuery()
+            ->whereNull('parent_category_id')
+            ->orderBy('sort_order')
+            ->get();
+
+        foreach ($mainCats as $cat) {
+            $subCount = $this->categoryRepository->newQuery()->where('parent_category_id', $cat->id)->count();
+            $cat->hasChild = !!$subCount;
+        }
+
+        return $mainCats;
+    }
 }
diff --git a/addons/default/visiosoft/cats-module/src/Category/CategoryRepository.php b/addons/default/visiosoft/cats-module/src/Category/CategoryRepository.php
index a403f9cbd..3f7995a8b 100644
--- a/addons/default/visiosoft/cats-module/src/Category/CategoryRepository.php
+++ b/addons/default/visiosoft/cats-module/src/Category/CategoryRepository.php
@@ -50,9 +50,16 @@ class CategoryRepository extends EntryRepository implements CategoryRepositoryIn
 
     public function getSubCatById($id)
     {
-        return $this->model->newQuery()
+        $cats = $this->model->newQuery()
             ->where('parent_category_id', $id)
             ->get();
+
+        foreach ($cats as $cat) {
+            $subCount = $this->model->newQuery()->where('parent_category_id', $cat->id)->count();
+            $cat->hasChild = !!$subCount;
+        }
+
+        return $cats;
     }
 
     public function getSingleCat($id)