Support change notification

Author: intitni

Core/Sources/CopilotService/CopilotService.swift

@@ -26,7 +26,10 @@ public protocol CopilotSuggestionServiceType {
     ) async throws -> [CopilotCompletion]
     func notifyAccepted(_ completion: CopilotCompletion) async
     func notifyRejected(_ completions: [CopilotCompletion]) async
-    func openTextDocument(fileURL: URL, content: String) async throws
+    func notifyOpenTextDocument(fileURL: URL, content: String) async throws
+    func notifyChangeTextDocument(fileURL: URL, content: String) async throws
+    func notifyCloseTextDocument(fileURL: URL) async throws
+    func notifySaveTextDocument(fileURL: URL) async throws
 }
 
 protocol CopilotLSP {
@@ -252,13 +255,13 @@ public final class CopilotSuggestionService: CopilotBaseService, CopilotSuggesti
         )
     }
 
-    public func openTextDocument(
+    public func notifyOpenTextDocument(
         fileURL: URL,
         content: String
     ) async throws {
         let languageId = languageIdentifierFromFileURL(fileURL)
         let uri = "file://\(fileURL.path)"
-        Logger.service.debug(uri)
+        Logger.service.debug("Open \(uri)")
         try await server.sendNotification(
             .didOpenTextDocument(
                 DidOpenTextDocumentParams(
@@ -272,6 +275,36 @@ public final class CopilotSuggestionService: CopilotBaseService, CopilotSuggesti
             )
         )
     }
+    
+    public func notifyChangeTextDocument(fileURL: URL, content: String) async throws {
+        let languageId = languageIdentifierFromFileURL(fileURL)
+        let uri = "file://\(fileURL.path)"
+        Logger.service.debug("Change \(uri)")
+        try await server.sendNotification(
+            .didOpenTextDocument(
+                DidOpenTextDocumentParams(
+                    textDocument: .init(
+                        uri: uri,
+                        languageId: languageId.rawValue,
+                        version: 0,
+                        text: content
+                    )
+                )
+            )
+        )
+    }
+    
+    public func notifySaveTextDocument(fileURL: URL) async throws {
+        let uri = "file://\(fileURL.path)"
+        Logger.service.debug("Save \(uri)")
+        try await server.sendNotification(.didSaveTextDocument(.init(uri: uri)))
+    }
+    
+    public func notifyCloseTextDocument(fileURL: URL) async throws {
+        let uri = "file://\(fileURL.path)"
+        Logger.service.debug("Close \(uri)")
+        try await server.sendNotification(.didCloseTextDocument(.init(uri: uri)))
+    }
 }
 
 extension InitializingServer: CopilotLSP {

Core/Sources/Service/RealtimeSuggestionController.swift

@@ -131,6 +131,7 @@ public class RealtimeSuggestionController {
                 switch notification.name {
                 case kAXValueChangedNotification:
                     self.triggerPrefetchDebounced()
+                    await self.notifyEditingFileChange(editor: focusElement)
                 default:
                     continue
                 }
@@ -207,7 +208,7 @@ public class RealtimeSuggestionController {
 
             guard UserDefaults.shared.value(for: \.realtimeSuggestionToggle)
             else { return }
-            
+
             if UserDefaults.shared.value(for: \.disableSuggestionFeatureGlobally),
                let fileURL = try? await Environment.fetchCurrentFileURL(),
                let (workspace, _) = try? await Workspace
@@ -216,7 +217,7 @@ public class RealtimeSuggestionController {
                 let isEnabled = workspace.isSuggestionFeatureEnabled
                 if !isEnabled { return }
             }
-            
+
             if Task.isCancelled { return }
 
             Logger.service.info("Prefetch suggestions.")
@@ -264,4 +265,12 @@ public class RealtimeSuggestionController {
         let application = AXUIElementCreateApplication(activeXcode.processIdentifier)
         return application.focusedWindow?.child(identifier: "_XC_COMPLETION_TABLE_") != nil
     }
+
+    func notifyEditingFileChange(editor: AXUIElement) async {
+        guard let fileURL = try? await Environment.fetchCurrentFileURL(),
+            let (workspace, filespace) = try? await Workspace
+            .fetchOrCreateWorkspaceIfNeeded(fileURL: fileURL)
+        else { return }
+        workspace.notifyUpdateFile(filespace: filespace, content: editor.value)
+    }
 }

Core/Sources/Service/SuggestionCommandHandler/WindowBaseCommandHandler.swift

@@ -147,7 +147,7 @@ struct WindowBaseCommandHandler: SuggestionCommandHandler {
         defer { presenter.markAsProcessing(false) }
 
         let fileURL = try await Environment.fetchCurrentFileURL()
-        let (workspace, _) = try await Workspace.fetchOrCreateWorkspaceIfNeeded(fileURL: fileURL)
+        let (workspace, filespace) = try await Workspace.fetchOrCreateWorkspaceIfNeeded(fileURL: fileURL)
 
         let injector = SuggestionInjector()
         var lines = editor.lines
@@ -199,11 +199,15 @@ struct WindowBaseCommandHandler: SuggestionCommandHandler {
 
             presenter.discardSuggestion(fileURL: fileURL)
 
-            return .init(
+            let result = UpdatedContent(
                 content: String(lines.joined(separator: "")),
                 newSelection: .cursor(cursorPosition),
                 modifications: extraInfo.modifications
             )
+            
+            workspace.notifyUpdateFile(filespace: filespace, content: result.content)
+            
+            return result
         }
 
         return nil

Core/Sources/Service/Workspace.swift

@@ -303,18 +303,30 @@ extension Workspace {
 
     func notifyOpenFile(filespace: Filespace) {
         Task {
-            try await copilotSuggestionService?.openTextDocument(
+            try await copilotSuggestionService?.notifyOpenTextDocument(
                 fileURL: filespace.fileURL,
                 content: try String(contentsOf: filespace.fileURL, encoding: .utf8)
             )
         }
     }
+    
+    func notifyUpdateFile(filespace: Filespace, content: String) {
+        Task {
+            try await copilotSuggestionService?.notifyChangeTextDocument(
+                fileURL: filespace.fileURL,
+                content: content
+            )
+        }
+    }
 }
 
 extension Workspace {
     func cleanUp() {
         for (fileURL, filespace) in filespaces {
             if filespace.isExpired {
+                Task {
+                    try await copilotSuggestionService?.notifyCloseTextDocument(fileURL: fileURL)
+                }
                 filespaces[fileURL] = nil
             }
         }