import {z} from 'zod'

export type EditCaptionsSpeakerReplacementV0 = {find: string, replace: string};

export type EditCaptionsReplaceExistingTracksV0 = 'replace' | 'fail';

export type EditCaptionsJobParametersV0 = {asset_id: string, track_id: string, auto_censor_profanity?: ({detection_method?: 'llm' | undefined, mode?: ('blank' | 'remove' | 'mask') | undefined, always_censor?: Array<string> | undefined, never_censor?: Array<string> | undefined}) | undefined, replacements?: (Array<{find: string, replace: string, case_sensitive?: boolean | undefined}>) | undefined, speaker_replacements?: Array<EditCaptionsSpeakerReplacementV0> | undefined, upload_to_mux?: boolean | undefined, replace_existing_tracks?: EditCaptionsReplaceExistingTracksV0 | undefined, track_name?: string | undefined, delete_original_track?: boolean | undefined, track_name_suffix?: string | undefined};

export type CreateEditCaptionsJobRequestV0 = {passthrough?: string | undefined, parameters: EditCaptionsJobParametersV0};

export type CreateApiRobotsV0JobsEditCaptionsArgs = {body: CreateEditCaptionsJobRequestV0};

export type CreateApiRobotsV0JobsEditCaptionsOptions = {baseUrl?: string | undefined, accessToken?: {username: string, password: string} | undefined, fetch?: typeof fetch | undefined};

export const jobDirectiveContextV0 = z.object({id: z.string(), run_id: z.string()});

export const editCaptionsSpeakerReplacementV0 = z.object({find: z.string().min(1), replace: z.string().min(1)});

export const editCaptionsReplaceExistingTracksV0 = z.enum(["replace", "fail"]);

export const editCaptionsJobParametersV0 = z.object({asset_id: z.string().min(1), track_id: z.string().min(1), auto_censor_profanity: z.object({detection_method: z.literal("llm").optional(), mode: z.enum(["blank", "remove", "mask"]).optional(), always_censor: z.array(z.string().min(1)).optional(), never_censor: z.array(z.string().min(1)).optional()}).optional(), replacements: z.array(z.object({find: z.string().min(1), replace: z.string(), case_sensitive: z.boolean().optional()})).optional(), speaker_replacements: z.array(editCaptionsSpeakerReplacementV0).optional(), upload_to_mux: z.boolean().optional(), replace_existing_tracks: editCaptionsReplaceExistingTracksV0.optional(), track_name: z.string().min(1).optional(), delete_original_track: z.boolean().optional(), track_name_suffix: z.string().min(1).optional()});

export const jobStatusV0 = z.enum(["pending", "processing", "completed", "errored", "cancelled"]);

export const jobErrorV0 = z.object({type: z.string(), message: z.string(), retryable: z.boolean().optional()});

export const slimlineAssetV0 = z.object({id: z.string(), meta: z.object({title: z.string().optional(), creator_id: z.string().optional(), external_id: z.string().optional()}).optional(), passthrough: z.string().optional(), _links: z.object({self: z.object({href: z.string()})})});

export const resourcesV0 = z.object({assets: z.array(slimlineAssetV0)});

export const replacedTrackV0 = z.object({id: z.string().min(1), type: z.enum(["text", "audio"]).optional(), name: z.string().optional(), language_code: z.string().optional(), text_source: z.string().optional(), passthrough: z.string().optional()});

export const replacedTracksV0 = z.array(replacedTrackV0);

export const editCaptionsJobOutputsV0 = z.object({total_replacement_count: z.number().int().gte(0), uploaded_track_id: z.string().min(1).optional(), temporary_vtt_url: z.string().min(1).optional(), replaced_tracks: replacedTracksV0.optional()});

export const editCaptionsJobV0 = z.object({id: z.string(), passthrough: z.string().optional(), units_consumed: z.number().int().gte(0), directive: jobDirectiveContextV0.optional(), created_at: z.number().int().gte(0), updated_at: z.number().int().gte(0), workflow: z.literal("edit-captions"), parameters: editCaptionsJobParametersV0, status: jobStatusV0, errors: z.array(jobErrorV0).optional(), resources: resourcesV0.optional(), outputs: editCaptionsJobOutputsV0.optional()});

export const editCaptionsJobResponseV0 = z.object({data: editCaptionsJobV0});

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
};

/**
 * Create an 'edit-captions' job
 * 
 * POST /robots/v0/jobs/edit-captions
 * 
 * Also accepts auth: authorizationToken
 */
export const createApiRobotsV0JobsEditCaptions = async (args: CreateApiRobotsV0JobsEditCaptionsArgs, options?: CreateApiRobotsV0JobsEditCaptionsOptions) => {
const {baseUrl = "https://api.mux.com", fetch: fetchFn = fetch, accessToken} = options ?? {}

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

if (accessToken !== undefined) headers.set('authorization', `Basic ${btoa(`${accessToken.username}:${accessToken.password}`)}`)

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

const res = await fetchFn(url, {method: "POST", headers, body: JSON.stringify(args.body)})

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

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