Add private contract delegate for PSES to handle idle on readline

PowerShell · Jun 25, 2020 · 882dfa8 · 882dfa8
1 parent fd99806
commit 882dfa8
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/PSReadLine.build.ps1 b/PSReadLine.build.ps1
@@ -130,6 +130,10 @@ task LayoutModule BuildMainModule, {
     $binPath = "PSReadLine/bin/$Configuration/$Framework/publish"
     Copy-Item $binPath/Microsoft.PowerShell.PSReadLine2.dll $targetDir
 
+    if ($Configuration -eq 'Debug') {
+      Copy-Item $binPath/*.pdb $targetDir
+    }
+
     if (Test-Path $binPath/System.Runtime.InteropServices.RuntimeInformation.dll)
     {
         Copy-Item $binPath/System.Runtime.InteropServices.RuntimeInformation.dll $targetDir

diff --git a/PSReadLine/ReadLine.cs b/PSReadLine/ReadLine.cs
@@ -44,6 +44,8 @@ public partial class PSConsoleReadLine : IPSConsoleReadLineMockableMethods
 
         private static readonly CancellationToken _defaultCancellationToken = new CancellationTokenSource().Token;
 
+        private static Action _handleIdleOverride;
+
         private bool _delayedOneTimeInitCompleted;
 
         private IPSConsoleReadLineMockableMethods _mockableMethods;
@@ -201,6 +203,12 @@ internal static PSKeyInfo ReadKey()
                     if (handleId != WaitHandle.WaitTimeout && handleId != EventProcessingRequested)
                         break;
 
+                    if (_handleIdleOverride != null)
+                    {
+                        _handleIdleOverride();
+                        continue;
+                    }
+
                     // If we timed out, check for event subscribers (which is just
                     // a hint that there might be an event waiting to be processed.)
                     var eventSubscribers = _singleton._engineIntrinsics?.Events.Subscribers;