Added a -UseHostReadKey switch to instruct LegacyReadLine to use the ReadKey provided by the host instead of Console.ReadKey

Author: dkattan

module/PowerShellEditorServices/Start-EditorServices.ps1

@@ -52,17 +52,20 @@ param(
     [ValidateSet("Diagnostic", "Verbose", "Normal", "Warning", "Error")]
     $LogLevel,
 
-	[Parameter(Mandatory=$true)]
-	[ValidateNotNullOrEmpty()]
-	[string]
-	$SessionDetailsPath,
+    [Parameter(Mandatory=$true)]
+    [ValidateNotNullOrEmpty()]
+    [string]
+    $SessionDetailsPath,
 
     [switch]
     $EnableConsoleRepl,
 
     [switch]
     $UseLegacyReadLine,
 
+    [switch]
+    $UseHostReadKey,
+
     [switch]
     $DebugServiceOnly,
 

src/PowerShellEditorServices.Hosting/Commands/StartEditorServicesCommand.cs

@@ -161,6 +161,12 @@ public StartEditorServicesCommand()
         [Parameter]
         public SwitchParameter UseLegacyReadLine { get; set; }
 
+        /// <summary>
+        /// When set and the console is enabled and legacy readline
+        /// is enabled, console operations will use PSHostreadline implementation will be used instead of PSReadLine.
+        /// </summary>
+        [Parameter]
+        public SwitchParameter UseHostReadKey { get; set; }
         /// <summary>
         /// When set, do not enable LSP service, only the debug adapter.
         /// </summary>
@@ -345,8 +351,9 @@ private EditorServicesConfig CreateConfigObject()
                 hostInfo,
                 Host,
                 SessionDetailsPath,
-                bundledModulesPath,
-                LogPath)
+                bundledModulesPath,                
+                LogPath,
+                UseHostReadKey)
             {
                 FeatureFlags = FeatureFlags,
                 LogLevel = LogLevel,

src/PowerShellEditorServices.Hosting/Configuration/EditorServicesConfig.cs

@@ -17,7 +17,7 @@ public enum ConsoleReplKind
         /// <summary>Use a REPL with the legacy readline implementation. This is generally used when PSReadLine is unavailable.</summary>
         LegacyReadLine = 1,
         /// <summary>Use a REPL with the PSReadLine module for console interaction.</summary>
-        PSReadLine = 2,
+        PSReadLine = 2
     }
 
     /// <summary>
@@ -39,13 +39,15 @@ public EditorServicesConfig(
             PSHost psHost,
             string sessionDetailsPath,
             string bundledModulePath,
-            string logPath)
+            string logPath,
+            bool useHostReadKey)
         {
             HostInfo = hostInfo;
             PSHost = psHost;
             SessionDetailsPath = sessionDetailsPath;
             BundledModulePath = bundledModulePath;
             LogPath = logPath;
+            UseHostReadKey = useHostReadKey;
         }
 
         /// <summary>
@@ -72,6 +74,7 @@ public EditorServicesConfig(
         /// The path to use for logging for Editor Services.
         /// </summary>
         public string LogPath { get; }
+        public bool UseHostReadKey { get; }
 
         /// <summary>
         /// Names of or paths to any additional modules to load on startup.
@@ -88,7 +91,7 @@ public EditorServicesConfig(
         /// (including none to disable the integrated console).
         /// </summary>
         public ConsoleReplKind ConsoleRepl { get; set; } = ConsoleReplKind.None;
-
+        
         /// <summary>
         /// The minimum log level to log events with.
         /// </summary>

src/PowerShellEditorServices.Hosting/EditorServicesLoader.cs

@@ -107,12 +107,12 @@ public static EditorServicesLoader Create(
                 // We only want the Editor Services DLL; the new ALC will lazily load its dependencies automatically
                 if (!string.Equals(asmName.Name, "Microsoft.PowerShell.EditorServices", StringComparison.Ordinal))
                 {
-                    return null;
+                    //return null;
                 }
 
                 string asmPath = Path.Combine(s_psesDependencyDirPath, $"{asmName.Name}.dll");
 
-                logger.Log(PsesLogLevel.Verbose, "Loading PSES DLL using new assembly load context");
+                logger.Log(PsesLogLevel.Verbose, $"Loading {asmName.Name}.dll using new assembly load context");
 
                 return psesLoadContext.LoadFromAssemblyPath(asmPath);
             };

src/PowerShellEditorServices.Hosting/Internal/EditorServicesRunner.cs

@@ -290,6 +290,7 @@ private HostStartupInfo CreateHostStartupInfo()
                 (int)_config.LogLevel,
                 consoleReplEnabled: _config.ConsoleRepl != ConsoleReplKind.None,
                 usesLegacyReadLine: _config.ConsoleRepl == ConsoleReplKind.LegacyReadLine,
+                useHostReadKey: _config.UseHostReadKey,
                 bundledModulePath: _config.BundledModulePath);
         }
 

