Update widget to use data source

Author: intitni

Core/Sources/SuggestionWidget/SuggestionPanelContent/CodeBlockSuggestionPanel.swift

@@ -1,17 +1,19 @@
 import SwiftUI
 
 struct CodeBlock: View {
-    var suggestion: SuggestionPanelViewModel.Suggestion
+    @ObservedObject var suggestion: SuggestionProvider
+    @Environment(\.colorScheme) var colorScheme
 
     var body: some View {
         VStack {
-            ForEach(0..<suggestion.code.endIndex, id: \.self) { index in
+            let code = suggestion.highlightedCode(colorScheme: colorScheme)
+            ForEach(0..<code.endIndex, id: \.self) { index in
                 HStack(alignment: .firstTextBaseline) {
                     Text("\(index + suggestion.startLineIndex + 1)")
                         .multilineTextAlignment(.trailing)
                         .foregroundColor(.secondary)
                         .frame(minWidth: 40)
-                    Text(AttributedString(suggestion.code[index]))
+                    Text(AttributedString(code[index]))
                         .foregroundColor(.white.opacity(0.1))
                         .frame(maxWidth: .infinity, alignment: .leading)
                         .multilineTextAlignment(.leading)
@@ -26,17 +28,15 @@ struct CodeBlock: View {
 }
 
 struct CodeBlockSuggestionPanel: View {
-    @ObservedObject var viewModel: SuggestionPanelViewModel
-    var suggestion: SuggestionPanelViewModel.Suggestion
+    @ObservedObject var suggestion: SuggestionProvider
 
     struct ToolBar: View {
-        @ObservedObject var viewModel: SuggestionPanelViewModel
-        var suggestion: SuggestionPanelViewModel.Suggestion
+        @ObservedObject var suggestion: SuggestionProvider
 
         var body: some View {
             HStack {
                 Button(action: {
-                    viewModel.onPreviousButtonTapped?()
+                    suggestion.selectPreviousSuggestion()
                 }) {
                     Image(systemName: "chevron.left")
                 }.buttonStyle(.plain)
@@ -47,21 +47,21 @@ struct CodeBlockSuggestionPanel: View {
                 .monospacedDigit()
 
                 Button(action: {
-                    viewModel.onNextButtonTapped?()
+                    suggestion.selectNextSuggestion()
                 }) {
                     Image(systemName: "chevron.right")
                 }.buttonStyle(.plain)
 
                 Spacer()
 
                 Button(action: {
-                    viewModel.onRejectButtonTapped?()
+                    suggestion.rejectSuggestion()
                 }) {
                     Text("Reject")
                 }.buttonStyle(CommandButtonStyle(color: .gray))
 
                 Button(action: {
-                    viewModel.onAcceptButtonTapped?()
+                    suggestion.acceptSuggestion()
                 }) {
                     Text("Accept")
                 }.buttonStyle(CommandButtonStyle(color: .indigo))
@@ -80,8 +80,116 @@ struct CodeBlockSuggestionPanel: View {
             }
             .background(Color.contentBackground)
 
-            ToolBar(viewModel: viewModel, suggestion: suggestion)
+            ToolBar(suggestion: suggestion)
         }
         .xcodeStyleFrame()
     }
 }
+
+// MARK: - Previews
+
+struct CodeBlockSuggestionPanel_Dark_Preview: PreviewProvider {
+    static var previews: some View {
+        CodeBlockSuggestionPanel(suggestion: SuggestionProvider(
+            code: """
+            LazyVGrid(columns: [GridItem(.fixed(30)), GridItem(.flexible())]) {
+            ForEach(0..<viewModel.suggestion.count, id: \\.self) { index in // lkjaskldjalksjdlkasjdlkajslkdjas
+                Text(viewModel.suggestion[index])
+                    .frame(maxWidth: .infinity, alignment: .leading)
+                    .multilineTextAlignment(.leading)
+            }
+            """,
+            language: "swift",
+            startLineIndex: 8,
+            suggestionCount: 2,
+            currentSuggestionIndex: 0
+        ))
+        .preferredColorScheme(.dark)
+        .frame(width: 450, height: 400)
+        .background {
+            HStack {
+                Color.red
+                Color.green
+                Color.blue
+            }
+        }
+    }
+}
+
+struct CodeBlockSuggestionPanel_Bright_Preview: PreviewProvider {
+    static var previews: some View {
+        CodeBlockSuggestionPanel(suggestion: SuggestionProvider(
+            code: """
+            LazyVGrid(columns: [GridItem(.fixed(30)), GridItem(.flexible())]) {
+            ForEach(0..<viewModel.suggestion.count, id: \\.self) { index in // lkjaskldjalksjdlkasjdlkajslkdjas
+                Text(viewModel.suggestion[index])
+                    .frame(maxWidth: .infinity, alignment: .leading)
+                    .multilineTextAlignment(.leading)
+            }
+            """,
+            language: "swift",
+            startLineIndex: 8,
+            suggestionCount: 2,
+            currentSuggestionIndex: 0
+        ))
+        .preferredColorScheme(.light)
+        .frame(width: 450, height: 400)
+        .background {
+            HStack {
+                Color.red
+                Color.green
+                Color.blue
+            }
+        }
+    }
+}
+
+struct CodeBlockSuggestionPanel_Dark_Objc_Preview: PreviewProvider {
+    static var previews: some View {
+        CodeBlockSuggestionPanel(suggestion: SuggestionProvider(
+            code: """
+            - (void)addSubview:(UIView *)view {
+                [self addSubview:view];
+            }
+            """,
+            language: "objective-c",
+            startLineIndex: 8,
+            suggestionCount: 2,
+            currentSuggestionIndex: 0
+        ))
+        .preferredColorScheme(.dark)
+        .frame(width: 450, height: 400)
+        .background {
+            HStack {
+                Color.red
+                Color.green
+                Color.blue
+            }
+        }
+    }
+}
+
+struct CodeBlockSuggestionPanel_Bright_Objc_Preview: PreviewProvider {
+    static var previews: some View {
+        CodeBlockSuggestionPanel(suggestion:SuggestionProvider(
+            code: """
+            - (void)addSubview:(UIView *)view {
+                [self addSubview:view];
+            }
+            """,
+            language: "objective-c",
+            startLineIndex: 8,
+            suggestionCount: 2,
+            currentSuggestionIndex: 0
+        ))
+        .preferredColorScheme(.light)
+        .frame(width: 450, height: 400)
+        .background {
+            HStack {
+                Color.red
+                Color.green
+                Color.blue
+            }
+        }
+    }
+}