import {z} from 'zod'

export type InstrumentType = 'etf' | 'fund' | 'stock';

export type GetApiInstrumentsArgs = {cursor?: string | undefined, limit?: number | undefined, type?: InstrumentType | undefined, 'LMG-Data-Privacy-Access-Principal': string, 'LMG-Data-Privacy-Access-Justification': string};

export type GetApiInstrumentsOptions = {baseUrl?: string | undefined, bearerAuth?: string | undefined, fetch?: typeof fetch | undefined};

export const instrumentType = z.enum(["etf", "fund", "stock"]);

export const instrumentFundProfitModel = z.enum(["accumulating", "distributing"]);

export const instrumentRegulatoryInformationResponse = z.object({kid: z.string().optional()});

export const side = z.enum(["buy", "sell"]);

export const tradingHaltResponse = z.object({valid_from: z.string(), valid_to: z.string().optional(), side: side.optional()});

export const instrumentComplexity = z.enum(["non_complex", "complex"]);

export const knowledgeAndExperienceProfile = z.enum(["basic", "extended"]);

export const productCategory = z.enum(["non_complex_products", "structured_investment_products", "swap_based_ucits_and_alternative_investment_funds"]);

export const tradingVenue = z.enum(["TGAT", "AFND"]);

export const tradingMinimumQuantity = z.string();

export const marketTradingConfigResponse = z.object({order_type: z.literal("market"), venue: tradingVenue, minimum_quantity: tradingMinimumQuantity});

export const timeWithTimezone = z.object({time: z.string(), timezone: z.string()});

export const batchTradingConfigResponse = z.object({order_type: z.literal("batch"), venue: tradingVenue, minimum_quantity: tradingMinimumQuantity, batch_cutoff: timeWithTimezone, prefunding_cutoff: timeWithTimezone.optional()});

export const tradingConfigResponse = z.discriminatedUnion("order_type", [marketTradingConfigResponse, batchTradingConfigResponse]);

export const instrumentResponse = z.object({isin: z.string(), title: z.string(), type: instrumentType, fund_profit_model: instrumentFundProfitModel.optional(), total_expense_ratio_pct: z.string().optional(), regulatory_information: instrumentRegulatoryInformationResponse, active_trading_halts: z.array(tradingHaltResponse).optional(), complexity: instrumentComplexity, knowledge_and_experience_profile: knowledgeAndExperienceProfile.optional(), product_category: productCategory, days_to_settle: z.number().int().gte(0), trading: z.array(tradingConfigResponse)});

export const pagination = z.object({next_cursor: z.string().optional()});

export const pageInstrumentResponse = z.object({data: z.array(instrumentResponse), pagination: pagination});

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

/**
 * List Instruments
 * 
 * GET /instruments
 */
export const getApiInstruments = async (args: GetApiInstrumentsArgs, options?: GetApiInstrumentsOptions) => {
const {baseUrl = "https://sandbox.api.lemon.markets/v1", fetch: fetchFn = fetch, bearerAuth} = options ?? {}

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

if (args.cursor != null) query.append("cursor", String(args.cursor))
if (args.limit != null) query.append("limit", String(args.limit))
if (args.type != null) query.append("type", String(args.type))
if (args['LMG-Data-Privacy-Access-Principal'] != null) headers.set("LMG-Data-Privacy-Access-Principal", String(args['LMG-Data-Privacy-Access-Principal']))
if (args['LMG-Data-Privacy-Access-Justification'] != null) headers.set("LMG-Data-Privacy-Access-Justification", String(args['LMG-Data-Privacy-Access-Justification']))
if (bearerAuth !== undefined) headers.set('authorization', `Bearer ${bearerAuth}`)

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