added completion after dot for StandardVO

Author: Traigor

server/src/completion.ts

@@ -1,30 +1,65 @@
 import {
   CompletionItem,
   CompletionItemKind,
+  InsertTextFormat,
+  InsertTextMode,
   TextDocumentPositionParams,
+  TextDocuments,
 } from 'vscode-languageserver/node.js';
 import { documentation } from './information/documentation.js';
 import { details } from './information/details.js';
 import { allKeywords, components, keywords } from './types/keywords.js';
+import { TextDocument } from 'vscode-languageserver-textdocument';
+
+//this would come from package
+const StandardVO = {
+  prefixes: ['StandardVO', 'StandardVO.Currency'],
+  completions: {
+    StandardVO: ['Currency'],
+    'StandardVO.Currency': ['Value', 'Create'],
+  },
+};
 
 export class CompletionItemProvider {
-  public static onCompletion(_textDocumentPosition: TextDocumentPositionParams) {
-    return allKeywords.map((symbol) => ({
-      label: symbol,
-      kind: CompletionItemKind.Keyword,
-      data: 1,
-    }));
+  public static onCompletion(
+    documents: TextDocuments<TextDocument>,
+    _textDocumentPosition: TextDocumentPositionParams,
+  ) {
+    const document = documents.get(_textDocumentPosition.textDocument.uri);
+    let text = documents.get(_textDocumentPosition.textDocument.uri).getText();
+    let position = _textDocumentPosition.position;
+    const offset = document.offsetAt(position);
+    for (const prefix of StandardVO.prefixes) {
+      const linePrefix = text.slice(offset - prefix.length - 1, offset);
+      if (linePrefix.endsWith(prefix + '.')) {
+        return [
+          ...StandardVO.completions[prefix].map((symbol) => ({
+            label: symbol,
+            kind: CompletionItemKind.Keyword,
+            data: 2,
+            //we could check for snippets too
+          })),
+        ];
+      }
+    }
+    return [
+      ...allKeywords.map((symbol) => ({
+        label: symbol,
+        kind: CompletionItemKind.Keyword,
+        data: 1,
+      })),
+    ];
   }
 
   public static onCompletionResolve(item: CompletionItem): CompletionItem {
-    const label: string = item.label;
-    if (documentation[item.label as keyof typeof documentation]) {
-      item.documentation = documentation[item.label as keyof typeof documentation];
+    if (item.data == 1) {
+      if (documentation[item.label as keyof typeof documentation]) {
+        item.documentation = documentation[item.label as keyof typeof documentation];
+      }
+      if (details[item.label as keyof typeof details]) {
+        item.detail = details[item.label as keyof typeof details];
+      }
     }
-    if (details[item.label as keyof typeof details]) {
-      item.detail = details[item.label as keyof typeof details];
-    }
-
     return item;
   }
 }

server/src/lsp-connection.ts

@@ -29,7 +29,7 @@ export class LspConnection {
 
     this.documents.onDidClose(server.onDidClose.bind(server));
     this.documents.onDidChangeContent(server.onDidChangeContent.bind(server));
-    this.connection.onCompletion(server.completion.bind(server));
+    this.connection.onCompletion(server.completion.bind(server, this.documents));
     this.connection.onCompletionResolve(server.completionResolve.bind(server));
     return this;
   }

server/src/server.ts

@@ -64,6 +64,7 @@ export class BitloopsServer {
         // Tell the client that this server supports code completion.
         completionProvider: {
           resolveProvider: true,
+          triggerCharacters: ['.'],
         },
       },
     };

server/src/types/keywords.ts

@@ -141,6 +141,9 @@ export const keywords = [
   'Operations',
   'Mutation',
   'Query',
+
+  'StandardVO',
 ];
 
+// 'StandardVO',
 export const allKeywords = Array.from(new Set([...Object.values(components), ...keywords]));