simstudioai--sim/apps/sim/tools/kalshi/get_orders.ts

import type {
  KalshiAuthParams,
  KalshiOrder,
  KalshiPaginationParams,
  KalshiPagingInfo,
} from '@/tools/kalshi/types'
import {
  buildKalshiAuthHeaders,
  buildKalshiUrl,
  handleKalshiError,
  KALSHI_ORDER_OUTPUT_PROPERTIES,
} from '@/tools/kalshi/types'
import type { ToolConfig } from '@/tools/types'

export interface KalshiGetOrdersParams extends KalshiAuthParams, KalshiPaginationParams {
  ticker?: string
  eventTicker?: string
  status?: string // resting, canceled, executed
}

export interface KalshiGetOrdersResponse {
  success: boolean
  output: {
    orders: KalshiOrder[]
    paging?: KalshiPagingInfo
  }
}

export const kalshiGetOrdersTool: ToolConfig<KalshiGetOrdersParams, KalshiGetOrdersResponse> = {
  id: 'kalshi_get_orders',
  name: 'Get Orders from Kalshi',
  description: 'Retrieve your orders from Kalshi with optional filtering',
  version: '1.0.0',

  params: {
    keyId: {
      type: 'string',
      required: true,
      visibility: 'user-only',
      description: 'Your Kalshi API Key ID',
    },
    privateKey: {
      type: 'string',
      required: true,
      visibility: 'user-only',
      description: 'Your RSA Private Key (PEM format)',
    },
    ticker: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'Filter by market ticker (e.g., "KXBTC-24DEC31")',
    },
    eventTicker: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description:
        'Filter by event ticker, max 10 comma-separated (e.g., "KXBTC-24DEC31,INX-25JAN03")',
    },
    status: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'Filter by order status: "resting", "canceled", or "executed"',
    },
    limit: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'Number of results to return (1-1000, default: 100)',
    },
    cursor: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'Pagination cursor from previous response for fetching next page',
    },
  },

  request: {
    url: (params) => {
      const queryParams = new URLSearchParams()
      if (params.ticker) queryParams.append('ticker', params.ticker)
      if (params.eventTicker) queryParams.append('event_ticker', params.eventTicker)
      if (params.status) queryParams.append('status', params.status)
      if (params.limit) queryParams.append('limit', params.limit)
      if (params.cursor) queryParams.append('cursor', params.cursor)

      const query = queryParams.toString()
      const url = buildKalshiUrl('/portfolio/orders')
      return query ? `${url}?${query}` : url
    },
    method: 'GET',
    headers: (params) => {
      const path = '/trade-api/v2/portfolio/orders'
      return buildKalshiAuthHeaders(params.keyId, params.privateKey, 'GET', path)
    },
  },

  transformResponse: async (response: Response) => {
    const data = await response.json()

    if (!response.ok) {
      handleKalshiError(data, response.status, 'get_orders')
    }

    const orders = data.orders || []

    return {
      success: true,
      output: {
        orders,
        paging: {
          cursor: data.cursor || null,
        },
      },
    }
  },

  outputs: {
    orders: {
      type: 'array',
      description: 'Array of order objects',
      items: {
        type: 'object',
        properties: KALSHI_ORDER_OUTPUT_PROPERTIES,
      },
    },
    paging: {
      type: 'object',
      description: 'Pagination cursor for fetching more results',
    },
  },
}

export interface KalshiGetOrdersV2Params extends KalshiAuthParams, KalshiPaginationParams {
  ticker?: string
  eventTicker?: string
  status?: string // resting, canceled, executed
  minTs?: string // Minimum timestamp filter (Unix timestamp)
  maxTs?: string // Maximum timestamp filter (Unix timestamp)
  subaccount?: string // Subaccount to filter orders
}

interface KalshiOrderV2 {
  order_id: string
  user_id: string | null
  client_order_id: string | null
  ticker: string
  side: string
  action: string
  type: string
  status: string
  yes_price: number | null
  no_price: number | null
  yes_price_dollars: string | null
  no_price_dollars: string | null
  fill_count: number | null
  fill_count_fp: string | null
  remaining_count: number | null
  remaining_count_fp: string | null
  initial_count: number | null
  initial_count_fp: string | null
  taker_fees: number | null
  maker_fees: number | null
  taker_fees_dollars: string | null
  maker_fees_dollars: string | null
  taker_fill_cost: number | null
  maker_fill_cost: number | null
  taker_fill_cost_dollars: string | null
  maker_fill_cost_dollars: string | null
  queue_position: number | null
  expiration_time: string | null
  created_time: string | null
  last_update_time: string | null
  self_trade_prevention_type: string | null
  order_group_id: string | null
  cancel_order_on_pause: boolean | null
}

export interface KalshiGetOrdersV2Response {
  success: boolean
  output: {
    orders: KalshiOrderV2[]
    cursor: string | null
  }
}

