Merge branch 'feature/rollback-suggestion-caching-machanism' into develop

Author: intitni

Core/Sources/Service/RealtimeSuggestionController.swift

@@ -134,27 +134,6 @@ public class RealtimeSuggestionController {
                 }
             }
         }
-
-        Task { // Get cache ready for real-time suggestions.
-            guard
-                let fileURL = try? await Environment.fetchCurrentFileURL(),
-                let (_, filespace) = try? await Workspace
-                .fetchOrCreateWorkspaceIfNeeded(fileURL: fileURL)
-            else { return }
-
-            if filespace.uti == nil {
-                Logger.service.info("Generate cache for file.")
-                // avoid the command get called twice
-                filespace.uti = ""
-                do {
-                    try await Environment.triggerAction("Real-time Suggestions")
-                } catch {
-                    if filespace.uti?.isEmpty ?? true {
-                        filespace.uti = nil
-                    }
-                }
-            }
-        }
     }
 
     func handleHIDEvent(event: CGEvent) async {

Core/Sources/Service/SuggestionCommandHandler/PseudoCommandHandler.swift

@@ -39,20 +39,28 @@ struct PseudoCommandHandler {
     }
 
     func generateRealtimeSuggestions() async {
+        // Can't use handler directly if content is not available.
         guard let editor = await getEditorContent() else {
             try? await Environment.triggerAction("Prefetch Suggestions")
             return
         }
+        
+        // If no cache is available, and completion panel is not displayed, try to get it with command.
+        if editor.uti.isEmpty, await Environment.frontmostXcodeWindowIsEditor() {
+            try? await Environment.triggerAction("Prefetch Suggestions")
+            return
+        }
+        
+        // Otherwise, get it from pseudo handler directly.
         let mode = UserDefaults.shared.value(for: \.suggestionPresentationMode)
-        let handler: SuggestionCommandHandler = {
-            switch mode {
-            case .comment:
-                return CommentBaseCommandHandler()
-            case .floatingWidget:
-                return WindowBaseCommandHandler()
-            }
-        }()
-        _ = try? await handler.generateRealtimeSuggestions(editor: editor)
+        switch mode {
+        case .comment:
+            let handler = CommentBaseCommandHandler()
+            _ = try? await handler.generateRealtimeSuggestions(editor: editor)
+        case .floatingWidget:
+            let handler = WindowBaseCommandHandler()
+            _ = try? await handler.generateRealtimeSuggestions(editor: editor)
+        }
     }
 
     func rejectSuggestions() async {

README.md

@@ -190,6 +190,7 @@ fi
 - The first run of the extension will be slow. Be patient.
 - The extension uses some dirty tricks to get the file and project/workspace paths. It may fail, it may be incorrect, especially when you have multiple Xcode windows running, and maybe even worse when they are in different displays. I am not sure about that though.
 - The suggestions are presented as C-style comments in comment mode, they may break your code if you are editing a JSON file or something.
+- When a real-time suggestion request is triggered, there is a chance that it may briefly block the editor. This can occur at most once for each file after each restart of the extension because the extension needs to initiate real-time suggestion by clicking an item from the menu bar. However, once a command has been executed and some information is cached, the extension will be able to trigger real-time suggestion using a different method.
 
 ## License