Support reading the whole file with @file

Author: intitni

Core/Sources/ChatContextCollector/ActiveDocumentChatContextCollector.swift

@@ -6,12 +6,20 @@ import XcodeInspector
 public struct ActiveDocumentChatContextCollector: ChatContextCollector {
     public init() {}
 
-    public func generateSystemPrompt(oldMessages: [String]) -> String {
+    public func generateSystemPrompt(history: [String], content prompt: String) -> String {
         let content = getEditorInformation()
         let relativePath = content.documentURL.path
             .replacingOccurrences(of: content.projectURL.path, with: "")
         let selectionRange = content.editorContent?.selections.first ?? .outOfScope
         let editorContent = {
+            if prompt.hasPrefix("@file") {
+                return """
+                File Content:```\(content.language.rawValue)
+                \(content.editorContent?.content ?? "")
+                ```
+                """
+            }
+            
             if selectionRange.start == selectionRange.end,
                UserDefaults.shared.value(for: \.embedFileContentInChatContextIfNoSelection)
             {

Core/Sources/ChatContextCollector/ChatContextCollector.swift

@@ -1,5 +1,5 @@
 import Foundation
 
 public protocol ChatContextCollector {
-    func generateSystemPrompt(oldMessages: [String]) -> String
+    func generateSystemPrompt(history: [String], content: String) -> String
 }

Core/Sources/ChatService/ChatService.swift

@@ -47,7 +47,7 @@ public final class ChatService: ObservableObject {
         try await contextController.updatePromptToMatchContent(systemPrompt: """
         \(systemPrompt)
         \(extraSystemPrompt)
-        """)
+        """, content: content)
 
         _ = try await chatGPTService.send(content: content, summary: nil)
     }

Core/Sources/ChatService/DynamicContextController.swift

@@ -13,7 +13,7 @@ final class DynamicContextController {
         self.contextCollectors = contextCollectors
     }
 
-    func updatePromptToMatchContent(systemPrompt: String) async throws {
+    func updatePromptToMatchContent(systemPrompt: String, content: String) async throws {
         let language = UserDefaults.shared.value(for: \.chatGPTLanguage)
         let oldMessages = (await chatGPTService.history).map(\.content)
         let contextualSystemPrompt = """
@@ -22,7 +22,7 @@ final class DynamicContextController {
 
         \(
             contextCollectors
-                .map { $0.generateSystemPrompt(oldMessages: oldMessages) }
+                .map { $0.generateSystemPrompt(history: oldMessages, content: content) }
                 .joined(separator: "\n")
         )
         """