Add global chat service

Author: intitni

Core/Sources/Preferences/Keys.swift

@@ -109,6 +109,13 @@ public struct UserDefaultPreferenceKeys {
         .init()
     }
 
+    public struct UseGlobalChat: UserDefaultPreferenceKey {
+        public let defaultValue = false
+        public let key = "UseGlobalChat"
+    }
+
+    public var useGlobalChat: UseGlobalChat { .init() }
+
     public var disableLazyVStack: FeatureFlags.DisableLazyVStack { .init() }
 }
 

Core/Sources/Service/GUI/WidgetDataSource.swift

@@ -4,6 +4,8 @@ import SuggestionWidget
 
 final class WidgetDataSource: SuggestionWidgetDataSource {
     static let shared = WidgetDataSource()
+    
+    var globalChat: ChatService? = nil
 
     private init() {}
 
@@ -49,6 +51,19 @@ final class WidgetDataSource: SuggestionWidgetDataSource {
     }
 
     func chatForFile(at url: URL) async -> ChatProvider? {
+        let useGlobalChat = UserDefaults.standard.value(for: \.useGlobalChat)
+        if useGlobalChat, let globalChat {
+            return .init(
+                service: globalChat,
+                fileURL: url,
+                onCloseChat: { [weak self] in
+                    Task { @ServiceActor [weak self] in
+                        self?.globalChat = nil
+                    }
+                }
+            )
+        }
+        
         for workspace in await workspaces.values {
             if let filespace = await workspace.filespaces[url],
                let service = await filespace.chatService

Core/Sources/Service/SuggestionCommandHandler/WindowBaseCommandHandler.swift

@@ -194,8 +194,7 @@ struct WindowBaseCommandHandler: SuggestionCommandHandler {
         let codeLanguage = languageIdentifierFromFileURL(fileURL)
         guard let selection = editor.selections.last else { return }
 
-        let chat = filespace.chatService ?? ChatService(chatGPTService: ChatGPTService())
-        filespace.chatService = chat
+        let chat = createChatServiceIfNeeded(filespace: filespace)
 
         await chat.mutateSystemPrompt(
             """
@@ -262,8 +261,7 @@ struct WindowBaseCommandHandler: SuggestionCommandHandler {
             """
         }()
 
-        let chat = filespace.chatService ?? ChatService(chatGPTService: ChatGPTService())
-        filespace.chatService = chat
+        let chat = createChatServiceIfNeeded(filespace: filespace)
 
         await chat.mutateSystemPrompt(prompt)
 
@@ -282,3 +280,19 @@ struct WindowBaseCommandHandler: SuggestionCommandHandler {
         presenter.presentChatGPTConversation(fileURL: fileURL)
     }
 }
+
+extension SuggestionCommandHandler {
+    @ServiceActor
+    func createChatServiceIfNeeded(filespace: Filespace) -> ChatService {
+        let chat: ChatService
+        if UserDefaults.shared.value(for: \.useGlobalChat) {
+            chat = WidgetDataSource.shared.globalChat
+                ?? ChatService(chatGPTService: ChatGPTService())
+            WidgetDataSource.shared.globalChat = chat
+        } else {
+            chat = filespace.chatService ?? ChatService(chatGPTService: ChatGPTService())
+            filespace.chatService = chat
+        }
+        return chat
+    }
+}