Move refreshUpdateTime calls to Workspace

Author: intitni

Tool/Sources/Workspace/Workspace.swift

@@ -123,24 +123,18 @@ public final class Workspace {
             fileURL: fileURL,
             onSave: { [weak self] filespace in
                 guard let self else { return }
-                for plugin in self.plugins.values {
-                    plugin.didSaveFilespace(filespace)
-                }
+                self.didSaveFilespace(filespace)
             },
             onClose: { [weak self] url in
                 guard let self else { return }
-                for plugin in self.plugins.values {
-                    plugin.didCloseFilespace(url)
-                }
+                self.didCloseFilespace(url)
             }
         )
         if filespaces[fileURL] == nil {
             filespaces[fileURL] = filespace
         }
         if existedFilespace == nil {
-            for plugin in plugins.values {
-                plugin.didOpenFilespace(filespace)
-            }
+            didOpenFilespace(filespace)
         } else {
             filespace.refreshUpdateTime()
         }
@@ -154,10 +148,37 @@ public final class Workspace {
 
     @WorkspaceActor
     public func didUpdateFilespace(fileURL: URL, content: String) {
+        refreshUpdateTime()
         guard let filespace = filespaces[fileURL] else { return }
+        filespace.refreshUpdateTime()
         for plugin in plugins.values {
             plugin.didUpdateFilespace(filespace, content: content)
         }
     }
+
+    @WorkspaceActor
+    func didOpenFilespace(_ filespace: Filespace) {
+        refreshUpdateTime()
+        openedFileRecoverableStorage.openFile(fileURL: filespace.fileURL)
+        for plugin in plugins.values {
+            plugin.didOpenFilespace(filespace)
+        }
+    }
+
+    @WorkspaceActor
+    func didCloseFilespace(_ fileURL: URL) {
+        for plugin in self.plugins.values {
+            plugin.didCloseFilespace(fileURL)
+        }
+    }
+
+    @WorkspaceActor
+    func didSaveFilespace(_ filespace: Filespace) {
+        refreshUpdateTime()
+        filespace.refreshUpdateTime()
+        for plugin in plugins.values {
+            plugin.didSaveFilespace(filespace)
+        }
+    }
 }
 

Tool/Sources/WorkspaceSuggestionService/SuggestionWorkspacePlugin.swift

@@ -95,8 +95,6 @@ public final class SuggestionServiceWorkspacePlugin: WorkspacePlugin {
     }
 
     public func notifyOpenFile(filespace: Filespace) {
-        workspace?.refreshUpdateTime()
-        workspace?.openedFileRecoverableStorage.openFile(fileURL: filespace.fileURL)
         Task {
             guard !(await filespace.isGitIgnored) else { return }
             // check if file size is larger than 15MB, if so, return immediately
@@ -114,8 +112,6 @@ public final class SuggestionServiceWorkspacePlugin: WorkspacePlugin {
     }
 
     public func notifyUpdateFile(filespace: Filespace, content: String) {
-        filespace.refreshUpdateTime()
-        workspace?.refreshUpdateTime()
         Task {
             guard !(await filespace.isGitIgnored) else { return }
             try await suggestionService?.notifyChangeTextDocument(
@@ -126,8 +122,6 @@ public final class SuggestionServiceWorkspacePlugin: WorkspacePlugin {
     }
 
     public func notifySaveFile(filespace: Filespace) {
-        filespace.refreshUpdateTime()
-        workspace?.refreshUpdateTime()
         Task {
             guard !(await filespace.isGitIgnored) else { return }
             try await suggestionService?.notifySaveTextDocument(fileURL: filespace.fileURL)