Add support for Enter-PSSession and Enter-PSHostProcess

This change adds support for remote sessions (PSv4+) and attaching to
processes on the same machine (PSv5+).

Author: daviwil

src/PowerShellEditorServices.Protocol/LanguageServer/PowerShellVersionRequest.cs

@@ -11,11 +11,11 @@ namespace Microsoft.PowerShell.EditorServices.Protocol.LanguageServer
     public class PowerShellVersionRequest
     {
         public static readonly
-            RequestType<object, PowerShellVersionResponse> Type =
-            RequestType<object, PowerShellVersionResponse>.Create("powerShell/getVersion");
+            RequestType<object, PowerShellVersion> Type =
+            RequestType<object, PowerShellVersion>.Create("powerShell/getVersion");
     }
 
-    public class PowerShellVersionResponse
+    public class PowerShellVersion
     {
         public string Version { get; set; }
 
@@ -25,11 +25,11 @@ public class PowerShellVersionResponse
 
         public string Architecture { get; set; }
 
-        public PowerShellVersionResponse()
+        public PowerShellVersion()
         {
         }
 
-        public PowerShellVersionResponse(PowerShellVersionDetails versionDetails)
+        public PowerShellVersion(PowerShellVersionDetails versionDetails)
         {
             this.Version = versionDetails.VersionString;
             this.DisplayVersion =  $"{versionDetails.Version.Major}.{versionDetails.Version.Minor}";

src/PowerShellEditorServices.Protocol/LanguageServer/RunspaceChanged.cs

@@ -0,0 +1,37 @@
+//
+// 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.Session;
+using Microsoft.PowerShell.EditorServices.Protocol.MessageProtocol;
+
+namespace Microsoft.PowerShell.EditorServices.Protocol.LanguageServer
+{
+    public class RunspaceChangedEvent
+    {
+        public static readonly
+            EventType<RunspaceDetails> Type =
+            EventType<RunspaceDetails>.Create("powerShell/runspaceChanged");
+    }
+
+    public class RunspaceDetails
+    {
+        public PowerShellVersion PowerShellVersion { get; set; }
+
+        public RunspaceType RunspaceType { get; set; }
+
+        public string ConnectionString { get; set; }
+
+        public RunspaceDetails()
+        {
+        }
+
+        public RunspaceDetails(RunspaceChangedEventArgs eventArgs)
+        {
+            this.PowerShellVersion = new PowerShellVersion(eventArgs.RunspaceVersion);
+            this.RunspaceType = eventArgs.RunspaceType;
+            this.ConnectionString = eventArgs.ConnectionString;
+        }
+    }
+}

src/PowerShellEditorServices.Protocol/PowerShellEditorServices.Protocol.csproj

@@ -61,6 +61,7 @@
     <Compile Include="LanguageServer\FindModuleRequest.cs" />
     <Compile Include="LanguageServer\InstallModuleRequest.cs" />
     <Compile Include="LanguageServer\PowerShellVersionRequest.cs" />
+    <Compile Include="LanguageServer\RunspaceChanged.cs" />
     <Compile Include="LanguageServer\SetPSSARulesRequest.cs" />
     <Compile Include="LanguageServer\ProjectTemplate.cs" />
     <Compile Include="MessageProtocol\Channel\NamedPipeClientChannel.cs" />

src/PowerShellEditorServices.Protocol/Server/LanguageServer.cs

@@ -52,6 +52,7 @@ public LanguageServer(HostDetails hostDetails, ProfilePaths profilePaths, Channe
             this.editorSession = new EditorSession();
             this.editorSession.StartSession(hostDetails, profilePaths);
             this.editorSession.ConsoleService.OutputWritten += this.powerShellContext_OutputWritten;
+            this.editorSession.PowerShellContext.RunspaceChanged += PowerShellContext_RunspaceChanged;
 
             // Attach to ExtensionService events
             this.editorSession.ExtensionService.CommandAdded += ExtensionService_ExtensionAdded;
@@ -910,10 +911,10 @@ protected async Task HandleWorkspaceSymbolRequest(
 
         protected async Task HandlePowerShellVersionRequest(
             object noParams,
-            RequestContext<PowerShellVersionResponse> requestContext)
+            RequestContext<PowerShellVersion> requestContext)
         {
             await requestContext.SendResult(
-                new PowerShellVersionResponse(
+                new PowerShellVersion(
                     this.editorSession.PowerShellContext.PowerShellVersionDetails));
         }
 
@@ -988,6 +989,13 @@ protected Task HandleEvaluateRequest(
 
         #region Event Handlers
 
+        private async void PowerShellContext_RunspaceChanged(object sender, RunspaceChangedEventArgs e)
+        {
+            await this.SendEvent(
+                RunspaceChangedEvent.Type,
+                new RunspaceDetails(e));
+        }
+
         private async void powerShellContext_OutputWritten(object sender, OutputWrittenEventArgs e)
         {
             // Queue the output for writing

src/PowerShellEditorServices/Language/CommandHelpers.cs

@@ -28,8 +28,8 @@ public static async Task<CommandInfo> GetCommandInfo(
             command.AddCommand(@"Microsoft.PowerShell.Core\Get-Command");
             command.AddArgument(commandName);
 
-            var results = await powerShellContext.ExecuteCommand<CommandInfo>(command, false, false);
-            return results.FirstOrDefault();
+            var results = await powerShellContext.ExecuteCommand<object>(command, false, false);
+            return results.OfType<CommandInfo>().FirstOrDefault();
         }
 
         /// <summary>

src/PowerShellEditorServices/Language/LanguageService.cs

@@ -346,7 +346,7 @@ await CommandHelpers.GetCommandInfo(
                         this.powerShellContext);
 
                 foundDefinition = 
-                    await FindDeclarationForBuiltinCommand(
+                    FindDeclarationForBuiltinCommand(
                         cmdInfo, 
                         foundSymbol,
                         workspace);
@@ -532,22 +532,17 @@ private ScriptFile[] GetBuiltinCommandScriptFiles(
             return new List<ScriptFile>().ToArray();
         }
 
-        private async Task<SymbolReference> FindDeclarationForBuiltinCommand(
-            CommandInfo cmdInfo, 
+        private SymbolReference FindDeclarationForBuiltinCommand(
+            CommandInfo commandInfo, 
             SymbolReference foundSymbol,
             Workspace workspace)
         {
             SymbolReference foundDefinition = null;
-            if (cmdInfo != null)
+            if (commandInfo != null)
             {
                 int index = 0;
                 ScriptFile[] nestedModuleFiles;
 
-                CommandInfo commandInfo =
-                    await CommandHelpers.GetCommandInfo(
-                        foundSymbol.SymbolName,
-                        this.powerShellContext);
-
                 nestedModuleFiles =
                     GetBuiltinCommandScriptFiles(
                         commandInfo.Module,

src/PowerShellEditorServices/Nano.PowerShellEditorServices.csproj

@@ -113,6 +113,7 @@
     <Compile Include="Session\SessionPSHost.cs" />
     <Compile Include="Session\SessionPSHostRawUserInterface.cs" />
     <Compile Include="Session\SessionPSHostUserInterface.cs" />
+    <Compile Include="Session\RunspaceChangedEventArgs.cs" />
     <Compile Include="Session\SessionStateChangedEventArgs.cs" />
     <Compile Include="Utility\AsyncContextThread.cs" />
     <Compile Include="Utility\AsyncDebouncer.cs" />
@@ -149,6 +150,7 @@
     <Compile Include="..\PowerShellEditorServices.Protocol\LanguageServer\FindModuleRequest.cs" />
     <Compile Include="..\PowerShellEditorServices.Protocol\LanguageServer\InstallModuleRequest.cs" />
     <Compile Include="..\PowerShellEditorServices.Protocol\LanguageServer\PowerShellVersionRequest.cs" />
+    <Compile Include="..\PowerShellEditorServices.Protocol\LanguageServer\RunspaceChanged.cs" />
     <Compile Include="..\PowerShellEditorServices.Protocol\MessageProtocol\Channel\NamedPipeClientChannel.cs" />
     <Compile Include="..\PowerShellEditorServices.Protocol\MessageProtocol\Channel\NamedPipeServerChannel.cs" />
     <Compile Include="..\PowerShellEditorServices.Protocol\MessageProtocol\Channel\TcpSocketClientChannel.cs" />

src/PowerShellEditorServices/PowerShellEditorServices.csproj

@@ -121,6 +121,7 @@
     <Compile Include="Session\PowerShellVersionDetails.cs" />
     <Compile Include="Session\ProfilePaths.cs" />
     <Compile Include="Session\ProgressDetails.cs" />
+    <Compile Include="Session\RunspaceChangedEventArgs.cs" />
     <Compile Include="Session\RunspaceHandle.cs" />
     <Compile Include="Session\SessionPSHost.cs" />
     <Compile Include="Session\SessionPSHostRawUserInterface.cs" />

src/PowerShellEditorServices/Session/PowerShell4Operations.cs

@@ -16,7 +16,10 @@ internal class PowerShell4Operations : IVersionSpecificOperations
         public void ConfigureDebugger(Runspace runspace)
         {
 #if !PowerShellv3
-            runspace.Debugger.SetDebugMode(DebugModes.LocalScript | DebugModes.RemoteScript);
+            if (runspace.Debugger != null)
+            {
+                runspace.Debugger.SetDebugMode(DebugModes.LocalScript | DebugModes.RemoteScript);
+            }
 #endif
         }
 

src/PowerShellEditorServices/Session/PowerShell5Operations.cs

@@ -12,7 +12,10 @@ internal class PowerShell5Operations : PowerShell4Operations
         public override void PauseDebugger(Runspace runspace)
         {
 #if !PowerShellv3 && !PowerShellv4
-            runspace.Debugger.SetDebuggerStepMode(true);
+            if (runspace.Debugger != null)
+            {
+                runspace.Debugger.SetDebuggerStepMode(true);
+            }
 #endif
         }
     }