export const kalshiGetOrdersV2Tool: ToolConfig<KalshiGetOrdersV2Params, KalshiGetOrdersV2Response> =
  {
    id: 'kalshi_get_orders_v2',
    name: 'Get Orders from Kalshi V2',
    description:
      'Retrieve your orders from Kalshi with optional filtering (V2 with full API response)',
    version: '2.0.0',

    params: {
      keyId: {
        type: 'string',
        required: true,
        visibility: 'user-only',
        description: 'Your Kalshi API Key ID',
      },
      privateKey: {
        type: 'string',
        required: true,
        visibility: 'user-only',
        description: 'Your RSA Private Key (PEM format)',
      },
      ticker: {
        type: 'string',
        required: false,
        visibility: 'user-or-llm',
        description: 'Filter by market ticker (e.g., "KXBTC-24DEC31")',
      },
      eventTicker: {
        type: 'string',
        required: false,
        visibility: 'user-or-llm',
        description:
          'Filter by event ticker, max 10 comma-separated (e.g., "KXBTC-24DEC31,INX-25JAN03")',
      },
      status: {
        type: 'string',
        required: false,
        visibility: 'user-or-llm',
        description: 'Filter by order status: "resting", "canceled", or "executed"',
      },
      minTs: {
        type: 'string',
        required: false,
        visibility: 'user-or-llm',
        description: 'Minimum timestamp filter (Unix timestamp, e.g., "1704067200")',
      },
      maxTs: {
        type: 'string',
        required: false,
        visibility: 'user-or-llm',
        description: 'Maximum timestamp filter (Unix timestamp, e.g., "1704153600")',
      },
      subaccount: {
        type: 'string',
        required: false,
        visibility: 'user-or-llm',
        description: 'Subaccount identifier to filter orders',
      },
      limit: {
        type: 'string',
        required: false,
        visibility: 'user-or-llm',
        description: 'Number of results to return (1-1000, default: 100)',
      },
      cursor: {
        type: 'string',
        required: false,
        visibility: 'user-or-llm',
        description: 'Pagination cursor from previous response for fetching next page',
      },
    },

    request: {
      url: (params) => {
        const queryParams = new URLSearchParams()
        if (params.ticker) queryParams.append('ticker', params.ticker)
        if (params.eventTicker) queryParams.append('event_ticker', params.eventTicker)
        if (params.status) queryParams.append('status', params.status)
        if (params.minTs) queryParams.append('min_ts', params.minTs)
        if (params.maxTs) queryParams.append('max_ts', params.maxTs)
        if (params.subaccount) queryParams.append('subaccount', params.subaccount)
        if (params.limit) queryParams.append('limit', params.limit)
        if (params.cursor) queryParams.append('cursor', params.cursor)

        const query = queryParams.toString()
        const url = buildKalshiUrl('/portfolio/orders')
        return query ? `${url}?${query}` : url
      },
      method: 'GET',
      headers: (params) => {
        const path = '/trade-api/v2/portfolio/orders'
        return buildKalshiAuthHeaders(params.keyId, params.privateKey, 'GET', path)
      },
    },

    transformResponse: async (response: Response) => {
      const data = await response.json()

      if (!response.ok) {
        handleKalshiError(data, response.status, 'get_orders_v2')
      }

      const rawOrders = data.orders || []
      const orders: KalshiOrderV2[] = rawOrders.map((order: any) => ({
        order_id: order.order_id ?? null,
        user_id: order.user_id ?? null,
        client_order_id: order.client_order_id ?? null,
        ticker: order.ticker ?? null,
        side: order.side ?? null,
        action: order.action ?? null,
        type: order.type ?? null,
        status: order.status ?? null,
        yes_price: order.yes_price ?? null,
        no_price: order.no_price ?? null,
        yes_price_dollars: order.yes_price_dollars ?? null,
        no_price_dollars: order.no_price_dollars ?? null,
        fill_count: order.fill_count ?? null,
        fill_count_fp: order.fill_count_fp ?? null,
        remaining_count: order.remaining_count ?? null,
        remaining_count_fp: order.remaining_count_fp ?? null,
        initial_count: order.initial_count ?? null,
        initial_count_fp: order.initial_count_fp ?? null,
        taker_fees: order.taker_fees ?? null,
        maker_fees: order.maker_fees ?? null,
        taker_fees_dollars: order.taker_fees_dollars ?? null,
        maker_fees_dollars: order.maker_fees_dollars ?? null,
        taker_fill_cost: order.taker_fill_cost ?? null,
        maker_fill_cost: order.maker_fill_cost ?? null,
        taker_fill_cost_dollars: order.taker_fill_cost_dollars ?? null,
        maker_fill_cost_dollars: order.maker_fill_cost_dollars ?? null,
        queue_position: order.queue_position ?? null,
        expiration_time: order.expiration_time ?? null,
        created_time: order.created_time ?? null,
        last_update_time: order.last_update_time ?? null,
        self_trade_prevention_type: order.self_trade_prevention_type ?? null,
        order_group_id: order.order_group_id ?? null,
        cancel_order_on_pause: order.cancel_order_on_pause ?? null,
      }))

      return {
        success: true,
        output: {
          orders,
          cursor: data.cursor ?? null,
        },
      }
    },

    outputs: {
      orders: {
        type: 'array',
        description: 'Array of order objects with full API response fields',
        items: {
          type: 'object',
          properties: KALSHI_ORDER_OUTPUT_PROPERTIES,
        },
      },
      cursor: {
        type: 'string',
        description: 'Pagination cursor for fetching more results',
      },
    },
  }