import {z} from 'zod'

export type GetApiV1AssetsAssetIdSegmentsArgs = {'App-ID': string, 'Auth-Token': string, asset_id: string, sort?: string | undefined, ids?: string | undefined, query?: string | undefined, includes?: string | undefined, per_page?: number | undefined, page?: number | undefined, scroll?: boolean | undefined, scroll_id?: string | undefined, transcription_id?: string | undefined, version_id?: string | undefined, segment_type?: string | undefined, segment_color?: string | undefined, time_start_milliseconds?: number | undefined, time_end_milliseconds?: number | undefined, time_start_milliseconds__gte?: number | undefined, time_end_milliseconds__lte?: number | undefined, status?: string | undefined, person_id?: string | undefined, share_id?: string | undefined, project_id?: string | undefined, include_users?: boolean | undefined, include_all_versions?: boolean | undefined};

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

export const drawingPointSchema = z.object({x: z.number(), y: z.number()});

export const drawingPrimitiveSchema = z.object({color: z.string().optional(), opacity: z.number().optional(), points: z.array(drawingPointSchema).optional(), text: z.string().optional(), type: z.enum(["LINE", "RECTANGLE", "ARROW", "ELLIPSE", "PENCIL", "TEXT", "CROSS"]), width: z.number().optional()});

export const drawingSchema = z.object({primitives: z.array(drawingPrimitiveSchema).optional(), version: z.number().int().nullable().optional()});

export const faceLandmark = z.object({x: z.number(), y: z.number()});

export const faceBoundingBoxSchema = z.object({bounding_box: z.array(z.number()).optional(), landmarks: z.array(faceLandmark).nullable().optional(), person_version_face_id: z.string(), timestamp_ms: z.number().int().gte(-9223372036854776000).lte(9223372036854776000).optional()});

export const semanticTranscriptionEntity = z.object({label: z.string().nullable().optional(), text: z.string()}).and(z.record(z.string(), z.unknown()));

