Add URI handler for one-click open

Author: kylecarbs

package.json

@@ -5,7 +5,7 @@
   "description": "Open any workspace with a single click.",
   "repository": "https://github.com/coder/vscode-coder",
   "preview": true,
-  "version": "0.0.3",
+  "version": "0.0.4",
   "engines": {
     "vscode": "^1.73.0"
   },

src/extension.ts

@@ -24,12 +24,28 @@ export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
     })
 
   vscode.window.registerUriHandler({
-    handleUri: (uri) => {
+    handleUri: async (uri) => {
       const params = new URLSearchParams(uri.query)
       if (uri.path === "/open") {
         const owner = params.get("owner")
-        const name = params.get("name")
-        vscode.commands.executeCommand("coder.open", owner, name)
+        const workspace = params.get("workspace")
+        const agent = params.get("agent")
+        if (!owner) {
+          throw new Error("owner must be specified as a query parameter")
+        }
+        if (!workspace) {
+          throw new Error("workspace must be specified as a query parameter")
+        }
+
+        const url = params.get("url")
+        const token = params.get("token")
+        if (url) {
+          await storage.setURL(url)
+        }
+        if (token) {
+          await storage.setSessionToken(token)
+        }
+        vscode.commands.executeCommand("coder.open", owner, workspace, agent)
       }
     },
   })