Fix #605 - Add "RECURSIVE" on build() for "WITH RECURSIVE" on the WithStatement class

Signed-off-by: William Desportes <[email protected]>

Author: williamdes

phpstan-baseline.neon

@@ -325,11 +325,6 @@ parameters:
 			count: 1
 			path: src/Components/OptionsArray.php
 
-		-
-			message: "#^Property PhpMyAdmin\\\\SqlParser\\\\Components\\\\OptionsArray\\:\\:\\$options \\(array\\<int, mixed\\>\\) does not accept array\\<int\\|string, mixed\\>\\.$#"
-			count: 8
-			path: src/Components/OptionsArray.php
-
 		-
 			message: "#^Right side of && is always true\\.$#"
 			count: 1

src/Statements/WithStatement.php

@@ -280,11 +280,17 @@ public function parse(Parser $parser, TokensList $list)
      */
     public function build()
     {
-        $str = 'WITH ';
+        $initial = true;
+        $str = 'WITH';
+
+        if ($this->options !== null && $this->options->options !== []) {
+            $str .= ' ' . OptionsArray::build($this->options);
+        }
 
         foreach ($this->withers as $wither) {
-            $str .= $str === 'WITH ' ? '' : ', ';
+            $str .= $initial ? ' ' : ', ';
             $str .= WithKeyword::build($wither);
+            $initial = false;
         }
 
         $str .= ' ';

tests/Builder/CreateStatementTest.php

@@ -15,6 +15,8 @@
 use PhpMyAdmin\SqlParser\Tests\TestCase;
 use PhpMyAdmin\SqlParser\TokensList;
 
+use function implode;
+
 class CreateStatementTest extends TestCase
 {
     public function testBuilder(): void
@@ -399,6 +401,33 @@ public function testBuilderView(): void
             . ' AS (SELECT 1 UNION ALL SELECT 2) SELECT col1 FROM cte AS `d`  ',
             $stmt->build()
         );
+
+        $parser = new Parser(
+            implode("\n", [
+                'CREATE VIEW number_sequence_view AS',
+                'WITH RECURSIVE number_sequence AS (',
+                '    SELECT 1 AS `number`',
+                '    UNION ALL',
+                '    SELECT `number` + 1',
+                '    FROM number_sequence',
+                '    WHERE `number` < 5',
+                ')',
+                'SELECT * FROM number_sequence;',
+            ])
+        );
+        $stmt = $parser->statements[0];
+        $this->assertEquals(
+            'CREATE VIEW number_sequence_view  AS'
+                . ' WITH RECURSIVE number_sequence AS ('
+                . 'SELECT 1 AS `number`'
+                . ' UNION ALL'
+                . ' SELECT `number`+ 1'
+                . ' FROM number_sequence'
+                . ' WHERE `number` < 5'
+                . ')'
+                . ' SELECT * FROM number_sequence ',
+            $stmt->build()
+        );
     }
 
     public function testBuilderViewComplex(): void

tests/Components/OptionsArrayTest.php

@@ -131,4 +131,17 @@ public function testBuild(): void
             'ALL SQL_CALC_FOUND_ROWS MAX_STATEMENT_TIME=42'
         );
     }
+
+    public function testBuildWithRecursive(): void
+    {
+        $component = OptionsArray::parse(
+            new Parser(),
+            $this->getTokensList('RECURSIVE'),
+            ['RECURSIVE' => 1]
+        );
+        $this->assertEquals(
+            OptionsArray::build($component),
+            'RECURSIVE'
+        );
+    }
 }

tests/Parser/WithStatementTest.php

@@ -74,6 +74,70 @@ public function testWith(): void
         $this->assertEquals($expected, $parser->statements[0]->build());
     }
 
+    public function testWithRecursive(): void
+    {
+        $sql = <<<SQL
+WITH RECURSIVE number_sequence AS (
+    SELECT 1 AS `number`
+    UNION ALL
+    SELECT `number` + 1
+    FROM number_sequence
+    WHERE `number` < 5
+)
+SELECT * FROM number_sequence;
+SQL;
+
+        $lexer = new Lexer($sql);
+
+        $lexerErrors = $this->getErrorsAsArray($lexer);
+        $this->assertCount(0, $lexerErrors);
+        $parser = new Parser($lexer->list);
+        $parserErrors = $this->getErrorsAsArray($parser);
+        $this->assertCount(0, $parserErrors);
+        $this->assertCount(1, $parser->statements);
+
+        // phpcs:disable Generic.Files.LineLength.TooLong
+        $expected = <<<SQL
+WITH RECURSIVE number_sequence AS (SELECT 1 AS `number` UNION ALL SELECT `number`+ 1 FROM number_sequence WHERE `number` < 5) SELECT * FROM number_sequence
+SQL;
+        // phpcs:enable
+        $this->assertEquals($expected, $parser->statements[0]->build());
+    }
+
+    public function testWithRecursiveWithers(): void
+    {
+        $sql = <<<SQL
+WITH RECURSIVE cte AS
+(
+    SELECT 1 AS n, CAST('abc' AS CHAR(20)) AS str
+    UNION ALL
+    SELECT n + 1, CONCAT(str, str) FROM cte WHERE n < 3
+), cte2 AS
+(
+    SELECT 1 AS n, CAST('def' AS CHAR(20)) AS str
+    UNION ALL
+    SELECT n + 1, CONCAT(str, str) FROM cte WHERE n < 3
+)
+SELECT * FROM cte UNION SELECT * FROM cte2;
+SQL;
+
+        $lexer = new Lexer($sql);
+
+        $lexerErrors = $this->getErrorsAsArray($lexer);
+        $this->assertCount(0, $lexerErrors);
+        $parser = new Parser($lexer->list);
+        $parserErrors = $this->getErrorsAsArray($parser);
+        $this->assertCount(0, $parserErrors);
+        $this->assertCount(1, $parser->statements);
+
+        // phpcs:disable Generic.Files.LineLength.TooLong
+        $expected = <<<SQL
+WITH RECURSIVE cte AS (SELECT 1 AS `n`, CAST('abc' AS CHAR(20)) AS `str` UNION ALL SELECT n+ 1, CONCAT(str, str) FROM cte WHERE n < 3), cte2 AS (SELECT 1 AS `n`, CAST('def' AS CHAR(20)) AS `str` UNION ALL SELECT n+ 1, CONCAT(str, str) FROM cte WHERE n < 3) SELECT * FROM cte UNION SELECT * FROM cte2
+SQL;
+        // phpcs:enable
+        $this->assertEquals($expected, $parser->statements[0]->build());
+    }
+
     public function testWithHasErrors(): void
     {
         $sql = <<<SQL