Add universal real-time suggestions toggle

Author: intitni

Core/Sources/Service/Environment.swift

@@ -43,6 +43,21 @@ enum Environment {
         }
         return !activeXcodes.isEmpty
     }
+    
+    static var frontmostXcodeWindowIsEditor: () async -> Bool = {
+        let appleScript = """
+        tell application "Xcode"
+            return path of document of the first window
+        end tell
+        """
+        do {
+            let result = try await runAppleScript(appleScript)
+            return !result.isEmpty
+        }
+        catch {
+            return false
+        }
+    }
 
     static var fetchCurrentProjectRootURL: (_ fileURL: URL?) async throws -> URL? = { fileURL in
         let appleScript = """
@@ -51,7 +66,7 @@ enum Environment {
         end tell
         """
 
-        let path = try await runAppleScript(appleScript)
+        let path = (try? await runAppleScript(appleScript)) ?? ""
         if !path.isEmpty {
             let trimmedNewLine = path.trimmingCharacters(in: .newlines)
             var url = URL(fileURLWithPath: trimmedNewLine)
@@ -101,19 +116,20 @@ enum Environment {
                     key: kAXFocusedWindowAttribute,
                     ofType: AXUIElement.self
                 )
-                var path = try frontmostWindow.copyValue(
+                var path = try? frontmostWindow.copyValue(
                     key: kAXDocumentAttribute,
                     ofType: String?.self
                 )
                 if path == nil {
-                    if let firstWindow = try application.copyValue(
+                    for window in try application.copyValue(
                         key: kAXWindowsAttribute,
                         ofType: [AXUIElement].self
-                    ).first {
-                        path = try firstWindow.copyValue(
+                    ) {
+                        path = try? window.copyValue(
                             key: kAXDocumentAttribute,
-                            ofType: String.self
+                            ofType: String?.self
                         )
+                        if path != nil { break }
                     }
                 }
                 if let path = path?.removingPercentEncoding {

Core/Sources/Service/Workspace.swift

@@ -46,9 +46,7 @@ final class Workspace {
 
     var filespaces = [URL: Filespace]()
     var isRealtimeSuggestionEnabled: Bool {
-        (UserDefaults.shared.dictionary(
-            forKey: SettingsKey.realtimeSuggestionState
-        )?[projectRootURL.absoluteString]) as? Bool ?? false
+        UserDefaults.shared.bool(forKey: SettingsKey.realtimeSuggestionToggle) as? Bool ?? false
     }
 
     var realtimeSuggestionRequests = Set<Task<Void, Error>>()

Core/Sources/Service/XPCService.swift

@@ -89,7 +89,6 @@ public class XPCService: NSObject, XPCServiceProtocol {
     ) {
         Task { @ServiceActor in
             do {
-                throw CancellationError()
                 let editor = try JSONDecoder().decode(EditorContent.self, from: editorContent)
                 let fileURL = try await Environment.fetchCurrentFileURL()
                 let workspace = try await fetchOrCreateWorkspaceIfNeeded(fileURL: fileURL)
@@ -265,19 +264,10 @@ public class XPCService: NSObject, XPCServiceProtocol {
             return
         }
         Task { @ServiceActor in
-            let fileURL = try await Environment.fetchCurrentFileURL()
-            let workspace = try await fetchOrCreateWorkspaceIfNeeded(fileURL: fileURL)
-            if var state = UserDefaults.shared
-                .dictionary(forKey: SettingsKey.realtimeSuggestionState)
-            {
-                state[workspace.projectRootURL.absoluteString] = enabled
-                UserDefaults.shared.set(state, forKey: SettingsKey.realtimeSuggestionState)
-            } else {
-                UserDefaults.shared.set(
-                    [workspace.projectRootURL.absoluteString: enabled],
-                    forKey: SettingsKey.realtimeSuggestionState
-                )
-            }
+            UserDefaults.shared.set(
+                enabled,
+                forKey: SettingsKey.realtimeSuggestionToggle
+            )
             reply(nil)
         }
     }
@@ -290,8 +280,8 @@ public class XPCService: NSObject, XPCServiceProtocol {
             do {
                 let editor = try JSONDecoder().decode(EditorContent.self, from: editorContent)
                 let fileURL = try await Environment.fetchCurrentFileURL()
-                try Task.checkCancellation()
                 let workspace = try await fetchOrCreateWorkspaceIfNeeded(fileURL: fileURL)
+                try Task.checkCancellation()
                 _ = workspace.getRealtimeSuggestedCode(
                     forFileAt: fileURL,
                     content: editor.content,

EditorExtension/TurnOffRealtimeSuggestionsCommand.swift

@@ -4,7 +4,7 @@ import Foundation
 import XcodeKit
 
 class TurnOffRealtimeSuggestionsCommand: NSObject, XCSourceEditorCommand, CommandType {
-    var name: String { "Turn Off Real-time Suggestions for Workspace" }
+    var name: String { "Turn Off Real-time Suggestions" }
 
     func perform(
         with invocation: XCSourceEditorCommandInvocation,

EditorExtension/TurnOnRealtimeSuggestionsCommand.swift

@@ -4,7 +4,7 @@ import Foundation
 import XcodeKit
 
 class TurnOnRealtimeSuggestionsCommand: NSObject, XCSourceEditorCommand, CommandType {
-    var name: String { "Turn On Real-time Suggestions for Workspace" }
+    var name: String { "Turn On Real-time Suggestions" }
 
     func perform(
         with invocation: XCSourceEditorCommandInvocation,