Merge pull request #117 from PowerShell/spike-dcf-psgallery

Add support for finding and installing PowerShell Gallery modules

Author: daviwil

src/PowerShellEditorServices.Protocol/LanguageServer/FindModuleRequest.cs

@@ -0,0 +1,24 @@
+//
+// 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;
+using System.Collections.Generic;
+
+namespace Microsoft.PowerShell.EditorServices.Protocol.LanguageServer
+{
+    public class FindModuleRequest
+    {
+        public static readonly
+            RequestType<List<PSModuleMessage>, object> Type =
+            RequestType<List<PSModuleMessage>, object>.Create("powerShell/findModule");
+    }
+
+
+    public class PSModuleMessage
+    {
+        public string Name { get; set; }
+        public string Description { get; set; }
+    }
+}

src/PowerShellEditorServices.Protocol/LanguageServer/InstallModuleRequest.cs

@@ -0,0 +1,16 @@
+//
+// 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
+{
+    class InstallModuleRequest
+    {
+        public static readonly
+            RequestType<string, object> Type =
+            RequestType<string, object>.Create("powerShell/installModule");
+    }
+}

src/PowerShellEditorServices.Protocol/PowerShellEditorServices.Protocol.csproj

@@ -52,6 +52,8 @@
     <Compile Include="DebugAdapter\AttachRequest.cs" />
     <Compile Include="DebugAdapter\Breakpoint.cs" />
     <Compile Include="DebugAdapter\ContinueRequest.cs" />
+    <Compile Include="LanguageServer\FindModuleRequest.cs" />
+    <Compile Include="LanguageServer\InstallModuleRequest.cs" />
     <Compile Include="Messages\PromptEvents.cs" />
     <Compile Include="Server\DebugAdapter.cs" />
     <Compile Include="Server\DebugAdapterBase.cs" />

src/PowerShellEditorServices.Protocol/Server/LanguageServer.cs

@@ -67,6 +67,10 @@ protected override void Initialize()
 
             this.SetRequestHandler(ShowOnlineHelpRequest.Type, this.HandleShowOnlineHelpRequest);
             this.SetRequestHandler(ExpandAliasRequest.Type, this.HandleExpandAliasRequest);
+
+            this.SetRequestHandler(FindModuleRequest.Type, this.HandleFindModuleRequest);
+            this.SetRequestHandler(InstallModuleRequest.Type, this.HandleInstallModuleRequest);
+
             this.SetEventHandler(CompleteChoicePromptNotification.Type, this.HandleCompleteChoicePromptNotification);
 
             this.SetRequestHandler(DebugAdapterMessages.EvaluateRequest.Type, this.HandleEvaluateRequest);
@@ -128,12 +132,28 @@ protected async Task HandleShowOnlineHelpRequest(
             psCommand.AddArgument(helpParams);
             psCommand.AddParameter("Online");
 
-            await editorSession.PowerShellContext.ExecuteCommand<object>(
-                    psCommand);
+            await editorSession.PowerShellContext.ExecuteCommand<object>(psCommand);
+
+            await requestContext.SendResult(null);
+        }
+
+        private async Task HandleInstallModuleRequest(
+            string moduleName,
+            RequestContext<object> requestContext
+        )
+        {
+            var script = string.Format("Install-Module -Name {0} -Scope CurrentUser", moduleName);
+
+            var executeTask =
+               editorSession.PowerShellContext.ExecuteScriptString(
+                   script,
+                   true,
+                   true).ConfigureAwait(false);
 
             await requestContext.SendResult(null);
         }
 
+
         private async Task HandleExpandAliasRequest(
             string content,
             RequestContext<string> requestContext)
@@ -172,6 +192,28 @@ Sort Start -Descending
             await requestContext.SendResult(result.First().ToString());
         }
 
+        private async Task HandleFindModuleRequest(
+            object param,
+            RequestContext<object> requestContext)
+        {
+            var psCommand = new PSCommand();
+            psCommand.AddScript("Find-Module | Select Name, Description");
+
+            var modules = await editorSession.PowerShellContext.ExecuteCommand<PSObject>(psCommand);
+
+            var moduleList = new List<PSModuleMessage>();
+
+            if (modules != null)
+            {
+                foreach (dynamic m in modules)
+                {
+                    moduleList.Add(new PSModuleMessage { Name = m.Name, Description = m.Description });
+                }
+            }
+
+            await requestContext.SendResult(moduleList);
+        }
+
         protected Task HandleCompleteChoicePromptNotification(
             CompleteChoicePromptNotification completeChoicePromptParams,
             EventContext eventContext)