Allow count with WHERE clause

Author: natanfelles

src/Model.php

@@ -15,6 +15,7 @@
 use Exception;
 use Framework\Cache\Cache;
 use Framework\Database\Database;
+use Framework\Database\Manipulation\Traits\Where;
 use Framework\Language\Language;
 use Framework\Pagination\Pager;
 use Framework\Validation\Debug\ValidationCollector;
@@ -401,23 +402,28 @@ protected function getDatabaseToWrite() : Database
     }
 
     /**
-     * A basic function to count all rows in the table.
+     * A basic function to count rows in the table.
+     *
+     * @param array<array<mixed>> $where Array in this format: `[['id', '=', 25]]`
+     *
+     * @see Where
      *
      * @return int
      */
-    public function count() : int
+    public function count(array $where = []) : int
     {
-        $result = $this->getDatabaseToRead()
+        $select = $this->getDatabaseToRead()
             ->select()
             ->expressions([
                 'count' => static function () : string {
                     return 'COUNT(*)';
                 },
             ])
-            ->from($this->getTable())
-            ->run()
-            ->fetch();
-        return $result->count; // @phpstan-ignore-line
+            ->from($this->getTable());
+        foreach ($where as $args) {
+            $select->where(...$args);
+        }
+        return $select->run()->fetch()->count; // @phpstan-ignore-line
     }
 
     /**

tests/ModelTest.php

@@ -397,6 +397,13 @@ public function testDeleteByWithCall() : void
     public function testCount() : void
     {
         self::assertSame(2, $this->model->count());
+        self::assertSame(1, $this->model->count([
+            ['id', '=', 1],
+        ]));
+        self::assertSame(2, $this->model->count([
+            ['id', 'is not null'],
+            ['id', '<', 3],
+        ]));
     }
 
     public function testPaginatedItems() : void