Use VS Code Remote SSH instead of custom installer

Fixes #15, #12, #5, #2, and #1.

Author: kylecarbs

package.json

@@ -21,11 +21,11 @@
     "Other"
   ],
   "activationEvents": [
-    "onResolveRemoteAuthority:coder",
+    "onResolveRemoteAuthority:ssh-remote",
     "onCommand:coder.connect",
     "onCommand:coder.open",
     "onCommand:coder.login",
-    "onView:coderWorkspaces",
+    "onView:coderRemote",
     "onUri"
   ],
   "extensionDependencies": [
@@ -45,7 +45,7 @@
     "views": {
       "coder": [
         {
-          "id": "coderWorkspaces",
+          "id": "coderRemote",
           "name": "",
           "visibility": "visible",
           "icon": "media/logo.svg",
@@ -55,12 +55,12 @@
     },
     "viewsWelcome": [
       {
-        "view": "coderWorkspaces",
+        "view": "coderRemote",
         "contents": "Coder is a platform that provisions remote development environments. \n[Login](command:coder.login)",
         "when": "!coder.authenticated && coder.loaded"
       },
       {
-        "view": "coderWorkspaces",
+        "view": "coderRemote",
         "contents": "You're logged in! \n[Open Workspace](command:coder.open)",
         "when": "coder.authenticated && coder.loaded"
       }
@@ -132,8 +132,11 @@
   "dependencies": {
     "axios": "0.26.1",
     "eventsource": "^2.0.2",
+    "find-process": "^1.4.7",
+    "jsonc-parser": "^3.2.0",
     "ndjson": "^2.0.0",
     "pretty-bytes": "^6.0.0",
+    "ssh-config": "4.1.6",
     "tar-fs": "^2.1.1",
     "which": "^2.0.2",
     "ws": "^8.11.0",

src/commands.ts

@@ -2,6 +2,7 @@ import axios from "axios"
 import { getUser, getWorkspaces } from "coder/site/src/api/api"
 import { Workspace } from "coder/site/src/api/typesGenerated"
 import * as vscode from "vscode"
+import { Remote } from "./remote"
 import { Storage } from "./storage"
 
 export class Commands {
@@ -33,7 +34,7 @@ export class Commands {
         title: "Coder API Key",
         password: true,
         placeHolder: "Copy your API key from the opened browser page.",
-        value: this.storage.getSessionToken(),
+        value: await this.storage.getSessionToken(),
         ignoreFocusOut: true,
         validateInput: (value) => {
           return axios
@@ -151,7 +152,7 @@ export class Commands {
 
     // A workspace can have multiple agents, but that's handled
     // when opening a workspace unless explicitly specified.
-    let uri = vscode.Uri.parse(`vscode-remote://coder+${workspaceOwner}.${workspaceName}/`)
+    let uri = vscode.Uri.parse(`vscode-remote://ssh-remote+${Remote.Prefix}${workspaceOwner}--${workspaceName}/`)
 
     const output: {
       workspaces: { folderUri: vscode.Uri; remoteAuthority: string }[]

src/extension.ts

@@ -1,21 +1,16 @@
 "use strict"
 
 import { getUser } from "coder/site/src/api/api"
-import { readFileSync } from "fs"
 import * as module from "module"
-import path from "path"
 import * as vscode from "vscode"
 import { Commands } from "./commands"
 import { Remote } from "./remote"
 import { Storage } from "./storage"
 
 export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
-  const productJSON = readFileSync(path.join(vscode.env.appRoot, "product.json"))
-  const product = JSON.parse(productJSON.toString())
-  const commit = product.commit
   const output = vscode.window.createOutputChannel("Coder")
-  const storage = new Storage(output, ctx.globalState, ctx.globalStorageUri)
-  storage.init()
+  const storage = new Storage(output, ctx.globalState, ctx.secrets, ctx.globalStorageUri, ctx.logUri)
+  await storage.init()
 
   getUser()
     .then(() => {
@@ -45,13 +40,16 @@ export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
   vscode.commands.registerCommand("coder.logout", commands.logout.bind(commands))
   vscode.commands.registerCommand("coder.open", commands.open.bind(commands))
 
-  // The remote SSH extension is required to provide the restricted
-  // proposed API for registering remote authority providers.
+  // The Remote SSH extension's proposed APIs are used to override
+  // the SSH host name in VS Code itself. It's visually unappealing
+  // having a lengthy name!
+  //
+  // This is janky, but that's alright since it provides such minimal
+  // functionality to the extension.
   const remoteSSHExtension = vscode.extensions.getExtension("ms-vscode-remote.remote-ssh")
   if (!remoteSSHExtension) {
     throw new Error("Remote SSH extension not found")
   }
-
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   const vscodeProposed: typeof vscode = (module as any)._load(
     "vscode",
@@ -61,8 +59,18 @@ export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
     false,
   )
 
-  const remote = new Remote(output, vscodeProposed, storage, commit)
-  ctx.subscriptions.push(remote)
+  // Since the "onResolveRemoteAuthority:ssh-remote" activation event exists
+  // in package.json we're able to perform actions before the authority is
+  // resolved by the remote SSH extension.
+  const activeRemotes = vscode.workspace.workspaceFolders?.filter((folder) => folder.uri.scheme === "vscode-remote")
+  // If the currently opened folder isn't remote we can return early!
+  if (activeRemotes?.length !== 1) {
+    return
+  }
+  const activeRemote = activeRemotes[0].uri
+
+  ctx.globalStorageUri
 
-  vscodeProposed.workspace.registerRemoteAuthorityResolver("coder", remote)
+  const remote = new Remote(vscodeProposed, storage, ctx.extensionMode)
+  await remote.setup(activeRemote)
 }