From 60d2876fbcdfa2ff11539f5d21c90264fbf2e3cd Mon Sep 17 00:00:00 2001
From: Alex Hoppen <ahoppen@apple.com>
Date: Wed, 12 Jun 2024 14:55:10 -0700
Subject: [PATCH] Add workaround for rdar://116221716

---
 Sources/SourceKitLSP/WorkDoneProgressManager.swift | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/Sources/SourceKitLSP/WorkDoneProgressManager.swift b/Sources/SourceKitLSP/WorkDoneProgressManager.swift
index 86645528..72aa5d63 100644
--- a/Sources/SourceKitLSP/WorkDoneProgressManager.swift
+++ b/Sources/SourceKitLSP/WorkDoneProgressManager.swift
@@ -51,6 +51,11 @@ final actor WorkDoneProgressManager {
   /// The last status that was sent to the client. Used so we don't send no-op updates to the client.
   private var lastStatus: Status? = nil
 
+  /// Needed to work around rdar://116221716
+  private static func getServerCapabilityRegistry(_ server: SourceKitLSPServer) async -> CapabilityRegistry? {
+    return await server.capabilityRegistry
+  }
+
   init?(
     server: SourceKitLSPServer,
     initialDebounce: Duration? = nil,
@@ -58,7 +63,7 @@ final actor WorkDoneProgressManager {
     message: String? = nil,
     percentage: Int? = nil
   ) async {
-    guard let capabilityRegistry = await server.capabilityRegistry else {
+    guard let capabilityRegistry = await Self.getServerCapabilityRegistry(server) else {
       return nil
     }
     self.init(