Merge pull request #6 from DrNightmare/feature/get-api-key

Take OpenAI API key as input before registering commands.
DrNightmare · Mar 15, 2023 · cd47cee · cd47cee
2 parents cac9f27 + 3037379
commit cd47cee
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/src/extension.ts b/src/extension.ts
@@ -7,7 +7,7 @@ import {
 import { Vortex } from "./vortex";
 
 const vortexConfig = vscode.workspace.getConfiguration("vortex");
-const vortex = new Vortex("openai-api-key");
+let vortex: Vortex;
 
 const displayOutput = async (
   output: string,
@@ -38,6 +38,21 @@ const displayOutput = async (
 };
 
 export async function activate(context: vscode.ExtensionContext) {
+  const openAIApiKey = await vscode.window.showInputBox({
+    title: "OpenAI API Key",
+    prompt: "Enter your OpenAI API Key here",
+    password: true,
+    ignoreFocusOut: true,
+  });
+
+  if (!openAIApiKey) {
+    vscode.window.showWarningMessage(
+      "OpenAI API Key not provided, Vortex is not activated"
+    );
+    return;
+  }
+  vortex = new Vortex(openAIApiKey);
+
   let disposable = vscode.commands.registerCommand(
     "vortex.editCode",
     async () => {