add types for dialect

Author: Light2Dark

demo/index.ts

@@ -8,6 +8,7 @@ import {
   DefaultSqlTooltipRenders,
   defaultSqlHoverTheme,
   NodeSqlParser,
+  type SupportedDialects,
   sqlExtension,
 } from "../src/index.js";
 import { type Schema, tableTooltipRenderer } from "./custom-renderers.js";
@@ -146,9 +147,9 @@ const getKeywordDocs = async () => {
   };
 };
 
-const setDatabase = StateEffect.define<string>();
+const setDatabase = StateEffect.define<SupportedDialects>();
 
-const databaseField = StateField.define({
+const databaseField = StateField.define<SupportedDialects>({
   create: () => "PostgreSQL",
   update: (prevValue, transaction) => {
     for (const effect of transaction.effects) {
@@ -162,7 +163,7 @@ const databaseField = StateField.define({
 
 // Initialize the SQL editor
 function initializeEditor() {
-  // Use the same parser for linter and gutter
+  // Use the same parser
   const parser = new NodeSqlParser({
     getParserOptions: (state: EditorState) => {
       return {
@@ -233,6 +234,7 @@ function initializeEditor() {
           table: tableTooltipRenderer,
         },
         theme: defaultSqlHoverTheme("light"),
+        parser,
       },
     }),
     dialect.language.data.of({
@@ -337,15 +339,15 @@ function setupExampleButtons() {
   });
 }
 
-function getDialect(state: EditorState): string {
+function getDialect(state: EditorState): SupportedDialects {
   return state.field(databaseField);
 }
 
 function setupDialectSelect() {
   const select = document.querySelector("#dialect-select");
   if (select) {
     select.addEventListener("change", (e) => {
-      const value = (e.target as HTMLSelectElement).value;
+      const value = (e.target as HTMLSelectElement).value as SupportedDialects;
       editor.dispatch({
         effects: [setDatabase.of(value)],
       });

src/index.ts

@@ -8,7 +8,7 @@ export type {
   SqlKeywordInfo,
 } from "./sql/hover.js";
 export { DefaultSqlTooltipRenders, defaultSqlHoverTheme, sqlHover } from "./sql/hover.js";
-export { NodeSqlParser } from "./sql/parser.js";
+export { NodeSqlParser, type SupportedDialects } from "./sql/parser.js";
 export type { SqlStatement } from "./sql/structure-analyzer.js";
 export { SqlStructureAnalyzer } from "./sql/structure-analyzer.js";
 export type { SqlGutterConfig } from "./sql/structure-extension.js";

src/sql/parser.ts

@@ -4,8 +4,12 @@ import { debug } from "../debug.js";
 import { lazy } from "../utils.js";
 import type { SqlParseError, SqlParseResult, SqlParser } from "./types.js";
 
+interface ParserOption extends Option {
+  database: SupportedDialects;
+}
+
 interface NodeSqlParserOptions {
-  getParserOptions?: (state: EditorState) => Option;
+  getParserOptions?: (state: EditorState) => ParserOption;
 }
 
 interface NodeSqlParserResult extends SqlParseResult {
@@ -97,7 +101,7 @@ export class NodeSqlParser implements SqlParser {
 
     // Otherwise, try standard parsing with PostgreSQL dialect
     try {
-      const postgresOptions = { ...parserOptions, database: "PostgresQL" };
+      const postgresOptions = { ...parserOptions, database: "PostgreSQL" };
       const ast = parser.astify(sql, postgresOptions);
       return {
         success: true,
@@ -202,3 +206,23 @@ export class NodeSqlParser implements SqlParser {
     }
   }
 }
+
+/**
+ * https://github.com/taozhi8833998/node-sql-parser?tab=readme-ov-file#supported-database-sql-syntax
+ * While DuckDB is not supported in the library, we perform some special handling for it and treat it as PostgreSQL.
+ */
+export type SupportedDialects =
+  | "Athena"
+  | "BigQuery"
+  | "DB2"
+  | "Hive"
+  | "MariaDB"
+  | "MySQL"
+  | "PostgreSQL"
+  | "DuckDB"
+  | "Redshift"
+  | "Sqlite"
+  | "TransactSQL"
+  | "FlinkSQL"
+  | "Snowflake"
+  | "Noql";