Add a open chat button to widget context menu

Author: intitni

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

@@ -10,6 +10,14 @@ public final class GraphicalUserInterfaceController {
     private nonisolated init() {
         Task { @MainActor in
             suggestionWidget.dataSource = WidgetDataSource.shared
+            suggestionWidget.onOpenChatClicked = {
+                Task {
+                    let fileURL = try await Environment.fetchCurrentFileURL()
+                    await WidgetDataSource.shared.createChatIfNeeded(for: fileURL)
+                    let presenter = PresentInWindowSuggestionPresenter()
+                    presenter.presentChatRoom(fileURL: fileURL)
+                }
+            }
         }
     }
 }

Core/Sources/SuggestionWidget/SuggestionPanelContent/PromptToCodePanel.swift

@@ -9,6 +9,15 @@ struct PromptToCodePanel: View {
         VStack(spacing: 0) {
             ScrollView {
                 VStack(spacing: 0) {
+                    if !provider.errorMessage.isEmpty {
+                        Text(provider.errorMessage)
+                            .multilineTextAlignment(.leading)
+                            .frame(maxWidth: .infinity, alignment: .leading)
+                            .foregroundColor(.white)
+                            .padding()
+                            .background(Color.red)
+                    }
+                    
                     if provider.code.isEmpty {
                         Text(
                             provider.isResponding
@@ -40,15 +49,6 @@ struct PromptToCodePanel: View {
                             .frame(maxWidth: .infinity)
                     }
 
-                    if !provider.errorMessage.isEmpty {
-                        Text(provider.errorMessage)
-                            .multilineTextAlignment(.leading)
-                            .frame(maxWidth: .infinity, alignment: .leading)
-                            .foregroundColor(.white)
-                            .padding()
-                            .background(Color.red)
-                    }
-
                     Spacer(minLength: 50)
                 }
             }
@@ -213,3 +213,24 @@ struct PromptToCodePanel_Bright_Preview: PreviewProvider {
         .frame(width: 450, height: 400)
     }
 }
+
+struct PromptToCodePanel_Error_Bright_Preview: PreviewProvider {
+    static var previews: some View {
+        PromptToCodePanel(provider: PromptToCodeProvider(
+            code: """
+            ForEach(0..<viewModel.suggestion.count, id: \\.self) { index in
+                Text(viewModel.suggestion[index])
+                    .frame(maxWidth: .infinity, alignment: .leading)
+                    .multilineTextAlignment(.leading)
+            }
+            """,
+            language: "swift",
+            description: "Hello world",
+            isResponding: false,
+            startLineIndex: 8,
+            errorMessage: "Error"
+        ))
+        .preferredColorScheme(.light)
+        .frame(width: 450, height: 400)
+    }
+}

Core/Sources/SuggestionWidget/SuggestionWidgetController.swift

@@ -36,7 +36,10 @@ public final class SuggestionWidgetController {
         it.contentView = NSHostingView(
             rootView: WidgetView(
                 viewModel: widgetViewModel,
-                panelViewModel: suggestionPanelViewModel
+                panelViewModel: suggestionPanelViewModel,
+                onOpenChatClicked: { [weak self] in
+                    self?.onOpenChatClicked()
+                }
             )
         )
         it.setIsVisible(true)
@@ -96,7 +99,8 @@ public final class SuggestionWidgetController {
     private var sourceEditorMonitorTask: Task<Void, Error>?
     private var currentFileURL: URL?
     private var colorScheme: ColorScheme = .light
-
+    
+    public var onOpenChatClicked: () -> Void = {}
     public var dataSource: SuggestionWidgetDataSource?
 
     public nonisolated init() {

Core/Sources/SuggestionWidget/WidgetView.swift

@@ -16,6 +16,7 @@ struct WidgetView: View {
     @ObservedObject var panelViewModel: SuggestionPanelViewModel
     @State var isHovering: Bool = false
     @State var processingProgress: Double = 0
+    var onOpenChatClicked: () -> Void = {}
 
     var body: some View {
         Circle().fill(isHovering ? .white.opacity(0.8) : .white.opacity(0.3))
@@ -74,7 +75,11 @@ struct WidgetView: View {
                     isHovering = yes
                 }
             }.contextMenu {
-                WidgetContextMenu(widgetViewModel: viewModel)
+                WidgetContextMenu(
+                    widgetViewModel: viewModel,
+                    isChatOpen: panelViewModel.isPanelDisplayed && panelViewModel.chat != nil,
+                    onOpenChatClicked: onOpenChatClicked
+                )
             }
     }
 
@@ -101,9 +106,21 @@ struct WidgetContextMenu: View {
     @AppStorage(\.suggestionFeatureEnabledProjectList) var suggestionFeatureEnabledProjectList
     @ObservedObject var widgetViewModel: WidgetViewModel
     @State var projectPath: String?
+    var isChatOpen: Bool
+    var onOpenChatClicked: () -> Void = {}
 
     var body: some View {
         Group {
+            if !isChatOpen {
+                Button(action: {
+                    onOpenChatClicked()
+                }) {
+                    Text("Open Chat")
+                }
+            }
+
+            Divider()
+
             Button(action: {
                 useGlobalChat.toggle()
             }) {
@@ -184,7 +201,7 @@ struct WidgetContextMenu: View {
             updateProjectPath(fileURL: fileURL)
         }
     }
-    
+
     func updateProjectPath(fileURL: URL?) {
         Task {
             let projectURL = try? await Environment.fetchCurrentProjectRootURL(fileURL)