import {z} from 'zod'

export type OrderID = string;

export type OrderHandle = string;

export type MarketSymbol = string;

export type TradingAccountId = string;

export type CancelOrderCommandV3 = {commandType: 'V3CancelOrder', orderId?: OrderID | undefined, clientOrderId?: OrderHandle | undefined, symbol: MarketSymbol, tradingAccountId: TradingAccountId};

export type CancelAllOrdersCommand = {commandType: 'V1CancelAllOrders', tradingAccountId: TradingAccountId};

export type CancelAllOrdersByMarketCommand = {commandType: 'V1CancelAllOrdersByMarket', symbol: MarketSymbol, tradingAccountId: TradingAccountId};

export type CancelId = string;

export type DelayBySeconds = '5' | '10' | '15' | '20' | '25' | '30' | '40' | '50' | '60';

export type DelayedCancelAllOrdersCommand = {commandType: 'V1DelayedCancelAllOrders', cancelId?: CancelId | undefined, delayBySeconds: DelayBySeconds, tradingAccountId: TradingAccountId};

export type UnsetDelayedCancelAllOrdersCommand = {commandType: 'V1UnsetDelayedCancelAllOrders', tradingAccountId: TradingAccountId};

export type AMMInstructionID = string;

export type TerminateAMMInstructionCommandV3 = {commandType: 'V3TerminateAMMInstruction', instructionId: AMMInstructionID, symbol: MarketSymbol, tradingAccountId: TradingAccountId};

export type AssetSymbol = string;

export type TransferAssetCommandV2 = {commandType: 'V2TransferAsset', assetSymbol: AssetSymbol, quantity: string, fromTradingAccountId: TradingAccountId, toTradingAccountId: TradingAccountId};

export type OrderTypeAsStringAmend = string;

export type AssetValue = string;

export type AmendOrderCommandV1 = {commandType: 'V1AmendOrder', orderId?: OrderID | undefined, symbol: MarketSymbol, type?: OrderTypeAsStringAmend | undefined, price?: AssetValue | undefined, clientOrderId?: string | undefined, quantity?: AssetValue | undefined, tradingAccountId: TradingAccountId};

export type CreateApiV2CommandArgs = {'BX-SIGNATURE': string, 'BX-TIMESTAMP': string, 'BX-NONCE': string, 'BX-NONCE-WINDOW-ENABLED'?: ('false' | 'true') | undefined, body: CancelOrderCommandV3 | CancelAllOrdersCommand | CancelAllOrdersByMarketCommand | DelayedCancelAllOrdersCommand | UnsetDelayedCancelAllOrdersCommand | TerminateAMMInstructionCommandV3 | TransferAssetCommandV2 | AmendOrderCommandV1};

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

export const requestID = z.string();

export const orderID = z.string();

export const cancelOrderCommandResponseV3 = z.object({commandType: z.literal("V3CancelOrder"), message: z.string(), requestId: requestID, orderId: orderID.optional(), clientOrderId: z.string().optional()});

export const cancelAllOrdersCommandResponse = z.object({commandType: z.literal("V1CancelAllOrders"), message: z.string(), requestId: requestID});

export const cancelAllOrdersByMarketResponse = z.object({commandType: z.literal("V1CancelAllOrdersByMarket"), message: z.string(), requestId: requestID});

export const delayedCancelAllOrdersResponse = z.object({commandType: z.literal("V1DelayedCancelAllOrders"), message: z.string(), requestId: requestID});

export const unsetDelayedCancelAllOrdersResponse = z.object({commandType: z.literal("V1UnsetDelayedCancelAllOrders"), message: z.string(), requestId: requestID});

export const aMMInstructionID = z.string();

export const terminateAMMInstructionCommandResponseV3 = z.object({commandType: z.literal("V3TerminateAMMInstruction"), message: z.string(), requestId: requestID, instructionId: aMMInstructionID});

export const tradingAccountTransferResponse = z.object({commandType: z.literal("V2TransferAsset"), message: z.string(), requestId: requestID});

export const amendOrderCommandResponseV1 = z.object({commandType: z.literal("V1AmendOrder"), message: z.string(), requestId: requestID, orderId: orderID, clientOrderId: z.string().optional()});

export const createApiV2CommandResponse = z.discriminatedUnion("commandType", [cancelOrderCommandResponseV3, cancelAllOrdersCommandResponse, cancelAllOrdersByMarketResponse, delayedCancelAllOrdersResponse, unsetDelayedCancelAllOrdersResponse, terminateAMMInstructionCommandResponseV3, tradingAccountTransferResponse, amendOrderCommandResponseV1]);

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

/**
 * Create Command
 * 
 * POST /v2/command
 */
export const createApiV2Command = async (args: CreateApiV2CommandArgs, options?: CreateApiV2CommandOptions) => {
const {baseUrl = "https://api.exchange.bullish.com/trading-api", fetch: fetchFn = fetch, jwtTokenAuth} = options ?? {}

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

if (args['BX-SIGNATURE'] != null) headers.set("BX-SIGNATURE", String(args['BX-SIGNATURE']))
if (args['BX-TIMESTAMP'] != null) headers.set("BX-TIMESTAMP", String(args['BX-TIMESTAMP']))
if (args['BX-NONCE'] != null) headers.set("BX-NONCE", String(args['BX-NONCE']))
if (args['BX-NONCE-WINDOW-ENABLED'] != null) headers.set("BX-NONCE-WINDOW-ENABLED", String(args['BX-NONCE-WINDOW-ENABLED']))
if (jwtTokenAuth !== undefined) headers.set('authorization', `Bearer ${jwtTokenAuth}`)

const url = `${baseUrl.replace(/\/+$/, '')}/v2/command${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "POST", headers, body: JSON.stringify(args.body)})

if (!res.ok) throw new Error(`Request failed with status ${res.status}`, {cause: {status: res.status, body: await res.text()}})

return validateResponse(createApiV2CommandResponse, await res.json())
};