Update custom prompt to code to support templates

Author: intitni

Core/Sources/ChatService/CustomCommandTemplateProcessor.swift

@@ -3,8 +3,10 @@ import Foundation
 import SuggestionModel
 import XcodeInspector
 
-struct CustomCommandTemplateProcessor {
-    func process(_ text: String) -> String {
+public struct CustomCommandTemplateProcessor {
+    public init() {}
+    
+    public func process(_ text: String) -> String {
         let info = getEditorInformation()
         let editorContent = info.editorContent
         let updatedText = text

Core/Sources/Service/SuggestionCommandHandler/WindowBaseCommandHandler.swift

@@ -409,6 +409,10 @@ extension WindowBaseCommandHandler {
         }() as (String, CursorRange)
 
         let viewStore = Service.shared.guiController.viewStore
+        
+        let customCommandTemplateProcessor = CustomCommandTemplateProcessor()
+        let newExtraSystemPrompt = extraSystemPrompt.map(customCommandTemplateProcessor.process)
+        let newPrompt = prompt.map(customCommandTemplateProcessor.process)
 
         _ = await Task { @MainActor in
             // if there is already a prompt to code presenting, we should not present another one
@@ -423,8 +427,8 @@ extension WindowBaseCommandHandler {
                 allCode: editor.content,
                 isContinuous: isContinuous,
                 commandName: name,
-                defaultPrompt: prompt ?? "",
-                extraSystemPrompt: extraSystemPrompt,
+                defaultPrompt: newPrompt ?? "",
+                extraSystemPrompt: newExtraSystemPrompt,
                 generateDescriptionRequirement: generateDescription
             ))))
         }.result

README.md

@@ -308,15 +308,15 @@ You can create custom commands that run Chat and Prompt to Code with personalize
 - Custom Chat: Open the chat window and immediately send a message, if provided. You can overwrite the entire system prompt through the system prompt field.
 - Single Round Dialog: Send a message to a temporary chat. Useful when you want to run a terminal command with `/run`.
 
-For Send Message, Single Round Dialog and Custom Chat commands, you can use the following template arguments:
+You can use the following template arguments in custom commands:
 
 | Argument                      | Description                                    |
 | ----------------------------- | ---------------------------------------------- |
 | `{{selected_code}}`           | The currently selected code in the editor.     |
 | `{{active_editor_language}}`  | The programming language of the active editor. |
 | `{{active_editor_file_url}}`  | The URL of the active file in the editor.      |
 | `{{active_editor_file_name}}` | The name of the active file in the editor.     |
-| `{{clipboard}}`               | The content in clipboard                       |
+| `{{clipboard}}`               | The content in clipboard.                      |
 
 ## Plus Features