Move pendingInput code to OnitPanelState

Author: timlenardo

macos/Onit/StateManagers/HighlightedText/HighlightedTextDelegate.swift

@@ -0,0 +1,17 @@
+//
+//  HighlightedTextDelegate.swift
+//  Onit
+//
+//  Created by TimL on 07/29/2025.
+//
+
+import Foundation
+
+@MainActor
+protocol HighlightedTextDelegate: AnyObject {
+    /// Called when highlighted text has changed
+    /// - Parameters:
+    ///   - selectedText: The newly selected text, or nil if text was deselected
+    ///   - application: The name of the application where the text was selected
+    func highlightedTextDidChange(selectedText: String?, application: String?)
+}

macos/Onit/StateManagers/HighlightedText/HighlightedTextManager.swift

@@ -26,6 +26,10 @@ class HighlightedTextManager: ObservableObject {
     // Published property for selected text that QuickEditManager can observe
     @Published var selectedText: String?
 
+    // MARK: - Delegates
+    
+    private var delegates = NSHashTable<AnyObject>.weakObjects()
+    
     // MARK: - Private initializer
 
     private init() {
@@ -39,6 +43,22 @@ class HighlightedTextManager: ObservableObject {
 
     // MARK: - Functions
 
+    // MARK: - Delegate Management
+    
+    func addDelegate(_ delegate: HighlightedTextDelegate) {
+        delegates.add(delegate)
+    }
+    
+    func removeDelegate(_ delegate: HighlightedTextDelegate) {
+        delegates.remove(delegate)
+    }
+    
+    private func notifyDelegates(selectedText: String?, application: String?) {
+        for case let delegate as HighlightedTextDelegate in delegates.allObjects {
+            delegate.highlightedTextDidChange(selectedText: selectedText, application: application)
+        }
+    }
+    
     func setCurrentSource(_ source: String?) {
         currentSource = source
     }
@@ -97,22 +117,19 @@ class HighlightedTextManager: ObservableObject {
               let selectedText = text,
               HighlightedTextValidator.isValid(text: selectedText) else {
 
-            PanelStateCoordinator.shared.state.pendingInput = nil
-            PanelStateCoordinator.shared.state.trackedPendingInput = nil
+            // Update the published selectedText property
             self.selectedText = nil
+            
+            // Notify delegates that text was deselected
+            notifyDelegates(selectedText: nil, application: currentSource)
             return
         }
 
         // Update the published selectedText property
         self.selectedText = selectedText
 
-        let input = Input(selectedText: selectedText, application: currentSource ?? "")
-        
-        if Defaults[.autoAddHighlightedTextToContext] {
-            PanelStateCoordinator.shared.state.pendingInput = input
-        } else {
-            PanelStateCoordinator.shared.state.trackedPendingInput = input
-        }
+        // Notify delegates about the text change
+        notifyDelegates(selectedText: selectedText, application: currentSource)
     }
 
     func handleCaretPositionChange(for element: AXUIElement) {

…te/OnitPanelState+NSWindowDelegate.swift …els/State/OnitPanelState+Delegates.swiftmacos/Onit/UI/Panels/State/OnitPanelState+NSWindowDelegate.swift renamed to macos/Onit/UI/Panels/State/OnitPanelState+Delegates.swift macos/Onit/UI/Panels/State/OnitPanelState+NSWindowDelegate.swift renamed to macos/Onit/UI/Panels/State/OnitPanelState+Delegates.swift

@@ -6,6 +6,7 @@
 //
 
 import AppKit
+import Defaults
 
 extension OnitPanelState: NSWindowDelegate {
 
@@ -29,3 +30,21 @@ extension OnitPanelState: NSWindowDelegate {
         }
     }
 }
+
+extension OnitPanelState: HighlightedTextDelegate {
+    func highlightedTextDidChange(selectedText: String?, application: String?) {
+        if let selectedText = selectedText {
+            let input = Input(selectedText: selectedText, application: application ?? "")
+
+            if Defaults[.autoAddHighlightedTextToContext] {
+                pendingInput = input
+            } else {
+                trackedPendingInput = input
+            }
+        } else {
+            // Text was deselected
+            pendingInput = nil
+            trackedPendingInput = nil
+        }
+    }
+}

macos/Onit/UI/Panels/State/OnitPanelState.swift

@@ -173,6 +173,7 @@ class OnitPanelState: NSObject {
     override init() {
         self.panelWidth = Defaults[.panelWidth]
         super.init()
+        setupHighlightedTextDelegate()
     }
 
     init(trackedWindow: TrackedWindow) {
@@ -181,6 +182,7 @@ class OnitPanelState: NSObject {
         super.init()
 
         self.promptSuggestionService = SystemPromptSuggestionService(state: self)
+        setupHighlightedTextDelegate()
     }
 
     init(screen: NSScreen) {
@@ -189,6 +191,7 @@ class OnitPanelState: NSObject {
         super.init()
 
         self.promptSuggestionService = SystemPromptSuggestionService(state: self)
+        setupHighlightedTextDelegate()
     }
 
     // MARK: - Functions
@@ -198,6 +201,12 @@ class OnitPanelState: NSObject {
         currentAnimationTask = nil
     }
 
+    // MARK: - Setup
+    
+    private func setupHighlightedTextDelegate() {
+        HighlightedTextManager.shared.addDelegate(self)
+    }
+    
     // MARK: - Delegates
 
     func addDelegate(_ delegate: OnitPanelStateDelegate) {

macos/Onit/UI/QuickEdit/QuickEditManager.swift

@@ -285,7 +285,7 @@ extension QuickEditManager {
     }
 
     func caretDidDisappear() {
-        if !hasTextSelection(PanelStateCoordinator.shared.state.pendingInput?.selectedText ?? PanelStateCoordinator.shared.state.trackedPendingInput?.selectedText) {
+        if !hasTextSelection(highlightedTextManager.selectedText) {
             hideHint()
         }
     }