feat: remove noReserved keywords in completions

Author: JackWang032

src/parser/flink/index.ts

@@ -29,6 +29,17 @@ export class FlinkSQL extends BasicSQL<FlinkSqlLexer, ProgramContext, FlinkSqlPa
         return new FlinkSqlParser(tokenStream);
     }
 
+    /**
+     * The rules that keywords you don't want to be suggested.
+     */
+    protected excludeKeywordRules = new Set([
+        FlinkSqlParser.RULE_nonReservedKeywords,
+        FlinkSqlParser.RULE_reservedKeywordsUsedAsFuncName,
+        FlinkSqlParser.RULE_reservedKeywordsFollowParamsUsedAsFuncName,
+        FlinkSqlParser.RULE_reservedKeywordsNoParamsUsedAsFuncName,
+        FlinkSqlParser.RULE_reservedKeywordsUsedAsFuncParam,
+    ]);
+
     protected preferredRules = new Set([
         FlinkSqlParser.RULE_catalogPath, // catalog name
         FlinkSqlParser.RULE_databasePath, // database name
@@ -47,6 +58,7 @@ export class FlinkSQL extends BasicSQL<FlinkSqlLexer, ProgramContext, FlinkSqlPa
         FlinkSqlParser.RULE_columnNameCreate,
         FlinkSqlParser.RULE_tablePropertyKey,
         FlinkSqlParser.RULE_tablePropertyValue,
+        ...this.excludeKeywordRules,
     ]);
 
     protected get splitListener() {

src/parser/hive/index.ts

@@ -30,6 +30,11 @@ export class HiveSQL extends BasicSQL<HiveSqlLexer, ProgramContext, HiveSqlParse
         return new HiveSqlParser(tokenStream);
     }
 
+    /**
+     * The rules that keywords you don't want to be suggested.
+     */
+    protected excludeKeywordRules = new Set([HiveSqlParser.RULE_nonReserved]);
+
     protected preferredRules: Set<number> = new Set([
         HiveSqlParser.RULE_dbSchemaName, // db or schema name
         HiveSqlParser.RULE_dbSchemaNameCreate, // db or schema name that will be created
@@ -43,6 +48,7 @@ export class HiveSQL extends BasicSQL<HiveSqlLexer, ProgramContext, HiveSqlParse
         HiveSqlParser.RULE_columnName,
         HiveSqlParser.RULE_columnNamePath,
         HiveSqlParser.RULE_columnNameCreate,
+        ...this.excludeKeywordRules,
     ]);
 
     protected get splitListener() {

src/parser/impala/index.ts

@@ -29,6 +29,11 @@ export class ImpalaSQL extends BasicSQL<ImpalaSqlLexer, ProgramContext, ImpalaSq
         return new ImpalaSqlParser(tokenStream);
     }
 
+    /**
+     * The rules that keywords you don't want to be suggested.
+     */
+    protected excludeKeywordRules = new Set([ImpalaSqlParser.RULE_nonReserved]);
+
     protected preferredRules: Set<number> = new Set([
         ImpalaSqlParser.RULE_functionNameCreate,
         ImpalaSqlParser.RULE_tableNameCreate,
@@ -41,6 +46,7 @@ export class ImpalaSQL extends BasicSQL<ImpalaSqlLexer, ProgramContext, ImpalaSq
         ImpalaSqlParser.RULE_databaseNamePath,
         ImpalaSqlParser.RULE_columnNamePath,
         ImpalaSqlParser.RULE_columnName,
+        ...this.excludeKeywordRules,
     ]);
 
     protected get splitListener() {

src/parser/mysql/index.ts

@@ -29,6 +29,11 @@ export class MySQL extends BasicSQL<MySqlLexer, ProgramContext, MySqlParser> {
         return new MySqlParser(tokenStream);
     }
 
+    /**
+     * The rules that keywords you don't want to be suggested.
+     */
+    protected excludeKeywordRules = new Set([MySqlParser.RULE_keywordsCanBeId]);
+
     protected preferredRules: Set<number> = new Set([
         MySqlParser.RULE_databaseName,
         MySqlParser.RULE_databaseNameCreate,
@@ -40,6 +45,7 @@ export class MySQL extends BasicSQL<MySqlLexer, ProgramContext, MySqlParser> {
         MySqlParser.RULE_functionNameCreate,
         MySqlParser.RULE_columnName,
         MySqlParser.RULE_columnNameCreate,
+        ...this.excludeKeywordRules,
     ]);
 
     protected get splitListener() {

src/parser/postgresql/index.ts

@@ -30,6 +30,11 @@ export class PostgreSQL extends BasicSQL<PostgreSqlLexer, ProgramContext, Postgr
         return new PostgreSqlParser(tokenStream);
     }
 
+    /**
+     * The rules that keywords you don't want to be suggested.
+     */
+    protected excludeKeywordRules = new Set([PostgreSqlParser.RULE_nonReservedWord]);
+
     protected preferredRules: Set<number> = new Set([
         PostgreSqlParser.RULE_tableNameCreate, // table name
         PostgreSqlParser.RULE_tableName, // table name that will be created
@@ -46,6 +51,7 @@ export class PostgreSQL extends BasicSQL<PostgreSqlLexer, ProgramContext, Postgr
         PostgreSqlParser.RULE_columnNameCreate, // column name that will be created
         PostgreSqlParser.RULE_columnName, // column name
         PostgreSqlParser.RULE_columnNamePath, // column name
+        ...this.excludeKeywordRules,
     ]);
 
     protected get splitListener() {

src/parser/spark/index.ts

@@ -29,6 +29,15 @@ export class SparkSQL extends BasicSQL<SparkSqlLexer, ProgramContext, SparkSqlPa
         return new SparkSqlParser(tokenStream);
     }
 
+    /**
+     * The rules that keywords you don't want to be suggested.
+     */
+    protected excludeKeywordRules = new Set([
+        SparkSqlParser.RULE_strictIdentifier,
+        SparkSqlParser.RULE_strictNonReserved,
+        SparkSqlParser.RULE_nonReserved,
+    ]);
+
     protected preferredRules: Set<number> = new Set([
         SparkSqlParser.RULE_namespaceName,
         SparkSqlParser.RULE_namespaceNameCreate,
@@ -41,6 +50,7 @@ export class SparkSQL extends BasicSQL<SparkSqlLexer, ProgramContext, SparkSqlPa
         SparkSqlParser.RULE_columnName,
         SparkSqlParser.RULE_columnNamePath,
         SparkSqlParser.RULE_columnNameCreate,
+        ...this.excludeKeywordRules,
     ]);
 
     protected get splitListener() {

src/parser/trino/index.ts

@@ -50,6 +50,11 @@ export class TrinoSQL extends BasicSQL<TrinoSqlLexer, ProgramContext, TrinoSqlPa
         return new TrinoSemanticContextCollector(input, caretPosition, allTokens, options);
     }
 
+    /**
+     * The rules that keywords you don't want to be suggested.
+     */
+    protected excludeKeywordRules = new Set([TrinoSqlParser.RULE_nonReserved]);
+
     protected preferredRules: Set<number> = new Set([
         TrinoSqlParser.RULE_catalogRef,
         TrinoSqlParser.RULE_catalogNameCreate,
@@ -64,6 +69,7 @@ export class TrinoSQL extends BasicSQL<TrinoSqlLexer, ProgramContext, TrinoSqlPa
         TrinoSqlParser.RULE_columnRef,
         TrinoSqlParser.RULE_columnName,
         TrinoSqlParser.RULE_columnNameCreate,
+        ...this.excludeKeywordRules,
     ]);
 
     protected processCandidates(