import {z} from 'zod'

export type GetApiQueryV2NotebooksCostOverviewArgs = {start_date: string, end_date: string, instance_id?: (number | null) | undefined, instance_ids?: (Array<number> | null) | undefined, account?: (string | null) | undefined, 'x-tenant': string, 'x-user-group-id'?: (string | null) | undefined};

export type GetApiQueryV2NotebooksCostOverviewOptions = {baseUrl: string, HTTPBearer?: string | undefined, fetch?: typeof fetch | undefined};

export const optionDescriptor = z.object({value: z.string(), label: z.string().nullable().optional()});

export const filterDescriptor = z.object({key: z.string(), label: z.string(), listable: z.boolean(), options: z.array(optionDescriptor).nullable().optional(), required: z.boolean().optional(), comparison: z.boolean().optional(), comparison_default: z.string().optional()});

export const relatedDescriptor = z.object({datasets: z.array(z.string()), charts: z.array(z.string()), summaries: z.array(z.string())});

export const paletteColorDescriptor = z.object({light: z.string(), dark: z.string()});

export const referenceLineDescriptor = z.object({of: z.string().nullable(), percentile: z.number().nullable(), label: z.string(), color: paletteColorDescriptor, value: z.number().nullable().optional()});

export const estimatePairDescriptor = z.object({actual: z.string(), estimate: z.string()});

export const chartDescriptor = z.object({name: z.string(), label: z.string(), dataset: z.string(), chart_type: z.string(), x: z.string(), series: z.array(z.string()), stacked: z.boolean(), filters: z.array(filterDescriptor), related: relatedDescriptor, reference: referenceLineDescriptor.nullable(), series_colors: z.record(z.string(), paletteColorDescriptor), series_from: z.string().nullable().optional(), show_change: z.boolean().optional(), show_total: z.boolean().optional(), show_zero: z.boolean().optional(), caption: z.string().nullable().optional(), stack_offset: z.string().optional(), secondary_axis_series: z.array(z.string()).optional(), estimate_pairs: z.array(estimatePairDescriptor).optional(), tooltip_series: z.array(z.string()).optional(), tooltip_estimate_pairs: z.array(estimatePairDescriptor).optional(), tooltip_decomposes: z.string().nullable().optional()});

export const chartColumn = z.object({key: z.string(), label: z.string(), type: z.string(), unit: z.string().nullable().optional(), order: z.number().int()});

export const referenceValue = z.object({label: z.string(), of: z.string().nullable(), value: z.number()});

export const chartMeta = z.object({row_count: z.number().int(), reference: referenceValue.nullable().optional(), last_event_time: z.string().nullable().optional()});

export const chartEnvelope = z.object({chart: chartDescriptor, columns: z.array(chartColumn), rows: z.array(z.record(z.string(), z.unknown())), meta: chartMeta});

export const metricValue = z.object({value: z.union([z.number(), z.number().int()]).nullable(), display: z.boolean(), label: z.string(), unit: z.string().nullable().optional(), info: z.string().nullable().optional(), order: z.number().int(), tone: z.string().nullable().optional(), compact: z.boolean().optional(), precision: z.number().int().nullable().optional()});

export const summaryEnvelope = z.object({metrics: z.record(z.string(), metricValue), last_event_time: z.string().nullable().optional()});

export const notebookCostDay = z.object({period: z.string(), status: z.enum(["available", "partial", "no_data", "restricted", "unavailable"]), unknown_credit_rows: z.number().int().nullable().optional(), unknown_cost_rows: z.number().int().nullable().optional(), unpriced_rows: z.number().int().nullable().optional()});

export const notebookCostSource = z.object({status: z.enum(["available", "partial", "no_data", "restricted", "unavailable"]), currency: z.string().nullable().optional(), source_rows: z.number().int().nullable().optional(), unknown_credit_rows: z.number().int().nullable().optional(), unknown_cost_rows: z.number().int().nullable().optional(), unpriced_rows: z.number().int().nullable().optional(), reason: z.string().nullable().optional(), days: z.array(notebookCostDay)});

export const notebookCostComparison = z.object({eligible: z.boolean(), currency: z.string().nullable().optional(), reason: z.string().nullable().optional()});

export const notebookCostsOverview = z.object({chart: chartEnvelope, summary: summaryEnvelope, sources: z.record(z.string(), notebookCostSource), comparison: notebookCostComparison});

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 Notebook Cost Overview
 * 
 * GET /query/v2/notebooks/cost-overview
 * 
 * Also accepts auth: HTTPBearer
 */
export const getApiQueryV2NotebooksCostOverview = async (args: GetApiQueryV2NotebooksCostOverviewArgs, options: GetApiQueryV2NotebooksCostOverviewOptions) => {
const {baseUrl, fetch: fetchFn = fetch, HTTPBearer} = options ?? {}

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

if (args.start_date != null) query.append("start_date", String(args.start_date))
if (args.end_date != null) query.append("end_date", String(args.end_date))
if (args.instance_id != null) query.append("instance_id", String(args.instance_id))
for (const item of args.instance_ids ?? []) query.append("instance_ids", String(item))
if (args.account != null) query.append("account", String(args.account))
if (args['x-tenant'] != null) headers.set("x-tenant", String(args['x-tenant']))
if (args['x-user-group-id'] != null) headers.set("x-user-group-id", String(args['x-user-group-id']))
if (HTTPBearer !== undefined) headers.set('authorization', `Bearer ${HTTPBearer}`)

const url = `${baseUrl.replace(/\/+$/, '')}/query/v2/notebooks/cost-overview${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(notebookCostsOverview, await res.json())
};