Disable text field when responding

Author: intitni

Core/Sources/SuggestionWidget/FeatureReducers/PromptToCode.swift

@@ -150,7 +150,6 @@ public struct PromptToCode: ReducerProtocol {
                 state.code = ""
                 state.description = ""
                 state.error = nil
-                state.prompt = ""
 
                 return .run { send in
                     do {
@@ -207,6 +206,7 @@ public struct PromptToCode: ReducerProtocol {
                 return .none
 
             case .modifyCodeFinished:
+                state.prompt = ""
                 state.isResponding = false
                 if state.code.isEmpty, state.description.isEmpty {
                     // if both code and description are empty, we treat it as failed

Core/Sources/SuggestionWidget/SuggestionPanelContent/PromptToCodePanel.swift

@@ -275,16 +275,6 @@ extension PromptToCodePanel {
                             .scaleEffect(x: 1, y: -1, anchor: .center)
                         }
                     }
-
-                    WithViewStore(store, observe: { $0.commandName }) { viewStore in
-                        if let name = viewStore.state {
-                            Text(name)
-                                .font(.footnote)
-                                .foregroundColor(.secondary)
-                                .padding(.top, 12)
-                                .scaleEffect(x: 1, y: -1, anchor: .center)
-                        }
-                    }
                 }
             }
             .scaleEffect(x: 1, y: -1, anchor: .center)
@@ -300,8 +290,9 @@ extension PromptToCodePanel {
             var canRevert: Bool
         }
 
-        struct Prompt: Equatable {
+        struct InputFieldState: Equatable {
             @BindingViewState var prompt: String
+            var isResponding: Bool
         }
 
         var body: some View {
@@ -366,7 +357,10 @@ extension PromptToCodePanel {
         }
 
         var inputField: some View {
-            WithViewStore(store, observe: { Prompt(prompt: $0.$prompt) }) { viewStore in
+            WithViewStore(
+                store,
+                observe: { InputFieldState(prompt: $0.$prompt, isResponding: $0.isResponding) }
+            ) { viewStore in
                 ZStack(alignment: .center) {
                     // a hack to support dynamic height of TextEditor
                     Text(viewStore.state.prompt.isEmpty ? "Hi" : viewStore.state.prompt)
@@ -380,10 +374,13 @@ extension PromptToCodePanel {
                     CustomTextEditor(
                         text: viewStore.$prompt,
                         font: .systemFont(ofSize: 14),
+                        isEditable: !viewStore.state.isResponding,
                         onSubmit: { viewStore.send(.modifyCodeButtonTapped) }
                     )
                     .padding(.top, 1)
                     .padding(.bottom, -1)
+                    .opacity(viewStore.state.isResponding ? 0.5 : 1)
+                    .disabled(viewStore.state.isResponding)
                 }
             }
             .focused($isInputAreaFocused)

Tool/Sources/SharedUIComponents/CustomTextEditor.swift

@@ -7,18 +7,21 @@ public struct CustomTextEditor: NSViewRepresentable {
 
     @Binding public var text: String
     public let font: NSFont
+    public let isEditable: Bool
     public let onSubmit: () -> Void
     public var completions: (_ text: String, _ words: [String], _ range: NSRange) -> [String]
 
     public init(
         text: Binding<String>,
         font: NSFont,
+        isEditable: Bool = true,
         onSubmit: @escaping () -> Void,
         completions: @escaping (_ text: String, _ words: [String], _ range: NSRange)
             -> [String] = { _, _, _ in [] }
     ) {
         _text = text
         self.font = font
+        self.isEditable = isEditable
         self.onSubmit = onSubmit
         self.completions = completions
     }
@@ -41,6 +44,7 @@ public struct CustomTextEditor: NSViewRepresentable {
     public func updateNSView(_ nsView: NSScrollView, context: Context) {
         context.coordinator.completions = completions
         let textView = (context.coordinator.theTextView.documentView as! NSTextView)
+        textView.isEditable = isEditable
         guard textView.string != text else { return }
         textView.string = text
         textView.undoManager?.removeAllActions()