Merge pull request #335 from PowerShell/rkeithhill/impl-getPSHostPorcesses

Implement getPSHostProcessesRequest.

Author: daviwil

src/PowerShellEditorServices.Protocol/LanguageServer/GetPSHostProcessesRequest.cs

@@ -0,0 +1,27 @@
+//
+// 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 GetPSHostProcessesRequest
+    {
+        public static readonly
+            RequestType<object, GetPSHostProcessesResponse[]> Type =
+                RequestType<object, GetPSHostProcessesResponse[]>.Create("powerShell/getPSHostProcesses");
+    }
+
+    public class GetPSHostProcessesResponse
+    {
+        public string ProcessName { get; set; }
+
+        public int ProcessId { get; set; }
+
+        public string AppDomainName { get; set; }
+
+        public string MainWindowTitle { get; set; }
+    }
+}

src/PowerShellEditorServices.Protocol/PowerShellEditorServices.Protocol.csproj

@@ -56,6 +56,7 @@
     <Compile Include="DebugAdapter\ContinueRequest.cs" />
     <Compile Include="DebugAdapter\SetFunctionBreakpointsRequest.cs" />
     <Compile Include="DebugAdapter\SetVariableRequest.cs" />
+    <Compile Include="LanguageServer\GetPSHostProcessesRequest.cs" />
     <Compile Include="LanguageServer\GetPSSARulesRequest.cs" />
     <Compile Include="LanguageServer\EditorCommands.cs" />
     <Compile Include="LanguageServer\CodeAction.cs" />

src/PowerShellEditorServices.Protocol/Server/LanguageServer.cs

@@ -123,6 +123,8 @@ protected override void Initialize()
             this.SetRequestHandler(GetPSSARulesRequest.Type, this.HandleGetPSSARulesRequest);
             this.SetRequestHandler(SetPSSARulesRequest.Type, this.HandleSetPSSARulesRequest);
 
+            this.SetRequestHandler(GetPSHostProcessesRequest.Type, this.HandleGetPSHostProcessesRequest);
+
             // Initialize the extension service
             // TODO: This should be made awaited once Initialize is async!
             this.editorSession.ExtensionService.Initialize(
@@ -924,6 +926,42 @@ await requestContext.SendResult(
                     this.editorSession.PowerShellContext.LocalPowerShellVersion));
         }
 
+        protected async Task HandleGetPSHostProcessesRequest(
+            object noParams,
+            RequestContext<GetPSHostProcessesResponse[]> requestContext)
+        {
+            var psHostProcesses = new List<GetPSHostProcessesResponse>();
+
+            if (this.editorSession.PowerShellContext.LocalPowerShellVersion.Version.Major >= 5)
+            {
+                int processId = System.Diagnostics.Process.GetCurrentProcess().Id;
+                var psCommand = new PSCommand();
+                psCommand.AddCommand("Get-PSHostProcessInfo");
+                psCommand.AddCommand("Where-Object")
+                    .AddParameter("Property", "ProcessId")
+                    .AddParameter("NE")
+                    .AddParameter("Value", processId.ToString());
+
+                var processes = await editorSession.PowerShellContext.ExecuteCommand<PSObject>(psCommand);
+                if (processes != null)
+                {
+                    foreach (dynamic p in processes)
+                    {
+                        psHostProcesses.Add(
+                            new GetPSHostProcessesResponse
+                            {
+                                ProcessName = p.ProcessName,
+                                ProcessId = p.ProcessId,
+                                AppDomainName = p.AppDomainName,
+                                MainWindowTitle = p.MainWindowTitle
+                            });
+                    }
+                }
+            }
+
+            await requestContext.SendResult(psHostProcesses.ToArray());
+        }
+
         private bool IsQueryMatch(string query, string symbolName)
         {
             return symbolName.IndexOf(query, StringComparison.OrdinalIgnoreCase) >= 0;