Add SuggestionServiceEventHandler

Author: intitni

Core/Sources/SuggestionService/SuggestionService.swift

@@ -1,7 +1,7 @@
 import BuiltinExtension
 import CodeiumService
-import struct CopilotForXcodeKit.WorkspaceInfo
 import enum CopilotForXcodeKit.SuggestionServiceError
+import struct CopilotForXcodeKit.WorkspaceInfo
 import Foundation
 import GitHubCopilotService
 import Preferences
@@ -17,18 +17,21 @@ import ProExtension
 public protocol SuggestionServiceType: SuggestionServiceProvider {}
 
 public actor SuggestionService: SuggestionServiceType {
+    typealias Middleware = SuggestionServiceMiddleware
+    typealias EventHandler = SuggestionServiceEventHandler
     public var configuration: SuggestionProvider.SuggestionServiceConfiguration {
         get async { await suggestionProvider.configuration }
     }
 
-    let middlewares: [SuggestionServiceMiddleware]
+    let middlewares: [Middleware]
+    let eventHandlers: [EventHandler]
 
     let suggestionProvider: SuggestionServiceProvider
 
     public init(
         provider: any SuggestionServiceProvider,
-        middlewares: [SuggestionServiceMiddleware] = SuggestionServiceMiddlewareContainer
-            .middlewares
+        middlewares: [Middleware] = SuggestionServiceMiddlewareContainer.middlewares,
+        eventHandlers: [EventHandler] = SuggestionServiceEventHandlerContainer.handlers
     ) {
         suggestionProvider = provider
         self.middlewares = middlewares
@@ -67,7 +70,7 @@ public extension SuggestionService {
         do {
             var getSuggestion = suggestionProvider.getSuggestions(_:workspaceInfo:)
             let configuration = await configuration
-            
+
             for middleware in middlewares.reversed() {
                 getSuggestion = { [getSuggestion] request, workspaceInfo in
                     try await middleware.getSuggestion(
@@ -79,7 +82,7 @@ public extension SuggestionService {
                     )
                 }
             }
-            
+
             return try await getSuggestion(request, workspaceInfo)
         } catch let error as SuggestionServiceError {
             throw error
@@ -92,13 +95,15 @@ public extension SuggestionService {
         _ suggestion: SuggestionBasic.CodeSuggestion,
         workspaceInfo: CopilotForXcodeKit.WorkspaceInfo
     ) async {
+        eventHandlers.forEach { $0.didAccept(suggestion, workspaceInfo: workspaceInfo) }
         await suggestionProvider.notifyAccepted(suggestion, workspaceInfo: workspaceInfo)
     }
 
     func notifyRejected(
         _ suggestions: [SuggestionBasic.CodeSuggestion],
         workspaceInfo: CopilotForXcodeKit.WorkspaceInfo
     ) async {
+        eventHandlers.forEach { $0.didReject(suggestion, workspaceInfo: workspaceInfo) }
         await suggestionProvider.notifyRejected(suggestions, workspaceInfo: workspaceInfo)
     }
 

Tool/Sources/SuggestionProvider/SuggestionServiceEventHandler.swift

@@ -0,0 +1,22 @@
+import Foundation
+import SuggestionBasic
+import CopilotForXcodeKit
+
+public protocol SuggestionServiceEventHandler {
+    func didAccept(_ suggestion: CodeSuggestion, workspaceInfo: WorkspaceInfo)
+    func didReject(_ suggestion: CodeSuggestion, workspaceInfo: WorkspaceInfo)
+}
+
+public enum SuggestionServiceEventHandlerContainer {
+    static var builtinHandlers: [SuggestionServiceEventHandler] = []
+
+    static var customHandlers: [SuggestionServiceEventHandler] = []
+
+    public static var handlers: [SuggestionServiceEventHandler] {
+        builtinHandlers + customHandlers
+    }
+
+    public static func addHandler(_ handler: SuggestionServiceEventHandler) {
+        customHandlers.append(handler)
+    }
+}