wip explorations

Author: JanJakes

tests/WP_SQLite_Driver_Translator_Tests.php

@@ -0,0 +1,50 @@
+<?php
+
+require_once __DIR__ . '/WP_SQLite_Translator_Tests.php';
+require_once __DIR__ . '/../wp-includes/sqlite-ast/class-wp-sqlite-driver.php';
+require_once __DIR__ . '/../wp-includes/sqlite-ast/class-wp-sqlite-expression.php';
+require_once __DIR__ . '/../wp-includes/sqlite-ast/class-wp-sqlite-token-factory.php';
+require_once __DIR__ . '/../wp-includes/sqlite-ast/class-wp-sqlite-token.php';
+require_once __DIR__ . '/../wp-includes/sqlite-ast/class-wp-sqlite-query-builder.php';
+
+use PHPUnit\Framework\TestCase;
+use WIP\WP_SQLite_Query_Builder;
+
+class WP_SQLite_Driver_Translator_Tests extends TestCase {
+	const GRAMMAR_PATH = __DIR__ . '/../wp-includes/mysql/mysql-grammar.php';
+
+	/**
+	 * @var WP_Parser_Grammar
+	 */
+	private static $grammar;
+
+	public static function setUpBeforeClass(): void {
+		self::$grammar = new WP_Parser_Grammar( include self::GRAMMAR_PATH );
+	}
+
+	public function testSelect(): void {
+		$this->assertTranslation( 'SELECT 1', 'SELECT 1' );
+		$this->assertTranslation( 'SELECT * FROM "t"', 'SELECT * FROM t' );
+		$this->assertTranslation( 'SELECT "c" FROM "t"', 'SELECT c FROM t' );
+		$this->assertTranslation( 'SELECT "t"."c" FROM "t"', 'SELECT t.c FROM t' );
+		$this->assertTranslation(
+			'SELECT * FROM "t1" LEFT JOIN "t2" ON "t1"."id" = "t2"."t1_id" WHERE "t1"."name" = \'abc\'',
+			"SELECT * FROM t1 LEFT JOIN t2 ON t1.id = t2.t1_id WHERE t1.name = 'abc'"
+		);
+	}
+
+	public function testInsert(): void {
+		$this->assertTranslation( 'INSERT INTO "t" ("c") VALUES (1)', 'INSERT INTO t (c) VALUES (1)' );
+	}
+
+	private function assertTranslation( string $expected, string $query ) {
+		$lexer  = new WP_MySQL_Lexer( $query );
+		$tokens = $lexer->remaining_tokens();
+
+		$parser = new WP_MySQL_Parser( self::$grammar, $tokens );
+		$ast    = $parser->parse();
+
+		$translator = new WIP\WP_SQLite_Translator();
+		$this->assertSame( $expected, WP_SQLite_Query_Builder::stringify( $translator->translate( $ast ) ) );
+	}
+}

tests/tools/dump-sqlite-query.php

@@ -17,5 +17,6 @@
 $driver = new WP_SQLite_Driver( new PDO( 'sqlite::memory:' ) );
 
 $query = "SELECT * FROM t1 LEFT JOIN t2 ON t1.id = t2.t1_id WHERE t1.name = 'abc'";
+//$query = 'CREATE TABLE t1 LIKE t2';
 
 echo $driver->query( $query );

wp-includes/parser/class-wp-parser-node.php

@@ -135,6 +135,19 @@ public function get_token( $token_id = null ) {
 		return null;
 	}
 
+	public function get_descendant_token( $token_id = null ) {
+		$parse_trees = array( $this );
+		while ( count( $parse_trees ) ) {
+			$parse_tree = array_pop( $parse_trees );
+			$token      = $parse_tree->get_token( $token_id );
+			if ( $token ) {
+				return $token;
+			}
+			array_push( $parse_trees, ...$parse_tree->get_children() );
+		}
+		return null;
+	}
+
 	public function get_child( $rule_name = null ) {
 		foreach ( $this->children as $child ) {
 			if ( $child instanceof WP_Parser_Node && (
@@ -158,7 +171,7 @@ public function get_descendant( $rule_name ) {
 		return null;
 	}
 
-	public function get_descendants( $rule_name ) {
+	public function get_descendants( $rule_name = null ) {
 		$parse_trees     = array( $this );
 		$all_descendants = array();
 		while ( count( $parse_trees ) ) {