Prevent getting multiple content from SourceEditor in a single run

Author: intitni

Core/Sources/ChatService/CustomCommandTemplateProcessor.swift

@@ -39,7 +39,7 @@ public struct CustomCommandTemplateProcessor {
     }
 
     func getEditorInformation() -> EditorInformation {
-        let editorContent = XcodeInspector.shared.focusedEditor?.content
+        let editorContent = XcodeInspector.shared.focusedEditor?.getContent()
         let documentURL = XcodeInspector.shared.activeDocumentURL
         let language = documentURL.map(languageIdentifierFromFileURL) ?? .plaintext
 

Core/Sources/Service/GUI/ChatTabFactory.swift

@@ -43,7 +43,7 @@ enum ChatTabFactory {
                 guard let editor = XcodeInspector.shared.focusedEditor else {
                     return .init(selectedText: "", language: "", fileContent: "")
                 }
-                let content = editor.content
+                let content = editor.getContent()
                 return .init(
                     selectedText: content.selectedContent,
                     language: (

Core/Sources/Service/SuggestionCommandHandler/PseudoCommandHandler.swift

@@ -78,9 +78,10 @@ struct PseudoCommandHandler {
                 editor: editor
             )
             if let sourceEditor {
+                let editorContent = sourceEditor.getContent()
                 _ = filespace.validateSuggestions(
-                    lines: sourceEditor.content.lines,
-                    cursorPosition: sourceEditor.content.cursorPosition
+                    lines: editorContent.lines,
+                    cursorPosition: editorContent.cursorPosition
                 )
             }
             if filespace.presentingSuggestion != nil {
@@ -98,9 +99,10 @@ struct PseudoCommandHandler {
         guard let (_, filespace) = try? await Service.shared.workspacePool
             .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL) else { return }
 
+        let content = sourceEditor.getContent()
         if !filespace.validateSuggestions(
-            lines: sourceEditor.content.lines,
-            cursorPosition: sourceEditor.content.cursorPosition
+            lines: content.lines,
+            cursorPosition: content.cursorPosition
         ) {
             PresentInWindowSuggestionPresenter().discardSuggestion(fileURL: fileURL)
         }
@@ -351,7 +353,7 @@ extension PseudoCommandHandler {
         guard let filespace = await getFilespace(),
               let sourceEditor = sourceEditor ?? XcodeInspector.shared.focusedEditor
         else { return nil }
-        let content = sourceEditor.content
+        let content = sourceEditor.getContent()
         let uti = filespace.codeMetadata.uti ?? ""
         let tabSize = filespace.codeMetadata.tabSize ?? 4
         let indentSize = filespace.codeMetadata.indentSize ?? 4

Pro

@@ -24,10 +24,25 @@ public class SourceEditor {
     var observeAXNotificationsTask: Task<Void, Never>?
     public let axNotifications = AsyncPassthroughSubject<AXNotification>()
 
-    /// The content of the source editor.
-    public var content: Content {
+
+    private var cachedContent: String?
+    private var cachedLines = [String]()
+    
+    /// Get the content of the source editor.
+    ///
+    /// - note: This method is expensive.
+    public func getContent() -> Content {
         let content = element.value
-        let split = content.breakLines(appendLineBreakToLastLine: false)
+                                                     
+        let split = if element.hashValue == cachedContent?.hashValue {
+            cachedLines
+        } else {
+            content.breakLines(appendLineBreakToLastLine: false)
+        }
+        
+        cachedContent = content
+        cachedLines = split
+        
         let lineAnnotationElements = element.children.filter { $0.identifier == "Line Annotation" }
         let lineAnnotations = lineAnnotationElements.map(\.description)
 

Tool/Sources/XcodeInspector/SourceEditor.swift

@@ -41,7 +41,7 @@ public final class XcodeInspector: ObservableObject {
               let projectURL = XcodeInspector.shared.activeProjectRootURL
         else { return nil }
 
-        let editorContent = XcodeInspector.shared.focusedEditor?.content
+        let editorContent = XcodeInspector.shared.focusedEditor?.getContent()
         let language = languageIdentifierFromFileURL(documentURL)
         let relativePath = documentURL.path.replacingOccurrences(of: projectURL.path, with: "")