From 4a95c6bea300ed44088f787e9fa646111e747297 Mon Sep 17 00:00:00 2001
From: Sergei Tigrov <rrr-r@ya.ru>
Date: Fri, 12 Apr 2024 18:44:49 +0700
Subject: [PATCH] Update test according to main PR (#289)

Co-authored-by: Sergei Predvoditelev <sergei@predvoditelev.ru>
---
 CHANGELOG.md                         |  4 +--
 rector.php                           |  5 +++-
 src/Builder/ExpressionBuilder.php    | 16 ++++++++++
 src/DQLQueryBuilder.php              |  3 ++
 src/SqlParser.php                    | 45 ++++++++++++++++++++++++++++
 tests/CommandTest.php                |  5 ++--
 tests/Provider/SqlParserProvider.php | 30 +++++++++++++++++++
 tests/QueryBuilderTest.php           |  7 +++--
 tests/SqlParserTest.php              | 25 ++++++++++++++++
 9 files changed, 132 insertions(+), 8 deletions(-)
 create mode 100644 src/Builder/ExpressionBuilder.php
 create mode 100644 src/SqlParser.php
 create mode 100644 tests/Provider/SqlParserProvider.php
 create mode 100644 tests/SqlParserTest.php

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 39a530e98..abb54123d 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,8 +1,8 @@
 # SQLite driver for Yii Database Change Log
 
-## 1.2.1 under development
+## 2.0.0 under development
 
-- no changes in this release.
+- Enh #289: Implement `SqlParser` and `ExpressionBuilder` driver classes (@Tigrov)
 
 ## 1.2.0 March 21, 2024
 
diff --git a/rector.php b/rector.php
index 6d4e4d49c..8cbdabbd1 100644
--- a/rector.php
+++ b/rector.php
@@ -11,7 +11,10 @@
 
     $rectorConfig->paths([
         __DIR__ . '/src',
-        __DIR__ . '/tests',
+        /**
+         * Disabled ./tests directory due to different branches with main package when testing
+         */
+        // __DIR__ . '/tests',
     ]);
 
     // register a single rule
diff --git a/src/Builder/ExpressionBuilder.php b/src/Builder/ExpressionBuilder.php
new file mode 100644
index 000000000..96ac22ee4
--- /dev/null
+++ b/src/Builder/ExpressionBuilder.php
@@ -0,0 +1,16 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Yiisoft\Db\Sqlite\Builder;
+
+use Yiisoft\Db\Expression\AbstractExpressionBuilder;
+use Yiisoft\Db\Sqlite\SqlParser;
+
+final class ExpressionBuilder extends AbstractExpressionBuilder
+{
+    protected function createSqlParser(string $sql): SqlParser
+    {
+        return new SqlParser($sql);
+    }
+}
diff --git a/src/DQLQueryBuilder.php b/src/DQLQueryBuilder.php
index 1faa2b0c0..6d41350ec 100644
--- a/src/DQLQueryBuilder.php
+++ b/src/DQLQueryBuilder.php
@@ -4,6 +4,7 @@
 
 namespace Yiisoft\Db\Sqlite;
 
+use Yiisoft\Db\Expression\Expression;
 use Yiisoft\Db\Expression\ExpressionBuilderInterface;
 use Yiisoft\Db\Expression\ExpressionInterface;
 use Yiisoft\Db\Expression\JsonExpression;
@@ -12,6 +13,7 @@
 use Yiisoft\Db\QueryBuilder\AbstractDQLQueryBuilder;
 use Yiisoft\Db\QueryBuilder\Condition\InCondition;
 use Yiisoft\Db\QueryBuilder\Condition\LikeCondition;
+use Yiisoft\Db\Sqlite\Builder\ExpressionBuilder;
 use Yiisoft\Db\Sqlite\Builder\InConditionBuilder;
 use Yiisoft\Db\Sqlite\Builder\JsonExpressionBuilder;
 use Yiisoft\Db\Sqlite\Builder\LikeConditionBuilder;
@@ -136,6 +138,7 @@ protected function defaultExpressionBuilders(): array
             LikeCondition::class => LikeConditionBuilder::class,
             InCondition::class => InConditionBuilder::class,
             JsonExpression::class => JsonExpressionBuilder::class,
+            Expression::class => ExpressionBuilder::class,
         ]);
     }
 }
