import {z} from 'zod'

export type FlightUpdatableAttributes = {startDate: string, endDate: string, expectedImpressions?: number | undefined, impressionCap?: number | undefined, cpm?: number | undefined, costPerSpot?: number | undefined, uid?: string | undefined};

export type CreateApiMeasurementDisplayAdsAdIdFlightsArgs = {ad_id: string, body: {measurementFlight: FlightUpdatableAttributes}};

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

export const adServerEnum = z.enum(["acast", "adarcade", "adswizz", "amperwave", "art19", "audioboom", "audiohook", "audiomeans", "barstool", "basis", "beeswax", "beeswax_pod", "beeswax_stream", "dax", "dovetail", "dv_360", "frequency_ads", "gam", "gt_impression", "iheart_pd", "iheart_stream", "mediamath", "megaphone", "omny", "podbean", "redcircle", "sas", "simplecast", "soundcast", "soundstack", "spgam", "spotify", "spreaker", "tap", "transistor", "triton", "triton_pd", "triton_streaming", "ttd", "xandr", "xandr_pod", "xandr_stream"]);

export const adCohort = z.object({cohortName: z.string(), subCohortName: z.string()});

export const displayAd = z.object({id: z.string().optional(), name: z.string().optional(), uid: z.string().optional(), adServer: adServerEnum.optional(), redirectUrl: z.string().optional(), cohorts: z.array(adCohort).optional()});

export const dynamicAd = z.object({id: z.string().optional(), name: z.string().optional(), uid: z.string().optional(), adServer: adServerEnum.optional(), streaming: z.boolean().optional(), cohorts: z.array(adCohort).optional()});

export const amazonAd = z.object({id: z.string().optional(), name: z.string().optional(), uid: z.string().optional(), adServer: adServerEnum.optional(), streaming: z.boolean().optional(), cohorts: z.array(adCohort).optional()});

export const dynamicFlightWithParent = z.object({id: z.number().int().optional(), startDate: z.string(), endDate: z.string(), expectedImpressions: z.number().int().optional(), impressionCap: z.number().int().optional(), cpm: z.number().optional(), costPerSpot: z.number().optional(), uid: z.string().optional(), ad: z.discriminatedUnion("type", [displayAd, dynamicAd, amazonAd]).optional()});

export const createApiMeasurementDisplayAdsAdIdFlightsResponse = z.object({measurementFlight: dynamicFlightWithParent.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
};

/**
 * Create a new flight
 * 
 * POST /measurement/display_ads/{ad_id}/flights
 */
export const createApiMeasurementDisplayAdsAdIdFlights = async (args: CreateApiMeasurementDisplayAdsAdIdFlightsArgs, options?: CreateApiMeasurementDisplayAdsAdIdFlightsOptions) => {
const {baseUrl = "https://api.magellan.ai/v2", fetch: fetchFn = fetch, apiKey} = options ?? {}

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

if (apiKey !== undefined) headers.set("Authorization", apiKey)

const url = `${baseUrl.replace(/\/+$/, '')}/measurement/display_ads/${encodeURIComponent(String(args.ad_id))}/flights${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "POST", 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(createApiMeasurementDisplayAdsAdIdFlightsResponse, await res.json())
};