Add comtext picker for custom command

Author: intitni

Core/Sources/HostApp/CustomCommandSettings/CustomCommandView.swift

@@ -275,7 +275,9 @@ struct CustomCommandView_Preview: PreviewProvider {
                     extraSystemPrompt: nil,
                     prompt: "Hello",
                     useExtraSystemPrompt: false
-                )
+                ),
+                ignoreExistingAttachments: false,
+                attachments: []
             ),
             .init(
                 commandId: "2",
@@ -285,7 +287,9 @@ struct CustomCommandView_Preview: PreviewProvider {
                     prompt: "Refactor",
                     continuousMode: false,
                     generateDescription: true
-                )
+                ),
+                ignoreExistingAttachments: false,
+                attachments: []
             ),
         ], "CustomCommandView_Preview"))
 
@@ -299,7 +303,9 @@ struct CustomCommandView_Preview: PreviewProvider {
                             extraSystemPrompt: nil,
                             prompt: "Hello",
                             useExtraSystemPrompt: false
-                        )
+                        ),
+                        ignoreExistingAttachments: false,
+                        attachments: [] as [CustomCommand.Attachment]
                     )))
                 ),
                 reducer: { CustomCommandFeature(settings: settings) }
@@ -319,7 +325,9 @@ struct CustomCommandView_NoEditing_Preview: PreviewProvider {
                     extraSystemPrompt: nil,
                     prompt: "Hello",
                     useExtraSystemPrompt: false
-                )
+                ),
+                ignoreExistingAttachments: false,
+                attachments: []
             ),
             .init(
                 commandId: "2",
@@ -329,7 +337,9 @@ struct CustomCommandView_NoEditing_Preview: PreviewProvider {
                     prompt: "Refactor",
                     continuousMode: false,
                     generateDescription: true
-                )
+                ),
+                ignoreExistingAttachments: false,
+                attachments: []
             ),
         ], "CustomCommandView_Preview"))
 

Core/Sources/HostApp/CustomCommandSettings/EditCustomCommand.swift

@@ -23,11 +23,16 @@ struct EditCustomCommand {
         var promptToCode = EditPromptToCodeCommand.State()
         var customChat = EditCustomChatCommand.State()
         var singleRoundDialog = EditSingleRoundDialogCommand.State()
+        var attachments = EditCustomCommandAttachment.State()
 
         init(_ command: CustomCommand?) {
             isNewCommand = command == nil
             commandId = command?.id ?? UUID().uuidString
             name = command?.name ?? "New Command"
+            attachments = .init(
+                attachments: command?.attachments ?? [],
+                ignoreExistingAttachments: command?.ignoreExistingAttachments ?? false
+            )
 
             switch command?.feature {
             case let .chatWithSelection(extraSystemPrompt, prompt, useExtraSystemPrompt):
@@ -83,6 +88,7 @@ struct EditCustomCommand {
         case promptToCode(EditPromptToCodeCommand.Action)
         case customChat(EditCustomChatCommand.Action)
         case singleRoundDialog(EditSingleRoundDialogCommand.Action)
+        case attachments(EditCustomCommandAttachment.Action)
     }
 
     let settings: CustomCommandView.Settings
@@ -106,6 +112,10 @@ struct EditCustomCommand {
             EditSingleRoundDialogCommand()
         }
 
+        Scope(state: \.attachments, action: \.attachments) {
+            EditCustomCommandAttachment()
+        }
+
         BindingReducer()
 
         Reduce { state, action in
@@ -151,7 +161,9 @@ struct EditCustomCommand {
                                 receiveReplyInNotification: state.receiveReplyInNotification
                             )
                         }
-                    }()
+                    }(),
+                    ignoreExistingAttachments: state.attachments.ignoreExistingAttachments,
+                    attachments: state.attachments.attachments
                 )
 
                 if state.isNewCommand {
@@ -184,6 +196,32 @@ struct EditCustomCommand {
                 return .none
             case .singleRoundDialog:
                 return .none
+            case .attachments:
+                return .none
+            }
+        }
+    }
+}
+
+@Reducer
+struct EditCustomCommandAttachment {
+    @ObservableState
+    struct State: Equatable {
+        var attachments: [CustomCommand.Attachment] = []
+        var ignoreExistingAttachments: Bool = false
+    }
+
+    enum Action: BindableAction, Equatable {
+        case binding(BindingAction<State>)
+    }
+
+    var body: some ReducerOf<Self> {
+        BindingReducer()
+
+        Reduce { _, action in
+            switch action {
+            case .binding:
+                return .none
             }
         }
     }