import {z} from 'zod'

export type GetApiV1FlightsSearchesSearchIdResultsArgs = {searchId: string, page?: number | undefined, pageSize?: number | undefined, sort?: ('score_desc' | 'price_asc' | 'duration_asc' | 'leg1_departure_time_asc' | 'leg1_departure_time_desc' | 'leg2_departure_time_asc' | 'leg2_departure_time_desc') | undefined, airlines?: Array<string> | undefined, 'airlines-match'?: ('any' | 'all') | undefined, 'same-airline'?: ('0' | '1' | 'true' | 'false') | undefined, alliances?: Array<string> | undefined, stops?: Array<number> | undefined, 'min-price'?: number | undefined, 'max-price'?: number | undefined, 'max-duration'?: number | undefined, 'min-stopover-duration'?: string | undefined, 'max-stopover-duration'?: string | undefined, 'outbound-departure-blocks'?: (Array<'midnight' | 'morning' | 'afternoon' | 'night'>) | undefined, 'outbound-departure-range'?: string | undefined, 'outbound-arrival-blocks'?: (Array<'midnight' | 'morning' | 'afternoon' | 'night'>) | undefined, 'outbound-arrival-range'?: string | undefined, 'return-departure-blocks'?: (Array<'midnight' | 'morning' | 'afternoon' | 'night'>) | undefined, 'return-departure-range'?: string | undefined, 'return-arrival-blocks'?: (Array<'midnight' | 'morning' | 'afternoon' | 'night'>) | undefined, 'return-arrival-range'?: string | undefined, 'outbound-min-duration'?: string | undefined, 'outbound-max-duration'?: string | undefined, 'return-min-duration'?: string | undefined, 'return-max-duration'?: string | undefined, 'booking-types'?: (Array<'wego' | 'airline'>) | undefined, 'booking-sites'?: Array<string> | undefined, 'stopover-airports'?: Array<string> | undefined, aircraft?: Array<string> | undefined, currency?: string | undefined, locale?: string | undefined, view?: 'card' | undefined};

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

export const getApiV1FlightsSearchesSearchIdResultsResponse = z.object({searchId: z.string(), currencyCode: z.string(), metadata: z.object({page: z.number().int().gte(1).lte(9007199254740991), pageSize: z.number().int().gte(1).lte(9007199254740991), resultCount: z.number().int().gte(0).lte(9007199254740991), totalCandidates: z.number().int().gte(0).lte(9007199254740991), hasMore: z.boolean(), filterOptions: z.object({alliances: z.array(z.object({code: z.string(), name: z.string().optional(), count: z.number().int().gte(0).lte(9007199254740991)})), airlines: z.array(z.object({code: z.string(), name: z.string().optional(), count: z.number().int().gte(0).lte(9007199254740991)})), bookingSites: z.array(z.object({code: z.string(), name: z.string().optional(), count: z.number().int().gte(0).lte(9007199254740991)})), stopoverAirports: z.array(z.object({code: z.string(), name: z.string().optional(), count: z.number().int().gte(0).lte(9007199254740991)})), aircraft: z.array(z.object({code: z.string(), name: z.string().optional(), count: z.number().int().gte(0).lte(9007199254740991)})), stopoverDurations: z.object({min: z.number().gte(0), max: z.number().gte(0)}).optional()}), snapshotTripCount: z.number().int().gte(0).lte(9007199254740991), snapshotFareCount: z.number().int().gte(0).lte(9007199254740991), createdAt: z.string().optional(), currencyCode: z.string(), currencyCodeSource: z.enum(["explicit", "default"]), locale: z.string(), localeSource: z.enum(["explicit", "default"])}), results: z.array(z.object({tripId: z.string(), badges: z.array(z.enum(["best_value", "cheapest", "cheapest_direct"])), stops: z.number(), durationMinutes: z.number(), price: z.object({total: z.number(), currency: z.string(), scope: z.literal("party"), websiteCount: z.number().int().gte(0).lte(9007199254740991), hasWegoFare: z.boolean()}), legs: z.array(z.object({from: z.string(), to: z.string(), departsAt: z.string(), arrivesAt: z.string(), arrivalDayOffset: z.number(), overnight: z.boolean(), durationMinutes: z.number(), stops: z.number(), via: z.array(z.string()), layoverMinutesByStop: z.array(z.number()).optional(), airlines: z.array(z.object({code: z.string(), name: z.string(), logoUrl: z.string()})), operatingAirlines: z.array(z.object({code: z.string(), name: z.string(), logoUrl: z.string()})).optional(), aircraft: z.array(z.string()), transportTypes: z.array(z.enum(["FLIGHT", "TRAIN", "BUS", "OTHER"]))}))}))});

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
};

