Supports esc to dismiss suggestion

Author: intitni

Core/Sources/Service/Service.swift

@@ -47,13 +47,14 @@ public final class Service {
         globalShortcutManager = .init(guiController: guiController)
 
         #if canImport(ProService)
-        proService = withDependencies { dependencyValues in
-            dependencyValues.proServiceAcceptSuggestion = {
+        proService = ProService(
+            acceptSuggestion: {
                 Task { await PseudoCommandHandler().acceptSuggestion() }
+            },
+            dismissSuggestion: {
+                Task { await PseudoCommandHandler().dismissSuggestion() }
             }
-        } operation: {
-            ProService()
-        }
+        )
         #endif
 
         scheduledCleaner.service = self
@@ -92,7 +93,7 @@ public extension Service {
             reply(nil, error)
             return
         }
-        
+
         reply(nil, XPCRequestNotHandledError())
     }
 }

Core/Sources/Service/SuggestionCommandHandler/PseudoCommandHandler.swift

@@ -241,6 +241,15 @@ struct PseudoCommandHandler {
             }
         }
     }
+    
+    func dismissSuggestion() async {
+        guard let documentURL = XcodeInspector.shared.activeDocumentURL else { return }
+        guard let (_, filespace) = try? await Service.shared.workspacePool
+            .fetchOrCreateWorkspaceAndFilespace(fileURL: documentURL) else { return }
+        
+        await filespace.reset()
+        PresentInWindowSuggestionPresenter().discardSuggestion(fileURL: documentURL)
+    }
 }
 
 extension PseudoCommandHandler {

Pro

@@ -334,6 +334,10 @@ public extension UserDefaultPreferenceKeys {
         .init(defaultValue: true, key: "AcceptSuggestionWithTab")
     }
 
+    var dismissSuggestionWithEsc: PreferenceKey<Bool> {
+        .init(defaultValue: true, key: "DismissSuggestionWithEsc")
+    }
+    
     var isSuggestionSenseEnabled: PreferenceKey<Bool> {
         .init(defaultValue: false, key: "IsSuggestionSenseEnabled")
     }