Support global chat

Author: intitni

Core/Sources/Service/GUI/ChatProvider+Service.swift

@@ -4,7 +4,12 @@ import OpenAIService
 import SuggestionWidget
 
 extension ChatProvider {
-    convenience init(service: ChatService, fileURL: URL, onCloseChat: @escaping () -> Void) {
+    convenience init(
+        service: ChatService,
+        fileURL: URL,
+        onCloseChat: @escaping () -> Void,
+        onSwitchContext: @escaping () -> Void
+    ) {
         self.init()
         let cancellable = service.objectWillChange.sink { [weak self] in
             guard let self else { return }
@@ -19,7 +24,7 @@ extension ChatProvider {
                 self.isReceivingMessage = await service.chatGPTService.isReceivingMessage
             }
         }
-        
+
         service.objectWillChange.send()
 
         onMessageSend = { [cancellable] message in
@@ -51,5 +56,9 @@ extension ChatProvider {
                 onCloseChat()
             }
         }
+        
+        self.onSwitchContext = {
+            onSwitchContext()
+        }
     }
 }

Core/Sources/Service/GUI/WidgetDataSource.swift

@@ -11,8 +11,9 @@ final class WidgetDataSource {
 
     private init() {}
 
+    @discardableResult
     func createChatIfNeeded(for url: URL) -> ChatService {
-        let useGlobalChat = UserDefaults.standard.value(for: \.useGlobalChat)
+        let useGlobalChat = UserDefaults.shared.value(for: \.useGlobalChat)
         let chat: ChatService
         if useGlobalChat {
             chat = globalChat ?? ChatService(chatGPTService: ChatGPTService())
@@ -68,30 +69,31 @@ extension WidgetDataSource: SuggestionWidgetDataSource {
     }
 
     func chatForFile(at url: URL) async -> ChatProvider? {
-        let useGlobalChat = UserDefaults.standard.value(for: \.useGlobalChat)
-        if useGlobalChat, let globalChat {
-            return .init(
-                service: globalChat,
+        let useGlobalChat = UserDefaults.shared.value(for: \.useGlobalChat)
+        let buildChatProvider = { (service: ChatService) in
+            return ChatProvider(
+                service: service,
                 fileURL: url,
                 onCloseChat: { [weak self] in
-                    Task { @ServiceActor [weak self] in
-                        self?.globalChat = nil
-                    }
+                    self?.globalChat = nil
+                },
+                onSwitchContext: { [weak self] in
+                    UserDefaults.shared.set(!useGlobalChat, for: \.useGlobalChat)
+                    self?.createChatIfNeeded(for: url)
+                    let presenter = PresentInWindowSuggestionPresenter()
+                    presenter.presentChatRoom(fileURL: url)
                 }
             )
         }
+        
+        if useGlobalChat, let globalChat {
+            return buildChatProvider(globalChat)
+        }
 
         if let service = chats[url] {
-            return .init(
-                service: service,
-                fileURL: url,
-                onCloseChat: { [weak self] in
-                    Task { @ServiceActor [weak self] in
-                        self?.chats[url] = nil
-                    }
-                }
-            )
+            return buildChatProvider(service)
         }
+        
         return nil
     }
 }

Core/Sources/Service/SuggestionCommandHandler/WindowBaseCommandHandler.swift

@@ -213,7 +213,7 @@ struct WindowBaseCommandHandler: SuggestionCommandHandler {
             )
         }
 
-        presenter.presentChatGPTConversation(fileURL: fileURL)
+        presenter.presentChatRoom(fileURL: fileURL)
     }
 
     func chatWithSelection(editor: EditorContent) async throws -> UpdatedContent? {
@@ -273,6 +273,6 @@ struct WindowBaseCommandHandler: SuggestionCommandHandler {
             }
         }
 
-        presenter.presentChatGPTConversation(fileURL: fileURL)
+        presenter.presentChatRoom(fileURL: fileURL)
     }
 }

Core/Sources/Service/SuggestionPresenter/PresentInWindowSuggestionPresenter.swift

@@ -48,7 +48,7 @@ struct PresentInWindowSuggestionPresenter {
         }
     }
 
