// Generated by scripts/gen-api.py from spec/openapi.json — do not edit.

import Foundation

// MARK: - Types

/// Used by POST /jobs.
public struct DomainAvailabilityBulkJob: Codable, Sendable, Hashable {
    public var domains: [String]

    public init(domains: [String]) {
        self.domains = domains
    }

    enum CodingKeys: String, CodingKey {
        case domains
    }
}

/// Used by POST /jobs.
public struct EmbeddingJob: Codable, Sendable, Hashable {
    public var input: String
    public var model: String?

    public init(input: String, model: String? = nil) {
        self.input = input
        self.model = model
    }

    enum CodingKeys: String, CodingKey {
        case input
        case model
    }
}

/// Values this build does not know decode as `.unrecognized`.
public enum ChatMessageRole: RawRepresentable, Codable, Sendable, Hashable {
    case system
    case user
    case assistant
    /// A value the server sent that this version of the package does not list.
    case unrecognized(String)

    /// Every value the spec listed when this file was generated.
    public static let known: [ChatMessageRole] = [.system, .user, .assistant]

    public init(rawValue: String) {
        switch rawValue {
        case "system": self = .system
        case "user": self = .user
        case "assistant": self = .assistant
        default: self = .unrecognized(rawValue)
        }
    }

    public var rawValue: String {
        switch self {
        case .system: return "system"
        case .user: return "user"
        case .assistant: return "assistant"
        case .unrecognized(let raw): return raw
        }
    }

    public init(from decoder: any Decoder) throws {
        self.init(rawValue: try decoder.singleValueContainer().decode(String.self))
    }

    public func encode(to encoder: any Encoder) throws {
        var container = encoder.singleValueContainer()
        try container.encode(rawValue)
    }
}

/// Used by POST /jobs.
public struct ChatMessage: Codable, Sendable, Hashable {
    public var role: ChatMessageRole
    public var content: String

    public init(role: ChatMessageRole, content: String) {
        self.role = role
        self.content = content
    }

    enum CodingKeys: String, CodingKey {
        case role
        case content
    }
}

/// Used by POST /jobs.
public struct LlmChatJob: Codable, Sendable, Hashable {
    public var messages: [ChatMessage]
    public var model: String?
    public var maxTokens: Int?
    public var temperature: Double?

    public init(messages: [ChatMessage], model: String? = nil, maxTokens: Int? = nil, temperature: Double? = nil) {
        self.messages = messages
        self.model = model
        self.maxTokens = maxTokens
        self.temperature = temperature
    }

    enum CodingKeys: String, CodingKey {
        case messages
        case model
        case maxTokens = "max_tokens"
        case temperature
    }
}

/// Used by POST /jobs.
public struct LlmRawCompletionJob: Codable, Sendable, Hashable {
    public var prompt: String
    public var model: String?
    public var maxTokens: Int?
    public var temperature: Double?

    public init(prompt: String, model: String? = nil, maxTokens: Int? = nil, temperature: Double? = nil) {
        self.prompt = prompt
        self.model = model
        self.maxTokens = maxTokens
        self.temperature = temperature
    }

    enum CodingKeys: String, CodingKey {
        case prompt
        case model
        case maxTokens = "max_tokens"
        case temperature
    }
}

/// Values this build does not know decode as `.unrecognized`.
public enum NlpAnalyzeJobLanguage: RawRepresentable, Codable, Sendable, Hashable {
    case auto
    case en
    case fr
    case ro
    /// A value the server sent that this version of the package does not list.
    case unrecognized(String)

    /// Every value the spec listed when this file was generated.
    public static let known: [NlpAnalyzeJobLanguage] = [.auto, .en, .fr, .ro]

    public init(rawValue: String) {
        switch rawValue {
        case "auto": self = .auto
        case "en": self = .en
        case "fr": self = .fr
        case "ro": self = .ro
        default: self = .unrecognized(rawValue)
        }
    }

    public var rawValue: String {
        switch self {
        case .auto: return "auto"
        case .en: return "en"
        case .fr: return "fr"
        case .ro: return "ro"
        case .unrecognized(let raw): return raw
        }
    }

    public init(from decoder: any Decoder) throws {
        self.init(rawValue: try decoder.singleValueContainer().decode(String.self))
    }

    public func encode(to encoder: any Encoder) throws {
        var container = encoder.singleValueContainer()
        try container.encode(rawValue)
    }
}

/// Used by POST /jobs.
public struct NlpAnalyzeJob: Codable, Sendable, Hashable {
    public var input: String
    public var language: NlpAnalyzeJobLanguage?

    public init(input: String, language: NlpAnalyzeJobLanguage? = nil) {
        self.input = input
        self.language = language
    }

    enum CodingKeys: String, CodingKey {
        case input
        case language
    }
}

