trading-cli 0.3.0__py3-none-any.whl

fastbooks_cli/commands/execution.py

@@ -0,0 +1,298 @@
+"""Execution command group — orders, balance, positions, leverage."""
+
+from __future__ import annotations
+
+import click
+
+from ..client import FastBooksClient
+from ..output import emit, emit_error, format_table
+
+VENUES = [
+    "binance", "bybit", "okx", "kraken", "coinbase", "kucoin",
+    "hyperliquid", "ibkr", "alpaca", "paper", "universal", "gmx",
+    "oanda", "oanda_paper", "saxo", "saxo_paper",
+]
+
+ORDER_TYPES = ["market", "limit", "stop", "stop_limit", "trailing_stop"]
+SIDES = ["buy", "sell"]
+
+
+@click.group()
+def execute():
+    """Order execution: place, cancel, list orders, balance, positions.
+
+    Supported venues: binance, bybit, okx, kraken, coinbase, kucoin,
+    hyperliquid, ibkr, alpaca, universal (internal OB).
+    """
+    pass
+
+
+# ── venues ───────────────────────────────────────────────────────────────
+
+@execute.command()
+@click.pass_context
+def venues(ctx):
+    """List all available venues and their capabilities.
+
+    Shows which operations each venue supports (orders, balance, positions).
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.venues()
+        emit(ctx, data, lambda d: _format_venues(d))
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+def _format_venues(data: dict) -> str:
+    items = data if isinstance(data, list) else data.get("venues", data.get("data", []))
+    if not isinstance(items, list):
+        return str(data)
+    return format_table(
+        items,
+        ["venue", "placeOrder", "cancelOrder", "getBalance", "getPositions"],
+        ["Venue", "Place", "Cancel", "Balance", "Positions"],
+    )
+
+
+# ── place order ──────────────────────────────────────────────────────────
+
+@execute.command()
+@click.argument("venue", type=click.Choice(VENUES, case_sensitive=False))
+@click.argument("symbol")
+@click.argument("side", type=click.Choice(SIDES, case_sensitive=False))
+@click.argument("order_type", type=click.Choice(ORDER_TYPES, case_sensitive=False))
+@click.argument("amount", type=float)
+@click.option("--price", "-p", type=float, default=None, help="Limit price (required for limit orders).")
+@click.option("--client-id", type=str, default=None, help="Client order ID for tracking.")
+@click.option("--confirm/--no-confirm", default=True, help="Require confirmation before placing.")
+@click.pass_context
+def order(ctx, venue, symbol, side, order_type, amount, price, client_id, confirm):
+    """Place an order on a venue.
+
+    VENUE: Exchange or broker to execute on.
+    SYMBOL: Trading pair (e.g. BTC/USDT, AAPL, ETH/USDC).
+    SIDE: buy or sell.
+    ORDER_TYPE: market, limit, stop, stop_limit, trailing_stop.
+    AMOUNT: Quantity to trade.
+
+    Examples:
+        fastbooks execute order binance BTC/USDT buy market 0.01
+        fastbooks execute order alpaca AAPL buy limit 10 --price 150.00
+        fastbooks execute order hyperliquid ETH/USDC sell market 1.5
+        fastbooks execute order ibkr MSFT buy market 100
+    """
+    client: FastBooksClient = ctx.obj
+
+    if order_type == "limit" and price is None:
+        emit_error(ctx, "Limit orders require --price")
+        ctx.exit(1)
+        return
+
+    if confirm and not ctx.find_root().params.get("json_output", False):
+        click.echo(f"\n  Order: {side.upper()} {amount} {symbol} @ {order_type} on {venue}")
+        if price:
+            click.echo(f"  Price: {price}")
+        if not click.confirm("\n  Confirm?"):
+            click.echo("  Cancelled.")
+            return
+
+    try:
+        data = client.place_order(venue, symbol, side, order_type, amount, price, client_id)
+        emit(ctx, data, lambda d: f"  Order placed: {d}")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── cancel order ─────────────────────────────────────────────────────────
+
+@execute.command()
+@click.argument("venue", type=click.Choice(VENUES, case_sensitive=False))
+@click.argument("order_id")
+@click.option("--symbol", "-s", type=str, default=None, help="Symbol (required for some venues).")
+@click.pass_context
+def cancel(ctx, venue, order_id, symbol):
+    """Cancel an open order.
+
+    Examples:
+        fastbooks execute cancel binance abc123 --symbol BTC/USDT
+        fastbooks execute cancel hyperliquid def456
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.cancel_order(venue, order_id, symbol)
+        emit(ctx, data, lambda d: f"  Order {order_id} cancelled.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── list orders ──────────────────────────────────────────────────────────
+
+@execute.command("orders")
+@click.argument("venue", type=click.Choice(VENUES, case_sensitive=False))
+@click.option("--symbol", "-s", type=str, default=None, help="Filter by symbol.")
+@click.pass_context
+def list_orders(ctx, venue, symbol):
+    """List open orders on a venue.
+
+    Examples:
+        fastbooks execute orders binance
+        fastbooks execute orders ibkr --symbol AAPL
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.list_orders(venue, symbol)
+        emit(ctx, data, lambda d: _format_orders(d))
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+def _format_orders(data) -> str:
+    items = data if isinstance(data, list) else data.get("orders", data.get("data", []))
+    if not isinstance(items, list):
+        return str(data)
+    if not items:
+        return "  No open orders."
+    return format_table(
+        items,
+        ["orderId", "symbol", "side", "type", "amount", "price", "status"],
+        ["Order ID", "Symbol", "Side", "Type", "Amount", "Price", "Status"],
+    )
+
+
+# ── get order ────────────────────────────────────────────────────────────
+
+@execute.command("order-status")
+@click.argument("venue", type=click.Choice(VENUES, case_sensitive=False))
+@click.argument("order_id")
+@click.option("--symbol", "-s", type=str, default=None)
+@click.pass_context
+def get_order(ctx, venue, order_id, symbol):
+    """Get status of a specific order.
+
+    Examples:
+        fastbooks execute order-status binance abc123
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.get_order(venue, order_id, symbol)
+        emit(ctx, data)
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── balance ──────────────────────────────────────────────────────────────
+
+@execute.command()
+@click.argument("venue", type=click.Choice(VENUES, case_sensitive=False))
+@click.option("--asset", "-a", type=str, default=None, help="Filter by asset (e.g. BTC, USD).")
+@click.pass_context
+def balance(ctx, venue, asset):
+    """Get account balance / holdings.
+
+    Examples:
+        fastbooks execute balance binance
+        fastbooks execute balance ibkr --asset USD
+        fastbooks execute balance alpaca
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.balance(venue, asset)
+        emit(ctx, data, lambda d: _format_balance(d))
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+def _format_balance(data) -> str:
+    items = data if isinstance(data, list) else data.get("balances", data.get("data", []))
+    if not isinstance(items, list):
+        return str(data)
+    return format_table(
+        items,
+        ["asset", "free", "locked", "total"],
+        ["Asset", "Free", "Locked", "Total"],
+    )
+
+
+# ── positions ────────────────────────────────────────────────────────────
+
+@execute.command()
+@click.argument("venue", type=click.Choice(VENUES, case_sensitive=False))
+@click.pass_context
+def positions(ctx, venue):
+    """Get open positions (perpetuals / margin).
+
+    Examples:
+        fastbooks execute positions hyperliquid
+        fastbooks execute positions binance
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.positions(venue)
+        emit(ctx, data, lambda d: _format_positions(d))
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+def _format_positions(data) -> str:
+    items = data if isinstance(data, list) else data.get("positions", data.get("data", []))
+    if not isinstance(items, list):
+        return str(data)
+    if not items:
+        return "  No open positions."
+    return format_table(
+        items,
+        ["symbol", "side", "size", "entryPrice", "unrealizedPnl", "leverage"],
+        ["Symbol", "Side", "Size", "Entry", "Unrealized PnL", "Leverage"],
+    )
+
+
+# ── leverage ─────────────────────────────────────────────────────────────
+
+@execute.command()
+@click.argument("venue", type=click.Choice(VENUES, case_sensitive=False))
+@click.argument("symbol")
+@click.argument("leverage_value", type=int)
+@click.option("--cross/--isolated", default=False, help="Cross or isolated margin mode.")
+@click.pass_context
+def leverage(ctx, venue, symbol, leverage_value, cross):
+    """Update leverage for a perpetual position.
+
+    Examples:
+        fastbooks execute leverage hyperliquid ETH/USDC 10
+        fastbooks execute leverage binance BTC/USDT 5 --cross
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.update_leverage(venue, symbol, leverage_value, cross)
+        emit(ctx, data, lambda d: f"  Leverage set to {leverage_value}x for {symbol} on {venue}")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── order history ────────────────────────────────────────────────────────
+
+@execute.command()
+@click.option("--venue", "-v", type=click.Choice(VENUES, case_sensitive=False), default="ibkr")
+@click.pass_context
+def history(ctx, venue):
+    """Get order history (filled, cancelled, rejected).
+
+    Examples:
+        fastbooks execute history --venue ibkr
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.order_history(venue)
+        emit(ctx, data, lambda d: _format_orders(d))
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)

