Handle dismiss suggestions event

Author: intitni

Core/Sources/SuggestionService/SuggestionService.swift

@@ -118,6 +118,13 @@ public extension SuggestionService {
         eventHandlers.forEach { $0.didReject(suggestions, workspaceInfo: workspaceInfo) }
         await suggestionProvider.notifyRejected(suggestions, workspaceInfo: workspaceInfo)
     }
+    
+    func notifyDismissed(
+        _ suggestions: [SuggestionBasic.CodeSuggestion],
+        workspaceInfo: CopilotForXcodeKit.WorkspaceInfo
+    ) async {
+        eventHandlers.forEach { $0.didDismiss(suggestions, workspaceInfo: workspaceInfo) }
+    }
 
     func cancelRequest(workspaceInfo: CopilotForXcodeKit.WorkspaceInfo) async {
         await suggestionProvider.cancelRequest(workspaceInfo: workspaceInfo)

Tool/Sources/SuggestionProvider/SuggestionServiceEventHandler.swift

@@ -5,6 +5,7 @@ import SuggestionBasic
 public protocol SuggestionServiceEventHandler {
     func didAccept(_ suggestion: SuggestionBasic.CodeSuggestion, workspaceInfo: WorkspaceInfo)
     func didReject(_ suggestions: [SuggestionBasic.CodeSuggestion], workspaceInfo: WorkspaceInfo)
+    func didDismiss(_ suggestions: [SuggestionBasic.CodeSuggestion], workspaceInfo: WorkspaceInfo)
 }
 
 public enum SuggestionServiceEventHandlerContainer {

Tool/Sources/WorkspaceSuggestionService/Workspace+SuggestionService.swift

@@ -19,6 +19,10 @@ public protocol SuggestionServiceType {
         _ suggestions: [CodeSuggestion],
         workspaceInfo: CopilotForXcodeKit.WorkspaceInfo
     ) async
+    func notifyDismissed(
+        _ suggestions: [CodeSuggestion],
+        workspaceInfo: CopilotForXcodeKit.WorkspaceInfo
+    ) async
     func cancelRequest(workspaceInfo: CopilotForXcodeKit.WorkspaceInfo) async
 
     var configuration: SuggestionServiceConfiguration { get async }
@@ -230,7 +234,26 @@ public extension Workspace {
     func dismissSuggestions(forFileAt fileURL: URL) {
         refreshUpdateTime()
         guard let filespace = filespaces[fileURL] else { return }
+        let displayedSuggestions = filespace.suggestionManager?.displaySuggestions.flatMap {
+            switch $0 {
+            case let .action(action):
+                return [action.suggestion]
+            case let .group(group):
+                return group.suggestions
+            }
+        } ?? []
+
         filespace.suggestionManager?.invalidateDisplaySuggestions()
+
+        Task {
+            await suggestionService?.notifyDismissed(
+                displayedSuggestions,
+                workspaceInfo: .init(
+                    workspaceURL: workspaceURL,
+                    projectURL: projectRootURL
+                )
+            )
+        }
     }
 }