Support save notification

Author: intitni

Core/Sources/CopilotService/CopilotService.swift

@@ -261,7 +261,7 @@ public final class CopilotSuggestionService: CopilotBaseService, CopilotSuggesti
     ) async throws {
         let languageId = languageIdentifierFromFileURL(fileURL)
         let uri = "file://\(fileURL.path)"
-        Logger.service.debug("Open \(uri)")
+//        Logger.service.debug("Open \(uri)")
         try await server.sendNotification(
             .didOpenTextDocument(
                 DidOpenTextDocumentParams(
@@ -279,7 +279,7 @@ 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)")
+//        Logger.service.debug("Change \(uri)")
         try await server.sendNotification(
             .didOpenTextDocument(
                 DidOpenTextDocumentParams(
@@ -312,3 +312,4 @@ extension InitializingServer: CopilotLSP {
         try await sendRequest(endpoint.request)
     }
 }
+

Core/Sources/Service/Workspace.swift

@@ -3,10 +3,10 @@ import CopilotModel
 import CopilotService
 import Environment
 import Foundation
+import Logger
 import Preferences
 import SuggestionInjector
 import XPCShared
-import Logger
 
 @ServiceActor
 final class Filespace {
@@ -40,8 +40,15 @@ final class Filespace {
         Environment.now().timeIntervalSince(lastSuggestionUpdateTime) > 60 * 60 * 8
     }
 
-    fileprivate init(fileURL: URL) {
+    let fileSaveWatcher: FileSaveWatcher
+
+    fileprivate init(fileURL: URL, onSave: @escaping (Filespace) -> Void) {
         self.fileURL = fileURL
+        fileSaveWatcher = .init(fileURL: fileURL)
+        fileSaveWatcher.changeHandler = { [weak self] in
+            guard let self else { return }
+            onSave(self)
+        }
     }
 
     func reset(resetSnapshot: Bool = true) {
@@ -51,7 +58,7 @@ final class Filespace {
             suggestionSourceSnapshot = .init(linesHash: -1, cursorPosition: .outOfScope)
         }
     }
-    
+
     func refreshUpdateTime() {
         lastSuggestionUpdateTime = Environment.now()
     }
@@ -71,7 +78,7 @@ final class Workspace {
             onChange?()
         }
     }
-    
+
     struct SuggestionFeatureDisabledError: Error, LocalizedError {
         var errorDescription: String? {
             "Suggestion feature is disabled for this project."
@@ -112,7 +119,7 @@ final class Workspace {
         }
         return _copilotSuggestionService
     }
-    
+
     var isSuggestionFeatureEnabled: Bool {
         let isSuggestionDisabledGlobally = UserDefaults.shared
             .value(for: \.disableSuggestionFeatureGlobally)
@@ -127,7 +134,7 @@ final class Workspace {
 
     private init(projectRootURL: URL) {
         self.projectRootURL = projectRootURL
-        
+
         Task {
             userDefaultsObserver.onChange = { [weak self] in
                 guard let self else { return }
@@ -140,7 +147,7 @@ final class Workspace {
                 options: .new,
                 context: nil
             )
-            
+
             UserDefaults.shared.addObserver(
                 userDefaultsObserver,
                 forKeyPath: UserDefaultPreferenceKeys().disableSuggestionFeatureGlobally.key,
@@ -171,22 +178,31 @@ final class Workspace {
                 return (workspace, filespace)
             }
         }
-        
+
         let projectURL = try await Environment.fetchCurrentProjectRootURL(fileURL)
         let workspaceURL = projectURL ?? fileURL
         let workspace = workspaces[workspaceURL] ?? Workspace(projectRootURL: workspaceURL)
-        let existedFilespace = workspace.filespaces[fileURL]
-        let filespace = existedFilespace ?? .init(fileURL: fileURL)
-        if workspace.filespaces[fileURL] == nil {
-            workspace.filespaces[fileURL] = filespace
-        }
+        let filespace = workspace.createFilespaceIfNeeded(fileURL: fileURL)
         workspaces[workspaceURL] = workspace
+        return (workspace, filespace)
+    }
+
+    func createFilespaceIfNeeded(fileURL: URL) -> Filespace {
+        let existedFilespace = filespaces[fileURL]
+        let filespace = existedFilespace ?? .init(fileURL: fileURL, onSave: { [weak self]
+            filespace in
+                guard let self else { return }
+                notifySaveFile(filespace: filespace)
+        })
+        if filespaces[fileURL] == nil {
+            filespaces[fileURL] = filespace
+        }
         if existedFilespace == nil {
-            workspace.notifyOpenFile(filespace: filespace)
+            notifyOpenFile(filespace: filespace)
         } else {
             filespace.refreshUpdateTime()
         }
-        return (workspace, filespace)
+        return filespace
     }
 }
 
@@ -202,7 +218,8 @@ extension Workspace {
         }
         lastTriggerDate = Environment.now()
 
-        let filespace = filespaces[fileURL] ?? .init(fileURL: fileURL)
+        let filespace = createFilespaceIfNeeded(fileURL: fileURL)
+
         if filespaces[fileURL] == nil {
             filespaces[fileURL] = filespace
         }
@@ -306,7 +323,7 @@ extension Workspace {
 
         return suggestion
     }
-    
+
     func notifyOpenFile(filespace: Filespace) {
         Task {
             try await copilotSuggestionService?.notifyOpenTextDocument(
@@ -315,7 +332,7 @@ extension Workspace {
             )
         }
     }
-    
+
     func notifyUpdateFile(filespace: Filespace, content: String) {
         Task {
             try await copilotSuggestionService?.notifyChangeTextDocument(
@@ -324,6 +341,12 @@ extension Workspace {
             )
         }
     }
+
+    func notifySaveFile(filespace: Filespace) {
+        Task {
+            try await copilotSuggestionService?.notifySaveTextDocument(fileURL: filespace.fileURL)
+        }
+    }
 }
 
 extension Workspace {
@@ -345,3 +368,41 @@ extension Workspace {
         realtimeSuggestionRequests = []
     }
 }
+
+final class FileSaveWatcher {
+    let url: URL
+    var fileHandle: FileHandle?
+    var source: DispatchSourceFileSystemObject?
+    var changeHandler: () -> Void = {}
+    
+    init(fileURL: URL) {
+        url = fileURL
+        startup()
+    }
+    
+    deinit {
+        source?.cancel()
+    }
+    
+    func startup() {
+        if let source = source {
+            source.cancel()
+        }
+        
+        fileHandle = try? FileHandle(forReadingFrom: url)
+        if let fileHandle {
+            source = DispatchSource.makeFileSystemObjectSource(
+                fileDescriptor: fileHandle.fileDescriptor,
+                eventMask: .link,
+                queue: .main
+            )
+
+            source?.setEventHandler { [weak self] in
+                self?.changeHandler()
+                self?.startup()
+            }
+            
+            source?.resume()
+        }
+    }
+}