export const understandingTag = z.object({label: z.string(), level: z.string().nullable().optional(), score: z.number().nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const semanticTranscription = z.object({cut_reasons: z.array(z.string()).nullable().optional(), embedding: z.string().nullable().optional(), embedding_dim: z.number().int().nullable().optional(), embedding_encoding: z.string().nullable().optional(), embedding_model: z.string().nullable().optional(), embedding_model_version: z.string().nullable().optional(), entities: z.array(semanticTranscriptionEntity).nullable().optional(), language: z.string().nullable().optional(), speaker_count: z.string().nullable().optional(), tone: z.string().nullable().optional(), topics: z.array(understandingTag).nullable().optional(), track_id: z.string().nullable().optional(), valence: z.string().nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const shot = z.object({actions: z.array(understandingTag).nullable().optional(), animal_species: z.array(understandingTag).nullable().optional(), brand_logo: z.array(understandingTag).nullable().optional(), camera_angle: understandingTag.nullable().optional(), caption: z.string().nullable().optional(), color_palette: z.array(understandingTag).nullable().optional(), composition: z.string().nullable().optional(), dog_breeds: z.array(understandingTag).nullable().optional(), dominant_hue: z.string().nullable().optional(), embedding: z.string().nullable().optional(), embedding_dim: z.number().int().nullable().optional(), embedding_encoding: z.string().nullable().optional(), embedding_model: z.string().nullable().optional(), embedding_model_version: z.string().nullable().optional(), facial_expression: z.array(understandingTag).nullable().optional(), keyframe_ts: z.number().int().gte(-9223372036854776000).lte(9223372036854776000).optional(), light_key: z.string().nullable().optional(), light_side: z.string().nullable().optional(), light_style: z.array(z.string()).nullable().optional(), light_time: z.string().nullable().optional(), mood: z.array(understandingTag).nullable().optional(), mood_valence: z.string().nullable().optional(), movement: z.array(z.string()).nullable().optional(), objects: z.array(understandingTag).nullable().optional(), ocr_text: z.string().nullable().optional(), orientation: z.string().nullable().optional(), people_age_range: z.array(z.string()).nullable().optional(), people_bucket: z.string().nullable().optional(), people_count: z.number().int().nullable().optional(), people_features: z.array(understandingTag).nullable().optional(), people_gender: z.array(z.string()).nullable().optional(), place: understandingTag.nullable().optional(), resolution: z.string().nullable().optional(), scale_rank: z.number().int().nullable().optional(), season: understandingTag.nullable().optional(), sensitive_content: z.array(understandingTag).nullable().optional(), setting: understandingTag.nullable().optional(), shot_scale: understandingTag.nullable().optional(), sport_type: understandingTag.nullable().optional(), subject_type: z.array(understandingTag).nullable().optional(), time_of_day: z.string().nullable().optional(), track_id: z.string().nullable().optional(), weather: understandingTag.nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const transcriptionElementType = z.object({end_ms: z.number().int().gte(-9223372036854776000).lte(9223372036854776000), score: z.number().optional(), start_ms: z.number().int().gte(-9223372036854776000).lte(9223372036854776000), value: z.string()});

export const transcriptionType = z.object({speaker: z.number().int().gte(-2147483648).lte(2147483647).nullable().optional(), words: z.array(transcriptionElementType)});

export const user = z.object({email: z.string().optional(), first_name: z.string().optional(), id: z.string().optional(), last_name: z.string().optional(), photo: z.string().optional(), photo_big: z.string().optional(), photo_small: z.string().optional()});

export const segmentElasticSchema = z.object({asset_id: z.string().optional(), attachments: z.array(z.object({})).optional(), date_created: z.string().optional(), date_modified: z.string().optional(), drawing: drawingSchema.nullable().optional(), external_id: z.string().nullable().optional(), face_bounding_boxes: z.array(faceBoundingBoxSchema).nullable().optional(), has_attachments: z.boolean().optional(), has_drawing: z.boolean().optional(), id: z.string().optional(), is_internal: z.boolean().nullable().optional(), keyframe_id: z.string().nullable().optional(), metadata: z.object({}).optional(), metadata_view_id: z.string().nullable().optional(), parent_id: z.string().optional(), path: z.string().optional(), person_id: z.string().nullable().optional(), project_id: z.string().nullable().optional(), segment_checked: z.boolean().optional(), segment_color: z.string().optional(), segment_text: z.string().nullable().optional(), segment_track: z.string().nullable().optional(), segment_type: z.enum(["MARKER", "QC", "GENERIC", "COMMENT", "TAG", "TRANSCRIPTION", "SCENE", "SHOT", "PERSON", "SEMANTIC_TRANSCRIPTION"]), semantic_transcription: semanticTranscription.nullable().optional(), share_id: z.string().nullable().optional(), share_session_id: z.string().nullable().optional(), share_user_email: z.string().nullable().optional(), shot: shot.nullable().optional(), status: z.enum(["ACTIVE", "DELETED", "AWAITED"]).optional(), subclip_id: z.string().optional(), time_end_milliseconds: z.number().int().gte(-9223372036854776000).lte(9223372036854776000).nullable().optional(), time_start_milliseconds: z.number().int().gte(-9223372036854776000).lte(9223372036854776000).nullable().optional(), top_level: z.boolean().optional(), transcription: transcriptionType.optional(), transcription_id: z.string().nullable().optional(), user_first_name: z.string().optional(), user_id: z.string().optional(), user_info: user.optional(), user_last_name: z.string().optional(), user_photo: z.string().optional(), version_id: z.string().optional()});

export const segmentsSchema = z.object({facets: z.object({}).optional(), first_url: z.string().optional(), last_url: z.string().optional(), next_url: z.string().optional(), objects: z.array(segmentElasticSchema).optional(), page: z.number().int().gte(-2147483648).lte(2147483647).optional(), pages: z.number().int().gte(-2147483648).lte(2147483647).optional(), per_page: z.number().int().gte(-2147483648).lte(2147483647).optional(), prev_url: z.string().optional(), scroll_id: z.string().optional(), search_after: z.array(z.unknown()).optional(), total: z.number().int().gte(-9223372036854776000).lte(9223372036854776000).optional()});

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 of segments
 * 
 * GET /v1/assets/{asset_id}/segments/
 */
export const getApiV1AssetsAssetIdSegments = async (args: GetApiV1AssetsAssetIdSegmentsArgs, options: GetApiV1AssetsAssetIdSegmentsOptions) => {
const {baseUrl, fetch: fetchFn = fetch} = options ?? {}

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

if (args['App-ID'] != null) headers.set("App-ID", String(args['App-ID']))
if (args['Auth-Token'] != null) headers.set("Auth-Token", String(args['Auth-Token']))
if (args.sort != null) query.append("sort", String(args.sort))
if (args.ids != null) query.append("ids", String(args.ids))
if (args.query != null) query.append("query", String(args.query))
if (args.includes != null) query.append("includes", String(args.includes))
if (args.per_page != null) query.append("per_page", String(args.per_page))
if (args.page != null) query.append("page", String(args.page))
if (args.scroll != null) query.append("scroll", String(args.scroll))
if (args.scroll_id != null) query.append("scroll_id", String(args.scroll_id))
if (args.transcription_id != null) query.append("transcription_id", String(args.transcription_id))
if (args.version_id != null) query.append("version_id", String(args.version_id))
if (args.segment_type != null) query.append("segment_type", String(args.segment_type))
if (args.segment_color != null) query.append("segment_color", String(args.segment_color))
if (args.time_start_milliseconds != null) query.append("time_start_milliseconds", String(args.time_start_milliseconds))
if (args.time_end_milliseconds != null) query.append("time_end_milliseconds", String(args.time_end_milliseconds))
if (args.time_start_milliseconds__gte != null) query.append("time_start_milliseconds__gte", String(args.time_start_milliseconds__gte))
if (args.time_end_milliseconds__lte != null) query.append("time_end_milliseconds__lte", String(args.time_end_milliseconds__lte))
if (args.status != null) query.append("status", String(args.status))
if (args.person_id != null) query.append("person_id", String(args.person_id))
if (args.share_id != null) query.append("share_id", String(args.share_id))
if (args.project_id != null) query.append("project_id", String(args.project_id))
if (args.include_users != null) query.append("include_users", String(args.include_users))
if (args.include_all_versions != null) query.append("include_all_versions", String(args.include_all_versions))

const url = `${baseUrl.replace(/\/+$/, '')}/v1/assets/${encodeURIComponent(String(args.asset_id))}/segments/${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(segmentsSchema, await res.json())
};