Support using {{selected_code}} in custom chat command

Author: intitni

Core/Sources/Service/CustomCommandTemplateProcessor.swift

@@ -0,0 +1,37 @@
+import Foundation
+import SuggestionModel
+import XcodeInspector
+
+struct CustomCommandTemplateProcessor {
+    func process(_ text: String) -> String {
+        let info = getEditorInformation()
+        if let editorContent = info.editorContent {
+            let updatedText = text.replacingOccurrences(of: "{{selected_code}}", with: """
+            ```\(info.language.rawValue)
+            \(editorContent.selectedContent.trimmingCharacters(in: ["\n"]))
+            ```
+            """)
+            return updatedText
+        } else {
+            let updatedText = text.replacingOccurrences(of: "{{selected_code}}", with: "")
+            return updatedText
+        }
+    }
+
+    struct EditorInformation {
+        let editorContent: SourceEditor.Content?
+        let language: CodeLanguage
+    }
+
+    func getEditorInformation() -> EditorInformation {
+        let editorContent = XcodeInspector.shared.focusedEditor?.content
+        let documentURL = XcodeInspector.shared.activeDocumentURL
+        let language = languageIdentifierFromFileURL(documentURL)
+
+        return .init(
+            editorContent: editorContent,
+            language: language
+        )
+    }
+}
+

Core/Sources/Service/SuggestionCommandHandler/WindowBaseCommandHandler.swift

@@ -1,12 +1,12 @@
 import ChatService
-import SuggestionModel
-import GitHubCopilotService
 import Environment
 import Foundation
+import GitHubCopilotService
 import LanguageServerProtocol
 import Logger
 import OpenAIService
 import SuggestionInjector
+import SuggestionModel
 import SuggestionWidget
 import XPCShared
 
@@ -53,7 +53,7 @@ struct WindowBaseCommandHandler: SuggestionCommandHandler {
             forFileAt: fileURL,
             editor: editor
         )
-        
+
         try Task.checkCancellation()
 
         if filespace.presentingSuggestion != nil {
@@ -388,7 +388,7 @@ extension WindowBaseCommandHandler {
 
         presenter.presentPromptToCode(fileURL: fileURL)
     }
-    
+
     private func startChat(
         specifiedSystemPrompt: String?,
         extraSystemPrompt: String?,
@@ -402,15 +402,16 @@ extension WindowBaseCommandHandler {
 
         let chat = WidgetDataSource.shared.createChatIfNeeded(for: focusedElementURI)
 
-        chat.mutateSystemPrompt(specifiedSystemPrompt)
-        chat.mutateExtraSystemPrompt(extraSystemPrompt ?? "")
+        let templateProcessor = CustomCommandTemplateProcessor()
+        chat.mutateSystemPrompt(specifiedSystemPrompt.map(templateProcessor.process))
+        chat.mutateExtraSystemPrompt(extraSystemPrompt.map(templateProcessor.process) ?? "")
 
         Task {
             let customCommandPrefix = {
                 if let name { return "[\(name)] " }
                 return ""
             }()
-            
+
             if specifiedSystemPrompt != nil || extraSystemPrompt != nil {
                 await chat.chatGPTService.mutateHistory { history in
                     history.append(.init(
@@ -422,10 +423,12 @@ extension WindowBaseCommandHandler {
             }
 
             if let sendingMessageImmediately, !sendingMessageImmediately.isEmpty {
-                try await chat.send(content: sendingMessageImmediately)
+                try await chat
+                    .send(content: templateProcessor.process(sendingMessageImmediately))
             }
         }
 
         presenter.presentChatRoom(fileURL: focusedElementURI)
     }
 }
+

Core/Sources/XcodeInspector/SourceEditor.swift

@@ -16,6 +16,22 @@ public class SourceEditor {
         public var cursorPosition: CursorPosition
         /// Line annotations of the source editor.
         public var lineAnnotations: [String]
+        
+        public var selectedContent: String {
+            if let range = selections.first {
+                let startIndex = min(
+                    max(0, range.start.line),
+                    lines.endIndex - 1
+                )
+                let endIndex = min(
+                    max(startIndex, range.end.line),
+                    lines.endIndex - 1
+                )
+                let selectedContent = lines[startIndex...endIndex]
+                return selectedContent.joined()
+            }
+            return ""
+        }
     }
 
     let runningApplication: NSRunningApplication