summaryrefslogtreecommitdiff
path: root/extension/src/activation/activate.ts
blob: 8ea08e89ad95c73a35fb498935f4249b89f9fac1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
import * as vscode from "vscode";
import { registerAllCommands } from "../commands";
import { registerAllCodeLensProviders } from "../lang-server/codeLens";
import { sendTelemetryEvent, TelemetryEvent } from "../telemetry";
import IdeProtocolClient from "../continueIdeClient";
import { getContinueServerUrl } from "../bridge";
import { ContinueGUIWebviewViewProvider } from "../debugPanel";
import {
  getExtensionVersion,
  startContinuePythonServer,
} from "./environmentSetup";
import fetch from "node-fetch";
import registerQuickFixProvider from "../lang-server/codeActions";
// import { CapturedTerminal } from "../terminal/terminalEmulator";

const PACKAGE_JSON_RAW_GITHUB_URL =
  "https://raw.githubusercontent.com/continuedev/continue/HEAD/extension/package.json";

export let extensionContext: vscode.ExtensionContext | undefined = undefined;

export let ideProtocolClient: IdeProtocolClient;

export async function activateExtension(context: vscode.ExtensionContext) {
  extensionContext = context;

  // Before anything else, check whether this is an out-of-date version of the extension
  // Do so by grabbing the package.json off of the GitHub respository for now.
  fetch(PACKAGE_JSON_RAW_GITHUB_URL)
    .then(async (res) => res.json())
    .then((packageJson) => {
      if (packageJson.version !== getExtensionVersion()) {
        vscode.window.showInformationMessage(
          `You are using an out-of-date version of the Continue extension. Please update to the latest version.`
        );
      }
    })
    .catch((e) => console.log("Error checking for extension updates: ", e));

  // Wrap the server start logic in a new Promise
  const serverStartPromise = new Promise((resolve, reject) => {
    let serverStarted = false;

    // Start the server and set serverStarted to true when done
    startContinuePythonServer().then(() => {
      serverStarted = true;
      resolve(null);
    });

    // Wait for 2 seconds
    setTimeout(() => {
      // If the server hasn't started after 2 seconds, show the notification
      if (!serverStarted) {
        vscode.window.withProgress(
          {
            location: vscode.ProgressLocation.Notification,
            title:
              "Starting Continue Server... (it may take a minute to download Python packages)",
            cancellable: false,
          },
          async (progress, token) => {
            // Wait for the server to start
            while (!serverStarted) {
              await new Promise((innerResolve) =>
                setTimeout(innerResolve, 1000)
              );
            }
            return Promise.resolve();
          }
        );
      }
    }, 2000);
  });

  // Await the server start promise
  await serverStartPromise;

  // Register commands and providers
  sendTelemetryEvent(TelemetryEvent.ExtensionActivated);
  registerAllCodeLensProviders(context);
  registerAllCommands(context);
  registerQuickFixProvider();

  // Initialize IDE Protocol Client
  const serverUrl = getContinueServerUrl();
  ideProtocolClient = new IdeProtocolClient(
    `${serverUrl.replace("http", "ws")}/ide/ws`,
    context
  );

  {
    const sessionIdPromise = await ideProtocolClient.getSessionId();
    const provider = new ContinueGUIWebviewViewProvider(sessionIdPromise);

    context.subscriptions.push(
      vscode.window.registerWebviewViewProvider(
        "continue.continueGUIView",
        provider,
        {
          webviewOptions: { retainContextWhenHidden: true },
        }
      )
    );
  }
}