src/PowerShellEditorServices/Hosting/HostStartupInfo.cs

@@ -80,7 +80,11 @@ public sealed class HostStartupInfo
         /// If the console REPL is not enabled, this setting will be ignored.
         /// </summary>
         public bool UsesLegacyReadLine { get; }
-
+        /// <summary>
+        /// If true, the legacy PSES readline implementation but calls ReadKey in the provided host
+        /// If the console REPL is not enabled, this setting will be ignored.
+        /// </summary>
+        public bool UseHostReadKey { get; }
         /// <summary>
         /// The PowerShell host to use with Editor Services.
         /// </summary>
@@ -153,6 +157,7 @@ public HostStartupInfo(
             int logLevel,
             bool consoleReplEnabled,
             bool usesLegacyReadLine,
+            bool useHostReadKey,
             string bundledModulePath)
         {
             Name = name ?? DefaultHostName;
@@ -167,6 +172,7 @@ public HostStartupInfo(
             LogLevel = logLevel;
             ConsoleReplEnabled = consoleReplEnabled;
             UsesLegacyReadLine = usesLegacyReadLine;
+            UseHostReadKey = useHostReadKey;
             BundledModulePath = bundledModulePath;
         }
 

src/PowerShellEditorServices/Services/PowerShell/Console/LegacyReadLine.cs

@@ -24,15 +24,19 @@ internal class LegacyReadLine : TerminalReadLine
 
         private readonly Action<CancellationToken> _onIdleAction;
 
+        private IConsoleOperations _consoleOperations;
+
         public LegacyReadLine(
             PsesInternalHost psesHost,
             Func<bool, ConsoleKeyInfo> readKeyFunc,
-            Action<CancellationToken> onIdleAction)
+            Action<CancellationToken> onIdleAction,
+            IConsoleOperations consoleOperations)
         {
             _psesHost = psesHost;
             _readKeyTasks = new Task[2];
             _readKeyFunc = readKeyFunc;
             _onIdleAction = onIdleAction;
+            _consoleOperations = consoleOperations;
         }
 
         public override string ReadLine(CancellationToken cancellationToken)
@@ -46,12 +50,12 @@ public override string ReadLine(CancellationToken cancellationToken)
 
             StringBuilder inputLine = new();
 
-            int initialCursorCol = Console.CursorLeft;
-            int initialCursorRow = Console.CursorTop;
+            int initialCursorCol = _consoleOperations?.GetCursorLeft(cancellationToken) ?? _psesHost.UI.RawUI.CursorPosition.X;
+            int initialCursorRow = _consoleOperations?.GetCursorTop(cancellationToken) ?? _psesHost.UI.RawUI.CursorPosition.Y; ;
 
             int currentCursorIndex = 0;
