check for virtual workspaces

Author: KTrain5169

package.json

@@ -605,7 +605,7 @@
 	"capabilities": {
 		"virtualWorkspaces": {
 			"supported": "limited",
-			"description": "In certain virtual workspace scenarios (namely Web), Terminal, Tasks, Git support and Copilot Chat mode is not available."
+			"description": "In virtual workspaces, Terminal, Tasks, Git support (restricted on web) and Copilot Chat mode (restricted on web) is not available."
 		},
 		"untrustedWorkspaces": {
 			"supported": "limited",

src/pseudoterminal.ts

@@ -7,7 +7,7 @@
 import * as vscode from 'vscode';
 import { spawn, ChildProcessWithoutNullStreams } from 'node:child_process';
 import { NEURO } from '~/constants';
-import { checkWorkspaceTrust } from '~/utils';
+import { checkWorkspaceTrust, checkVirtualWorkspace } from '~/utils';
 import { logOutput, delayAsync, getFence } from '~/utils';
 import { ActionData, actionValidationAccept, actionValidationFailure, ActionValidationResult, RCEAction, contextFailure, stripToActions } from '~/neuro_client_helper';
 import { CONFIG, PERMISSIONS, getPermissionLevel } from '~/config';
@@ -49,7 +49,7 @@ export const terminalAccessHandlers = {
         },
         permissions: [PERMISSIONS.terminalAccess],
         handler: handleRunCommand,
-        validator: [checkWorkspaceTrust],
+        validator: [checkVirtualWorkspace, checkWorkspaceTrust],
         promptGenerator: (actionData: ActionData) => `run "${actionData.params?.command}" in the "${actionData.params?.shell}" shell.`,
     },
     'kill_terminal_process': {
@@ -64,15 +64,15 @@ export const terminalAccessHandlers = {
         },
         permissions: [PERMISSIONS.terminalAccess],
         handler: handleKillTerminal,
-        validator: [checkLiveTerminals, checkWorkspaceTrust],
+        validator: [checkLiveTerminals, checkVirtualWorkspace, checkWorkspaceTrust],
         promptGenerator: (actionData: ActionData) => `kill the "${actionData.params?.shell}" shell.`,
     },
     'get_currently_running_shells': {
         name: 'get_currently_running_shells',
         description: 'Get the list of terminal processes that are spawned.',
         permissions: [PERMISSIONS.terminalAccess],
         handler: handleGetCurrentlyRunningShells,
-        validator: [checkWorkspaceTrust],
+        validator: [checkVirtualWorkspace, checkWorkspaceTrust],
         promptGenerator: 'get the list of currently running shells.',
     },
 } satisfies Record<string, RCEAction>;

src/rce.ts

@@ -6,7 +6,7 @@
 import * as vscode from 'vscode';
 import { ActionData, RCEAction } from '~/neuro_client_helper';
 import { NEURO } from '~/constants';
-import { checkWorkspaceTrust, logOutput } from '~/utils';
+import { checkVirtualWorkspace, checkWorkspaceTrust, logOutput } from '~/utils';
 import { CONFIG, getPermissionLevel, PermissionLevel, PERMISSIONS } from '~/config';
 import { handleRunTask } from '~/tasks';
 import { validate } from 'jsonschema';
@@ -252,7 +252,7 @@ export async function RCEActionHandler(actionData: ActionData, actionList: Recor
                 description: task.description,
                 permissions: [PERMISSIONS.runTasks],
                 handler: handleRunTask,
-                validator: [checkWorkspaceTrust],
+                validator: [checkVirtualWorkspace, checkWorkspaceTrust],
                 promptGenerator: () => `run the task "${task.id}".`,
             };
         }

src/tasks.ts

@@ -7,7 +7,7 @@
 import * as vscode from 'vscode';
 
 import { NEURO } from '~/constants';
-import { logOutput, formatActionID, getFence, checkWorkspaceTrust } from '~/utils';
+import { logOutput, formatActionID, getFence, checkWorkspaceTrust, checkVirtualWorkspace } from '~/utils';
 import { ActionData, RCEAction, actionValidationAccept, actionValidationFailure, stripToActions } from '~/neuro_client_helper';
 import { CONFIG, PERMISSIONS, getPermissionLevel } from '~/config';
 
@@ -19,7 +19,7 @@ export const taskHandlers = {
         permissions: [PERMISSIONS.runTasks],
         handler: handleTerminateTask,
         promptGenerator: 'terminate the currently running task.',
-        validator: [checkWorkspaceTrust, () => NEURO.currentTaskExecution !== null
+        validator: [checkVirtualWorkspace, checkWorkspaceTrust, () => NEURO.currentTaskExecution !== null
             ? actionValidationAccept()
             : actionValidationFailure('No task to terminate.')],
     },

src/utils.ts

@@ -390,3 +390,13 @@ export function checkWorkspaceTrust(_actionData: ActionData): ActionValidationRe
     }
     return actionValidationFailure('You are in an untrusted workspace.');
 }
+
+/**
+ * Checks if the extension is currently on a virtual file system.
+ */
+export function checkVirtualWorkspace(_actionData: ActionData): ActionValidationResult {
+    if (vscode.workspace.workspaceFolders?.every(f => f.uri.scheme !== 'file')) {
+        return actionValidationFailure('You are currently in a virtual workspace.');
+    }
+    return actionValidationAccept();
+}