Fix that ChatProvider is recreated and cause UI blinks

Author: intitni

Core/Sources/Service/GUI/WidgetDataSource.swift

@@ -8,13 +8,16 @@ final class WidgetDataSource {
 
     var globalChat: ChatService?
     var chats = [URL: ChatService]()
+    var globalChatProvider: ChatProvider?
+    var chatProviders = [URL: ChatProvider]()
 
     private init() {}
 
     @discardableResult
     func createChatIfNeeded(for url: URL) -> ChatService {
         let useGlobalChat = UserDefaults.shared.value(for: \.useGlobalChat)
         let chat: ChatService
+        
         if useGlobalChat {
             chat = globalChat ?? ChatService(chatGPTService: ChatGPTService())
             globalChat = chat
@@ -77,8 +80,10 @@ extension WidgetDataSource: SuggestionWidgetDataSource {
                 onCloseChat: { [weak self] in
                     if UserDefaults.shared.value(for: \.useGlobalChat) {
                         self?.globalChat = nil
+                        self?.globalChatProvider = nil
                     } else {
                         self?.chats[url] = nil
+                        self?.chatProviders[url] = nil
                     }
                     let presenter = PresentInWindowSuggestionPresenter()
                     presenter.closeChatRoom(fileURL: url)
@@ -94,12 +99,20 @@ extension WidgetDataSource: SuggestionWidgetDataSource {
         }
 
         if useGlobalChat {
-            if let globalChat {
-                return buildChatProvider(globalChat)
+            if let globalChatProvider {
+                return globalChatProvider
+            } else if let globalChat {
+                let new = buildChatProvider(globalChat)
+                self.globalChatProvider = new
+                return new
             }
         } else {
-            if let service = chats[url] {
-                return buildChatProvider(service)
+            if let provider = chatProviders[url] {
+                return provider
+            } else if let service = chats[url] {
+                let new = buildChatProvider(service)
+                self.chatProviders[url] = new
+                return new
             }
         }
 

Core/Sources/Service/ScheduledCleaner.swift

@@ -15,6 +15,7 @@ public final class ScheduledCleaner {
                         for (url, filespace) in filespaces {
                             if filespace.isExpired {
                                 WidgetDataSource.shared.chats[url] = nil
+                                WidgetDataSource.shared.chatProviders[url] = nil
                             }
                         }
                         // cleanup workspace

Core/Sources/SuggestionWidget/ChatProvider.swift

@@ -2,6 +2,7 @@ import Foundation
 import SwiftUI
 
 public final class ChatProvider: ObservableObject {
+    let id = UUID()
     @Published public var history: [ChatMessage] = []
     @Published public var isReceivingMessage = false
     public var onMessageSend: (String) -> Void

Core/Sources/SuggestionWidget/SuggestionWidgetController.swift

@@ -444,7 +444,9 @@ extension SuggestionWidgetController {
         }
 
         if let chat = await dataSource?.chatForFile(at: fileURL) {
-            suggestionPanelViewModel.chat = chat
+            if suggestionPanelViewModel.chat?.id != chat.id {
+                suggestionPanelViewModel.chat = chat
+            }
         } else {
             suggestionPanelViewModel.chat = nil
         }