PromptToCodeAgent.swift

import ComposableArchitecture
import Foundation
import SuggestionBasic

public enum PromptToCodeAgentResponse {
    case code(String)
    case description(String)
}

public struct PromptToCodeAgentRequest {
    var code: String
    var requirement: String
    var source: PromptToCodeSource
    var isDetached: Bool
    var extraSystemPrompt: String?
    var generateDescriptionRequirement: Bool?

    public struct PromptToCodeSource {
        public var language: CodeLanguage
        public var documentURL: URL
        public var projectRootURL: URL
        public var content: String
        public var lines: [String]
        public var range: CursorRange

        public init(
            language: CodeLanguage,
            documentURL: URL,
            projectRootURL: URL,
            content: String,
            lines: [String],
            range: CursorRange
        ) {
            self.language = language
            self.documentURL = documentURL
            self.projectRootURL = projectRootURL
            self.content = content
            self.lines = lines
            self.range = range
        }
    }
}

public protocol PromptToCodeAgent {
    typealias Request = PromptToCodeAgentRequest
    typealias Response = PromptToCodeAgentResponse

    func send(_ request: Request) -> AsyncThrowingStream<Response, any Error>
}

public struct PromptToCodeSnippet: Equatable, Identifiable {
    public let id = UUID()
    public var startLineIndex: Int
    public var originalCode: String
    public var modifiedCode: String
    public var description: String
    public var error: String?
    public var attachedRange: CursorRange

    public init(
        startLineIndex: Int,
        originalCode: String,
        modifiedCode: String,
        description: String,
        error: String?,
        attachedRange: CursorRange
    ) {
        self.startLineIndex = startLineIndex
        self.originalCode = originalCode
        self.modifiedCode = modifiedCode
        self.description = description
        self.error = error
        self.attachedRange = attachedRange
    }
}

public enum PromptToCodeAttachedTarget: Equatable {
    case file(URL, projectURL: URL, code: String, lines: [String])
    case dynamic
}

public struct PromptToCodeHistoryNode: Equatable {
    public var snippets: IdentifiedArrayOf<PromptToCodeSnippet>
    public var instruction: String

    public init(snippets: IdentifiedArrayOf<PromptToCodeSnippet>, instruction: String) {
        self.snippets = snippets
        self.instruction = instruction
    }
}