feat: add support for 'select_complete' context in SQL intellisense

Author: Krishnachaitanyakc

src/renderer/lib/intellisense/IntellisenseProvider.ts

@@ -69,6 +69,11 @@ export abstract class IntellisenseProvider {
         suggestions.push(...this.getColumnSuggestions(context, schema, position, range))
         suggestions.push(...this.getFunctionSuggestions(context, schema, position, range))
         break
+      case 'select_complete':
+        suggestions.push(...this.getKeywordSuggestions(context, schema, position, range))
+        suggestions.push(...this.getColumnSuggestions(context, schema, position, range))
+        suggestions.push(...this.getFunctionSuggestions(context, schema, position, range))
+        break
       case 'function':
         suggestions.push(...this.getFunctionSuggestions(context, schema, position, range))
         break

src/renderer/lib/intellisense/parsers/SQLContextParser.ts

@@ -94,7 +94,16 @@ export class SQLContextParser {
 
     switch (lastKeywordMatch.type) {
       case 'select':
-        context.type = hasContent ? 'column' : 'select'
+        if (hasContent) {
+          // Check if we're at the end of the SELECT clause and should suggest keywords
+          const endsWithSpace = textBefore.endsWith(' ')
+          const hasFrom = this.hasFromClause(lowerText)
+          const shouldSuggestKeywords = endsWithSpace && !hasFrom
+
+          context.type = shouldSuggestKeywords ? 'select_complete' : 'column'
+        } else {
+          context.type = 'select'
+        }
         context.currentClause = 'SELECT'
         break
 
@@ -136,6 +145,10 @@ export class SQLContextParser {
     }
   }
 
+  private hasFromClause(text: string): boolean {
+    return this.keywordPatterns.from.test(text)
+  }
+
   private findLastKeyword(text: string): { type: string; keyword: string; index: number } | null {
     let lastMatch: { type: string; keyword: string; index: number } | null = null
     let lastIndex = -1

src/renderer/lib/intellisense/types.ts

@@ -22,7 +22,16 @@ export interface Position {
 }
 
 export interface SQLContext {
-  type: 'select' | 'from' | 'where' | 'join' | 'column' | 'function' | 'keyword' | 'unknown'
+  type:
+    | 'select'
+    | 'from'
+    | 'where'
+    | 'join'
+    | 'column'
+    | 'function'
+    | 'keyword'
+    | 'unknown'
+    | 'select_complete'
   currentClause?: string
   tableAliases: Map<string, string>
   availableTables: string[]