import {z} from 'zod'

export type GetApiJobsArgs = {status_filter?: (string | null) | undefined};

export type GetApiJobsOptions = {baseUrl: string, fetch?: typeof fetch | undefined};

export const jobStatus = z.enum(["pending", "processing", "completed", "failed"]);

export const ingestionPhase = z.enum(["extracting", "indexing", "detecting_edges", "propagating_scores"]);

export const jobProgress = z.object({current_step: z.string(), chunks_total: z.number().int().nullable().optional(), chunks_done: z.number().int().nullable().optional(), facts_extracted: z.number().int().nullable().optional(), phase: ingestionPhase.nullable().optional(), phase_index: z.number().int().gte(1).lte(4).nullable().optional(), phases_total: z.number().int().nullable().optional(), phase_detail: z.string().nullable().optional(), overall_percent: z.number().int().gte(0).lte(100).nullable().optional()});

export const documentContext = z.object({title: z.string(), key_entities: z.array(z.string()), main_themes: z.array(z.string()), document_type: z.enum(["technical", "conceptual", "reference", "guide", "other"]), language: z.string().nullable().optional(), scope_hint: z.enum(["own", "third_party", "mixed_context", "unknown"]).nullable().optional()});

export const piiRunSummary = z.object({mode: z.enum(["off", "anonymize", "block"]), provider: z.string(), provider_version: z.string().nullable().optional(), entitlement: z.enum(["premium", "floor"]), facts_scanned: z.number().int().gte(0).optional(), facts_anonymized: z.number().int().gte(0).optional(), facts_blocked: z.number().int().gte(0).optional(), by_type: z.record(z.string(), z.number().int()).optional(), coverage: z.enum(["full", "partial", "none"]).optional(), language: z.string().nullable().optional(), units: z.number().int().gte(0).optional(), free_units: z.number().int().gte(0).optional(), chars: z.number().int().gte(0).optional(), processing_ms: z.number().int().gte(0).optional()});

export const clashPair = z.object({new_fact_id: z.string(), other_fact_id: z.string(), relation: z.enum(["contradicts", "supersedes"]), new_fact_role: z.enum(["replaces", "replaced", "contradicts"]).optional()});

export const chunkError = z.object({chunk_index: z.number().int().gte(0), code: z.string(), message: z.string(), retryable: z.boolean().optional()});

export const jobError = z.object({code: z.string(), message: z.string(), step: z.string().nullable().optional(), retryable: z.boolean().optional()});

export const tokenUsage = z.object({llm_calls: z.number().int().gte(0).optional(), prompt_tokens: z.number().int().gte(0).optional(), completion_tokens: z.number().int().gte(0).optional(), cached_prompt_tokens: z.number().int().gte(0).optional(), cache_write_tokens: z.number().int().gte(0).optional(), total_tokens: z.number().int()});

export const jobUsage = z.object({extraction: tokenUsage.optional(), edge_detection: tokenUsage.nullable().optional(), total_tokens: z.number().int()});

export const jobResult = z.object({chunks_processed: z.number().int().gte(0), facts_created: z.number().int().gte(0), fact_ids: z.array(z.string()).optional(), facts_duplicated: z.number().int().gte(0), document_context: documentContext, processing_time_ms: z.number().int().gte(0), phase_timings_ms: z.record(z.string(), z.number().int()).optional(), extraction_audit: z.unknown(), detail: z.string().nullable().optional(), detail_code: z.enum(["all_filtered", "no_facts_found", "all_withheld", "all_duplicates"]).nullable().optional(), pii: piiRunSummary.nullable().optional(), edges_created: z.number().int().gte(0).optional(), contradictions_created: z.number().int().gte(0).optional(), supersedes_created: z.number().int().gte(0).optional(), clash_sample: z.array(clashPair).optional(), scores_updated: z.number().int().gte(0).optional(), chunks_failed: z.number().int().gte(0).optional(), errors: z.array(chunkError).optional(), facts_indexed: z.number().int().gte(0).nullable().optional(), indexing_error: jobError.nullable().optional(), usage: jobUsage.nullable().optional(), partial: z.boolean()});

export const repropagationJobResult = z.object({trigger: z.string(), source_type: z.string().nullable().optional(), document_id: z.string().nullable().optional(), previous_weight: z.number().nullable().optional(), weight: z.number().nullable().optional(), facts_recomputed: z.number().int().gte(0).optional(), facts_saved: z.number().int().gte(0).optional(), skipped_curated: z.number().int().gte(0).optional(), skipped_endorsed: z.number().int().gte(0).optional(), skipped_unknown_start: z.number().int().gte(0).optional(), processing_time_ms: z.number().int().gte(0)});

export const jobResponse = z.object({job_id: z.string(), pot_id: z.string(), status: jobStatus, created_at: z.string(), updated_at: z.string(), progress: jobProgress.nullable().optional(), result: z.union([jobResult, repropagationJobResult]).nullable().optional(), error: jobError.nullable().optional(), poll_interval_ms: z.number().int().nullable().optional()});

export const jobListResponse = z.object({jobs: z.array(jobResponse), total: z.number().int()});

export const validateResponse = async <Output>(
  schema: {
    readonly '~standard': {
      readonly validate: (value: unknown) => ({ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray<PropertyKey | { readonly key: PropertyKey }> | undefined }> }) | Promise<{ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray<PropertyKey | { readonly key: PropertyKey }> | undefined }> }>
    }
  },
  value: unknown
): Promise<Output> => {
  const result = await schema['~standard'].validate(value)

  if (result.issues !== undefined) {
    const detail = result.issues
      .map(({ message, path }) =>
        path?.length
          ? `${path.map((segment) => String(typeof segment === 'object' ? segment.key : segment)).join('.')}: ${message}`
          : message
      )
      .join('; ')

    throw new Error(`Response does not match the schema: ${detail}`, { cause: { issues: result.issues } })
  }

  return result.value
};

/**
 * List jobs for workspace
 * 
 * GET /jobs
 */
export const getApiJobs = async (args: GetApiJobsArgs, options: GetApiJobsOptions) => {
const {baseUrl, fetch: fetchFn = fetch} = options ?? {}

const query = new URLSearchParams()
const headers = new Headers()
headers.set('accept', "application/json")

if (args.status_filter != null) query.append("status_filter", String(args.status_filter))

const url = `${baseUrl.replace(/\/+$/, '')}/jobs${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "GET", headers})

if (!res.ok) throw new Error(`Request failed with status ${res.status}`, {cause: {status: res.status, body: await res.text()}})

return validateResponse(jobListResponse, await res.json())
};