-
-            Console.TreatControlCAsInput = true;
+            if(_consoleOperations != null)
+                Console.TreatControlCAsInput = true;
 
             try
             {
@@ -63,7 +67,8 @@ public override string ReadLine(CancellationToken cancellationToken)
                     // because the window could have been resized before then
                     int promptStartCol = initialCursorCol;
                     int promptStartRow = initialCursorRow;
-                    int consoleWidth = Console.WindowWidth;
+
+                    int consoleWidth = _consoleOperations is not null ? Console.WindowWidth : _psesHost.UI.RawUI.WindowSize.Width;
 
                     switch (keyInfo.Key)
                     {

src/PowerShellEditorServices/Services/PowerShell/Console/PsrlReadLine.cs

@@ -17,19 +17,22 @@ internal class PsrlReadLine : TerminalReadLine
         private readonly PsesInternalHost _psesHost;
 
         private readonly EngineIntrinsics _engineIntrinsics;
+        private IConsoleOperations _consoleOperations;
 
         public PsrlReadLine(
             PSReadLineProxy psrlProxy,
             PsesInternalHost psesHost,
             EngineIntrinsics engineIntrinsics,
             Func<bool, ConsoleKeyInfo> readKeyFunc,
-            Action<CancellationToken> onIdleAction)
+            Action<CancellationToken> onIdleAction,
+            IConsoleOperations consoleOperations)
         {
             _psrlProxy = psrlProxy;
             _psesHost = psesHost;
             _engineIntrinsics = engineIntrinsics;
             _psrlProxy.OverrideReadKey(readKeyFunc);
             _psrlProxy.OverrideIdleHandler(onIdleAction);
+            _consoleOperations = consoleOperations;
         }
 
         public override string ReadLine(CancellationToken cancellationToken) => _psesHost.InvokeDelegate(
@@ -40,6 +43,15 @@ public override string ReadLine(CancellationToken cancellationToken) => _psesHos
 
         protected override ConsoleKeyInfo ReadKey(CancellationToken cancellationToken) => _psesHost.ReadKey(intercept: true, cancellationToken);
 
+        protected override ConsoleKeyInfo ReadKey(CancellationToken cancellationToken)
+        {
+            return _consoleOperations.ReadKey(intercept: true, cancellationToken);
+        }
+
+        #endregion
+
+        #region Private Methods
+
         private string InvokePSReadLine(CancellationToken cancellationToken)
         {
             EngineIntrinsics engineIntrinsics = _psesHost.IsRunspacePushed ? null : _engineIntrinsics;

src/PowerShellEditorServices/Services/PowerShell/Console/TerminalReadLine.cs

@@ -9,6 +9,8 @@ namespace Microsoft.PowerShell.EditorServices.Services.PowerShell.Console
 
     internal abstract class TerminalReadLine : IReadLine
     {
+        private IConsoleOperations _consoleOperations;
+
         public abstract string ReadLine(CancellationToken cancellationToken);
 
         protected abstract ConsoleKeyInfo ReadKey(CancellationToken cancellationToken);

src/PowerShellEditorServices/Services/PowerShell/Host/EditorServicesConsolePSHostRawUserInterface.cs

@@ -24,10 +24,12 @@ internal class EditorServicesConsolePSHostRawUserInterface : PSHostRawUserInterf
         /// </summary>
         public EditorServicesConsolePSHostRawUserInterface(
             ILoggerFactory loggerFactory,
-            PSHostRawUserInterface internalRawUI)
+            PSHostRawUserInterface internalRawUI,
+            IConsoleOperations consoleOperations)
         {
             _logger = loggerFactory.CreateLogger<EditorServicesConsolePSHostRawUserInterface>();
             _internalRawUI = internalRawUI;
+            _consoleOperations = consoleOperations;
         }
 
         #endregion
@@ -39,17 +41,57 @@ public EditorServicesConsolePSHostRawUserInterface(
         /// </summary>
         public override ConsoleColor BackgroundColor
         {
-            get => System.Console.BackgroundColor;
-            set => System.Console.BackgroundColor = value;
+            get
+            {
+                if (_consoleOperations is not null)
+                {
+                    return System.Console.BackgroundColor;
+                }
+                else
+                {
+                    return _internalRawUI.BackgroundColor;
+                }
+            }
+            set
+            {
+                if (_consoleOperations is not null)
+                {
+                    System.Console.BackgroundColor = value;
+                }
+                else
+                {
+                    _internalRawUI.BackgroundColor = value;
+                }
+            }
         }
 
         /// <summary>
         /// Gets or sets the foreground color of the console.
         /// </summary>
         public override ConsoleColor ForegroundColor
         {
-            get => System.Console.ForegroundColor;
-            set => System.Console.ForegroundColor = value;
+            get
+            {
+                if (_consoleOperations is not null)
+                {
+                    return System.Console.ForegroundColor;
+                }
+                else
+                {
+                    return _internalRawUI.ForegroundColor;
+                }
+            }
+            set
+            {
+                if (_consoleOperations is not null)
+                {
+                    System.Console.ForegroundColor = value;
+                }
+                else
+                {
+                    _internalRawUI.ForegroundColor = value;
+                }
+            }
         }
 
         /// <summary>

src/PowerShellEditorServices/Services/PowerShell/Host/EditorServicesConsolePSHostUserInterface.cs

@@ -27,7 +27,14 @@ public EditorServicesConsolePSHostUserInterface(
             PSHostUserInterface underlyingHostUI)
         {
             _underlyingHostUI = underlyingHostUI;
-            RawUI = new EditorServicesConsolePSHostRawUserInterface(loggerFactory, underlyingHostUI.RawUI);
+            RawUI = new EditorServicesConsolePSHostRawUserInterface(loggerFactory, underlyingHostUI.RawUI, consoleOperations);
+
+            _consoleHostUI = GetConsoleHostUI(_underlyingHostUI);
+
+            if (_consoleHostUI != null)
+            {
+                SetConsoleHostUIToInteractive(_consoleHostUI);
+            }
         }
 
         public override bool SupportsVirtualTerminal => _underlyingHostUI.SupportsVirtualTerminal;