PrivateNode

@classmethod
def new(
  cls, mnemonic: str | None = None, *, url: str = OEGS_GRPC_URL,
  rest_indexer: str = INDEXER_HTTP_URL,
  websocket_indexer: str = INDEXER_WS_URL,
):
  if mnemonic is None:
    mnemonic = os.environ['DYDX_MNEMONIC']
  node_client = make_node_client(url, rest_indexer=rest_indexer, websocket_indexer=websocket_indexer)
  return cls(node_client=node_client, mnemonic=mnemonic)

def build_order(
  self, wallet: Wallet, *, market: PerpetualMarket, order: Order,
  good_til_block: int | None = None, good_til_block_time: int | None = None,
  subaccount: int = 0,
):
  mkt = Market(market) # type: ignore
  client_id = order.get('client_id') or rand_id()
  order_id = mkt.order_id(
    address=wallet.address, subaccount_number=subaccount,
    client_id=client_id, order_flags=parse_flags(order['flags']),
  )
  return OrderProto(
    order_id=order_id,
    side=parse_side(order['side']),
    quantums=mkt.calculate_quantums(float(order['size'])),
    subticks=mkt.calculate_subticks(float(order['price'])),
    good_til_block=good_til_block,
    good_til_block_time=good_til_block_time,
    time_in_force=parse_tif(order.get('time_in_force')),
    reduce_only=order.get('reduce_only') or False,
    client_metadata=order.get('client_metadata'),
    condition_type=order.get('condition_type'),
    conditional_order_trigger_subticks=order.get('conditional_order_trigger_subticks'),
  )

async def build_order_now(
  self, market: PerpetualMarket, order: Order, *, subaccount: int = 0,
  gtb_delta: int | None = None, gtbt_delta: int | None = None
):
  if (gtb := order.get('good_til_block')) is None and order['flags'] == 'SHORT_TERM':
    latest_block = await self.node_client.latest_block()
    if gtb_delta is None:
      gtb_delta = SHORT_BLOCK_WINDOW
    gtb = latest_block.block.header.height + gtb_delta

  if (gtbt := order.get('good_til_block_time')) is None and order['flags'] == 'LONG_TERM':
    latest_block = await self.node_client.latest_block()
    if gtbt_delta is None:
      gtbt_delta = STATEFUL_ORDER_TIME_WINDOW
    gtbt = latest_block.block.header.time.seconds + gtbt_delta
    gtb = None

  wallet = await self.wallet
  return self.build_order(wallet=wallet, market=market, order=order, good_til_block=gtb, good_til_block_time=gtbt, subaccount=subaccount)

async def place_order(
  self, market: PerpetualMarket, order: Order, *, subaccount: int = 0,
  mode: BroadcastMode = BroadcastMode.BROADCAST_MODE_SYNC, tx_options: TxOptions | None = None,
  gtb_delta: int | None = None, gtbt_delta: int | None = None,
  good_til_block: int | None = None, good_til_block_time: int | None = None,
) -> OrderResponse:
  """Place an order.

  - `market`: market to place the order on
  - `order`: order to place
  - `subaccount`: subaccount to place the order on
  - `gtb_delta`: sets good-til-block for short-term orders to `current_block() + gtb_delta`
  - `gtbt_delta`: sets good-til-block-time (in seconds) for long-term-orders to  `current_block().time.seconds + gtbt_delta`

  > [dYdX API docs](https://docs.dydx.xyz/node-client/private#place-order)
  """
  wallet = await self.wallet
  if good_til_block is not None or good_til_block_time is not None:
    order_proto = self.build_order(wallet=wallet, market=market, order=order, good_til_block=good_til_block, good_til_block_time=good_til_block_time, subaccount=subaccount)
  else:
    order_proto = await self.build_order_now(market=market, order=order, subaccount=subaccount, gtb_delta=gtb_delta, gtbt_delta=gtbt_delta)
  tx: BroadcastTxResponse = await self.node_client.broadcast_message(
    wallet, MsgPlaceOrder(order=order_proto), # type: ignore
    mode=mode, tx_options=tx_options
  )
  if tx.tx_response.code != 0:
    raise ApiError(tx.tx_response.code, tx.tx_response, tx.tx_response.raw_log)
  return {
    'tx': tx,
    'order': order_proto,
  }

async def cancel_order(
  self, order_id: OrderId, *,
  good_til_block: int | None = None,
  good_til_block_time: int | None = None,
  mode: BroadcastMode = BroadcastMode.BROADCAST_MODE_SYNC,
  tx_options: TxOptions | None = None,
) -> BroadcastTxResponse:
  """Cancel an order.

  - `order_id`: order to cancel
  - `good_til_block`: block number to cancel the order at (defaults to the current one)
  - `good_til_block_time`: timestamp to cancel the order at (defaults to now)

  > [dYdX API docs](https://docs.dydx.xyz/node-client/private#cancel-order)
  """
  # SHORT-TERM orders required good_til_block
  if good_til_block is None and order_id.order_flags == OrderFlags.SHORT_TERM:
    latest_block = await self.node_client.latest_block()
    good_til_block = latest_block.block.header.height + SHORT_BLOCK_WINDOW
  # LONG-TERM orders required good_til_block_time
  elif good_til_block_time is None and order_id.order_flags == OrderFlags.LONG_TERM:
    latest_block = await self.node_client.latest_block()
    good_til_block_time = latest_block.block.header.time.seconds + STATEFUL_ORDER_TIME_WINDOW

  wallet = await self.wallet
  tx = MsgCancelOrder(order_id=order_id, good_til_block=good_til_block, good_til_block_time=good_til_block_time)
  r: BroadcastTxResponse = await self.node_client.broadcast_message(
    wallet, tx, mode=mode, tx_options=tx_options # type: ignore
  )
  if r.tx_response.code != 0:
    raise ApiError(r.tx_response.code, r.tx_response)
  return r

async def batch_cancel_orders(
  self, order_ids: Sequence[OrderId], *,
  good_til_block: int | None = None,
) -> BroadcastTxResponse:
  """Cancel an order.

  - `order_ids`: orders to cancel
  - `good_til_block`: block number to cancel the orders at (defaults to the current one)

  > [dYdX API docs](https://docs.dydx.xyz/node-client/private#cancel-order)
  """
  subaccount_id = order_ids[0].subaccount_id
  orders_by_pair = defaultdict[int, list[int]](list)
  for order_id in order_ids:
    if order_id.order_flags != OrderFlags.SHORT_TERM:
      raise BadRequest('Only short-term orders can be cancelled in a batch')
    orders_by_pair[order_id.clob_pair_id].append(order_id.client_id)

  batches = [
    OrderBatch(clob_pair_id=pair, client_ids=orders)
    for pair, orders in orders_by_pair.items()
  ]

  if good_til_block is None:
    latest_block = await self.node_client.latest_block()
    good_til_block = latest_block.block.header.height + SHORT_BLOCK_WINDOW

  wallet = await self.wallet
  r: BroadcastTxResponse = await self.node_client.batch_cancel_orders(
    wallet=wallet, subaccount_id=subaccount_id,
    short_term_cancels=batches,
    good_til_block=good_til_block,
  )
  if r.tx_response.code != 0:
    raise ApiError(r.tx_response.code, r.tx_response)
  return r