Reset progress messages at end of REPL

This seems to work. We needed to hold onto processed progress records,
and then at the end of the REPL mark each as completed and re-write it
(which uses the underlying host and effectively clears it).

Author: andyleejordan

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

@@ -22,6 +22,8 @@ internal class EditorServicesConsolePSHostUserInterface : PSHostUserInterface
 
         private readonly PSHostUserInterface _consoleHostUI;
 
+        private readonly HashSet<(long, ProgressRecord)> _currentProgressRecords = new();
+
         public EditorServicesConsolePSHostUserInterface(
             ILoggerFactory loggerFactory,
             IReadLineProvider readLineProvider,
@@ -103,7 +105,30 @@ public override PSCredential PromptForCredential(string caption, string message,
 
         public override void WriteLine(string value) => _underlyingHostUI.WriteLine(value);
 
-        public override void WriteProgress(long sourceId, ProgressRecord record) => _underlyingHostUI.WriteProgress(sourceId, record);
+        public override void WriteProgress(long sourceId, ProgressRecord record)
+        {
+            if (record.RecordType == ProgressRecordType.Completed)
+            {
+                _currentProgressRecords.Remove((sourceId, record));
+            }
+            else
+            {
+                _currentProgressRecords.Add((sourceId, record));
+            }
+            _underlyingHostUI.WriteProgress(sourceId, record);
+        }
+
+        public void ResetProgress()
+        {
+            // Mark all processed progress records as completed.
+            foreach ((long id, ProgressRecord record) in _currentProgressRecords)
+            {
+                record.RecordType = ProgressRecordType.Completed;
+                _underlyingHostUI.WriteProgress(id, record);
+                _currentProgressRecords.Clear();
+            }
+            // TODO: Maybe send the OSC sequence to turn off progress indicator.
+        }
 
         public override void WriteVerboseLine(string message) => _underlyingHostUI.WriteVerboseLine(message);
 

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

@@ -662,6 +662,14 @@ private void DoOneRepl(CancellationToken cancellationToken)
                 UI.WriteErrorLine($"An error occurred while running the REPL loop:{Environment.NewLine}{e}");
                 _logger.LogError(e, "An error occurred while running the REPL loop");
             }
+            finally
+            {
+                // After the REPL we need to complete all progress records.
+                if (UI is EditorServicesConsolePSHostUserInterface ui)
+                {
+                    ui.ResetProgress();
+                }
+            }
         }
 
         private string GetPrompt(CancellationToken cancellationToken)