Add CursorPositionTracker

Author: intitni

Core/Sources/SuggestionWidget/CursorPositionTracker.swift

@@ -0,0 +1,53 @@
+import Combine
+import Foundation
+import Perception
+import SuggestionModel
+import XcodeInspector
+
+@Perceptible
+final class CursorPositionTracker {
+    @MainActor
+    var cursorPosition: CursorPosition = .zero
+
+    @PerceptionIgnored var editorObservationTask: Set<AnyCancellable> = []
+    @PerceptionIgnored var eventObservationTask: Task<Void, Never>?
+
+    init() {
+        observeAppChange()
+    }
+
+    deinit {
+        eventObservationTask?.cancel()
+    }
+
+    private func observeAppChange() {
+        editorObservationTask = []
+        Task {
+            await XcodeInspector.shared.safe.$focusedEditor.sink { [weak self] editor in
+                guard let editor, let self else { return }
+                Task { @MainActor in
+                    self.observeAXNotifications(editor)
+                }
+            }.store(in: &editorObservationTask)
+        }
+    }
+
+    private func observeAXNotifications(_ editor: SourceEditor) {
+        eventObservationTask?.cancel()
+        let content = editor.getLatestEvaluatedContent()
+        Task { @MainActor in
+            self.cursorPosition = content.cursorPosition
+        }
+        eventObservationTask = Task { [weak self] in
+            for await event in await editor.axNotifications.notifications() {
+                guard let self else { return }
+                guard event.kind == .evaluatedContentChanged else { continue }
+                let content = editor.getLatestEvaluatedContent()
+                Task { @MainActor in
+                    self.cursorPosition = content.cursorPosition
+                }
+            }
+        }
+    }
+}
+

Core/Sources/SuggestionWidget/WidgetWindowsController.swift

@@ -599,6 +599,7 @@ public final class WidgetWindows {
     let store: StoreOf<WidgetFeature>
     let chatTabPool: ChatTabPool
     weak var controller: WidgetWindowsController?
+    let cursorPositionTracker = CursorPositionTracker()
 
     // you should make these window `.transient` so they never show up in the mission control.
 
@@ -670,7 +671,7 @@ public final class WidgetWindows {
                     state: \.sharedPanelState,
                     action: \.sharedPanel
                 )
-            )
+            ).environment(cursorPositionTracker)
         )
         it.setIsVisible(true)
         it.canBecomeKeyChecker = { [store] in
@@ -704,7 +705,7 @@ public final class WidgetWindows {
                     state: \.suggestionPanelState,
                     action: \.suggestionPanel
                 )
-            )
+            ).environment(cursorPositionTracker)
         )
         it.canBecomeKeyChecker = { false }
         it.setIsVisible(true)