From b2fbe075b5ba91636d643f88be1c04ea576e2622 Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Sat, 3 Jun 2023 11:33:05 -0400
Subject: API tokens set through global Vsc settings

---
 extension/package.json                             | 10 ++++++++
 .../react-app/src/components/StepContainer.tsx     | 10 ++++----
 extension/src/continueIdeClient.ts                 | 27 ++++++++++++++++++++++
 3 files changed, 42 insertions(+), 5 deletions(-)

(limited to 'extension')

diff --git a/extension/package.json b/extension/package.json
index 13086954..cc8e18c4 100644
--- a/extension/package.json
+++ b/extension/package.json
@@ -39,6 +39,16 @@
           "type": "string",
           "default": "http://localhost:8000",
           "description": "The URL of the Continue server to use."
+        },
+        "continue.OPENAI_API_KEY": {
+          "type": "string",
+          "default": "",
+          "description": "The OpenAI API key to use for code generation."
+        },
+        "continue.HUGGING_FACE_TOKEN": {
+          "type": "string",
+          "default": "",
+          "description": "The Hugging Face API token to use for code generation."
         }
       }
     },
diff --git a/extension/react-app/src/components/StepContainer.tsx b/extension/react-app/src/components/StepContainer.tsx
index 5e979b34..fd29f21b 100644
--- a/extension/react-app/src/components/StepContainer.tsx
+++ b/extension/react-app/src/components/StepContainer.tsx
@@ -114,12 +114,12 @@ function StepContainer(props: StepContainerProps) {
       hidden={props.historyNode.step.hide as any}
     >
       <GradientBorder
-        className="m-2 overflow-hidden cursor-pointer"
-        onClick={() => setOpen((prev) => !prev)}
+        className="m-2 overflow-hidden"
+        // onClick={() => setOpen((prev) => !prev)}
       >
         <StepContainerDiv open={open}>
           <HeaderDiv>
-            <h4 className="m-2 cursor-pointer">
+            <h4 className="m-2">
               {open ? (
                 <ChevronDown size="1.4em" />
               ) : (
@@ -127,14 +127,14 @@ function StepContainer(props: StepContainerProps) {
               )}
               {props.historyNode.step.name as any}:
             </h4>
-            <HeaderButton
+            {/* <HeaderButton
               onClick={(e) => {
                 e.stopPropagation();
                 props.onReverse();
               }}
             >
               <Backward size="1.6em" onClick={props.onReverse}></Backward>
-            </HeaderButton>
+            </HeaderButton> */}
           </HeaderDiv>
 
           <ReactMarkdown key={1} className="overflow-scroll">
diff --git a/extension/src/continueIdeClient.ts b/extension/src/continueIdeClient.ts
index 03e5fbc5..a5a1c5dc 100644
--- a/extension/src/continueIdeClient.ts
+++ b/extension/src/continueIdeClient.ts
@@ -74,6 +74,12 @@ class IdeProtocolClient {
         this.messenger?.send("workspaceDirectory", {
           workspaceDirectory: this.getWorkspaceDirectory(),
         });
+        break;
+      case "getUserSecret":
+        this.messenger?.send("getUserSecret", {
+          value: await this.getUserSecret(data.key),
+        });
+        break;
       case "openFiles":
         this.messenger?.send("openFiles", {
           openFiles: this.getOpenFiles(),
@@ -130,6 +136,27 @@ class IdeProtocolClient {
     openEditorAndRevealRange(filepath, undefined, vscode.ViewColumn.One);
   }
 
+  async getUserSecret(key: string) {
+    // Check if secret already exists in VS Code settings (global)
+    let secret = vscode.workspace.getConfiguration("continue").get(key);
+    if (secret && secret !== "") return secret;
+
+    // If not, ask user for secret
+    while (typeof secret === "undefined" || secret === "") {
+      secret = await vscode.window.showInputBox({
+        prompt: `Enter secret for ${key}`,
+        password: true,
+      });
+    }
+
+    // Add secret to VS Code settings
+    vscode.workspace
+      .getConfiguration("continue")
+      .update(key, secret, vscode.ConfigurationTarget.Global);
+
+    return secret;
+  }
+
   // ------------------------------------ //
   // Initiate Request
 
-- 
cgit v1.2.3-70-g09d2