rawsql-ts
Version:
[beta]High-performance SQL parser and AST analyzer written in TypeScript. Provides fast parsing and advanced transformation capabilities.
74 lines • 3.76 kB
JavaScript
import { CommonTable } from "../models/Clause";
import { TokenType } from "../models/Lexeme";
import { SqlTokenizer } from "./SqlTokenizer";
import { SelectQueryParser } from "./SelectQueryParser";
import { SourceAliasExpressionParser } from "./SourceAliasExpressionParser";
export class CommonTableParser {
// Parse SQL string to AST (was: parse)
static parse(query) {
const tokenizer = new SqlTokenizer(query); // Initialize tokenizer
const lexemes = tokenizer.readLexmes(); // Get tokens
// Parse
const result = this.parseFromLexeme(lexemes, 0);
// Error if there are remaining tokens
if (result.newIndex < lexemes.length) {
throw new Error(`Syntax error: Unexpected token "${lexemes[result.newIndex].value}" at position ${result.newIndex}. The CommonTable definition is complete but there are additional tokens.`);
}
return result.value;
}
// Parse from lexeme array (was: parse)
static parseFromLexeme(lexemes, index) {
let idx = index;
// Capture comments from the CTE name token (before parsing alias)
const cteNameComments = idx < lexemes.length ? lexemes[idx].comments : null;
// Parse alias and optional column aliases
// SourceAliasExpressionParser already handles column aliases if present
const aliasResult = SourceAliasExpressionParser.parseFromLexeme(lexemes, idx);
idx = aliasResult.newIndex;
if (idx < lexemes.length && lexemes[idx].value !== "as") {
throw new Error(`Syntax error at position ${idx}: Expected 'AS' keyword after CTE name but found "${lexemes[idx].value}".`);
}
idx++; // Skip 'AS' keyword
// Materialized flag
let materialized = null;
// Parse optional MATERIALIZED or NOT MATERIALIZED keywords
if (idx < lexemes.length) {
const currentValue = lexemes[idx].value;
if (currentValue === "materialized") {
materialized = true;
idx++;
}
else if (currentValue === "not materialized") {
materialized = false;
idx++;
}
}
if (idx < lexemes.length && lexemes[idx].type !== TokenType.OpenParen) {
throw new Error(`Syntax error at position ${idx}: Expected '(' after CTE name but found "${lexemes[idx].value}".`);
}
// Capture comments from the opening parenthesis for the CTE inner query
const cteQueryComments = lexemes[idx].comments;
idx++; // Skip opening parenthesis
const queryResult = SelectQueryParser.parseFromLexeme(lexemes, idx);
idx = queryResult.newIndex;
// If there are comments from the opening parenthesis, add them to the inner query
if (cteQueryComments && cteQueryComments.length > 0) {
if (queryResult.value.comments) {
// Prepend the CTE query comments to existing comments
queryResult.value.comments = [...cteQueryComments, ...queryResult.value.comments];
}
else {
queryResult.value.comments = cteQueryComments;
}
}
if (idx < lexemes.length && lexemes[idx].type !== TokenType.CloseParen) {
throw new Error(`Syntax error at position ${idx}: Expected ')' after CTE query but found "${lexemes[idx].value}".`);
}
idx++; // Skip closing parenthesis
const value = new CommonTable(queryResult.value, aliasResult.value, materialized);
// Set comments on the CommonTable from the CTE name token
value.comments = cteNameComments;
return { value, newIndex: idx };
}
}
//# sourceMappingURL=CommonTableParser.js.map