7 files changed, 61 insertions, 16 deletions
diff --git a/extension/.vscodeignore b/extension/.vscodeignore
index d3326fdc..51d66585 100644
--- a/extension/.vscodeignore
+++ b/extension/.vscodeignore
@@ -24,4 +24,6 @@ react-app/src
 
 scripts/data
 scripts/env
-scripts/.continue_env_installed
-\ No newline at end of file
+scripts/.continue_env_installed
+
+server/exe/**
+\ No newline at end of file
diff --git a/extension/package.json b/extension/package.json
index fb4a2df8..2e47e80d 100644
--- a/extension/package.json
+++ b/extension/package.json
@@ -194,7 +194,7 @@
     "lint": "eslint src --ext ts",
     "build-test": "tsc && node esbuild.test.mjs",
     "test": "npm run build-test && node ./out/test-runner/runTestOnVSCodeHost.js",
-    "package": "npm install && npm run typegen && cd react-app && npm install && npm run build && cd .. && mkdir -p ./build && vsce package --out ./build"
+    "package": "node scripts/package.js"
   },
   "devDependencies": {
     "@nestjs/common": "^8.4.7",
diff --git a/extension/react-app/src/redux/slices/serverStateReducer.ts b/extension/react-app/src/redux/slices/serverStateReducer.ts
index bd60f1c7..a20476b2 100644
--- a/extension/react-app/src/redux/slices/serverStateReducer.ts
+++ b/extension/react-app/src/redux/slices/serverStateReducer.ts
@@ -9,9 +9,9 @@ const initialState: FullState = {
           name: "Welcome to Continue",
           hide: false,
           description: `- Highlight code section and ask a question or give instructions
-    - Use \`cmd+m\` (Mac) / \`ctrl+m\` (Windows) to open Continue
-    - Use \`/help\` to ask questions about how to use Continue
-    - [Customize Continue](https://continue.dev/docs/customization) (e.g. use your own API key) by typing '/config'.`,
+- Use \`cmd+m\` (Mac) / \`ctrl+m\` (Windows) to open Continue
+- Use \`/help\` to ask questions about how to use Continue
+- [Customize Continue](https://continue.dev/docs/customization) (e.g. use your own API key) by typing '/config'.`,
           system_message: null,
           chat_context: [],
           manage_own_chat_context: false,
diff --git a/extension/scripts/package.js b/extension/scripts/package.js
new file mode 100644
index 00000000..ae9a4d94
--- /dev/null
+++ b/extension/scripts/package.js
@@ -0,0 +1,35 @@
+const { exec } = require("child_process");
+const fs = require("fs");
+
+exec("npm install", (error) => {
+  if (error) throw error;
+  console.log("npm install completed");
+
+  exec("npm run typegen", (error) => {
+    if (error) throw error;
+    console.log("npm run typegen completed");
+
+    process.chdir("react-app");
+
+    exec("npm install", (error) => {
+      if (error) throw error;
+      console.log("npm install in react-app completed");
+
+      exec("npm run build", (error) => {
+        if (error) throw error;
+        console.log("npm run build in react-app completed");
+
+        process.chdir("..");
+
+        if (!fs.existsSync("build")) {
+          fs.mkdirSync("build");
+        }
+
+        exec("vsce package --out ./build", (error) => {
+          if (error) throw error;
+          console.log("vsce package completed");
+        });
+      });
+    });
+  });
+});
diff --git a/extension/src/continueIdeClient.ts b/extension/src/continueIdeClient.ts
index d89093ca..5b9e285d 100644
--- a/extension/src/continueIdeClient.ts
+++ b/extension/src/continueIdeClient.ts
@@ -12,7 +12,7 @@ import {
   rejectSuggestionCommand,
 } from "./suggestions";
 import { FileEditWithFullContents } from "../schema/FileEditWithFullContents";
-import * as fs from 'fs';
+import * as fs from "fs";
 import { WebsocketMessenger } from "./util/messenger";
 import { diffManager } from "./diffs";
 const os = require("os");
@@ -30,13 +30,12 @@ class IdeProtocolClient {
   private _lastReloadTime: number = 16;
   private _reconnectionTimeouts: NodeJS.Timeout[] = [];
 
-  private _sessionId: string | null = null;
+  sessionId: string | null = null;
   private _serverUrl: string;
 
   private _newWebsocketMessenger() {
     const requestUrl =
-      this._serverUrl +
-      (this._sessionId ? `?session_id=${this._sessionId}` : "");
+      this._serverUrl + (this.sessionId ? `?session_id=${this.sessionId}` : "");
     const messenger = new WebsocketMessenger(requestUrl);
     this.messenger = messenger;
 
@@ -383,7 +382,9 @@ class IdeProtocolClient {
   async getUserSecret(key: string) {
     // Check if secret already exists in VS Code settings (global)
     let secret = vscode.workspace.getConfiguration("continue").get(key);
-    if (typeof secret !== "undefined" && secret !== null) {return secret;}
+    if (typeof secret !== "undefined" && secret !== null) {
+      return secret;
+    }
 
     // If not, ask user for secret
     secret = await vscode.window.showInputBox({
@@ -420,7 +421,7 @@ class IdeProtocolClient {
     console.log("Getting session ID");
     const resp = await this.messenger?.sendAndReceive("getSessionId", {});
     console.log("New Continue session with ID: ", resp.sessionId);
-    this._sessionId = resp.sessionId;
+    this.sessionId = resp.sessionId;
     return resp.sessionId;
   }
 
diff --git a/extension/src/test-runner/runTestOnVSCodeHost.ts b/extension/src/test-runner/runTestOnVSCodeHost.ts
index 2a542ffc..21267c2d 100644
--- a/extension/src/test-runner/runTestOnVSCodeHost.ts
+++ b/extension/src/test-runner/runTestOnVSCodeHost.ts
@@ -11,7 +11,10 @@ async function main() {
 
     // The path to test runner
     // Passed to --extensionTestsPath
-    const extensionTestsPath = path.resolve(extensionDevelopmentPath, "out/test-runner/mochaRunner");
+    const extensionTestsPath = path.resolve(
+      extensionDevelopmentPath,
+      "out/test-runner/mochaRunner"
+    );
 
     // Download VS Code, unzip it and run the integration test
     await runTests({ extensionDevelopmentPath, extensionTestsPath });
diff --git a/extension/src/test-suite/environmentSetup.test.ts b/extension/src/test-suite/environmentSetup.test.ts
index 9a478522..8099e9ac 100644
--- a/extension/src/test-suite/environmentSetup.test.ts
+++ b/extension/src/test-suite/environmentSetup.test.ts
@@ -2,18 +2,22 @@ import { test, describe } from "mocha";
 import * as assert from "assert";
 
 import { getContinueServerUrl } from "../bridge";
-import { startContinuePythonServer } from "../activation/environmentSetup";
+import { ideProtocolClient } from "../activation/activate";
 import fetch from "node-fetch";
+import fkill from "fkill";
 
 describe("Can start python server", () => {
   test("Can start python server in under 10 seconds", async function () {
-    this.timeout(17_000);
-    await startContinuePythonServer();
+    const allowedTime = 15_000;
+    this.timeout(allowedTime + 10_000);
+    fkill(65432, { force: true, silent: true });
 
-    await new Promise((resolve) => setTimeout(resolve, 15_000));
+    // If successful, the server is started by the extension while we wait
+    await new Promise((resolve) => setTimeout(resolve, allowedTime));
 
     // Check if server is running
     const serverUrl = getContinueServerUrl();
+    console.log("Server URL: ", serverUrl);
     const response = await fetch(`${serverUrl}/health`);
     assert.equal(response.status, 200);
   });