/// Used by POST /jobs.
public struct NlpDetectLanguageJob: Codable, Sendable, Hashable {
    public var input: String

    public init(input: String) {
        self.input = input
    }

    enum CodingKeys: String, CodingKey {
        case input
    }
}

/// Discriminated by `type`.
public enum JobSpec: Codable, Sendable, Hashable {
    case domainAvailabilityBulk(DomainAvailabilityBulkJob)
    case embedding(EmbeddingJob)
    case llmChat(LlmChatJob)
    case llmRawCompletion(LlmRawCompletionJob)
    case nlpAnalyze(NlpAnalyzeJob)
    case nlpDetectLanguage(NlpDetectLanguageJob)

    /// The wire discriminator.
    public var type: String {
        switch self {
        case .domainAvailabilityBulk: return "domain-availability-bulk"
        case .embedding: return "embedding"
        case .llmChat: return "llm-chat"
        case .llmRawCompletion: return "llm-raw-completion"
        case .nlpAnalyze: return "nlp-analyze"
        case .nlpDetectLanguage: return "nlp-detect-language"
        }
    }

    private enum DiscriminatorKey: String, CodingKey { case type = "type" }

    public init(from decoder: any Decoder) throws {
        let kind = try decoder.container(keyedBy: DiscriminatorKey.self).decode(String.self, forKey: .type)
        switch kind {
        case "domain-availability-bulk": self = .domainAvailabilityBulk(try DomainAvailabilityBulkJob(from: decoder))
        case "embedding": self = .embedding(try EmbeddingJob(from: decoder))
        case "llm-chat": self = .llmChat(try LlmChatJob(from: decoder))
        case "llm-raw-completion": self = .llmRawCompletion(try LlmRawCompletionJob(from: decoder))
        case "nlp-analyze": self = .nlpAnalyze(try NlpAnalyzeJob(from: decoder))
        case "nlp-detect-language": self = .nlpDetectLanguage(try NlpDetectLanguageJob(from: decoder))
        default: throw DecodingError.dataCorrupted(.init(codingPath: decoder.codingPath, debugDescription: "unknown type \(kind)"))
        }
    }

    public func encode(to encoder: any Encoder) throws {
        var container = encoder.container(keyedBy: DiscriminatorKey.self)
        try container.encode(type, forKey: .type)
        switch self {
        case .domainAvailabilityBulk(let payload): try payload.encode(to: encoder)
        case .embedding(let payload): try payload.encode(to: encoder)
        case .llmChat(let payload): try payload.encode(to: encoder)
        case .llmRawCompletion(let payload): try payload.encode(to: encoder)
        case .nlpAnalyze(let payload): try payload.encode(to: encoder)
        case .nlpDetectLanguage(let payload): try payload.encode(to: encoder)
        }
    }
}

/// Values this build does not know decode as `.unrecognized`.
public enum JobStatus: RawRepresentable, Codable, Sendable, Hashable {
    case queued
    case processing
    case succeeded
    case failed
    /// A value the server sent that this version of the package does not list.
    case unrecognized(String)

    /// Every value the spec listed when this file was generated.
    public static let known: [JobStatus] = [.queued, .processing, .succeeded, .failed]

    public init(rawValue: String) {
        switch rawValue {
        case "queued": self = .queued
        case "processing": self = .processing
        case "succeeded": self = .succeeded
        case "failed": self = .failed
        default: self = .unrecognized(rawValue)
        }
    }

    public var rawValue: String {
        switch self {
        case .queued: return "queued"
        case .processing: return "processing"
        case .succeeded: return "succeeded"
        case .failed: return "failed"
        case .unrecognized(let raw): return raw
        }
    }

    public init(from decoder: any Decoder) throws {
        self.init(rawValue: try decoder.singleValueContainer().decode(String.self))
    }

    public func encode(to encoder: any Encoder) throws {
        var container = encoder.singleValueContainer()
        try container.encode(rawValue)
    }
}

/// Used by POST /jobs.
public struct QueuedJob: Codable, Sendable, Hashable, Identifiable {
    public var id: String
    public var type: String
    public var status: JobStatus
    public var createdAt: String

    public init(id: String, type: String, status: JobStatus, createdAt: String) {
        self.id = id
        self.type = type
        self.status = status
        self.createdAt = createdAt
    }

    enum CodingKeys: String, CodingKey {
        case id
        case type
        case status
        case createdAt = "created_at"
    }
}

/// Used by POST /jobs.
public struct EnqueueResponse: Codable, Sendable, Hashable {
    public var job: QueuedJob

    public init(job: QueuedJob) {
        self.job = job
    }

    enum CodingKeys: String, CodingKey {
        case job
    }
}

