Use pseudo command handler in suggestion panel

Author: intitni

Core/Sources/Service/GUI/GraphicalUserInterfaceController.swift.swift

@@ -1,10 +1,40 @@
 import AppKit
 import SuggestionWidget
+import Environment
 
 @MainActor
 public final class GraphicalUserInterfaceController {
     public nonisolated static let shared = GraphicalUserInterfaceController()
     nonisolated let realtimeSuggestionIndicatorController = RealtimeSuggestionIndicatorController()
     nonisolated let suggestionWidget = SuggestionWidgetController()
-    private nonisolated init() {}
+    private nonisolated init() {
+        Task { @MainActor in
+            suggestionWidget.onNextButtonTapped = {
+                Task { @ServiceActor in
+                    let handler = PseudoCommandHandler()
+                    await handler.presentNextSuggestion()
+                }
+            }
+            
+            suggestionWidget.onPreviousButtonTapped = {
+                Task { @ServiceActor in
+                    let handler = PseudoCommandHandler()
+                    await handler.presentPreviousSuggestion()
+                }
+            }
+            
+            suggestionWidget.onRejectButtonTapped = {
+                Task { @ServiceActor in
+                    let handler = PseudoCommandHandler()
+                    await handler.rejectSuggestions()
+                }
+            }
+            
+            suggestionWidget.onAcceptButtonTapped = {
+                Task {
+                    try await Environment.triggerAction("Accept Suggestion")
+                }
+            }
+        }
+    }
 }

Core/Sources/SuggestionWidget/SuggestionPanelView.swift

@@ -9,18 +9,31 @@ final class SuggestionPanelViewModel: ObservableObject {
     @Published var suggestionCount: Int
     @Published var currentSuggestionIndex: Int
 
+    var onAcceptButtonTapped: (() -> Void)?
+    var onRejectButtonTapped: (() -> Void)?
+    var onPreviousButtonTapped: (() -> Void)?
+    var onNextButtonTapped: (() -> Void)?
+
     public init(
         startLineIndex: Int = 0,
         suggestion: [String] = [],
         isPanelDisplayed: Bool = false,
         suggestionCount: Int = 0,
-        currentSuggestionIndex: Int = 0
+        currentSuggestionIndex: Int = 0,
+        onAcceptButtonTapped: (() -> Void)? = nil,
+        onRejectButtonTapped: (() -> Void)? = nil,
+        onPreviousButtonTapped: (() -> Void)? = nil,
+        onNextButtonTapped: (() -> Void)? = nil
     ) {
         self.startLineIndex = startLineIndex
         self.suggestion = suggestion
         self.isPanelDisplayed = isPanelDisplayed
         self.suggestionCount = suggestionCount
         self.currentSuggestionIndex = currentSuggestionIndex
+        self.onAcceptButtonTapped = onAcceptButtonTapped
+        self.onRejectButtonTapped = onRejectButtonTapped
+        self.onPreviousButtonTapped = onPreviousButtonTapped
+        self.onNextButtonTapped = onNextButtonTapped
     }
 }
 
@@ -107,35 +120,47 @@ struct CodeBlock: View {
         }))
     }
 }
-        
+
 struct ToolBar: View {
     @ObservedObject var viewModel: SuggestionPanelViewModel
 
     var body: some View {
         HStack {
             Button(action: {
                 Task {
+                    if let block = viewModel.onPreviousButtonTapped {
+                        block()
+                        return
+                    }
                     try await Environment.triggerAction("Previous Suggestion")
                 }
             }) {
                 Image(systemName: "chevron.left")
             }.buttonStyle(.plain)
-            
+
             Text("\(viewModel.currentSuggestionIndex + 1) / \(viewModel.suggestionCount)")
                 .monospacedDigit()
-            
+
             Button(action: {
                 Task {
+                    if let block = viewModel.onNextButtonTapped {
+                        block()
+                        return
+                    }
                     try await Environment.triggerAction("Next Suggestion")
                 }
             }) {
-                Image(systemName: "chevron.right")     
+                Image(systemName: "chevron.right")
             }.buttonStyle(.plain)
 
             Spacer()
 
             Button(action: {
                 Task {
+                    if let block = viewModel.onRejectButtonTapped {
+                        block()
+                        return
+                    }
                     try await Environment.triggerAction("Reject Suggestion")
                 }
             }) {
@@ -144,6 +169,10 @@ struct ToolBar: View {
 
             Button(action: {
                 Task {
+                    if let block = viewModel.onAcceptButtonTapped {
+                        block()
+                        return
+                    }
                     try await Environment.triggerAction("Accept Suggestion")
                 }
             }) {

Core/Sources/SuggestionWidget/SuggestionWidgetController.swift

@@ -70,6 +70,26 @@ public final class SuggestionWidgetController {
     private var xcode: NSRunningApplication?
     private var suggestionForFiles: [URL: Suggestion] = [:]
 
+    public var onAcceptButtonTapped: (() -> Void)? {
+        get { suggestionPanelViewModel.onAcceptButtonTapped }
+        set { suggestionPanelViewModel.onAcceptButtonTapped = newValue }
+    }
+
+    public var onRejectButtonTapped: (() -> Void)? {
+        get { suggestionPanelViewModel.onRejectButtonTapped }
+        set { suggestionPanelViewModel.onRejectButtonTapped = newValue }
+    }
+
+    public var onPreviousButtonTapped: (() -> Void)? {
+        get { suggestionPanelViewModel.onPreviousButtonTapped }
+        set { suggestionPanelViewModel.onPreviousButtonTapped = newValue }
+    }
+
+    public var onNextButtonTapped: (() -> Void)? {
+        get { suggestionPanelViewModel.onNextButtonTapped }
+        set { suggestionPanelViewModel.onNextButtonTapped = newValue }
+    }
+
     enum Suggestion {
         case code([String], startLineIndex: Int, currentSuggestionIndex: Int, suggestionCount: Int)
     }