diff --git a/src/SqlParser.php b/src/SqlParser.php
new file mode 100644
index 000000000..9698d6d68
--- /dev/null
+++ b/src/SqlParser.php
@@ -0,0 +1,45 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Yiisoft\Db\Sqlite;
+
+use Yiisoft\Db\Syntax\AbstractSqlParser;
+
+final class SqlParser extends AbstractSqlParser
+{
+    public function getNextPlaceholder(int|null &$position = null): string|null
+    {
+        $result = null;
+        $length = $this->length - 1;
+
+        while ($this->position < $length) {
+            $pos = $this->position++;
+
+            match ($this->sql[$pos]) {
+                ':' => ($word = $this->parseWord()) === ''
+                    ? $this->skipChars(':')
+                    : $result = ':' . $word,
+                '"', "'", '`' => $this->skipQuotedWithoutEscape($this->sql[$pos]),
+                '[' => $this->sql[$this->position] === '['
+                    ? $this->skipToAfterString(']]')
+                    : $this->skipQuotedWithoutEscape(']'),
+                '-' => $this->sql[$this->position] === '-'
+                    ? ++$this->position && $this->skipToAfterChar("\n")
+                    : null,
+                '/' => $this->sql[$this->position] === '*'
+                    ? ++$this->position && $this->skipToAfterString('*/')
+                    : null,
+                default => null,
+            };
+
+            if ($result !== null) {
+                $position = $pos;
+
+                return $result;
+            }
+        }
+
+        return null;
+    }
+}
diff --git a/tests/CommandTest.php b/tests/CommandTest.php
index 7f3dbd4d5..32486c272 100644
--- a/tests/CommandTest.php
+++ b/tests/CommandTest.php
@@ -465,9 +465,10 @@ public function testUpdate(
         array $columns,
         array|string $conditions,
         array $params,
-        string $expected
+        array $expectedValues,
+        int $expectedCount,
     ): void {
-        parent::testUpdate($table, $columns, $conditions, $params, $expected);
+        parent::testUpdate($table, $columns, $conditions, $params, $expectedValues, $expectedCount);
     }
 
     /**
diff --git a/tests/Provider/SqlParserProvider.php b/tests/Provider/SqlParserProvider.php
new file mode 100644
index 000000000..77912c609
--- /dev/null
+++ b/tests/Provider/SqlParserProvider.php
@@ -0,0 +1,30 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Yiisoft\Db\Sqlite\Tests\Provider;
+
+class SqlParserProvider extends \Yiisoft\Db\Tests\Provider\SqlParserProvider
+{
+    public static function getNextPlaceholder(): array
+    {
+        return [
+            ...parent::getNextPlaceholder(),
+            [
+                '`:field` = :name AND age = :age',
+                ':name',
+                11,
+            ],
+            [
+                '[:field] = :name AND age = :age',
+                ':name',
+                11,
+            ],
+            [
+                '[[:field]] = :name AND age = :age',
+                ':name',
+                13,
+            ],
+        ];
+    }
+}
diff --git a/tests/QueryBuilderTest.php b/tests/QueryBuilderTest.php
index a54377eac..a14c9116f 100644
--- a/tests/QueryBuilderTest.php
+++ b/tests/QueryBuilderTest.php
@@ -703,10 +703,11 @@ public function testUpdate(
         string $table,
         array $columns,
         array|string $condition,
-        string $expectedSQL,
-        array $expectedParams
+        array $params,
+        string $expectedSql,
+        array $expectedParams,
     ): void {
-        parent::testUpdate($table, $columns, $condition, $expectedSQL, $expectedParams);
+        parent::testUpdate($table, $columns, $condition, $params, $expectedSql, $expectedParams);
     }
 
     /**
diff --git a/tests/SqlParserTest.php b/tests/SqlParserTest.php
new file mode 100644
index 000000000..512b770d6
--- /dev/null
+++ b/tests/SqlParserTest.php
@@ -0,0 +1,25 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Yiisoft\Db\Sqlite\Tests;
+
+use Yiisoft\Db\Sqlite\SqlParser;
+use Yiisoft\Db\Tests\AbstractSqlParserTest;
+
+/**
+ * @group sqlite
+ */
+final class SqlParserTest extends AbstractSqlParserTest
+{
+    protected function createSqlParser(string $sql): SqlParser
+    {
+        return new SqlParser($sql);
+    }
+
+    /** @dataProvider \Yiisoft\Db\Sqlite\Tests\Provider\SqlParserProvider::getNextPlaceholder */
+    public function testGetNextPlaceholder(string $sql, string|null $expectedPlaceholder, int|null $expectedPosition): void
+    {
+        parent::testGetNextPlaceholder($sql, $expectedPlaceholder, $expectedPosition);
+    }
+}