Implement CancelRequest for Codeium

Author: intitni

Core/Sources/CodeiumService/CodeiumRequest.swift

@@ -48,12 +48,25 @@ enum CodeiumRequest {
         }
     }
 
-    struct AcceptCompletion: CodeiumRequestType {
-        struct Response: Codable {
-            var state: State
-            var completionItems: [CodeiumCompletionItem]?
+    struct CancelRequest: CodeiumRequestType {
+        struct Response: Codable {}
+
+        struct RequestBody: Codable {
+            var request_id: UInt64
+            var session_id: String
         }
 
+        var requestBody: RequestBody
+
+        func makeURLRequest(server: String) -> URLRequest {
+            let data = (try? JSONEncoder().encode(requestBody)) ?? Data()
+            return assembleURLRequest(server: server, method: "CancelRequest", body: data)
+        }
+    }
+    
+    struct AcceptCompletion: CodeiumRequestType {
+        struct Response: Codable {}
+
         struct RequestBody: Codable {
             var metadata: Metadata
             var completion_id: String

Core/Sources/CodeiumService/CodeiumService.swift

@@ -257,13 +257,13 @@ extension CodeiumSuggestionService: CodeiumSuggestionServiceType {
                     )
                 }
         ))
-        
+
         if request.requestBody.metadata.request_id <= cancellationCounter {
             throw CancellationError()
         }
 
         let result = try await (try await setupServerIfNeeded()).sendRequest(request)
-        
+
         if request.requestBody.metadata.request_id <= cancellationCounter {
             throw CancellationError()
         }
@@ -292,8 +292,16 @@ extension CodeiumSuggestionService: CodeiumSuggestionServiceType {
             )
         } ?? []
     }
-    
+
     public func cancelRequest() async {
+        Task {
+            try await server?.sendRequest(
+                CodeiumRequest.CancelRequest(requestBody: .init(
+                    request_id: requestCounter,
+                    session_id: CodeiumSuggestionService.sessionId
+                ))
+            )
+        }
         cancellationCounter = requestCounter
     }
 
@@ -326,7 +334,7 @@ extension CodeiumSuggestionService: CodeiumSuggestionServiceType {
     public func notifyCloseTextDocument(fileURL: URL) async throws {
         await openedDocumentPool.closeDocument(url: fileURL)
     }
-    
+
     public func terminate() {
         server?.terminate()
         server = nil