feat(vscode): adding auto-import during accept inline completion item

clients/vscode/src/Commands.ts

@@ -13,6 +13,8 @@ import {
   QuickPickItem,
   ViewColumn,
   Range,
+  CodeAction,
+  CodeActionKind,
 } from "vscode";
 import os from "os";
 import path from "path";
@@ -223,7 +225,19 @@ export class Commands {
       commands.executeCommand("editor.action.inlineSuggest.trigger");
     },
     "inlineCompletion.accept": () => {
+      const editor = window.activeTextEditor;
+      if (!editor) {
+        return;
+      }
+
       commands.executeCommand("editor.action.inlineSuggest.commit");
+
+      const uri = editor.document.uri;
+      const range = this.inlineCompletionProvider.getCurrentDisplayRange();
+      if (!range) {
+        return;
+      }
+      applyQuickFixes(uri, range);
     },
     "inlineCompletion.acceptNextWord": () => {
       this.inlineCompletionProvider.handleEvent("accept_word");
@@ -452,3 +466,23 @@ export class Commands {
     },
   };
 }
+
+async function applyQuickFixes(uri: Uri, range: Range): Promise<void> {
+  const codeActions = await commands.executeCommand<CodeAction[]>("vscode.executeCodeActionProvider", uri, range);
+  const quickFixActions = codeActions.filter(
+    (action) =>
+      action.kind && action.kind.contains(CodeActionKind.QuickFix) && action.title.toLowerCase().includes("import"),
+  );
+  quickFixActions.forEach(async (action) => {
+    try {
+      if (action.edit) {
+        await workspace.applyEdit(action.edit);
+      }
+      if (action.command) {
+        await commands.executeCommand(action.command.command, action.command.arguments);
+      }
+    } catch (error) {
+      // ignore errors
+    }
+  });
+}

clients/vscode/src/InlineCompletionProvider.ts

@@ -211,4 +211,28 @@ export class InlineCompletionProvider extends EventEmitter implements InlineComp
     // await not required
     this.client.telemetry.postEvent(params);
   }
+
+  /**
+   * Get current display range with the current completion item
+   * @return {Range | undefined} - The range with the current completion item
+   */
+  public getCurrentDisplayRange(): Range | undefined {
+    const item = this.displayedCompletion?.completions.items[this.displayedCompletion.index];
+    const range = item?.range;
+    if (!range) {
+      return undefined;
+    }
+    if (!item) {
+      return undefined;
+    }
+    const length = (item.insertText as string).split("\n").length - 1; //remove current line count;
+    const completionEnd = range.end.line + length;
+    const completionRange = new Range(
+      new Position(range.start.line, range.start.character),
+      new Position(completionEnd == 0 ? completionEnd : completionEnd + 1, 0),
+    );
+    this.logger.info("current range with displayed completion item:", completionRange);
+
+    return completionRange;
+  }
 }