Build insert statement with context's quote style

Author: MoonE

phpstan-baseline.neon

@@ -485,6 +485,11 @@ parameters:
 			count: 1
 			path: src/Components/IntoKeyword.php
 
+		-
+			message: "#^Parameter \\#2 \\$pieces of function implode expects array, array\\<string\\>\\|string given\\.$#"
+			count: 1
+			path: src/Components/IntoKeyword.php
+
 		-
 			message: "#^Construct empty\\(\\) is not allowed\\. Use more strict comparison\\.$#"
 			count: 1

psalm-baseline.xml

@@ -247,6 +247,9 @@
     <MoreSpecificImplementedParamType occurrences="1">
       <code>$component</code>
     </MoreSpecificImplementedParamType>
+    <PossiblyInvalidArgument occurrences="1">
+      <code>Context::escape($component-&gt;columns)</code>
+    </PossiblyInvalidArgument>
     <PossiblyInvalidPropertyFetch occurrences="1">
       <code>ArrayObj::parse($parser, $list)-&gt;values</code>
     </PossiblyInvalidPropertyFetch>

src/Components/IntoKeyword.php

@@ -5,6 +5,7 @@
 namespace PhpMyAdmin\SqlParser\Components;
 
 use PhpMyAdmin\SqlParser\Component;
+use PhpMyAdmin\SqlParser\Context;
 use PhpMyAdmin\SqlParser\Parser;
 use PhpMyAdmin\SqlParser\Token;
 use PhpMyAdmin\SqlParser\TokensList;
@@ -271,7 +272,9 @@ public function parseFileOptions(Parser $parser, TokensList $list, $keyword = 'F
     public static function build($component, array $options = [])
     {
         if ($component->dest instanceof Expression) {
-            $columns = ! empty($component->columns) ? '(`' . implode('`, `', $component->columns) . '`)' : '';
+            $columns = $component->columns === null || $component->columns === []
+                 ? ''
+                 : '(' . implode(', ', Context::escape($component->columns)) . ')';
 
             return $component->dest . $columns;
         }

tests/Builder/InsertStatementTest.php

@@ -4,64 +4,60 @@
 
 namespace PhpMyAdmin\SqlParser\Tests\Builder;
 
+use Generator;
+use PhpMyAdmin\SqlParser\Context;
 use PhpMyAdmin\SqlParser\Parser;
 use PhpMyAdmin\SqlParser\Tests\TestCase;
 
 class InsertStatementTest extends TestCase
 {
-    public function testBuilder(): void
+    /** @var int */
+    private $sqlMode;
+
+    public function setUp(): void
     {
-        /* Assertion 1 */
-        $parser = new Parser('INSERT INTO tbl(`col1`, `col2`, `col3`) VALUES (1, "str", 3.14)');
-        $stmt = $parser->statements[0];
-        $this->assertEquals(
-            'INSERT INTO tbl(`col1`, `col2`, `col3`) VALUES (1, "str", 3.14)',
-            $stmt->build()
-        );
-
-        /* Assertion 2 */
-        /* Reserved keywords (with backquotes as field name) */
-        $parser = new Parser('INSERT INTO tbl(`order`) VALUES (1)');
-        $stmt = $parser->statements[0];
-        $this->assertEquals(
-            'INSERT INTO tbl(`order`) VALUES (1)',
-            $stmt->build()
-        );
-
-        /* Assertion 3 */
-        /* INSERT ... SET ... */
-        $parser = new Parser('INSERT INTO tbl SET FOO = 1');
-        $stmt = $parser->statements[0];
-        $this->assertEquals(
-            'INSERT INTO tbl SET FOO = 1',
-            $stmt->build()
-        );
-
-        /* Assertion 4 */
-        /* INSERT ... SELECT ... */
-        $parser = new Parser('INSERT INTO tbl SELECT * FROM bar');
-        $stmt = $parser->statements[0];
-        $this->assertEquals(
-            'INSERT INTO tbl SELECT * FROM bar',
-            $stmt->build()
-        );
-
-        /* Assertion 5 */
-        /* INSERT ... ON DUPLICATE KEY UPDATE ... */
-        $parser = new Parser('INSERT INTO tbl SELECT * FROM bar ON DUPLICATE KEY UPDATE baz = 1');
+        parent::setUp();
+
+        $this->sqlMode = Context::getMode();
+    }
+
+    protected function tearDown(): void
+    {
+        Context::setMode($this->sqlMode);
+
+        parent::tearDown();
+    }
+
+    /** @dataProvider providerForTestBuilder */
+    public function testBuilder(string $sql): void
+    {
+        $parser = new Parser($sql);
         $stmt = $parser->statements[0];
-        $this->assertEquals(
-            'INSERT INTO tbl SELECT * FROM bar ON DUPLICATE KEY UPDATE baz = 1',
-            $stmt->build()
-        );
-
-        /* Assertion 6 */
-        /* INSERT [OPTIONS] INTO ... */
-        $parser = new Parser('INSERT DELAYED IGNORE INTO tbl SELECT * FROM bar');
+        $this->assertEquals($sql, $stmt->build());
+    }
+
+    /** @return Generator<string, list<string>> */
+    public function providerForTestBuilder(): Generator
+    {
+        yield 'INSERT ... VALUES ...' => ['INSERT INTO tbl(`col1`, `col2`, `col3`) VALUES (1, "str", 3.14)'];
+
+        yield 'Reserved keywords (with backquotes as field name)' => ['INSERT INTO tbl(`order`) VALUES (1)'];
+
+        yield 'INSERT ... SET ...' => ['INSERT INTO tbl SET FOO = 1'];
+
+        yield 'INSERT ... SELECT ... ' => ['INSERT INTO tbl SELECT * FROM bar'];
+
+        yield 'INSERT ... ON DUPLICATE KEY UPDATE ...' => ['INSERT INTO tbl SELECT * FROM bar ON DUPLICATE KEY UPDATE baz = 1'];
+
+        yield 'INSERT [OPTIONS] INTO ...' => ['INSERT DELAYED IGNORE INTO tbl SELECT * FROM bar'];
+    }
+
+    public function testBuilderAnsi(): void
+    {
+        Context::setMode(Context::SQL_MODE_ANSI);
+        $sql = "INSERT INTO foo (bar, baz) VALUES ('bar', 'baz')";
+        $parser = new Parser($sql);
         $stmt = $parser->statements[0];
-        $this->assertEquals(
-            'INSERT DELAYED IGNORE INTO tbl SELECT * FROM bar',
-            $stmt->build()
-        );
+        $this->assertEquals('INSERT INTO foo("bar", "baz") VALUES (\'bar\', \'baz\')', $stmt->build());
     }
 }