Add alias

Author: erikaraujo

src/Tempest/Database/src/Builder/QueryBuilders/CountQueryBuilder.php

@@ -36,9 +36,15 @@ public function __construct(string|object $model, ?string $column = null)
 
     public function execute(mixed ...$bindings): int
     {
-        $key = "COUNT({$this->count->countArgument})";
+        return $this->build()->fetchFirst(...$bindings)[$this->count->getKey()];
+    }
 
-        return $this->build()->fetchFirst(...$bindings)[$key];
+    /** @return self<TModelClass> */
+    public function as(string $alias): self
+    {
+        $this->count->alias = $alias;
+
+        return $this;
     }
 
     /** @return self<TModelClass> */

src/Tempest/Database/src/QueryStatements/CountStatement.php

@@ -13,6 +13,8 @@ final class CountStatement implements QueryStatement
 {
     public readonly string $countArgument;
 
+    public ?string $alias = null;
+
     public function __construct(
         public readonly TableDefinition $table,
         public ?string $column = null,
@@ -26,7 +28,11 @@ public function __construct(
     public function compile(DatabaseDialect $dialect): string
     {
         $query = arr([
-            sprintf('SELECT COUNT(%s)', $this->countArgument),
+            sprintf(
+                'SELECT COUNT(%s)%s',
+                $this->countArgument,
+                $this->alias ? " AS `{$this->alias}`" : '',
+            ),
             sprintf('FROM `%s`', $this->table->name),
         ]);
 
@@ -38,4 +44,13 @@ public function compile(DatabaseDialect $dialect): string
 
         return $query->implode(PHP_EOL);
     }
+
+    public function getKey(): string
+    {
+        if ($this->alias !== null) {
+            return $this->alias;
+        }
+
+        return "COUNT({$this->countArgument})";
+    }
 }

src/Tempest/Database/tests/QueryStatements/CountStatementTest.php

@@ -37,6 +37,27 @@ public function test_count_statement(): void
         $this->assertSame($expected, $statement->compile(DatabaseDialect::SQLITE));
     }
 
+    public function test_count_statement_with_alias(): void
+    {
+        $tableDefinition = new TableDefinition('foo', 'bar');
+
+        $statement = new CountStatement(
+            table: $tableDefinition,
+            column: null,
+        );
+
+        $statement->alias = 'foobar';
+
+        $expected = <<<SQL
+        SELECT COUNT(*) AS `foobar`
+        FROM `foo`
+        SQL;
+
+        $this->assertSame($expected, $statement->compile(DatabaseDialect::MYSQL));
+        $this->assertSame($expected, $statement->compile(DatabaseDialect::POSTGRESQL));
+        $this->assertSame($expected, $statement->compile(DatabaseDialect::SQLITE));
+    }
+
     public function test_count_statement_with_specified_column(): void
     {
         $tableDefinition = new TableDefinition('foo', 'bar');

tests/Integration/Database/Builder/CountQueryBuilderTest.php

@@ -15,7 +15,7 @@
  */
 final class CountQueryBuilderTest extends FrameworkIntegrationTestCase
 {
-    public function test_count_query(): void
+    public function test_simple_count_query(): void
     {
         $query = query('chapters')
             ->count()
@@ -39,6 +39,31 @@ public function test_count_query(): void
         $this->assertSame(['Timeline Taxi', '1', '2025-01-01'], $bindings);
     }
 
+    public function test_count_query_with_alias(): void
+    {
+        $query = query('chapters')
+            ->count()
+            ->as('total')
+            ->where('`title` = ?', 'Timeline Taxi')
+            ->andWhere('`index` <> ?', '1')
+            ->orWhere('`createdAt` > ?', '2025-01-01')
+            ->build();
+
+        $expected = <<<SQL
+        SELECT COUNT(*) AS `total`
+        FROM `chapters`
+        WHERE `title` = ?
+        AND `index` <> ?
+        OR `createdAt` > ?
+        SQL;
+
+        $sql = $query->getSql();
+        $bindings = $query->bindings;
+
+        $this->assertSame($expected, $sql);
+        $this->assertSame(['Timeline Taxi', '1', '2025-01-01'], $bindings);
+    }
+
     public function test_count_query_with_specified_field(): void
     {
         $query = query('chapters')->count('title')->build();