add quoting and logging for run as task and background

Author: eleanorjboyd

src/features/execution/execUtils.ts

@@ -1,10 +1,10 @@
-export function quoteArg(arg: string): string {
+export function quoteStringIfNecessary(arg: string): string {
     if (arg.indexOf(' ') >= 0 && !(arg.startsWith('"') && arg.endsWith('"'))) {
         return `"${arg}"`;
     }
     return arg;
 }
 
 export function quoteArgs(args: string[]): string[] {
-    return args.map(quoteArg);
+    return args.map(quoteStringIfNecessary);
 }

src/features/execution/runAsTask.ts

@@ -9,10 +9,10 @@ import {
     WorkspaceFolder,
 } from 'vscode';
 import { PythonEnvironment, PythonTaskExecutionOptions } from '../../api';
-import { traceInfo } from '../../common/logging';
+import { traceInfo, traceWarn } from '../../common/logging';
 import { executeTask } from '../../common/tasks.apis';
 import { getWorkspaceFolder } from '../../common/workspace.apis';
-import { quoteArg } from './execUtils';
+import { quoteStringIfNecessary } from './execUtils';
 
 function getWorkspaceFolderOrDefault(uri?: Uri): WorkspaceFolder | TaskScope {
     const workspace = uri ? getWorkspaceFolder(uri) : undefined;
@@ -26,8 +26,14 @@ export async function runAsTask(
 ): Promise<TaskExecution> {
     const workspace: WorkspaceFolder | TaskScope = getWorkspaceFolderOrDefault(options.project?.uri);
 
-    let executable = environment.execInfo?.activatedRun?.executable ?? environment.execInfo?.run.executable ?? 'python';
-    executable = quoteArg(executable);
+    let executable = environment.execInfo?.activatedRun?.executable ?? environment.execInfo?.run.executable;
+    if (!executable) {
+        traceWarn('No Python executable found in environment; falling back to "python".');
+        executable = 'python';
+    }
+    // Check and quote the executable path if necessary
+    executable = quoteStringIfNecessary(executable);
+
     const args = environment.execInfo?.activatedRun?.args ?? environment.execInfo?.run.args ?? [];
     const allArgs = [...args, ...options.args];
     traceInfo(`Running as task: ${executable} ${allArgs.join(' ')}`);

src/features/execution/runInBackground.ts

@@ -1,13 +1,19 @@
 import * as cp from 'child_process';
 import { PythonBackgroundRunOptions, PythonEnvironment, PythonProcess } from '../../api';
-import { traceError, traceInfo } from '../../common/logging';
+import { traceError, traceInfo, traceWarn } from '../../common/logging';
+import { quoteStringIfNecessary } from './execUtils';
 
 export async function runInBackground(
     environment: PythonEnvironment,
     options: PythonBackgroundRunOptions,
 ): Promise<PythonProcess> {
-    const executable =
-        environment.execInfo?.activatedRun?.executable ?? environment.execInfo?.run.executable ?? 'python';
+    let executable = environment.execInfo?.activatedRun?.executable ?? environment.execInfo?.run.executable;
+    if (!executable) {
+        traceWarn('No Python executable found in environment; falling back to "python".');
+        executable = 'python';
+    }
+    // Check and quote the executable path if necessary
+    executable = quoteStringIfNecessary(executable);
     const args = environment.execInfo?.activatedRun?.args ?? environment.execInfo?.run.args ?? [];
     const allArgs = [...args, ...options.args];
     traceInfo(`Running in background: ${executable} ${allArgs.join(' ')}`);