import {z} from 'zod'

export type GetApiV1AssetsAssetIdSegmentsSegmentIdArgs = {'App-ID': string, 'Auth-Token': string, asset_id: string, segment_id: string};

export type GetApiV1AssetsAssetIdSegmentsSegmentIdOptions = {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 segmentSchema = z.object({asset_id: z.string().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 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
};

/**
 * Get a segment by ID
 * 
 * GET /v1/assets/{asset_id}/segments/{segment_id}/
 */
export const getApiV1AssetsAssetIdSegmentsSegmentId = async (args: GetApiV1AssetsAssetIdSegmentsSegmentIdArgs, options: GetApiV1AssetsAssetIdSegmentsSegmentIdOptions) => {
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']))

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