import {z} from 'zod'

export type DrawingPointSchema = {x: number, y: number};

export type DrawingPrimitiveSchema = {color?: string | undefined, opacity?: number | undefined, points?: Array<DrawingPointSchema> | undefined, text?: string | undefined, type: 'LINE' | 'RECTANGLE' | 'ARROW' | 'ELLIPSE' | 'PENCIL' | 'TEXT' | 'CROSS', width?: number | undefined};

export type DrawingSchema = {primitives?: Array<DrawingPrimitiveSchema> | undefined, version?: (number | null) | undefined};

export type FaceLandmark = {x: number, y: number};

export type FaceBoundingBox = {bounding_box?: Array<number> | undefined, landmarks?: (Array<FaceLandmark> | null) | undefined, person_version_face_id: string, timestamp_ms?: number | undefined};

export type SemanticTranscriptionEntity = {label?: (string | null) | undefined, text: string} | {[key: string]: unknown};

export type UnderstandingTag = {label: string, level?: (string | null) | undefined, score?: (number | null) | undefined} | {[key: string]: unknown};

export type SemanticTranscription = {cut_reasons?: (Array<string> | null) | undefined, embedding?: (string | null) | undefined, embedding_dim?: (number | null) | undefined, embedding_encoding?: (string | null) | undefined, embedding_model?: (string | null) | undefined, embedding_model_version?: (string | null) | undefined, entities?: (Array<SemanticTranscriptionEntity> | null) | undefined, language?: (string | null) | undefined, speaker_count?: (string | null) | undefined, tone?: (string | null) | undefined, topics?: (Array<UnderstandingTag> | null) | undefined, track_id?: (string | null) | undefined, valence?: (string | null) | undefined} | {[key: string]: unknown};

export type Shot = {actions?: (Array<UnderstandingTag> | null) | undefined, animal_species?: (Array<UnderstandingTag> | null) | undefined, brand_logo?: (Array<UnderstandingTag> | null) | undefined, camera_angle?: (UnderstandingTag | null) | undefined, caption?: (string | null) | undefined, color_palette?: (Array<UnderstandingTag> | null) | undefined, composition?: (string | null) | undefined, dog_breeds?: (Array<UnderstandingTag> | null) | undefined, dominant_hue?: (string | null) | undefined, embedding?: (string | null) | undefined, embedding_dim?: (number | null) | undefined, embedding_encoding?: (string | null) | undefined, embedding_model?: (string | null) | undefined, embedding_model_version?: (string | null) | undefined, facial_expression?: (Array<UnderstandingTag> | null) | undefined, keyframe_ts?: number | undefined, light_key?: (string | null) | undefined, light_side?: (string | null) | undefined, light_style?: (Array<string> | null) | undefined, light_time?: (string | null) | undefined, mood?: (Array<UnderstandingTag> | null) | undefined, mood_valence?: (string | null) | undefined, movement?: (Array<string> | null) | undefined, objects?: (Array<UnderstandingTag> | null) | undefined, ocr_text?: (string | null) | undefined, orientation?: (string | null) | undefined, people_age_range?: (Array<string> | null) | undefined, people_bucket?: (string | null) | undefined, people_count?: (number | null) | undefined, people_features?: (Array<UnderstandingTag> | null) | undefined, people_gender?: (Array<string> | null) | undefined, place?: (UnderstandingTag | null) | undefined, resolution?: (string | null) | undefined, scale_rank?: (number | null) | undefined, season?: (UnderstandingTag | null) | undefined, sensitive_content?: (Array<UnderstandingTag> | null) | undefined, setting?: (UnderstandingTag | null) | undefined, shot_scale?: (UnderstandingTag | null) | undefined, sport_type?: (UnderstandingTag | null) | undefined, subject_type?: (Array<UnderstandingTag> | null) | undefined, time_of_day?: (string | null) | undefined, track_id?: (string | null) | undefined, weather?: (UnderstandingTag | null) | undefined} | {[key: string]: unknown};

export type TranscriptionElementType = {end_ms: number, score?: number | undefined, start_ms: number, value: string};

export type TranscriptionType = {speaker?: (number | null) | undefined, words: Array<TranscriptionElementType>};

export type User = {email?: string | undefined, first_name?: string | undefined, id?: string | undefined, last_name?: string | undefined, photo?: string | undefined, photo_big?: string | undefined, photo_small?: string | undefined};

export type EditSegmentSchema = {asset_id?: string | undefined, date_created?: string | undefined, date_modified?: string | undefined, drawing?: (DrawingSchema | null) | undefined, external_id?: (string | null) | undefined, face_bounding_boxes?: (Array<FaceBoundingBox> | null) | undefined, has_attachments?: boolean | undefined, id?: string | undefined, is_internal?: (boolean | null) | undefined, keyframe_id?: (string | null) | undefined, metadata?: Record<string, never> | undefined, metadata_view_id?: (string | null) | undefined, parent_id?: string | undefined, path?: string | undefined, person_id?: (string | null) | undefined, project_id?: (string | null) | undefined, segment_checked?: boolean | undefined, segment_color?: string | undefined, segment_text?: (string | null) | undefined, segment_track?: (string | null) | undefined, segment_type: 'MARKER' | 'QC' | 'GENERIC' | 'COMMENT' | 'TAG' | 'TRANSCRIPTION' | 'SCENE' | 'SHOT' | 'PERSON' | 'SEMANTIC_TRANSCRIPTION', semantic_transcription?: (SemanticTranscription | null) | undefined, share_id?: (string | null) | undefined, share_session_id?: (string | null) | undefined, share_user_email?: (string | null) | undefined, shot?: (Shot | null) | undefined, status?: ('ACTIVE' | 'DELETED' | 'AWAITED') | undefined, subclip_id?: string | undefined, time_end_milliseconds?: (number | null) | undefined, time_start_milliseconds?: (number | null) | undefined, top_level?: boolean | undefined, transcription?: TranscriptionType | undefined, transcription_id?: (string | null) | undefined, user_first_name?: string | undefined, user_id?: string | undefined, user_info?: User | undefined, user_last_name?: string | undefined, user_photo?: string | undefined, version_id?: string | undefined};

export type PatchApiV1AssetsAssetIdSegmentsSegmentIdArgs = {'App-ID': string, 'Auth-Token': string, asset_id: string, segment_id: string, generate_subclip_keyframes?: boolean | undefined, index_immediately?: boolean | undefined, body: EditSegmentSchema};

export type PatchApiV1AssetsAssetIdSegmentsSegmentIdOptions = {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
};

/**
 * Update segment
 * 
 * PATCH /v1/assets/{asset_id}/segments/{segment_id}/
 */
export const patchApiV1AssetsAssetIdSegmentsSegmentId = async (args: PatchApiV1AssetsAssetIdSegmentsSegmentIdArgs, options: PatchApiV1AssetsAssetIdSegmentsSegmentIdOptions) => {
const {baseUrl, fetch: fetchFn = fetch} = options ?? {}

const query = new URLSearchParams()
const headers = new Headers()
headers.set('content-type', "application/json")
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.generate_subclip_keyframes != null) query.append("generate_subclip_keyframes", String(args.generate_subclip_keyframes))
if (args.index_immediately != null) query.append("index_immediately", String(args.index_immediately))

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: "PATCH", 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(segmentSchema, await res.json())
};