Add Request.sendAsync() function

jpsim · Feb 20, 2023 · e89cebf · e89cebf
1 parent 0849eb7
commit e89cebf
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -6,7 +6,8 @@
 
 ##### Enhancements
 
-* None.
+* Add `Request.sendAsync()` function.  
+  [JP Simard](https://github.com/jpsim)
 
 ##### Bug Fixes
 

diff --git a/Source/SourceKittenFramework/Request.swift b/Source/SourceKittenFramework/Request.swift
@@ -333,6 +333,19 @@ public enum Request {
     /**
     Sends the request to SourceKit and return the response as an [String: SourceKitRepresentable].
 
+    - returns: SourceKit output as a dictionary.
+    - throws: Request.Error on fail ()
+    */
+    public func asyncSend() async throws -> [String: SourceKitRepresentable] {
+        initializeSourceKitFailable
+        let response = try await sourcekitObject.sendAsync()
+        defer { sourcekitd_response_dispose(response) }
+        return fromSourceKit(sourcekitd_response_get_value(response)) as! [String: SourceKitRepresentable]
+    }
+
+    /**
+    Sends the request to SourceKit and return the response as an [String: SourceKitRepresentable].
+
     - returns: SourceKit output as a dictionary.
     - throws: Request.Error on fail ()
     */

diff --git a/Source/SourceKittenFramework/SourceKitObject.swift b/Source/SourceKittenFramework/SourceKitObject.swift
@@ -100,6 +100,31 @@ public final class SourceKitObject {
     func sendSync() -> sourcekitd_response_t? {
         return sourcekitd_send_request_sync(sourcekitdObject)
     }
+
+    func sendAsync() async throws -> sourcekitd_response_t {
+        let handle = UnsafeMutablePointer<sourcekitd_request_handle_t?>.allocate(capacity: 1)
+
+        return try await withTaskCancellationHandler {
+            try await withUnsafeThrowingContinuation { continuation in
+                sourcekitd_send_request(sourcekitdObject, handle) { response in
+                    enum SourceKitSendError: Error { case error, noResponse }
+
+                    guard let response else {
+                        continuation.resume(throwing: SourceKitSendError.noResponse)
+                        return
+                    }
+
+                    if sourcekitd_response_is_error(response) {
+                        continuation.resume(throwing: SourceKitSendError.error)
+                    } else {
+                        continuation.resume(returning: response)
+                    }
+                }
+            }
+        } onCancel: {
+            sourcekitd_cancel_request(handle)
+        }
+    }
 }
 
 extension SourceKitObject: SourceKitObjectConvertible {