import {z} from 'zod'

export type GetApiPoolsArgs = {'accept-version'?: ('stable' | 'next') | undefined, address?: string | undefined, name?: string | undefined, priceOracleAddress?: string | undefined, chainId?: string | undefined, order?: string | undefined, limit?: number | undefined, page?: number | undefined};

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

export const rewardsDistributor = z.object({id: z.string(), rewardTokenAddress: z.string(), marketAddress: z.string(), chainId: z.string(), lastRewardingSupplyBlockOrTimestamp: z.string(), lastRewardingBorrowBlockOrTimestamp: z.string(), supplySpeed: z.string(), borrowSpeed: z.string(), priceMantissa: z.string().optional(), rewardsDistributorContractAddress: z.string(), rewardType: z.enum(["venus", "merkl", "intrinsic", "off-chain", "yield-to-maturity"]), rewardDetails: z.object({name: z.string().optional(), description: z.string().optional(), merklCampaignId: z.string().optional(), apr: z.string().optional(), apy: z.string().optional()}).nullable().optional(), isActive: z.boolean()});

export const pointsDistribution = z.object({id: z.string(), pointsProgram: z.enum(["ethena", "etherfi", "kelp", "solv", "aster"]), marketAddress: z.string(), chainId: z.string(), action: z.enum(["supply", "borrow"]), title: z.string(), incentive: z.string().nullable().optional(), description: z.string().nullable().optional(), extraInfoUrl: z.string().nullable().optional(), startDate: z.string().nullable().optional(), endDate: z.string().nullable().optional()});

export const eModeSetting = z.object({id: z.string(), chainId: z.string(), marketAddress: z.string(), poolId: z.number().int(), collateralFactorMantissa: z.string(), liquidationThresholdMantissa: z.string(), liquidationIncentiveMantissa: z.string(), canBeCollateral: z.boolean(), isBorrowable: z.boolean()});

export const tokenPrice = z.object({id: z.string(), tokenAddress: z.string(), tokenWrappedAddress: z.string().nullable().optional(), priceMantissa: z.string().nullable().optional(), chainId: z.string(), priceSource: z.enum(["oracle", "coingecko", "merkl"]), priceOracleAddress: z.string().nullable().optional(), mainOracleAddress: z.string().nullable().optional(), mainOracleName: z.string().nullable().optional(), isPriceInvalid: z.boolean().optional(), hasErrorFetchingPrice: z.boolean().optional()});

export const tokenMetadata = z.object({address: z.string(), name: z.string().nullable().optional(), symbol: z.string().nullable().optional(), chainId: z.string(), decimals: z.number().int(), maturityDate: z.string().nullable().optional(), tokenPrices: z.array(tokenPrice).optional()});

export const market = z.object({address: z.string(), chainId: z.string(), symbol: z.string().optional(), name: z.string().optional(), underlyingAddress: z.string().nullable().optional(), underlyingName: z.string().optional(), underlyingSymbol: z.string().optional(), underlyingDecimal: z.number().int().optional(), borrowerDailyXvsMantissa: z.string().nullable().optional(), supplierDailyXvsMantissa: z.string().nullable().optional(), xvsBorrowIndex: z.string().nullable().optional(), xvsSupplyIndex: z.string().nullable().optional(), borrowRatePerBlock: z.string().optional(), supplyRatePerBlock: z.string().optional(), exchangeRateMantissa: z.string().optional(), underlyingPriceMantissa: z.string().optional(), totalBorrowsMantissa: z.string().optional(), totalSupplyMantissa: z.string().optional(), cashMantissa: z.string().optional(), totalReservesMantissa: z.string().optional(), reserveFactorMantissa: z.string().optional(), collateralFactorMantissa: z.string().optional(), liquidationThresholdMantissa: z.string().optional(), liquidationIncentiveMantissa: z.string().optional(), isBorrowable: z.boolean().optional(), canBeCollateral: z.boolean().optional(), borrowApy: z.string().optional(), supplyApy: z.string().optional(), borrowApyDecimal: z.string().nullable().optional(), supplyApyDecimal: z.string().nullable().optional(), totalBorrowApyDecimal: z.string().nullable().optional(), totalSupplyApyDecimal: z.string().nullable().optional(), borrowXvsApy: z.string().nullable().optional(), supplyXvsApy: z.string().nullable().optional(), borrowXvsApr: z.string().nullable().optional(), supplyXvsApr: z.string().nullable().optional(), totalSupplyUnderlyingCents: z.number().int().optional(), totalBorrowCents: z.number().int().optional(), liquidityCents: z.string().optional(), tokenPriceCents: z.string().optional(), totalDistributedMantissa: z.string().nullable().optional(), lastCalculatedXvsAccruedBlockNumber: z.string().nullable().optional(), supplyCapsMantissa: z.string().nullable().optional(), borrowCapsMantissa: z.string().optional(), borrowerCount: z.number().int().optional(), supplierCount: z.number().int().optional(), estimatedPrimeBorrowApyBoost: z.string().nullable().optional(), estimatedPrimeSupplyApyBoost: z.string().nullable().optional(), pausedActionsBitmap: z.number().int().optional(), isListed: z.boolean().optional(), poolComptrollerAddress: z.string(), badDebtMantissa: z.string().optional(), isPriceInvalid: z.boolean().optional(), rewardsDistributors: z.array(rewardsDistributor).optional(), pointsDistributions: z.array(pointsDistribution).optional(), eModeSettings: z.array(eModeSetting).optional(), underlyingTokenMetadata: z.array(tokenMetadata).optional()});

export const eModeGroup = z.object({id: z.string(), comptrollerAddress: z.string(), chainId: z.string(), label: z.string(), poolId: z.number().int(), isActive: z.boolean(), allowCorePoolFallback: z.boolean(), eModeSettings: z.array(eModeSetting).optional()});

export const pool = z.object({chainId: z.string(), address: z.string(), name: z.string(), description: z.string().nullable().optional(), priceOracleAddress: z.string(), closeFactorMantissa: z.string().nullable().optional(), liquidationIncentiveMantissa: z.string().nullable().optional(), minLiquidatableCollateralMantissa: z.string().nullable().optional(), markets: z.array(market).optional(), eModeGroups: z.array(eModeGroup).optional()});

export const poolsStableResponse = z.object({limit: z.number(), page: z.number(), total: z.number(), result: z.array(pool), tokens: z.array(tokenMetadata)});

export const poolsNextResponse = z.object({pools: z.record(z.string(), z.array(pool)), markets: z.object({limit: z.number(), page: z.number(), total: z.number(), result: z.array(market)})});

export const getApiPoolsResponse = z.union([poolsStableResponse, poolsNextResponse]);

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

/**
 * Returns pool data for all listed pools, optionally filterable by the pool's comptroller address.
 * 
 * GET /pools
 */
export const getApiPools = async (args: GetApiPoolsArgs, options: GetApiPoolsOptions) => {
const {baseUrl, fetch: fetchFn = fetch} = options ?? {}

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

if (args['accept-version'] != null) headers.set("accept-version", String(args['accept-version']))
if (args.address != null) query.append("address", String(args.address))
if (args.name != null) query.append("name", String(args.name))
if (args.priceOracleAddress != null) query.append("priceOracleAddress", String(args.priceOracleAddress))
if (args.chainId != null) query.append("chainId", String(args.chainId))
if (args.order != null) query.append("order", String(args.order))
if (args.limit != null) query.append("limit", String(args.limit))
if (args.page != null) query.append("page", String(args.page))

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