import {z} from 'zod'

export type MediaType = 'photography' | 'illustration';

export type GetApiV3AiGenerationHistoryArgs = {date_start?: string | undefined, date_end?: string | undefined, media_type?: MediaType | undefined, page_size?: number | undefined, page?: number | undefined, product_ids?: Array<number> | undefined, company_generations?: boolean | undefined};

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

export const generationType = z.enum(["text_to_image", "variation", "extend", "refine", "background_removal", "influence_color_by_image", "influence_composition_by_image", "object_removal", "influence_outline_by_image", "background_generation"]);

export const mediaType = z.enum(["photography", "illustration"]);

export const mood = z.enum(["black_and_white", "warm", "cool", "natural", "vivid", "dramatic", "bold"]);

export const generationOptions = z.object({media_type: mediaType.optional(), aspect_ratio: z.string().nullable().optional(), mood: mood.optional()});

export const sourceGenerationRequest = z.object({source_generation_request_id: z.string().nullable().optional(), source_generation_result_index: z.number().int().optional()});

export const assetDetail = z.object({id: z.string().nullable().optional()});

export const previewUrl = z.object({preview_size: z.string().nullable().optional(), image_url: z.string().nullable().optional(), width: z.number().int().optional(), height: z.number().int().optional()});

export const generationHistoryResult = z.object({index: z.number().int().optional(), is_blocked: z.boolean().optional(), preview_urls: z.array(previewUrl).nullable().optional()});

export const user = z.object({username: z.string().nullable().optional(), first_name: z.string().nullable().optional(), middle_name: z.string().nullable().optional(), last_name: z.string().nullable().optional()});

export const generationHistoryItemResponse = z.object({generation_request_id: z.string().nullable().optional(), seed: z.number().int().nullable().optional(), generation_date: z.string().optional(), product_revoked_date: z.string().nullable().optional(), prompt: z.string().nullable().optional(), negative_prompt: z.string().nullable().optional(), product_id: z.number().int().optional(), notes: z.string().nullable().optional(), project_code: z.string().nullable().optional(), generation_type: generationType.optional(), generation_options: generationOptions.optional(), source_generation_request: sourceGenerationRequest.optional(), original_asset: assetDetail.optional(), results: z.array(generationHistoryResult).nullable().optional(), user: user.optional()});

export const generationHistoryResponse = z.object({generations: z.array(generationHistoryItemResponse).nullable().optional(), result_count: z.number().int().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
};

/**
 * Retrieve history of AI generations
 * 
 * GET /v3/ai/generation-history
 * 
 * Also accepts auth: OAuth2
 */
export const getApiV3AiGenerationHistory = async (args: GetApiV3AiGenerationHistoryArgs, options: GetApiV3AiGenerationHistoryOptions) => {
const {baseUrl, fetch: fetchFn = fetch, ApiKey} = options ?? {}

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

if (args.date_start != null) query.append("date_start", String(args.date_start))
if (args.date_end != null) query.append("date_end", String(args.date_end))
if (args.media_type != null) query.append("media_type", String(args.media_type))
if (args.page_size != null) query.append("page_size", String(args.page_size))
if (args.page != null) query.append("page", String(args.page))
for (const item of args.product_ids ?? []) query.append("product_ids", String(item))
if (args.company_generations != null) query.append("company_generations", String(args.company_generations))
if (ApiKey !== undefined) headers.set("Api-Key", ApiKey)

const url = `${baseUrl.replace(/\/+$/, '')}/v3/ai/generation-history${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(generationHistoryResponse, await res.json())
};