/**
 * Read ranked flight results
 * 
 * GET /v1/flights/searches/{searchId}/results
 * 
 * Also accepts auth: bearerAuth
 */
export const getApiV1FlightsSearchesSearchIdResults = async (args: GetApiV1FlightsSearchesSearchIdResultsArgs, options?: GetApiV1FlightsSearchesSearchIdResultsOptions) => {
const {baseUrl = "https://api.wego.com", fetch: fetchFn = fetch, oauth2} = options ?? {}

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

if (args.page != null) query.append("page", String(args.page))
if (args.pageSize != null) query.append("pageSize", String(args.pageSize))
if (args.sort != null) query.append("sort", String(args.sort))
for (const item of args.airlines ?? []) query.append("airlines", String(item))
if (args['airlines-match'] != null) query.append("airlines-match", String(args['airlines-match']))
if (args['same-airline'] != null) query.append("same-airline", String(args['same-airline']))
for (const item of args.alliances ?? []) query.append("alliances", String(item))
for (const item of args.stops ?? []) query.append("stops", String(item))
if (args['min-price'] != null) query.append("min-price", String(args['min-price']))
if (args['max-price'] != null) query.append("max-price", String(args['max-price']))
if (args['max-duration'] != null) query.append("max-duration", String(args['max-duration']))
if (args['min-stopover-duration'] != null) query.append("min-stopover-duration", String(args['min-stopover-duration']))
if (args['max-stopover-duration'] != null) query.append("max-stopover-duration", String(args['max-stopover-duration']))
for (const item of args['outbound-departure-blocks'] ?? []) query.append("outbound-departure-blocks", String(item))
if (args['outbound-departure-range'] != null) query.append("outbound-departure-range", String(args['outbound-departure-range']))
for (const item of args['outbound-arrival-blocks'] ?? []) query.append("outbound-arrival-blocks", String(item))
if (args['outbound-arrival-range'] != null) query.append("outbound-arrival-range", String(args['outbound-arrival-range']))
for (const item of args['return-departure-blocks'] ?? []) query.append("return-departure-blocks", String(item))
if (args['return-departure-range'] != null) query.append("return-departure-range", String(args['return-departure-range']))
for (const item of args['return-arrival-blocks'] ?? []) query.append("return-arrival-blocks", String(item))
if (args['return-arrival-range'] != null) query.append("return-arrival-range", String(args['return-arrival-range']))
if (args['outbound-min-duration'] != null) query.append("outbound-min-duration", String(args['outbound-min-duration']))
if (args['outbound-max-duration'] != null) query.append("outbound-max-duration", String(args['outbound-max-duration']))
if (args['return-min-duration'] != null) query.append("return-min-duration", String(args['return-min-duration']))
if (args['return-max-duration'] != null) query.append("return-max-duration", String(args['return-max-duration']))
for (const item of args['booking-types'] ?? []) query.append("booking-types", String(item))
for (const item of args['booking-sites'] ?? []) query.append("booking-sites", String(item))
for (const item of args['stopover-airports'] ?? []) query.append("stopover-airports", String(item))
for (const item of args.aircraft ?? []) query.append("aircraft", String(item))
if (args.currency != null) query.append("currency", String(args.currency))
if (args.locale != null) query.append("locale", String(args.locale))
if (args.view != null) query.append("view", String(args.view))
if (oauth2 !== undefined) headers.set('authorization', `Bearer ${oauth2}`)

const url = `${baseUrl.replace(/\/+$/, '')}/v1/flights/searches/${encodeURIComponent(String(args.searchId))}/results${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(getApiV1FlightsSearchesSearchIdResultsResponse, await res.json())
};