rawsql-ts
Version:
[beta]High-performance SQL parser and AST analyzer written in TypeScript. Provides fast parsing and advanced transformation capabilities.
83 lines • 3.55 kB
JavaScript
// filepath: src/parsers/InsertQueryParser.ts
// Provides parsing for INSERT queries, supporting optional columns and WITH/SELECT/VALUES structure.
import { InsertQuery } from "../models/InsertQuery";
import { TokenType } from "../models/Lexeme";
import { SqlTokenizer } from "./SqlTokenizer";
import { SelectQueryParser } from "./SelectQueryParser";
import { WithClauseParser } from "./WithClauseParser";
import { SimpleSelectQuery } from "../models/SimpleSelectQuery";
import { SourceExpressionParser } from "./SourceExpressionParser";
import { InsertClause } from "../models/Clause";
export class InsertQueryParser {
/**
* Parse SQL string to InsertQuery AST.
* @param query SQL string
*/
static parse(query) {
const tokenizer = new SqlTokenizer(query);
const lexemes = tokenizer.readLexmes();
const result = this.parseFromLexeme(lexemes, 0);
if (result.newIndex < lexemes.length) {
throw new Error(`Syntax error: Unexpected token "${lexemes[result.newIndex].value}" at position ${result.newIndex}. The INSERT query is complete but there are additional tokens.`);
}
return result.value;
}
/**
* Parse from lexeme array (for internal use and tests)
*/
static parseFromLexeme(lexemes, index) {
var _a, _b, _c;
let idx = index;
let withclause = null;
if (lexemes[idx].value === "with") {
const result = WithClauseParser.parseFromLexeme(lexemes, idx);
withclause = result.value;
idx = result.newIndex;
}
// Expect INSERT INTO
if (lexemes[idx].value !== "insert into") {
throw new Error(`Syntax error at position ${idx}: Expected 'INSERT INTO' but found '${lexemes[idx].value}'.`);
}
idx++;
// Parse table and optional alias/schema using SourceExpressionParser
const sourceResult = SourceExpressionParser.parseTableSourceFromLexemes(lexemes, idx);
idx = sourceResult.newIndex;
// Optional columns
let columns = [];
if (((_a = lexemes[idx]) === null || _a === void 0 ? void 0 : _a.type) === TokenType.OpenParen) {
idx++;
while (idx < lexemes.length && lexemes[idx].type === TokenType.Identifier) {
columns.push(lexemes[idx].value);
idx++;
if (((_b = lexemes[idx]) === null || _b === void 0 ? void 0 : _b.type) === TokenType.Comma) {
idx++;
}
else {
break;
}
}
if (((_c = lexemes[idx]) === null || _c === void 0 ? void 0 : _c.type) !== TokenType.CloseParen) {
throw new Error(`Syntax error at position ${idx}: Expected ')' after column list.`);
}
idx++;
}
const selectResult = SelectQueryParser.parseFromLexeme(lexemes, idx);
if (withclause) {
if (selectResult.value instanceof SimpleSelectQuery) {
selectResult.value.withClause = withclause;
}
else {
throw new Error(`WITH clause is not supported in this context.`);
}
}
idx = selectResult.newIndex;
return {
value: new InsertQuery({
insertClause: new InsertClause(sourceResult.value, columns),
selectQuery: selectResult.value
}),
newIndex: idx
};
}
}
//# sourceMappingURL=InsertQueryParser.js.map