Add Debug Runspace command

package.json

@@ -36,7 +36,8 @@
     "onCommand:PowerShell.ShowSessionConsole",
     "onCommand:PowerShell.ShowSessionMenu",
     "onCommand:PowerShell.RestartSession",
-    "onView:PowerShellCommands"
+    "onView:PowerShellCommands",
+    "onCommand:PowerShell.PickRunspace"
   ],
   "dependencies": {
     "vscode-languageclient": "~5.1.1"
@@ -318,6 +319,7 @@
         "runtime": "node",
         "variables": {
           "PickPSHostProcess": "PowerShell.PickPSHostProcess",
+          "PickRunspace": "PowerShell.PickRunspace",
           "SpecifyScriptArgs": "PowerShell.SpecifyScriptArgs"
         },
         "languages": [
@@ -407,6 +409,17 @@
               "request": "launch",
               "cwd": ""
             }
+          },
+          {
+            "label": "PowerShell: Debug Runspace",
+            "description": "Open runspace picker to select runspace to attach debugger to",
+            "body": {
+              "name": "PowerShell Debug Runspace",
+              "type": "PowerShell",
+              "request": "attach",
+              "processId": "current",
+              "localRunspaceId": "^\"\\${command:PickRunspace}\""
+            }
           }
         ],
         "configurationAttributes": {
@@ -451,6 +464,11 @@
                 "type": "number",
                 "description": "Optional: The ID of the runspace to debug in the attached process.  Defaults to 1.  Works only on PowerShell 5 and above.",
                 "default": 1
+              },
+              "localRunspaceId": {
+                "type": "string",
+                "description": "The ID of the runspace to debug.  Defaults to 1.  Works only on PowerShell 5 and above.",
+                "default": "${command:PickRunspace}"
               }
             }
           }
@@ -495,6 +513,13 @@
             "type": "powershell",
             "request": "launch",
             "cwd": ""
+          },
+          {
+            "name": "PowerShell Debug Runspace",
+            "type": "powershell",
+            "request": "attach",
+            "processId": "current",
+            "localRunspaceId": "${command:PickRunspace}"
           }
         ]
       }

src/features/DebugSession.ts

@@ -381,3 +381,130 @@ export class PickPSHostProcessFeature implements IFeature {
         }
     }
 }
+
+interface IRunspaceItem extends vscode.QuickPickItem {
+    id: string;	// payload for the QuickPick UI
+}
+
+interface IRunspace {
+    id: string;
+    name: string;
+    availability: string;
+}
+
+export const GetRunspaceRequestType =
+    new RequestType<any, IGetRunspaceResponseBody, string, void>("powerShell/getRunspace");
+
+interface IGetRunspaceResponseBody {
+    runspaces: IRunspace[];
+}
+
+export class PickRunspaceFeature implements IFeature {
+
+    private command: vscode.Disposable;
+    private languageClient: LanguageClient;
+    private waitingForClientToken: vscode.CancellationTokenSource;
+    private getLanguageClientResolve: (value?: LanguageClient | Thenable<LanguageClient>) => void;
+
+    constructor() {
+
+        this.command =
+            vscode.commands.registerCommand("PowerShell.PickRunspace", () => {
+                return this.getLanguageClient()
+                           .then((_) => this.pickRunspace(), (_) => undefined);
+            });
+    }
+
+    public setLanguageClient(languageClient: LanguageClient) {
+        this.languageClient = languageClient;
+
+        if (this.waitingForClientToken) {
+            this.getLanguageClientResolve(this.languageClient);
+            this.clearWaitingToken();
+        }
+    }
+
+    public dispose() {
+        this.command.dispose();
+    }
+
+    private getLanguageClient(): Thenable<LanguageClient> {
+        if (this.languageClient) {
+            return Promise.resolve(this.languageClient);
+        } else {
+            // If PowerShell isn't finished loading yet, show a loading message
+            // until the LanguageClient is passed on to us
+            this.waitingForClientToken = new vscode.CancellationTokenSource();
+
+            return new Promise<LanguageClient>(
+                (resolve, reject) => {
+                    this.getLanguageClientResolve = resolve;
+
+                    vscode.window
+                        .showQuickPick(
+                            ["Cancel"],
+                            { placeHolder: "Attach to PowerShell host process: Please wait, starting PowerShell..." },
+                            this.waitingForClientToken.token)
+                        .then((response) => {
+                            if (response === "Cancel") {
+                                this.clearWaitingToken();
+                                reject();
+                            }
+                        });
+
+                    // Cancel the loading prompt after 60 seconds
+                    setTimeout(() => {
+                        if (this.waitingForClientToken) {
+                            this.clearWaitingToken();
+                            reject();
+
+                            vscode.window.showErrorMessage(
+                                "Attach to PowerShell host process: PowerShell session took too long to start.");
+                        }
+                    }, 60000);
+                },
+            );
+        }
+    }
+
+    private pickRunspace(): Thenable<string> {
+        return this.languageClient.sendRequest(GetRunspaceRequestType, null).then((runspaces) => {
+            const items: IRunspaceItem[] = [];
+
+            for (const p in runspaces) {
+                if (runspaces.hasOwnProperty(p)) {
+
+                    // Skip default runspace
+                    if (runspaces[p].id === 1) {
+                        continue;
+                    }
+
+                    items.push({
+                        label: runspaces[p].name,
+                        description: `ID: ${runspaces[p].id} - ${runspaces[p].availability}`,
+                        id: runspaces[p].id,
+                    });
+                }
+            }
+
+            const options: vscode.QuickPickOptions = {
+                placeHolder: "Select PowerShell runspace to debug",
+                matchOnDescription: true,
+                matchOnDetail: true,
+            };
+
+            return vscode.window.showQuickPick(items, options).then((item) => {
+                // Return undefined when user presses Esc.
+                // This prevents VSCode from opening launch.json in this case which happens if we return "".
+                return item ? `${item.id}` : undefined;
+            });
+        });
+    }
+
+    private clearWaitingToken() {
+        if (this.waitingForClientToken) {
+            this.waitingForClientToken.dispose();
+            this.waitingForClientToken = undefined;
+        }
+    }
+}

src/main.ts

@@ -13,6 +13,7 @@ import { ConsoleFeature } from "./features/Console";
 import { CustomViewsFeature } from "./features/CustomViews";
 import { DebugSessionFeature } from "./features/DebugSession";
 import { PickPSHostProcessFeature } from "./features/DebugSession";
+import { PickRunspaceFeature } from "./features/DebugSession";
 import { SpecifyScriptArgsFeature } from "./features/DebugSession";
 import { DocumentFormatterFeature } from "./features/DocumentFormatter";
 import { ExamplesFeature } from "./features/Examples";
@@ -142,6 +143,7 @@ export function activate(context: vscode.ExtensionContext): void {
         new SpecifyScriptArgsFeature(context),
         new HelpCompletionFeature(logger),
         new CustomViewsFeature(),
+        new PickRunspaceFeature(),
     ];
 
     sessionManager.setExtensionFeatures(extensionFeatures);