import {z} from 'zod'

export type GetApiV1ViewTraderPubkeyArgs = {pubkey: string};

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

export const capabilityAccessView = z.object({immediate: z.boolean(), viaColdActivation: z.boolean().optional()});

export const traderCapabilitiesView = z.object({depositCollateral: capabilityAccessView, placeLimitOrder: capabilityAccessView, placeMarketOrder: capabilityAccessView, riskIncreasingTrade: capabilityAccessView, riskReducingTrade: capabilityAccessView, withdrawCollateral: capabilityAccessView});

export const traderCapabilityFlags = z.number().int().gte(0);

export const side = z.enum(["bid", "ask"]);

export const limitOrder = z.object({initialTradeSize: z.string(), isConditionalOrder: z.boolean().optional(), isReduceOnly: z.boolean(), isStopLoss: z.boolean().optional(), isStopLossDirection: z.boolean().optional(), marginFactor: z.string(), marginRequirement: z.string(), orderSequenceNumber: z.string(), price: z.string(), side: side, tradeSizeRemaining: z.string()});

export const traderPositionView = z.object({accumulatedFunding: z.string(), backstopMargin: z.string(), discountedUnrealizedPnl: z.string(), entryPrice: z.string(), initialMargin: z.string(), limitOrderMargin: z.string(), liquidationPrice: z.string(), maintenanceMargin: z.string(), positionInitialMargin: z.string(), positionSize: z.string(), positionValue: z.string(), stopLossPrice: z.string().nullable().optional(), symbol: z.string(), takeProfitPrice: z.string().nullable().optional(), unrealizedPnl: z.string(), unsettledFunding: z.string(), virtualQuotePosition: z.string()});

export const riskState = z.enum(["healthy", "unhealthy", "underwater", "zeroCollateralNoPositions"]);

export const riskTier = z.enum(["safe", "atRisk", "cancellable", "liquidatable", "backstopLiquidatable", "highRisk"]);

export const traderActivityStateView = z.enum(["uninitialized", "cold", "active", "reduceOnly", "frozen"]);

export const traderView = z.object({accumulatedFunding: z.string(), authority: z.string(), cancelMargin: z.string(), capabilities: traderCapabilitiesView, collateralBalance: z.string(), discountedUnrealizedPnl: z.string(), effectiveCollateral: z.string(), effectiveCollateralForWithdrawals: z.string(), flags: traderCapabilityFlags, initialMargin: z.string(), initialMarginForWithdrawals: z.string(), isInActiveTraders: z.boolean(), lastDepositSlot: z.number().int().gte(0), limitOrders: z.record(z.string(), z.array(limitOrder)), maintenanceMargin: z.string(), makerFeeOverrideMultiplier: z.number(), maxPositions: z.number().int().gte(0), numMarketsWithSplines: z.number().int().gte(0), portfolioValue: z.string(), positions: z.array(traderPositionView), riskState: riskState, riskTier: riskTier, slot: z.number().int().gte(0), slotIndex: z.number().int().gte(0), state: traderActivityStateView, takerFeeOverrideMultiplier: z.number(), traderKey: z.string(), traderPdaIndex: z.number().int().gte(0), traderSubaccountIndex: z.number().int().gte(0), unrealizedPnl: z.string(), unsettledFundingOwed: z.string()});

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 trader
 * 
 * GET /v1/view/trader/{pubkey}
 */
export const getApiV1ViewTraderPubkey = async (args: GetApiV1ViewTraderPubkeyArgs, options?: GetApiV1ViewTraderPubkeyOptions) => {
const {baseUrl = "https://perp-api.phoenix.trade", fetch: fetchFn = fetch, PhoenixBearerAuth} = options ?? {}

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

if (PhoenixBearerAuth !== undefined) headers.set('authorization', `Bearer ${PhoenixBearerAuth}`)

const url = `${baseUrl.replace(/\/+$/, '')}/v1/view/trader/${encodeURIComponent(String(args.pubkey))}${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(traderView, await res.json())
};