Support open document notification

Author: intitni

Core/Sources/CopilotService/CopilotService.swift

@@ -4,6 +4,7 @@ import LanguageClient
 import LanguageServerProtocol
 import Preferences
 import XPCShared
+import Logger
 
 public protocol CopilotAuthServiceType {
     func checkStatus() async throws -> CopilotStatus
@@ -25,10 +26,12 @@ public protocol CopilotSuggestionServiceType {
     ) async throws -> [CopilotCompletion]
     func notifyAccepted(_ completion: CopilotCompletion) async
     func notifyRejected(_ completions: [CopilotCompletion]) async
+    func openTextDocument(fileURL: URL, content: String) async throws
 }
 
 protocol CopilotLSP {
     func sendRequest<E: CopilotRequestType>(_ endpoint: E) async throws -> E.Response
+    func sendNotification(_ notif: ClientNotification) async throws
 }
 
 public class CopilotBaseService {
@@ -57,7 +60,7 @@ public class CopilotBaseService {
             }
             let executionParams: Process.ExecutionParameters
             let runner = UserDefaults.shared.value(for: \.runNodeWith)
-            
+
             switch runner {
             case .bash:
                 let nodePath = UserDefaults.shared.value(for: \.nodePath)
@@ -248,6 +251,27 @@ public final class CopilotSuggestionService: CopilotBaseService, CopilotSuggesti
             CopilotRequest.NotifyRejected(completionUUIDs: completions.map(\.uuid))
         )
     }
+
+    public func openTextDocument(
+        fileURL: URL,
+        content: String
+    ) async throws {
+        let languageId = languageIdentifierFromFileURL(fileURL)
+        let uri = "file://\(fileURL.path)"
+        Logger.service.debug(uri)
+        try await server.sendNotification(
+            .didOpenTextDocument(
+                DidOpenTextDocumentParams(
+                    textDocument: .init(
+                        uri: uri,
+                        languageId: languageId.rawValue,
+                        version: 0,
+                        text: content
+                    )
+                )
+            )
+        )
+    }
 }
 
 extension InitializingServer: CopilotLSP {

Core/Sources/Service/Workspace.swift

@@ -6,6 +6,7 @@ import Foundation
 import Preferences
 import SuggestionInjector
 import XPCShared
+import Logger
 
 @ServiceActor
 final class Filespace {
@@ -39,7 +40,7 @@ final class Filespace {
         Environment.now().timeIntervalSince(lastSuggestionUpdateTime) > 60 * 60 * 8
     }
 
-    init(fileURL: URL) {
+    fileprivate init(fileURL: URL) {
         self.fileURL = fileURL
     }
 
@@ -170,11 +171,15 @@ final class Workspace {
         let projectURL = try await Environment.fetchCurrentProjectRootURL(fileURL)
         let workspaceURL = projectURL ?? fileURL
         let workspace = workspaces[workspaceURL] ?? Workspace(projectRootURL: workspaceURL)
-        let filespace = workspace.filespaces[fileURL] ?? .init(fileURL: fileURL)
+        let existedFilespace = workspace.filespaces[fileURL]
+        let filespace = existedFilespace ?? .init(fileURL: fileURL)
         if workspace.filespaces[fileURL] == nil {
             workspace.filespaces[fileURL] = filespace
         }
         workspaces[workspaceURL] = workspace
+        if existedFilespace == nil {
+            workspace.notifyOpenFile(filespace: filespace)
+        }
         return (workspace, filespace)
     }
 }
@@ -295,6 +300,15 @@ extension Workspace {
 
         return suggestion
     }
+    
+    func notifyOpenFile(filespace: Filespace) {
+        Task {
+            try await copilotSuggestionService?.openTextDocument(
+                fileURL: filespace.fileURL,
+                content: try String(contentsOf: filespace.fileURL, encoding: .utf8)
+            )
+        }
+    }
 }
 
 extension Workspace {