-    func presentChatGPTConversation(fileURL: URL) {
+    func presentChatRoom(fileURL: URL) {
         Task { @MainActor in
             let controller = GraphicalUserInterfaceController.shared.suggestionWidget
             controller.presentChatRoom(fileURL: fileURL)

Core/Sources/SuggestionWidget/ChatProvider.swift

@@ -1,37 +1,38 @@
 import Foundation
+import SwiftUI
 
-public final class ChatProvider: ObservableObject, Equatable {
+public final class ChatProvider: ObservableObject {
     @Published public var history: [ChatMessage] = []
     @Published public var isReceivingMessage = false
     public var onMessageSend: (String) -> Void
     public var onStop: () -> Void
     public var onClear: () -> Void
     public var onClose: () -> Void
+    public var onSwitchContext: () -> Void
 
     public init(
         history: [ChatMessage] = [],
         isReceivingMessage: Bool = false,
         onMessageSend: @escaping (String) -> Void = { _ in },
         onStop: @escaping () -> Void = {},
         onClear: @escaping () -> Void = {},
-        onClose: @escaping () -> Void = {}
+        onClose: @escaping () -> Void = {},
+        onSwitchContext: @escaping () -> Void = {}
     ) {
         self.history = history
         self.isReceivingMessage = isReceivingMessage
         self.onMessageSend = onMessageSend
         self.onStop = onStop
         self.onClear = onClear
         self.onClose = onClose
+        self.onSwitchContext = onSwitchContext
     }
 
-    public static func == (lhs: ChatProvider, rhs: ChatProvider) -> Bool {
-        lhs.history == rhs.history && lhs.isReceivingMessage == rhs.isReceivingMessage
-    }
-    
     public func send(_ message: String) { onMessageSend(message) }
     public func stop() { onStop() }
     public func clear() { onClear() }
     public func close() { onClose() }
+    public func switchContext() { onSwitchContext() }
 }
 
 public struct ChatMessage: Equatable {

Core/Sources/SuggestionWidget/SuggestionPanelContent/ChatPanel.swift

@@ -2,7 +2,6 @@ import MarkdownUI
 import SwiftUI
 
 struct ChatPanel: View {
-    @ObservedObject var viewModel: SuggestionPanelViewModel
     @ObservedObject var chat: ChatProvider
     @Namespace var inputAreaNamespace
     @State var typedMessage = ""
@@ -13,8 +12,7 @@ struct ChatPanel: View {
             Divider()
             ChatPanelMessages(
                 chat: chat,
-                inputAreaNamespace: inputAreaNamespace,
-                colorScheme: viewModel.colorScheme
+                inputAreaNamespace: inputAreaNamespace
             )
             Divider()
             ChatPanelInputArea(
@@ -30,10 +28,18 @@ struct ChatPanel: View {
 }
 
 struct ChatPanelToolbar: View {
-    let chat: ChatProvider
+    @ObservedObject var chat: ChatProvider
+    @AppStorage(\.useGlobalChat) var useGlobalChat
 
     var body: some View {
         HStack {
+            Toggle(isOn: .init(get: {
+                useGlobalChat
+            }, set: { _ in
+                chat.switchContext()
+            })) { EmptyView() }
+                .toggleStyle(GlobalChatSwitchToggleStyle())
+
             Spacer()
 
             Button(action: {
@@ -45,7 +51,8 @@ struct ChatPanelToolbar: View {
             }
             .buttonStyle(.plain)
         }
-        .padding(.horizontal, 4)
+        .padding(.leading, 8)
+        .padding(.trailing, 4)
         .padding(.vertical, 4)
         .background(.regularMaterial)
     }
@@ -54,7 +61,7 @@ struct ChatPanelToolbar: View {
 struct ChatPanelMessages: View {
     @ObservedObject var chat: ChatProvider
     var inputAreaNamespace: Namespace.ID
-    var colorScheme: ColorScheme
+    @Environment(\.colorScheme) var colorScheme
     @AppStorage(\.disableLazyVStack) var disableLazyVStack
 
     @ViewBuilder
@@ -74,7 +81,7 @@ struct ChatPanelMessages: View {
         ScrollView {
             vstack {
                 let r = 6 as Double
-                
+
                 Spacer()
 
                 if chat.isReceivingMessage {
@@ -112,7 +119,6 @@ struct ChatPanelMessages: View {
                 ForEach(chat.history.reversed(), id: \.id) { message in
                     let text = message.text.isEmpty && !message.isUser ? "..." : message
                         .text
-                    
 
                     if message.isUser {
                         Markdown(text)
@@ -310,6 +316,42 @@ struct RoundedCorners: Shape {
     }
 }
 
+struct GlobalChatSwitchToggleStyle: ToggleStyle {
+    func makeBody(configuration: Configuration) -> some View {
+        HStack(spacing: 4) {
+            RoundedRectangle(cornerRadius: 10, style: .circular)
+                .foregroundColor(configuration.isOn ? Color.indigo : .gray.opacity(0.5))
+                .frame(width: 30, height: 20, alignment: .center)
+                .overlay(
+                    Circle()
+                        .fill(.regularMaterial)
+                        .padding(.all, 2)
+                        .overlay(
+                            Image(
+                                systemName: configuration
+                                    .isOn ? "globe" : "doc.circle"
+                            )
+                            .resizable()
+                            .aspectRatio(contentMode: .fit)
+                            .frame(width: 12, height: 12, alignment: .center)
+                            .foregroundStyle(.secondary)
+                        )
+                        .offset(x: configuration.isOn ? 5 : -5, y: 0)
+                        .animation(.linear(duration: 0.1), value: configuration.isOn)
+
+                )
+                .onTapGesture { configuration.isOn.toggle() }
+                .overlay {
+                    RoundedRectangle(cornerRadius: 10, style: .circular)
+                        .stroke(.black.opacity(0.2), lineWidth: 1)
+                }
+            
+            Text(configuration.isOn ? "Global Chat" : "File Chat")
+                .foregroundStyle(.tertiary)
+        }
+    }
+}
+
 // MARK: - Previews
 
 struct ChatPanel_Preview: PreviewProvider {
@@ -343,9 +385,7 @@ struct ChatPanel_Preview: PreviewProvider {
     ]
 
     static var previews: some View {
-        ChatPanel(viewModel: .init(
-            isPanelDisplayed: true
-        ), chat: .init(
+        ChatPanel(chat: .init(
             history: ChatPanel_Preview.history,
             isReceivingMessage: true
         ))
@@ -356,9 +396,7 @@ struct ChatPanel_Preview: PreviewProvider {
 
 struct ChatPanel_EmptyChat_Preview: PreviewProvider {
     static var previews: some View {
-        ChatPanel(viewModel: .init(
-            isPanelDisplayed: true
-        ), chat: .init(
+        ChatPanel(chat: .init(
             history: [],
             isReceivingMessage: false
         ))
@@ -388,9 +426,7 @@ struct ChatCodeSyntaxHighlighter: CodeSyntaxHighlighter {
 
 struct ChatPanel_InputText_Preview: PreviewProvider {
     static var previews: some View {
-        ChatPanel(viewModel: .init(
-            isPanelDisplayed: true
-        ), chat: .init(
+        ChatPanel(chat: .init(
             history: ChatPanel_Preview.history,
             isReceivingMessage: false
         ))
@@ -403,9 +439,6 @@ struct ChatPanel_InputText_Preview: PreviewProvider {
 struct ChatPanel_InputMultilineText_Preview: PreviewProvider {
     static var previews: some View {
         ChatPanel(
-            viewModel: .init(
-                isPanelDisplayed: true
-            ),
             chat: .init(
                 history: ChatPanel_Preview.history,
                 isReceivingMessage: false
@@ -420,10 +453,7 @@ struct ChatPanel_InputMultilineText_Preview: PreviewProvider {
 
 struct ChatPanel_Light_Preview: PreviewProvider {
     static var previews: some View {
-        ChatPanel(viewModel: .init(
-            isPanelDisplayed: true,
-            colorScheme: .light
-        ), chat: .init(
+        ChatPanel(chat: .init(
             history: ChatPanel_Preview.history,
             isReceivingMessage: true
         ))

Core/Sources/SuggestionWidget/SuggestionPanelView.swift

@@ -124,7 +124,7 @@ struct SuggestionPanelView: View {
 
                 if let chat = viewModel.chat {
                     if case .chat = viewModel.activeTab {
-                        ChatPanel(viewModel: viewModel, chat: chat)
+                        ChatPanel(chat: chat)
                             .frame(maxWidth: .infinity, maxHeight: Style.panelHeight)
                             .fixedSize(horizontal: false, vertical: true)
                             .allowsHitTesting(viewModel.isPanelDisplayed)
@@ -146,7 +146,6 @@ struct SuggestionPanelView: View {
             return 1
         }())
         .animation(.easeInOut(duration: 0.2), value: viewModel.content?.contentHash)
-        .animation(.easeInOut(duration: 0.2), value: viewModel.chat)
         .animation(.easeInOut(duration: 0.2), value: viewModel.activeTab)
         .animation(.easeInOut(duration: 0.2), value: viewModel.isPanelDisplayed)
         .frame(maxWidth: Style.panelWidth, maxHeight: Style.panelHeight)