Add attach to code to prompt to code service

Author: intitni

Core/Package.swift

@@ -157,6 +157,7 @@ let package = Package(
         .target(
             name: "PromptToCodeService",
             dependencies: [
+                .product(name: "FocusedCodeFinder", package: "Tool"),
                 .product(name: "SuggestionModel", package: "Tool"),
                 .product(name: "Environment", package: "Tool"),
                 .product(name: "OpenAIService", package: "Tool"),

Core/Sources/PromptToCodeService/PromptToCodeService.swift

@@ -1,14 +1,16 @@
-import SuggestionModel
+import FocusedCodeFinder
 import Foundation
 import OpenAIService
+import SuggestionModel
+import XcodeInspector
 
 public final class PromptToCodeService: ObservableObject {
     var designatedPromptToCodeAPI: PromptToCodeAPI?
     var promptToCodeAPI: PromptToCodeAPI {
         if let designatedPromptToCodeAPI {
             return designatedPromptToCodeAPI
         }
-        
+
         return OpenAIPromptToCodeAPI()
     }
 
@@ -70,7 +72,7 @@ public final class PromptToCodeService: ObservableObject {
         self.projectRootURL = projectRootURL
         self.fileURL = fileURL
         self.allCode = allCode
-        self.history = .empty
+        history = .empty
         self.extraSystemPrompt = extraSystemPrompt
         self.generateDescriptionRequirement = generateDescriptionRequirement
     }
@@ -126,6 +128,48 @@ public final class PromptToCodeService: ObservableObject {
         runningAPI?.stopResponding()
         isResponding = false
     }
+
+    public func toggleAttachOrDetachToCode() {
+        if selectionRange != nil {
+            selectionRange = nil
+        } else {
+            let inspector = XcodeInspector.shared
+            guard let editor = XcodeInspector.shared.focusedEditorContent
+            else { return }
+            if let range = editor.editorContent?.selections.first, range.start != range.end {
+                selectionRange = range
+            } else {
+                let focusedCodeFinder: FocusedCodeFinder = {
+                    switch language {
+                    case .builtIn(.swift):
+                        return SwiftFocusedCodeFinder()
+                    default:
+                        return UnknownLanguageFocusedCodeFinder(proposedSearchRange: 10)
+                    }
+                }()
+
+                let cursorPosition = editor.editorContent?.cursorPosition ?? .zero
+                let codeContext = focusedCodeFinder.findFocusedCode(
+                    containingRange: .init(start: cursorPosition, end: cursorPosition),
+                    activeDocumentContext: .init(
+                        filePath: editor.documentURL.path,
+                        relativePath: editor.relativePath,
+                        language: editor.language,
+                        fileContent: editor.editorContent?.content ?? "",
+                        lines: editor.editorContent?.lines ?? [],
+                        selectedCode: editor.editorContent?.selectedContent ?? "",
+                        selectionRange: editor.editorContent?.selections.first ?? .zero,
+                        lineAnnotations: [],
+                        imports: []
+                    )
+                )
+                
+                fileURL = editor.documentURL
+                projectRootURL = editor.projectURL
+                selectionRange = codeContext.contextRange
+            }
+        }
+    }
 }
 
 protocol PromptToCodeAPI {
@@ -144,3 +188,4 @@ protocol PromptToCodeAPI {
 
     func stopResponding()
 }
+