Close stray processes on exit

src/PowerShellEditorServices.Host/EditorServicesHost.cs

@@ -70,7 +70,7 @@ public class EditorServicesHost
         private IServerListener languageServiceListener;
         private IServerListener debugServiceListener;
 
-        private TaskCompletionSource<bool> serverCompletedTask;
+        private TaskCompletionSource<bool> serverCompletedTask = new TaskCompletionSource<bool>();
 
         #endregion
 
@@ -226,6 +226,7 @@ private async void OnLanguageServiceClientConnect(
                     this.editorSession,
                     messageDispatcher,
                     protocolEndpoint,
+                    this.serverCompletedTask,
                     this.logger);
 
             await this.editorSession.PowerShellContext.ImportCommandsModule(
@@ -348,7 +349,6 @@ public void StopServices()
         public void WaitForCompletion()
         {
             // TODO: We need a way to know when to complete this task!
-            this.serverCompletedTask = new TaskCompletionSource<bool>();
             this.serverCompletedTask.Task.Wait();
         }
 

src/PowerShellEditorServices.Protocol/Server/LanguageServer.cs

@@ -40,22 +40,31 @@ public class LanguageServer
         private Dictionary<string, Dictionary<string, MarkerCorrection>> codeActionsPerFile =
             new Dictionary<string, Dictionary<string, MarkerCorrection>>();
 
+        private TaskCompletionSource<bool> serverCompletedTask;
+
         public IEditorOperations EditorOperations
         {
             get { return this.editorOperations; }
         }
 
-        /// <param name="hostDetails">
-        /// Provides details about the host application.
-        /// </param>
+        /// <summary>
+        /// Initializes a new language server that is used for handing language server protocol messages 
+        /// </summary>
+        /// <param name="editorSession">The editor session that handles the PowerShell runspace</param>
+        /// <param name="messageHandlers">An object that manages all of the message handlers</param>
+        /// <param name="messageSender">The message sender</param>
+        /// <param name="serverCompletedTask">A TaskCompletionSource<bool> that will be completed to stop the running process</param>
+        /// <param name="logger">The logger</param>
         public LanguageServer(
             EditorSession editorSession,
             IMessageHandlers messageHandlers,
             IMessageSender messageSender,
+            TaskCompletionSource<bool> serverCompletedTask,
             ILogger logger)
         {
             this.Logger = logger;
             this.editorSession = editorSession;
+            this.serverCompletedTask = serverCompletedTask;
             // Attach to the underlying PowerShell context to listen for changes in the runspace or execution status
             this.editorSession.PowerShellContext.RunspaceChanged += PowerShellContext_RunspaceChanged;
             this.editorSession.PowerShellContext.ExecutionStatusChanged += PowerShellContext_ExecutionStatusChanged;
@@ -146,6 +155,7 @@ protected Task Stop()
             Logger.Write(LogLevel.Normal, "Language service is shutting down...");
 
             // TODO: Raise an event so that the host knows to shut down
+            this.serverCompletedTask.SetResult(true);
 
             return Task.FromResult(true);
         }
@@ -156,7 +166,6 @@ private async Task HandleShutdownRequest(
             RequestContext<object> requestContext)
         {
             // Allow the implementor to shut down gracefully
-            await this.Stop();
 
             await requestContext.SendResult(new object());
         }