Indicator follows the editing cursor if possible

Author: intitni

Core/Sources/Service/Environment.swift

@@ -204,4 +204,13 @@ extension AXUIElement {
         }
         throw error
     }
+    
+    func copyParameterizedValue<T>(key: String, parameters: AnyObject, ofType _: T.Type = T.self) throws -> T {
+        var value: AnyObject?
+        let error = AXUIElementCopyParameterizedAttributeValue(self, key as CFString, parameters as CFTypeRef, &value)
+        if error == .success, let value = value as? T {
+            return value
+        }
+        throw error
+    }
 }

Core/Sources/Service/RealtimeSuggestionController.swift

@@ -173,9 +173,9 @@ final class RealtimeSuggestionIndicatorController {
         @ObservedObject var viewModel: IndicatorContentViewModel
         @State var progress: CGFloat = 1
         var opacityA: CGFloat { progress }
-        var opacityB: CGFloat { (1 - progress) }
+        var opacityB: CGFloat { 1 - progress }
         var scaleA: CGFloat { progress / 2 + 0.5 }
-        var scaleB: CGFloat { 1 - progress }
+        var scaleB: CGFloat { max(1 - progress, 0.01) }
 
         var body: some View {
             Circle()
@@ -315,6 +315,36 @@ final class RealtimeSuggestionIndicatorController {
                 return
             }
 
+            if let activeXcode = NSRunningApplication
+                .runningApplications(withBundleIdentifier: "com.apple.dt.Xcode")
+                .first(where: \.isActive)
+            {
+                let application = AXUIElementCreateApplication(activeXcode.processIdentifier)
+                if let focusElement: AXUIElement = try? application
+                    .copyValue(key: kAXFocusedUIElementAttribute),
+                    let selectedRange: AXValue = try? focusElement
+                    .copyValue(key: kAXSelectedTextRangeAttribute),
+                    let rect: AXValue = try? focusElement.copyParameterizedValue(
+                        key: kAXBoundsForRangeParameterizedAttribute,
+                        parameters: selectedRange
+                    )
+                {
+                    var frame: CGRect = .zero
+                    let found = AXValueGetValue(rect, .cgRect, &frame)
+                    let screen = NSScreen.screens.first
+                    if found, let screen {
+                        frame.origin = .init(
+                            x: frame.maxX + 5,
+                            y: screen.frame.height - frame.minY - 12
+                        )
+                        frame.size = .init(width: 10, height: 10)
+                        window.setFrame(frame, display: false)
+                        window.makeKey()
+                        return
+                    }
+                }
+            }
+
             var frame = window.frame
             let location = NSEvent.mouseLocation
             frame.origin = .init(x: location.x + 15, y: location.y + 15)

EditorExtension/SourceEditorExtension.swift

@@ -9,6 +9,8 @@ class SourceEditorExtension: NSObject, XCSourceEditorExtension {
             RejectSuggestionCommand(),
             NextSuggestionCommand(),
             PreviousSuggestionCommand(),
+            TurnOnRealtimeSuggestionsCommand(),
+            TurnOffRealtimeSuggestionsCommand(),
             RealtimeSuggestionsCommand(),
             PrefetchSuggestionsCommand(),
         ].map(makeCommandDefinition)

EditorExtension/TurnOnRealtimeSuggestionsCommand.swift

@@ -3,8 +3,8 @@ import CopilotModel
 import Foundation
 import XcodeKit
 
-class TurnOnRealtimeSuggestionsCommand: NSObject, XCSourceEditorCommand, CommandType {
-    var name: String { "Turn On Real-time Suggestions" }
+class ToggleRealtimeSuggestionsCommand: NSObject, XCSourceEditorCommand, CommandType {
+    var name: String { "Toggle Real-time Suggestions" }
 
     func perform(
         with invocation: XCSourceEditorCommandInvocation,