Display the command name for custom prompt to code command

Author: intitni

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

@@ -3,13 +3,18 @@ import PromptToCodeService
 import SuggestionWidget
 
 extension PromptToCodeProvider {
-    convenience init(service: PromptToCodeService, onClosePromptToCode: @escaping () -> Void) {
+    convenience init(
+        service: PromptToCodeService,
+        name: String?,
+        onClosePromptToCode: @escaping () -> Void
+    ) {
         self.init(
             code: service.code,
             language: service.language.rawValue,
             description: "",
             startLineIndex: service.selectionRange.start.line,
-            startLineColumn: service.selectionRange.start.character
+            startLineColumn: service.selectionRange.start.character,
+            name: name
         )
 
         var cancellables = Set<AnyCancellable>()
@@ -44,18 +49,18 @@ extension PromptToCodeProvider {
                 }
             }
         }
-        
+
         onStopRespondingTap = {
             service.stopResponding()
         }
-        
+
         onAcceptSuggestionTapped = {
             Task { @ServiceActor in
                 let handler = PseudoCommandHandler()
                 await handler.acceptSuggestion()
             }
         }
-        
+
         onContinuousToggleClick = {
             service.isContinuous.toggle()
         }

Core/Sources/Service/GUI/WidgetDataSource.swift

@@ -109,6 +109,7 @@ final class WidgetDataSource {
             )
             let provider = PromptToCodeProvider(
                 service: service,
+                name: name,
                 onClosePromptToCode: { [weak self] in
                     self?.removePromptToCode(for: url)
                     let presenter = PresentInWindowSuggestionPresenter()

Core/Sources/Service/SuggestionCommandHandler/WindowBaseCommandHandler.swift

@@ -252,7 +252,8 @@ struct WindowBaseCommandHandler: SuggestionCommandHandler {
                     editor: editor,
                     extraSystemPrompt: nil,
                     prompt: nil,
-                    isContinuous: false
+                    isContinuous: false,
+                    name: nil
                 )
             } catch {
                 presenter.presentError(error)
@@ -303,7 +304,8 @@ extension WindowBaseCommandHandler {
                 editor: editor,
                 extraSystemPrompt: extraSystemPrompt,
                 prompt: prompt,
-                isContinuous: continuousMode ?? false
+                isContinuous: continuousMode ?? false,
+                name: command.name
             )
         }
     }
@@ -312,7 +314,8 @@ extension WindowBaseCommandHandler {
         editor: EditorContent,
         extraSystemPrompt: String?,
         prompt: String?,
-        isContinuous: Bool
+        isContinuous: Bool,
+        name: String?
     ) async throws {
         presenter.markAsProcessing(true)
         defer { presenter.markAsProcessing(false) }
@@ -368,7 +371,7 @@ extension WindowBaseCommandHandler {
             selectionRange: selection,
             language: codeLanguage,
             extraSystemPrompt: extraSystemPrompt,
-            name: nil
+            name: name
         )
 
         promptToCode.isContinuous = isContinuous

Core/Sources/SuggestionWidget/PromptToCodeProvider.swift

@@ -3,6 +3,7 @@ import SwiftUI
 
 public final class PromptToCodeProvider: ObservableObject {
     let id = UUID()
+    let name: String?
 
     @Published public var code: String
     @Published public var language: String
@@ -33,6 +34,7 @@ public final class PromptToCodeProvider: ObservableObject {
         errorMessage: String = "",
         canRevert: Bool = false,
         isContinuous: Bool = false,
+        name: String? = nil,
         onRevertTapped: @escaping () -> Void = {},
         onStopRespondingTap: @escaping () -> Void = {},
         onCancelTapped: @escaping () -> Void = {},
@@ -50,6 +52,7 @@ public final class PromptToCodeProvider: ObservableObject {
         self.errorMessage = errorMessage
         self.canRevert = canRevert
         self.isContinuous = isContinuous
+        self.name = name
         self.onRevertTapped = onRevertTapped
         self.onStopRespondingTap = onStopRespondingTap
         self.onCancelTapped = onCancelTapped

Core/Sources/SuggestionWidget/SuggestionPanelContent/PromptToCodePanel.swift

@@ -109,7 +109,7 @@ struct PromptToCodePanelContent: View {
                         .padding(.vertical, 2)
                         .background(
                             Color.red,
-                            in: RoundedRectangle(cornerRadius: 16, style: .continuous)
+                            in: RoundedRectangle(cornerRadius: 8, style: .continuous)
                         )
                         .scaleEffect(x: -1, y: -1, anchor: .center)
                 }
@@ -147,6 +147,14 @@ struct PromptToCodePanelContent: View {
                     .frame(maxWidth: .infinity)
                     .scaleEffect(x: -1, y: -1, anchor: .center)
                 }
+                
+                if let name = provider.name {
+                    Text(name)
+                        .font(.footnote)
+                        .foregroundColor(.secondary)
+                        .padding(.top, 12)
+                        .scaleEffect(x: -1, y: -1, anchor: .center)
+                }
             }
             .scaleEffect(x: -1, y: 1, anchor: .center)
         }