Add ChatPanel

Author: intitni

Core/Sources/SuggestionWidget/Styles.swift

@@ -1,7 +1,40 @@
+import AppKit
+import SwiftUI
+
 enum Style {
     static let panelHeight: Double = 300
     static let panelWidth: Double = 450
     static let widgetHeight: Double = 30
     static var widgetWidth: Double { widgetHeight }
     static let widgetPadding: Double = 4
 }
+
+extension Color {
+    static var contentBackground: Color {
+        Color(nsColor: NSColor(name: nil, dynamicProvider: { appearance in
+            if appearance.isDarkMode {
+                return #colorLiteral(red: 0.1580096483, green: 0.1730263829, blue: 0.2026666105, alpha: 1)
+            }
+            return .white
+        }))
+    }
+    
+    static var userChatContentBackground: Color {
+        Color(nsColor: NSColor(name: nil, dynamicProvider: { appearance in
+            if appearance.isDarkMode {
+                return #colorLiteral(red: 0.1777971793, green: 0.1670255649, blue: 0.2501594388, alpha: 1)
+            }
+            return #colorLiteral(red: 0.896820749, green: 0.8709097223, blue: 0.9766687925, alpha: 1)
+        }))
+    }
+}
+
+extension NSAppearance {
+    var isDarkMode: Bool {
+        if self.bestMatch(from: [.darkAqua, .aqua]) == .darkAqua {
+            return true
+        } else {
+            return false
+        }
+    }
+}

Core/Sources/SuggestionWidget/SuggestionPanelContent/ChatPanel.swift

@@ -0,0 +1,118 @@
+import MarkdownUI
+import SwiftUI
+
+struct ChatPanel: View {
+    @ObservedObject var viewModel: SuggestionPanelViewModel
+    var chat: SuggestionPanelViewModel.Chat
+
+    var body: some View {
+        ZStack(alignment: .topTrailing) {
+            ScrollView {
+                LazyVStack {
+                    ForEach(chat.history.reversed(), id: \.id) { message in
+                        Markdown(message.text)
+                            .markdownTheme(.gitHub.text {
+                                BackgroundColor(Color.clear)
+                            })
+                            .frame(maxWidth: .infinity, alignment: .leading)
+                            .padding()
+                            .background(
+                                RoundedRectangle(cornerRadius: 9, style: .continuous)
+                                    .fill(
+                                        message.isUser
+                                            ? Color.userChatContentBackground
+                                            : Color.contentBackground
+                                    )
+                            )
+                            .rotationEffect(Angle(degrees: 180))
+                    }
+                }
+            }
+            .rotationEffect(Angle(degrees: 180))
+
+            // close button
+            Button(action: {
+                viewModel.isPanelDisplayed = false
+                viewModel.content = .empty
+            }) {
+                Image(systemName: "xmark")
+                    .padding([.leading, .bottom], 16)
+                    .padding([.top, .trailing], 8)
+                    .foregroundColor(.white)
+            }
+            .buttonStyle(.plain)
+        }
+        .colorScheme(viewModel.colorScheme)
+    }
+}
+
+struct ChatPanel_Preview: PreviewProvider {
+    static var previews: some View {
+        ChatPanel(viewModel: .init(
+            content: .empty,
+            isPanelDisplayed: true
+        ), chat: .init(
+            history: [
+                .init(
+                    id: "1",
+                    isUser: true,
+                    text: "**Hello**"
+                ),
+                .init(id: "2", isUser: false, text: "**Hey**! What can I do for you?"),
+                .init(
+                    id: "3",
+                    isUser: true,
+                    text: #"""
+                    Please buy me a coffee!
+                    | Coffee | Milk |
+                    |--------|------|
+                    | Espresso | No |
+                    | Latte | Yes |
+
+                    ```swift
+                    func foo() {}
+                    ```
+                    """#
+                ),
+            ],
+            isReceivingMessage: false
+        ))
+        .frame(width: 450, height: 500)
+    }
+}
+
+struct ChatPanel_Light_Preview: PreviewProvider {
+    static var previews: some View {
+        ChatPanel(viewModel: .init(
+            content: .empty,
+            isPanelDisplayed: true,
+            colorScheme: .light
+        ), chat: .init(
+            history: [
+                .init(
+                    id: "1",
+                    isUser: true,
+                    text: "**Hello**"
+                ),
+                .init(id: "2", isUser: false, text: "**Hey**! What can I do for you?"),
+                .init(
+                    id: "3",
+                    isUser: true,
+                    text: #"""
+                    Please buy me a coffee!
+                    | Coffee | Milk |
+                    |--------|------|
+                    | Espresso | No |
+                    | Latte | Yes |
+
+                    ```swift
+                    func foo() {}
+                    ```
+                    """#
+                ),
+            ],
+            isReceivingMessage: false
+        ))
+        .frame(width: 450, height: 500)
+    }
+}

Core/Sources/SuggestionWidget/SuggestionPanelContent/CodeBlockSuggestionPanel.swift

@@ -78,16 +78,7 @@ struct CodeBlockSuggestionPanel: View {
                 CodeBlock(suggestion: suggestion)
                     .frame(maxWidth: .infinity)
             }
-            .background(Color(nsColor: {
-                switch viewModel.colorScheme {
-                case .dark:
-                    return #colorLiteral(red: 0.1580096483, green: 0.1730263829, blue: 0.2026666105, alpha: 1)
-                case .light:
-                    return .white
-                @unknown default:
-                    return .white
-                }
-            }()))
+            .background(Color.contentBackground)
 
             ToolBar(viewModel: viewModel, suggestion: suggestion)
         }

Core/Sources/SuggestionWidget/SuggestionPanelContent/ErrorPanel.swift

@@ -1,7 +1,7 @@
 import SwiftUI
 
 struct ErrorPanel: View {
-    @ObservedObject var viewModel: SuggestionPanelViewModel
+    var viewModel: SuggestionPanelViewModel
     var description: String
 
     var body: some View {

Core/Sources/SuggestionWidget/SuggestionPanelView.swift

@@ -16,9 +16,21 @@ final class SuggestionPanelViewModel: ObservableObject {
         )
     }
 
+    struct Chat: Equatable {
+        struct ChatMessage: Equatable {
+            var id: String
+            var isUser: Bool
+            var text: String
+        }
+
+        var history: [ChatMessage]
+        var isReceivingMessage: Bool
+    }
+
     enum Content: Equatable {
         case empty
         case suggestion(Suggestion)
+        case chat(Chat)
         case error(String)
     }
 
@@ -70,6 +82,8 @@ struct SuggestionPanelView: View {
                     CodeBlockSuggestionPanel(viewModel: viewModel, suggestion: suggestion)
                 case let .error(description):
                     ErrorPanel(viewModel: viewModel, description: description)
+                case let .chat(chat):
+                    ChatPanel(viewModel: viewModel, chat: chat)
                 }
             }
             .frame(maxWidth: .infinity, maxHeight: Style.panelHeight)