feat: add workspace folder selection for project creation (#501)

fixes microsoft/vscode-python-environments#465

Author: eleanorjboyd

src/common/pickers/managers.ts

@@ -1,4 +1,4 @@
-import { commands, QuickInputButtons, QuickPickItem, QuickPickItemKind } from 'vscode';
+import { commands, QuickInputButtons, QuickPickItem, QuickPickItemKind, workspace, WorkspaceFolder } from 'vscode';
 import { PythonProjectCreator } from '../../api';
 import { InternalEnvironmentManager, InternalPackageManager } from '../../internal.api';
 import { Common, Pickers } from '../localize';
@@ -125,6 +125,31 @@ export async function pickPackageManager(
     return (item as QuickPickItem & { id: string })?.id;
 }
 
+export async function pickWorkspaceFolder(showBackButton = true): Promise<WorkspaceFolder | undefined> {
+    const folders = workspace.workspaceFolders;
+    if (!folders || folders.length === 0) {
+        return undefined;
+    }
+    if (folders.length === 1) {
+        return folders[0];
+    }
+    const items = folders.map((f) => ({
+        label: f.name,
+        description: f.uri.fsPath,
+        folder: f,
+    }));
+
+    const selected = await showQuickPickWithButtons(items, {
+        placeHolder: 'Select a workspace folder',
+        ignoreFocusOut: true,
+        showBackButton,
+    });
+    if (!selected) {
+        return undefined;
+    }
+    const selectedItem = Array.isArray(selected) ? selected[0] : selected;
+    return selectedItem?.folder;
+}
 export async function pickCreator(creators: PythonProjectCreator[]): Promise<PythonProjectCreator | undefined> {
     if (creators.length === 0) {
         return;

src/features/envCommands.ts

@@ -1,4 +1,4 @@
-import { commands, QuickInputButtons, TaskExecution, TaskRevealKind, Terminal, Uri } from 'vscode';
+import { commands, QuickInputButtons, TaskExecution, TaskRevealKind, Terminal, Uri, workspace } from 'vscode';
 import {
     CreateEnvironmentOptions,
     PythonEnvironment,
@@ -20,7 +20,12 @@ import { removePythonProjectSetting, setEnvironmentManager, setPackageManager }
 import { clipboardWriteText } from '../common/env.apis';
 import {} from '../common/errors/utils';
 import { pickEnvironment } from '../common/pickers/environments';
-import { pickCreator, pickEnvironmentManager, pickPackageManager } from '../common/pickers/managers';
+import {
+    pickCreator,
+    pickEnvironmentManager,
+    pickPackageManager,
+    pickWorkspaceFolder,
+} from '../common/pickers/managers';
 import { pickProject, pickProjectMany } from '../common/pickers/projects';
 import { activeTextEditor, showErrorMessage, showInformationMessage } from '../common/window.apis';
 import { quoteArgs } from './execution/execUtils';
@@ -451,6 +456,20 @@ export async function addPythonProjectCommand(
         return;
     }
 
+    // if multiroot, prompt the user to select which workspace to create the project in
+    const workspaceFolders = workspace.workspaceFolders;
+    if (!resource && workspaceFolders && workspaceFolders.length > 1) {
+        try {
+            const workspace = await pickWorkspaceFolder(true);
+            resource = workspace?.uri;
+        } catch (ex) {
+            if (ex === QuickInputButtons.Back) {
+                return addPythonProjectCommand(resource, wm, em, pc);
+            }
+            throw ex;
+        }
+    }
+
     try {
         await creator.create(options);
     } catch (ex) {