Add a shared widget data source

Author: intitni

Core/Sources/Service/GUI/GraphicalUserInterfaceController.swift.swift

@@ -9,33 +9,7 @@ public final class GraphicalUserInterfaceController {
     nonisolated let suggestionWidget = SuggestionWidgetController()
     private nonisolated init() {
         Task { @MainActor in
-            suggestionWidget.onNextButtonTapped = {
-                Task { @ServiceActor in
-                    let handler = PseudoCommandHandler()
-                    await handler.presentNextSuggestion()
-                }
-            }
-
-            suggestionWidget.onPreviousButtonTapped = {
-                Task { @ServiceActor in
-                    let handler = PseudoCommandHandler()
-                    await handler.presentPreviousSuggestion()
-                }
-            }
-
-            suggestionWidget.onRejectButtonTapped = {
-                Task { @ServiceActor in
-                    let handler = PseudoCommandHandler()
-                    await handler.rejectSuggestions()
-                }
-            }
-
-            suggestionWidget.onAcceptButtonTapped = {
-                Task { @ServiceActor in
-                    let handler = PseudoCommandHandler()
-                    await handler.acceptSuggestion()
-                }
-            }
+            suggestionWidget.dataSource = WidgetDataSource.shared
         }
     }
 }

Core/Sources/Service/GUI/WidgetDataSource.swift

@@ -0,0 +1,78 @@
+import ChatService
+import Foundation
+import SuggestionWidget
+
+final class WidgetDataSource: SuggestionWidgetDataSource {
+    static let shared = WidgetDataSource()
+
+    private init() {}
+
+    func suggestionForFile(at url: URL) async -> SuggestionProvider? {
+        for workspace in await workspaces.values {
+            if let filespace = await workspace.filespaces[url],
+               let suggestion = await filespace.presentingSuggestion
+            {
+                return .init(
+                    code: suggestion.text,
+                    language: await filespace.language,
+                    startLineIndex: suggestion.position.line,
+                    suggestionCount: await filespace.suggestions.count,
+                    currentSuggestionIndex: await filespace.suggestionIndex,
+                    onSelectPreviousSuggestionTapped: {
+                        Task { @ServiceActor in
+                            let handler = PseudoCommandHandler()
+                            await handler.presentPreviousSuggestion()
+                        }
+                    },
+                    onSelectNextSuggestionTapped: {
+                        Task { @ServiceActor in
+                            let handler = PseudoCommandHandler()
+                            await handler.presentNextSuggestion()
+                        }
+                    },
+                    onRejectSuggestionTapped: {
+                        Task { @ServiceActor in
+                            let handler = PseudoCommandHandler()
+                            await handler.rejectSuggestions()
+                        }
+                    },
+                    onAcceptSuggestionTapped: {
+                        Task { @ServiceActor in
+                            let handler = PseudoCommandHandler()
+                            await handler.acceptSuggestion()
+                        }
+                    }
+                )
+            }
+        }
+        return nil
+    }
+
+    func chatForFile(at url: URL) async -> ChatProvider? {
+        for workspace in await workspaces.values {
+            if let filespace = await workspace.filespaces[url],
+               let service = await filespace.chatService
+            {
+                return .init(
+                    service: service,
+                    fileURL: url,
+                    onCloseChat: { [weak filespace] in
+                        Task { @ServiceActor [weak filespace] in
+                            filespace?.chatService = nil
+                        }
+                    }
+                )
+            }
+        }
+        return nil
+    }
+
+    func chatServiceForFile(at url: URL) async -> ChatService? {
+        for workspace in await workspaces.values {
+            if let filespace = await workspace.filespaces[url] {
+                return await filespace.chatService
+            }
+        }
+        return nil
+    }
+}

Core/Sources/SuggestionWidget/SuggestionWidgetDataSource.swift

@@ -1,12 +1,12 @@
 import Foundation
 
 public protocol SuggestionWidgetDataSource {
-    func suggestionForFile(at path: URL) -> (SuggestionProvider)?
-    func chatForFile(at path: URL) -> ChatProvider?
+    func suggestionForFile(at url: URL) async -> (SuggestionProvider)?
+    func chatForFile(at url: URL) async -> ChatProvider?
 }
 
 struct MockWidgetDataSource: SuggestionWidgetDataSource {
-    func suggestionForFile(at path: URL) -> (SuggestionProvider)? {
+    func suggestionForFile(at url: URL) async -> (SuggestionProvider)? {
         return SuggestionProvider(
             code: """
             func test() {
@@ -22,7 +22,7 @@ struct MockWidgetDataSource: SuggestionWidgetDataSource {
         )
     }
 
-    func chatForFile(at path: URL) -> ChatProvider? {
+    func chatForFile(at url: URL) async -> ChatProvider? {
         return nil
     }
 }