/// Used by GET /jobs/{id}.
public struct Job: Codable, Sendable, Hashable, Identifiable {
    public var id: String
    public var type: String
    public var status: JobStatus
    public var error: String?
    public var attempts: Int
    public var createdAt: String
    public var startedAt: String?
    public var finishedAt: String?
    public var bytesIn: Int?
    public var bytesOut: Int?
    public var durationMs: Int?
    public var resultExpiresAt: String?

    public init(id: String, type: String, status: JobStatus, error: String? = nil, attempts: Int, createdAt: String, startedAt: String? = nil, finishedAt: String? = nil, bytesIn: Int? = nil, bytesOut: Int? = nil, durationMs: Int? = nil, resultExpiresAt: String? = nil) {
        self.id = id
        self.type = type
        self.status = status
        self.error = error
        self.attempts = attempts
        self.createdAt = createdAt
        self.startedAt = startedAt
        self.finishedAt = finishedAt
        self.bytesIn = bytesIn
        self.bytesOut = bytesOut
        self.durationMs = durationMs
        self.resultExpiresAt = resultExpiresAt
    }

    enum CodingKeys: String, CodingKey {
        case id
        case type
        case status
        case error
        case attempts
        case createdAt = "created_at"
        case startedAt = "started_at"
        case finishedAt = "finished_at"
        case bytesIn = "bytes_in"
        case bytesOut = "bytes_out"
        case durationMs = "duration_ms"
        case resultExpiresAt = "result_expires_at"
    }

    /// Explicit so required-but-nullable fields encode as `null` rather than vanishing.
    public func encode(to encoder: any Encoder) throws {
        var container = encoder.container(keyedBy: CodingKeys.self)
        try container.encode(id, forKey: .id)
        try container.encode(type, forKey: .type)
        try container.encode(status, forKey: .status)
        try container.encode(error, forKey: .error)
        try container.encode(attempts, forKey: .attempts)
        try container.encode(createdAt, forKey: .createdAt)
        try container.encode(startedAt, forKey: .startedAt)
        try container.encode(finishedAt, forKey: .finishedAt)
        try container.encode(bytesIn, forKey: .bytesIn)
        try container.encode(bytesOut, forKey: .bytesOut)
        try container.encode(durationMs, forKey: .durationMs)
        try container.encodeIfPresent(resultExpiresAt, forKey: .resultExpiresAt)
    }
}

/// Used by GET /jobs/{id}.
public struct JobGetResponse: Codable, Sendable, Hashable {
    public var job: Job

    public init(job: Job) {
        self.job = job
    }

    enum CodingKeys: String, CodingKey {
        case job
    }
}

/// Used by GET /jobs/{id}/result.
public struct JobResult: Codable, Sendable, Hashable {
    public var jobId: String
    public var type: String
    public var result: [String: JSONValue]

    public init(jobId: String, type: String, result: [String: JSONValue]) {
        self.jobId = jobId
        self.type = type
        self.result = result
    }

    enum CodingKeys: String, CodingKey {
        case jobId = "job-id"
        case type
        case result
    }
}

// MARK: - Operations

extension IrdOperation {
    public static let jobs_enqueue = IrdOperation(method: .post, pathTemplate: "/jobs", module: "jobs", name: "enqueue", security: .keyOrJwt)
    public static let jobs_get = IrdOperation(method: .get, pathTemplate: "/jobs/{id}", module: "jobs", name: "get", security: .keyOrJwt)
    public static let jobs_getResult = IrdOperation(method: .get, pathTemplate: "/jobs/{id}/result", module: "jobs", name: "getResult", security: .keyOrJwt)

    static let jobsOperations: [IrdOperation] = [.jobs_enqueue, .jobs_get, .jobs_getResult]
}

/// The `jobs` operations (3).
public struct JobsAPI: Sendable {
    let client: IrdClient

    /// Enqueue a job
    /// Submits a job for asynchronous processing and returns `202 Accepted` immediately.
    /// `POST /jobs` — 202
    public func enqueue(body: JobSpec) async throws -> EnqueueResponse {
        try await client.request(.jobs_enqueue, body: body)
    }

    /// Poll job status
    /// Returns the job's status and usage metrics — never submitted content or results.
    /// `GET /jobs/{id}` — 200
    public func get(id: String) async throws -> JobGetResponse {
        try await client.request(.jobs_get, params: ["id": id])
    }

    /// Fetch job result
    /// Returns the result of a succeeded job. Results are held for 24 hours after completion, then expire.
    /// `GET /jobs/{id}/result` — 200
    public func getResult(id: String) async throws -> JobResult {
        try await client.request(.jobs_getResult, params: ["id": id])
    }
}

extension IrdClient {
    public var jobs: JobsAPI { JobsAPI(client: self) }
}