import {z} from 'zod'

export type GetApiJobsArgs = {created_before?: string | undefined, limit?: number | undefined, include_deleted?: boolean | undefined};

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

export const jobType = z.literal("transcription");

export const dataFetchConfig = z.object({url: z.string(), auth_headers: z.array(z.string()).optional()});

export const model = z.enum(["standard", "enhanced", "melia-1"]);

export const operatingPoint = z.enum(["standard", "enhanced", "melia-1"]);

export const speakersInputItem = z.object({label: z.string().min(1), speaker_identifiers: z.array(z.string())});

export const transcriptionConfig = z.object({language: z.string(), domain: z.string().optional(), output_locale: z.string().optional(), model: model.optional(), operating_point: operatingPoint.optional(), additional_vocab: z.array(z.object({content: z.string(), sounds_like: z.array(z.string()).optional()})).optional(), punctuation_overrides: z.object({sensitivity: z.number().gte(0).lte(1).optional(), permitted_marks: z.array(z.string()).optional()}).optional(), diarization: z.enum(["none", "speaker", "channel"]).optional(), channel_diarization_labels: z.array(z.string()).optional(), enable_entities: z.boolean().optional(), max_delay_mode: z.enum(["fixed", "flexible"]).optional(), audio_filtering_config: z.object({volume_threshold: z.number().gte(0).lte(100).optional()}).optional(), transcript_filtering_config: z.object({remove_disfluencies: z.boolean().optional(), replacements: z.array(z.object({from: z.string(), to: z.string()})).optional()}).optional(), speaker_diarization_config: z.object({prefer_current_speaker: z.boolean().optional(), speaker_sensitivity: z.number().gte(0).lte(1).optional(), get_speakers: z.boolean().optional(), speakers: z.array(speakersInputItem).optional()}).optional()});

export const notificationConfig = z.object({url: z.string(), contents: z.array(z.enum(["jobinfo", "transcript", "transcript.json-v2", "transcript.txt", "transcript.srt", "data", "text"])).optional(), method: z.enum(["post", "put"]).optional(), auth_headers: z.array(z.string()).optional()});

export const trackingData = z.object({title: z.string().optional(), reference: z.string().optional(), tags: z.array(z.string()).optional(), details: z.object({}).optional()});

export const outputConfig = z.object({srt_overrides: z.object({max_line_length: z.number().int().optional(), max_lines: z.number().int().optional()}).optional()});

export const translationConfig = z.object({target_languages: z.array(z.string())});

export const languageIdentificationConfig = z.object({expected_languages: z.array(z.string()).optional(), low_confidence_action: z.enum(["allow", "reject", "use_default_language"]).optional(), default_language: z.string().optional()});

export const summarizationConfig = z.object({content_type: z.enum(["auto", "informative", "conversational"]).optional(), summary_length: z.enum(["brief", "detailed"]).optional(), summary_type: z.enum(["paragraphs", "bullets"]).optional()});

export const sentimentAnalysisConfig = z.object({});

export const topicDetectionConfig = z.object({topics: z.array(z.string()).optional()});

export const autoChaptersConfig = z.object({});

export const audioEventsConfig = z.object({types: z.array(z.string()).optional()});

export const jobConfig = z.object({type: jobType, fetch_data: dataFetchConfig.optional(), fetch_text: dataFetchConfig.optional(), transcription_config: transcriptionConfig.optional(), notification_config: z.array(notificationConfig).optional(), tracking: trackingData.optional(), output_config: outputConfig.optional(), translation_config: translationConfig.optional(), language_identification_config: languageIdentificationConfig.optional(), summarization_config: summarizationConfig.optional(), sentiment_analysis_config: sentimentAnalysisConfig.optional(), topic_detection_config: topicDetectionConfig.optional(), auto_chapters_config: autoChaptersConfig.optional(), audio_events_config: audioEventsConfig.optional()});

export const jobDetailError = z.object({timestamp: z.string(), message: z.string()});

export const jobDetails = z.object({created_at: z.string(), data_name: z.string(), text_name: z.string().optional(), duration: z.number().int().gte(0).optional(), id: z.string(), status: z.enum(["running", "done", "rejected", "deleted", "expired"]), config: jobConfig.optional(), lang: z.string().optional(), errors: z.array(jobDetailError).optional()});

export const retrieveJobsResponse = z.object({jobs: z.array(jobDetails)});

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 all jobs
 * 
 * GET /jobs
 */
export const getApiJobs = async (args: GetApiJobsArgs, options?: GetApiJobsOptions) => {
const {baseUrl = "https://eu1.asr.api.speechmatics.com/v2", fetch: fetchFn = fetch, bearerAuth} = options ?? {}

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

if (args.created_before != null) query.append("created_before", String(args.created_before))
if (args.limit != null) query.append("limit", String(args.limit))
if (args.include_deleted != null) query.append("include_deleted", String(args.include_deleted))
if (bearerAuth !== undefined) headers.set("Authorization", bearerAuth)

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(retrieveJobsResponse, await res.json())
};