fastbooks_cli/commands/init_cmd.py

@@ -0,0 +1,149 @@
+"""`init` — one-command onboarding for controlling FastBooks with an AI agent.
+
+Sets up, in ~/.fastbooks/config.json (0600): the FastBooks data key (so the bot is
+throttled), which exchanges are activated, per-exchange trading keys, and an MCP host
+config. Interactive by default; fully non-interactive for agents/scripts.
+"""
+
+from __future__ import annotations
+
+import json
+
+import click
+
+from ..client import FastBooksClient
+from ..output import emit, emit_error
+from .. import config_store
+
+# Curated activation catalog: (title, exchange ids, backend kind)
+CATALOG = [
+    ("Crypto CEX — spot + perps (ccxt: apiKey + secret [+ password])",
+     ["binance", "bybit", "okx", "kraken", "coinbase", "kucoin", "bitget", "gate", "mexc", "htx"], "ccxt"),
+    ("On-chain perps / DEX — via ccxt (hyperliquid/dydx/… use a wallet key as 'secret')",
+     ["hyperliquid", "dydx", "paradex", "apex", "woofipro"], "ccxt"),
+    ("Stocks — FastBooks platform (connect: `fastbooks --token <JWT> wallet add ...`)",
+     ["alpaca", "ibkr"], "platform"),
+    ("Forex — FastBooks platform",
+     ["oanda", "saxo"], "platform"),
+    ("Prediction markets — FastBooks platform (Polymarket tradable via `execute`)",
+     ["polymarket", "limitless"], "platform"),
+]
+
+
+def _kind_of(ex: str) -> str:
+    for _title, ids, kind in CATALOG:
+        if ex in ids:
+            return kind
+    return "ccxt"  # default: treat unknown as a ccxt exchange id
+
+
+@click.command("init")
+@click.option("--activate", "activate_csv", default=None, help="Comma list of exchanges to activate (non-interactive).")
+@click.option("--mint-key", is_flag=True, help="Mint a free FastBooks data key (x-api-key throttle).")
+@click.option("--api-key", "api_key", default=None, help="Store this existing FastBooks data key instead of minting.")
+@click.option("--mcp-config/--no-mcp-config", default=True, help="Write an MCP host config for agents (~/.fastbooks/mcp.json).")
+@click.option("--yes", "-y", is_flag=True, help="Non-interactive: apply flags with no prompts.")
+@click.pass_context
+def init_cmd(ctx, activate_csv, mint_key, api_key, mcp_config, yes):
+    """Set up FastBooks for your AI agent — data key, exchange activation, MCP config.
+
+    Interactive:      fastbooks init
+    Agent/scripted:   fastbooks --json init -y --mint-key --activate binance,hyperliquid,polymarket
+    """
+    client: FastBooksClient = ctx.obj
+    json_mode = bool(ctx.find_root().params.get("json_output"))
+    interactive = not (yes or json_mode or activate_csv or mint_key or api_key)
+
+    summary = {
+        "config_path": str(config_store.CONFIG_PATH),
+        "data_key": None,
+        "activated": [],
+        "credentials_set": [],
+        "mcp_config": None,
+        "next_steps": [],
+    }
+
+    # ── Step 1: FastBooks data key (per-bot throttle) ──
+    key = api_key
+    if not key and (mint_key or interactive and click.confirm(
+            "Mint a free FastBooks data key now (lets the platform rate-throttle your bot)?", default=True)):
+        try:
+            res = client.create_fastbooks_api_key("cli-agent")
+            key = res.get("apiKey") or res.get("api_key")
+            if interactive and key:
+                click.echo(f"  minted {key}")
+        except Exception as e:  # noqa: BLE001
+            if interactive:
+                click.echo(f"  (could not mint — {e}; is the backend up?)")
+    if interactive and not key:
+        entered = click.prompt("  Or paste an existing fb_live_ key (blank to skip)", default="", show_default=False)
+        key = entered.strip() or None
+    if key:
+        config_store.set_api_key(key)
+        summary["data_key"] = "set"
+
+    # ── Step 2: choose exchanges to activate ──
+    chosen: list[str] = []
+    if activate_csv:
+        chosen = [x.strip().lower() for x in activate_csv.split(",") if x.strip()]
+    elif interactive:
+        click.echo("\n  Available venues:")
+        for title, ids, _kind in CATALOG:
+            click.echo(f"  • {title}\n      {', '.join(ids)}")
+        raw = click.prompt("\n  Activate which? (comma list, 'all-cex', or blank)", default="", show_default=False)
+        if raw.strip().lower() == "all-cex":
+            chosen = list(CATALOG[0][1])
+        else:
+            chosen = [x.strip().lower() for x in raw.split(",") if x.strip()]
+
+    for ex in chosen:
+        config_store.activate(ex)
+        summary["activated"].append(ex)
+
+    # ── Step 3: trading keys for activated ccxt venues ──
+    for ex in chosen:
+        kind = _kind_of(ex)
+        if kind == "platform":
+            summary["next_steps"].append(
+                f"Connect {ex} (platform venue): fastbooks --token <JWT> wallet add {ex} --api-key ... --secret ..."
+            )
+            continue
+        if interactive and click.confirm(f"  Add trading keys for '{ex}' now? (skip = data-only)", default=False):
+            ak = click.prompt(f"    {ex} apiKey", default="", show_default=False).strip()
+            sk = click.prompt(f"    {ex} secret/wallet-key", default="", show_default=False, hide_input=True).strip()
+            pw = click.prompt(f"    {ex} password/passphrase (blank if none)", default="", show_default=False, hide_input=True).strip()
+            config_store.set_credentials(ex, ak or None, sk or None, pw or None)
+            summary["credentials_set"].append(ex)
+
+    # ── Step 4: MCP host config ──
+    if mcp_config:
+        env = {"FASTBOOKS_API_KEY": key} if key else {}
+        host_cfg = {"mcpServers": {"fastbooks": {"command": "fastbooks-mcp", **({"env": env} if env else {})}}}
+        config_store.CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+        path = config_store.CONFIG_DIR / "mcp.json"
+        path.write_text(json.dumps(host_cfg, indent=2))
+        summary["mcp_config"] = str(path)
+
+    summary["next_steps"] += [
+        "Data (keyless):    fastbooks --json direct ticker hyperliquid BTC/USDC:USDC",
+        "Trade (activated): fastbooks direct order binance BTC/USDT buy 0.001 --type limit --price 60000",
+        "Agent (MCP):       add ~/.fastbooks/mcp.json to your MCP host — or run `fastbooks-mcp`",
+    ]
+    emit(ctx, summary, _format)
+
+
+def _format(s: dict) -> str:
+    lines = [
+        "",
+        "  FastBooks onboarding complete",
+        "  =============================",
+        f"  Config:     {s['config_path']}",
+        f"  Data key:   {s['data_key'] or '(none — mint later with `fastbooks apikey create`)'}",
+        f"  Activated:  {', '.join(s['activated']) or '(none)'}",
+        f"  Keys set:   {', '.join(s['credentials_set']) or '(none — data-only)'}",
+        f"  MCP config: {s['mcp_config'] or '(skipped)'}",
+        "",
+        "  Next:",
+    ]
+    lines += [f"    • {n}" for n in s["next_steps"]]
+    return "\n".join(lines)