[vscode-extension] Handling for window os and devbox not installed (#835)

## Summary
- Added handling for open in vscode on windows (show as unsupported will
implement handling WSL in future).
- Added handling for devbox not installed for open in vscode by
downloading and using launcher script.

 
## How was it tested?
See steps in #804

Author: mohsenari

vscode-extension/package.json

@@ -108,6 +108,7 @@
     "@types/node": "16.x",
     "@types/node-fetch": "^2",
     "@types/vscode": "^1.72.0",
+    "@types/which": "^3.0.0",
     "@typescript-eslint/eslint-plugin": "^5.38.1",
     "@typescript-eslint/parser": "^5.38.1",
     "@vscode/test-electron": "^2.1.5",
@@ -117,7 +118,9 @@
     "typescript": "^4.8.4"
   },
   "dependencies": {
+    "@types/node": "16.x",
     "form-data": "^4.0.0",
-    "node-fetch": "^2"
+    "node-fetch": "^2",
+    "which": "^3.0.0"
   }
-}
+}

vscode-extension/src/openinvscode.ts

@@ -1,7 +1,9 @@
-import { Uri, commands, window } from 'vscode';
+import * as os from 'os';
+import * as which from 'which';
 import fetch from 'node-fetch';
 import { exec } from 'child_process';
 import * as FormData from 'form-data';
+import { Uri, commands, window } from 'vscode';
 import { chmod, open, writeFile } from 'fs/promises';
 
 type VmInfo = {
@@ -17,17 +19,21 @@ export async function handleOpenInVSCode(uri: Uri) {
     const queryParams = new URLSearchParams(uri.query);
 
     if (queryParams.has('vm_id') && queryParams.has('token')) {
-        window.showInformationMessage('Setting up devbox');
-
-        // getting ssh keys
-        const response = await getVMInfo(queryParams.get('token'), queryParams.get('vm_id'));
-        const res = await response.json() as VmInfo;
-        console.debug("data:");
-        console.debug(res);
-        // set ssh config
-        await setupSSHConfig(res.vm_id, res.private_key);
-        // connect to remote vm
-        connectToRemote(res.username, res.vm_id, res.working_directory);
+        //Not yet supported for windows + WSL - will be added in future
+        if (os.platform() !== 'win32') {
+            window.showInformationMessage('Setting up devbox');
+            // getting ssh keys
+            const response = await getVMInfo(queryParams.get('token'), queryParams.get('vm_id'));
+            const res = await response.json() as VmInfo;
+            console.debug("data:");
+            console.debug(res);
+            // set ssh config
+            await setupSSHConfig(res.vm_id, res.private_key);
+            // connect to remote vm
+            connectToRemote(res.username, res.vm_id, res.working_directory);
+        } else {
+            window.showErrorMessage('This function is not yet supported in Windows operating system.');
+        }
     } else {
         window.showErrorMessage('Error parsing information for remote environment.');
         console.debug(queryParams.toString());
@@ -49,13 +55,39 @@ async function getVMInfo(token: string | null, vmId: string | null): Promise<any
     return response;
 }
 
+async function setupDevboxLauncher(): Promise<any> {
+    // download devbox launcher script
+    const gatewayHost = 'https://releases.jetpack.io/devbox';
+    const response = await fetch(gatewayHost, {
+        method: 'get',
+    });
+    const launcherPath = `${process.env['HOME']}/.config/devbox/launcher.sh`;
+
+    try {
+        const launcherScript = await response.text();
+        const launcherData = new Uint8Array(Buffer.from(launcherScript));
+        const fileHandler = await open(launcherPath, 'w');
+        await writeFile(fileHandler, launcherData, { flag: 'w' });
+        await chmod(launcherPath, 0o711);
+        await fileHandler.close();
+    } catch (err) {
+        console.error(err);
+    }
+    return launcherPath;
+}
+
 async function setupSSHConfig(vmId: string, prKey: string) {
-    // TODO: change this back before to devbox generate ssh-config
-    // This requires a release for devbox that has generate ssh-config included in it
+
+    const devboxBinary = await which('devbox', { nothrow: true });
+    let devboxPath = 'devbox';
+    if (devboxBinary === null) {
+        devboxPath = await setupDevboxLauncher();
+    }
     // For testing change devbox to path to a compiled devbox binary or add --config
-    exec('devbox generate ssh-config', (error, stdout, stderr) => {
+    exec(`${devboxPath} generate ssh-config`, (error, stdout, stderr) => {
         if (error) {
-            console.error(`exec error: ${error}`);
+            window.showErrorMessage('Failed to setup ssh config. Run VSCode in debug mode to see logs.');
+            console.error(`Failed to setup ssh config: ${error}`);
             return;
         }
         console.debug(`stdout: ${stdout}`);
@@ -71,6 +103,7 @@ async function setupSSHConfig(vmId: string, prKey: string) {
         await fileHandler.close();
     } catch (err) {
         // When a request is aborted - err is an AbortError
+        window.showErrorMessage('Failed to setup ssh config. Run VSCode in debug mode to see logs.');
         console.error(err);
     }
 }