Add top bar to prompt to code panel

Author: intitni

Core/Sources/Service/GUI/PromptToCodeProvider+Service.swift

@@ -13,8 +13,7 @@ extension PromptToCodeProvider {
             code: service.code,
             language: service.language.rawValue,
             description: "",
-            startLineIndex: service.selectionRange.start.line,
-            startLineColumn: service.selectionRange.start.character,
+            attachedToRange: service.selectionRange,
             name: name
         )
 

Core/Sources/SuggestionWidget/Providers/PromptToCodeProvider.swift

@@ -1,16 +1,18 @@
 import Foundation
+import SuggestionModel
 import SwiftUI
 
 public final class PromptToCodeProvider: ObservableObject {
     let id = UUID()
     let name: String?
-    
+
     @Published public var code: String
     @Published public var language: String
     @Published public var description: String
     @Published public var isResponding: Bool
-    @Published public var startLineIndex: Int
-    @Published public var startLineColumn: Int
+    public var startLineIndex: Int { attachedToRange?.start.line ?? 0 }
+    public var startLineColumn: Int { attachedToRange?.start.character ?? 0 }
+    @Published public var attachedToRange: CursorRange?
     @Published public var requirement: String
     @Published public var errorMessage: String
     @Published public var canRevert: Bool
@@ -28,8 +30,7 @@ public final class PromptToCodeProvider: ObservableObject {
         language: String = "",
         description: String = "",
         isResponding: Bool = false,
-        startLineIndex: Int = 0,
-        startLineColumn: Int = 0,
+        attachedToRange: CursorRange? = nil,
         requirement: String = "",
         errorMessage: String = "",
         canRevert: Bool = false,
@@ -46,8 +47,7 @@ public final class PromptToCodeProvider: ObservableObject {
         self.language = language
         self.description = description
         self.isResponding = isResponding
-        self.startLineIndex = startLineIndex
-        self.startLineColumn = startLineColumn
+        self.attachedToRange = attachedToRange
         self.requirement = requirement
         self.errorMessage = errorMessage
         self.canRevert = canRevert
@@ -65,11 +65,14 @@ public final class PromptToCodeProvider: ObservableObject {
         onRevertTapped()
         errorMessage = ""
     }
+
     func stopResponding() {
         onStopRespondingTap()
         errorMessage = ""
     }
+
     func cancel() { onCancelTapped() }
+    
     func sendRequirement() {
         guard !isResponding else { return }
         guard !requirement.isEmpty else { return }
@@ -79,6 +82,7 @@ public final class PromptToCodeProvider: ObservableObject {
     }
 
     func acceptSuggestion() { onAcceptSuggestionTapped() }
-    
+
     func toggleContinuous() { onContinuousToggleClick() }
 }
+

Core/Sources/SuggestionWidget/SharedPanelView.swift

@@ -97,20 +97,21 @@ struct SharedPanelView: View {
 }
 
 struct CommandButtonStyle: ButtonStyle {
-    let color: Color
+    var color: Color
+    var cornerRadius: Double = 4
 
     func makeBody(configuration: Configuration) -> some View {
         configuration.label
             .padding(.vertical, 4)
             .padding(.horizontal, 8)
             .foregroundColor(.white)
             .background(
-                RoundedRectangle(cornerRadius: 4, style: .continuous)
+                RoundedRectangle(cornerRadius: cornerRadius, style: .continuous)
                     .fill(color.opacity(configuration.isPressed ? 0.8 : 1))
                     .animation(.easeOut(duration: 0.1), value: configuration.isPressed)
             )
             .overlay {
-                RoundedRectangle(cornerRadius: 4, style: .continuous)
+                RoundedRectangle(cornerRadius: cornerRadius, style: .continuous)
                     .stroke(Color.white.opacity(0.2), style: .init(lineWidth: 1))
             }
     }