Add attach to local runspace.

src/PowerShellEditorServices.Protocol/DebugAdapter/AttachRequest.cs

@@ -20,7 +20,7 @@ public class AttachRequestArguments
 
         public string ProcessId { get; set; }
 
-        public int RunspaceId { get; set; }
+        public string RunspaceId { get; set; }
 
         public string CustomPipeName { get; set; }
     }

src/PowerShellEditorServices.Protocol/LanguageServer/GetRunspaceRequest.cs

@@ -0,0 +1,25 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+using Microsoft.PowerShell.EditorServices.Protocol.MessageProtocol;
+
+namespace Microsoft.PowerShell.EditorServices.Protocol.LanguageServer
+{
+    public class GetRunspaceRequest
+    {
+        public static readonly
+            RequestType<object, GetRunspaceResponse[], object, object> Type =
+                RequestType<object, GetRunspaceResponse[], object, object>.Create("powerShell/getRunspace");
+    }
+
+    public class GetRunspaceResponse
+    {
+        public int Id { get; set; }
+
+        public string Name { get; set; }
+
+        public string Availability { get; set; }
+    }
+}

src/PowerShellEditorServices.Protocol/Server/DebugAdapter.cs

@@ -417,16 +417,18 @@ await requestContext.SendErrorAsync(
                     return;
                 }
 
-                await _editorSession.PowerShellContext.ExecuteScriptStringAsync(
+                if (attachParams.ProcessId != "current") {
+                    await _editorSession.PowerShellContext.ExecuteScriptStringAsync(
                     $"Enter-PSHostProcess -Id {processId}",
                     errorMessages);
 
-                if (errorMessages.Length > 0)
-                {
-                    await requestContext.SendErrorAsync(
-                        $"Could not attach to process '{processId}'");
+                    if (errorMessages.Length > 0)
+                    {
+                        await requestContext.SendErrorAsync(
+                            $"Could not attach to process '{processId}'");
 
-                    return;
+                        return;
+                    }
                 }
             }
             else if (customPipeNameIsSet)
@@ -450,7 +452,7 @@ await requestContext.SendErrorAsync(
                     return;
                 }
             }
-            else
+            else if (attachParams.ProcessId != "current")
             {
                 Logger.Write(
                     LogLevel.Error,
@@ -469,7 +471,24 @@ await requestContext.SendErrorAsync(
             // will block the debug adapter initialization process.  The
             // InitializedEvent will be sent as soon as the RunspaceChanged
             // event gets fired with the attached runspace.
-            int runspaceId = attachParams.RunspaceId > 0 ? attachParams.RunspaceId : 1;
+
+            int runspaceId = 1;
+            if (int.TryParse(attachParams.RunspaceId, out runspaceId))
+            {
+                runspaceId = runspaceId > 0 ? runspaceId : 1;
+            }
+            else
+            {
+                Logger.Write(
+                    LogLevel.Error,
+                    $"Attach request failed, '{attachParams.RunspaceId}' is an invalid value for the processId.");
+
+                await requestContext.SendErrorAsync(
+                    "A positive integer must be specified for the RunspaceId field.");
+
+                return;
+            }
+
             _waitingForAttach = true;
             Task nonAwaitedTask = _editorSession.PowerShellContext
                 .ExecuteScriptStringAsync($"\nDebug-Runspace -Id {runspaceId}")

src/PowerShellEditorServices.Protocol/Server/LanguageServer.cs

@@ -18,6 +18,7 @@
 using System.Linq;
 using System.Management.Automation;
 using System.Management.Automation.Language;
+using System.Management.Automation.Runspaces;
 using System.Text;
 using System.Text.RegularExpressions;
 using System.Threading;
@@ -161,6 +162,8 @@ public void Start()
             this.messageHandlers.SetRequestHandler(GetPSHostProcessesRequest.Type, this.HandleGetPSHostProcessesRequestAsync);
             this.messageHandlers.SetRequestHandler(CommentHelpRequest.Type, this.HandleCommentHelpRequestAsync);
 
+            this.messageHandlers.SetRequestHandler(GetRunspaceRequest.Type, this.HandleGetRunspaceRequestAsync);
+
             // Initialize the extension service
             // TODO: This should be made awaited once Initialize is async!
             this.editorSession.ExtensionService.InitializeAsync(
@@ -1218,6 +1221,36 @@ protected async Task HandleCommentHelpRequestAsync(
             await requestContext.SendResultAsync(result);
         }
 
+        protected async Task HandleGetRunspaceRequestAsync(
+            object noParams,
+            RequestContext<GetRunspaceResponse[]> requestContext)
+        {
+            var runspaceResponses = new List<GetRunspaceResponse>();
+
+            if (this.editorSession.PowerShellContext.LocalPowerShellVersion.Version.Major >= 5)
+            {
+                var psCommand = new PSCommand();
+                psCommand.AddCommand("Get-Runspace");
+
+                var runspaces = await editorSession.PowerShellContext.ExecuteCommandAsync<PSObject>(psCommand);
+                if (runspaces != null)
+                {
+                    foreach (dynamic p in runspaces)
+                    {
+                        runspaceResponses.Add(
+                            new GetRunspaceResponse
+                            {
+                                Id = p.Id,
+                                Name = p.Name,
+                                Availability = p.RunspaceAvailability.ToString()
+                            });
+                    }
+                }
+            }
+
+            await requestContext.SendResultAsync(runspaceResponses.ToArray());
+        }
+
         private bool IsQueryMatch(string query, string symbolName)
         {
             return symbolName.IndexOf(query, StringComparison.OrdinalIgnoreCase) >= 0;