feat(database): add HasConditions to query builders (#1154)

Co-authored-by: Brent Roose <[email protected]>

Author: erikaraujo

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

@@ -7,6 +7,7 @@
 use Tempest\Database\Query;
 use Tempest\Database\QueryStatements\DeleteStatement;
 use Tempest\Database\QueryStatements\WhereStatement;
+use Tempest\Support\Conditions\HasConditions;
 
 use function Tempest\Database\model;
 
@@ -15,6 +16,8 @@
  */
 final class DeleteQueryBuilder
 {
+    use HasConditions;
+
     private DeleteStatement $delete;
 
     private array $bindings = [];

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

@@ -17,6 +17,7 @@
 use Tempest\Database\QueryStatements\WhereStatement;
 use Tempest\Database\Virtual;
 use Tempest\Support\Arr\ImmutableArray;
+use Tempest\Support\Conditions\HasConditions;
 
 use function Tempest\map;
 use function Tempest\reflect;
@@ -27,6 +28,8 @@
  */
 final class SelectQueryBuilder
 {
+    use HasConditions;
+
     /** @var class-string<TModelClass> $modelClass */
     private readonly string $modelClass;
 

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

@@ -10,12 +10,15 @@
 use Tempest\Mapper\SerializerFactory;
 use Tempest\Reflection\ClassReflector;
 use Tempest\Support\Arr\ImmutableArray;
+use Tempest\Support\Conditions\HasConditions;
 
 use function Tempest\Database\model;
 use function Tempest\Support\arr;
 
 final class UpdateQueryBuilder
 {
+    use HasConditions;
+
     private UpdateStatement $update;
 
     private array $bindings = [];

tests/Integration/Database/Builder/DeleteQueryBuilderTest.php

@@ -2,6 +2,7 @@
 
 namespace Tests\Tempest\Integration\Database\Builder;
 
+use Tempest\Database\Builder\QueryBuilders\DeleteQueryBuilder;
 use Tempest\Database\Id;
 use Tests\Tempest\Fixtures\Modules\Books\Models\Author;
 use Tests\Tempest\Integration\FrameworkIntegrationTestCase;
@@ -68,4 +69,32 @@ public function test_delete_on_model_object(): void
             $query->bindings['id']->id,
         );
     }
+
+    public function test_delete_on_plain_table_with_conditions(): void
+    {
+        $query = query('foo')
+            ->delete()
+            ->when(
+                true,
+                fn (DeleteQueryBuilder $query) => $query->where('`bar` = ?', 'boo'),
+            )
+            ->when(
+                false,
+                fn (DeleteQueryBuilder $query) => $query->where('`bar` = ?', 'foo'),
+            )
+            ->build();
+
+        $this->assertSame(
+            <<<SQL
+            DELETE FROM `foo`
+            WHERE `bar` = ?
+            SQL,
+            $query->getSql(),
+        );
+
+        $this->assertSame(
+            'boo',
+            $query->bindings[0],
+        );
+    }
 }

tests/Integration/Database/Builder/SelectQueryBuilderTest.php

@@ -4,6 +4,7 @@
 
 namespace Tests\Tempest\Integration\Database\Builder;
 
+use Tempest\Database\Builder\QueryBuilders\SelectQueryBuilder;
 use Tempest\Database\Migrations\CreateMigrationsTable;
 use Tests\Tempest\Fixtures\Migrations\CreateAuthorTable;
 use Tests\Tempest\Fixtures\Migrations\CreateBookTable;
@@ -192,4 +193,41 @@ public function test_raw(): void
         $this->assertCount(1, $books);
         $this->assertSame('A', $books[0]->title);
     }
+
+    public function test_select_query_with_conditions(): void
+    {
+        $query = query('chapters')
+            ->select('title', 'index')
+            ->when(
+                true,
+                fn (SelectQueryBuilder $query) => $query
+                    ->where('`title` = ?', 'Timeline Taxi')
+                    ->andWhere('`index` <> ?', '1')
+                    ->orWhere('`createdAt` > ?', '2025-01-01'),
+            )
+            ->when(
+                false,
+                fn (SelectQueryBuilder $query) => $query
+                    ->where('`title` = ?', 'Timeline Uber')
+                    ->andWhere('`index` <> ?', '2')
+                    ->orWhere('`createdAt` > ?', '2025-01-02'),
+            )
+            ->orderBy('`index` ASC')
+            ->build();
+
+        $expected = <<<SQL
+        SELECT `title`, `index`
+        FROM `chapters`
+        WHERE `title` = ?
+        AND `index` <> ?
+        OR `createdAt` > ?
+        ORDER BY `index` ASC
+        SQL;
+
+        $sql = $query->getSql();
+        $bindings = $query->bindings;
+
+        $this->assertSame($expected, $sql);
+        $this->assertSame(['Timeline Taxi', '1', '2025-01-01'], $bindings);
+    }
 }

tests/Integration/Database/Builder/UpdateQueryBuilderTest.php

@@ -2,6 +2,7 @@
 
 namespace Tests\Tempest\Integration\Database\Builder;
 
+use Tempest\Database\Builder\QueryBuilders\UpdateQueryBuilder;
 use Tempest\Database\Exceptions\CannotUpdateHasManyRelation;
 use Tempest\Database\Exceptions\InvalidUpdateStatement;
 use Tempest\Database\Id;
@@ -213,4 +214,36 @@ public function test_attach_new_has_many_relation_on_update(): void
         //            )
         //            ->build();
     }
+
+    public function test_update_on_plain_table_with_conditions(): void
+    {
+        $query = query('chapters')
+            ->update(
+                title: 'Chapter 01',
+                index: 1,
+            )
+            ->when(
+                true,
+                fn (UpdateQueryBuilder $query) => $query->where('`id` = ?', 10),
+            )
+            ->when(
+                false,
+                fn (UpdateQueryBuilder $query) => $query->where('`id` = ?', 20),
+            )
+            ->build();
+
+        $this->assertSame(
+            <<<SQL
+            UPDATE `chapters`
+            SET `title` = ?, `index` = ?
+            WHERE `id` = ?
+            SQL,
+            $query->getSql(),
+        );
+
+        $this->assertSame(
+            ['Chapter 01', 1, 10],
+            $query->bindings,
+        );
+    }
 }