Explicitly shut down the build server when SourceKit-LSP is shut down

The build server is automatically shut down using a background task when `BuildSystemManager` is deallocated.

This, however, leads to possible race conditions where the shutdown task might not finish before the test is done, which could result in the connection being reported as a leak. To avoid this problem, we want to explicitly shut, down the build server when the `SourceKitLSPServer` gets shut down.

Author: ahoppen

Sources/BuildSystemIntegration/BuildSystemManager.swift

@@ -410,6 +410,23 @@ package actor BuildSystemManager: QueueBasedMessageHandler {
     }
   }
 
+  /// Explicitly shut down the build server.
+  ///
+  /// The build server is automatically shut down using a background task when `BuildSystemManager` is deallocated.
+  /// This, however, leads to possible race conditions where the shutdown task might not finish before the test is done,
+  /// which could result in the connection being reported as a leak. To avoid this problem, we want to explicitly shut
+  /// down the build server when the `SourceKitLSPServer` gets shut down.
+  package func shutdown() async {
+    guard let buildSystemAdapter = await self.buildSystemAdapterAfterInitialized else {
+      return
+    }
+    await orLog("Sending shutdown request to build server") {
+      _ = try await buildSystemAdapter.send(BuildShutdownRequest())
+      await buildSystemAdapter.send(OnBuildExitNotification())
+    }
+    self.buildSystemAdapter = nil
+  }
+
   deinit {
     // Shut down the build server before closing the connection to it
     Task { [buildSystemAdapter, initializeResult] in

Sources/BuildSystemIntegration/ExternalBuildSystemAdapter.swift

@@ -118,7 +118,7 @@ actor ExternalBuildSystemAdapter {
   }
 
   /// Send a notification to the build server.
-  func send(_ notification: some NotificationType) async {
+  func send(_ notification: some NotificationType) {
     guard let connectionToBuildServer else {
       logger.error("Dropping notification because BSP server has crashed: \(notification.forLogging)")
       return

Sources/SourceKitLSP/SourceKitLSPServer.swift

@@ -1216,6 +1216,11 @@ extension SourceKitLSPServer {
           await service.shutdown()
         }
       }
+      for workspace in workspaces {
+        taskGroup.addTask {
+          await workspace.buildSystemManager.shutdown()
+        }
+      }
     }
 
     // We have a semantic guarantee that no request or notification should be