trading-cli 0.3.0__py3-none-any.whl

fastbooks_cli/commands/swap.py

@@ -0,0 +1,320 @@
+"""Swap command group — on-chain DEX swap execution (Jupiter, Uniswap, SushiSwap)."""
+
+from __future__ import annotations
+
+import json
+import os
+import time
+
+import click
+
+from ..client import FastBooksClient
+from ..output import emit, emit_error, format_table, require_experimental
+
+# ── Solana token address registry ────────────────────────────────────────
+
+SOLANA_TOKENS = {
+    "SOL": "So11111111111111111111111111111111111111112",
+    "USDC": "EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v",
+    "USDT": "Es9vMFrzaCERmJfrF4H2FYD4KCoNkY11McCe8BenwNYB",
+    "BONK": "DezXAZ8z7PnrnRJjz3wXBoRgixCa6xjnB7YaB1pPB263",
+    "JUP": "JUPyiwrYJFskUPiHa7hkeR8VUtAeFoSYbKedZNsDvCN",
+    "RAY": "4k3Dyjzvzp8eMZWUXbBCjEvwSkkk59S5iCNLY3QrkX6R",
+    "ORCA": "orcaEKTdK7LKz57vaAYr9QeNsVEPfiu6QeMU1kektZE",
+    "MSOL": "mSoLzYCxHdYgdzU16g5QSh3i5K3z3KZK7ytfqcJm7So",
+    "JITOSOL": "J1toso1uCk3RLmjorhTtrVwY9HJ7X8V9yYac6Y7kGCPn",
+    "WIF": "EKpQGSJtjMFqKZ9KQanSqYXRcF8fBopzLHYxdM65zcjm",
+}
+
+# ── EVM token address registry (Ethereum mainnet) ───────────────────────
+
+EVM_TOKENS = {
+    "ETH": "0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE",  # native
+    "WETH": "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2",
+    "USDC": "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48",
+    "USDT": "0xdAC17F958D2ee523a2206206994597C13D831ec7",
+    "DAI": "0x6B175474E89094C44Da98b954EedeAC495271d0F",
+    "WBTC": "0x2260FAC5E5542a773Aa44fBCfeDf7C193bc2C599",
+    "UNI": "0x1f9840a85d5aF5bf1D1762F925BDADdC4201F984",
+    "LINK": "0x514910771AF9Ca656af840dff83E8264EcF986CA",
+    "AAVE": "0x7Fc66500c84A76Ad7e9c93437bFc5Ac33E2DDaE9",
+    "ARB": "0xB50721BCf8d664c30412Cfbc6cf7a15145234ad1",
+}
+
+CHAINS = ["solana", "ethereum", "arbitrum", "polygon", "base", "optimism"]
+
+
+@click.group()
+@click.pass_context
+def swap(ctx):
+    """[experimental] On-chain DEX swaps: Jupiter (Solana), Uniswap (EVM), SushiSwap.
+
+    NOTE: requires an on-chain swap router not enabled on standard FastBooks
+    deployments. Gated behind FASTBOOKS_EXPERIMENTAL=1.
+
+    Execute token swaps directly on-chain through DEX aggregators.
+    Requires a wallet private key (stored via `fastbooks wallet add`).
+
+    Supported DEX aggregators:
+        Jupiter:   Solana tokens (5000+). Best price routing.
+        Uniswap:   Ethereum/L2 tokens. V3 concentrated liquidity.
+        SushiSwap: Multi-chain (Ethereum, Arbitrum, Polygon).
+    """
+    require_experimental(ctx, "swap")
+
+
+# ── Jupiter swap (Solana) ───────────────────────────────────────────────
+
+@swap.command("jupiter")
+@click.argument("input_token", type=str)
+@click.argument("output_token", type=str)
+@click.argument("amount", type=float)
+@click.option("--slippage", type=float, default=1.0, help="Max slippage in percent (default: 1%).")
+@click.option("--wallet-key", envvar="SOLANA_PRIVATE_KEY", default=None, help="Solana private key (or set SOLANA_PRIVATE_KEY).")
+@click.option("--rpc-url", envvar="SOLANA_RPC_URL", default="https://api.mainnet-beta.solana.com", help="Solana RPC URL.")
+@click.option("--quote-only", is_flag=True, help="Only get quote, don't execute swap.")
+@click.option("--priority-fee", type=int, default=50000, help="Priority fee in micro-lamports.")
+@click.pass_context
+def jupiter_swap(ctx, input_token, output_token, amount, slippage, wallet_key, rpc_url, quote_only, priority_fee):
+    """Swap tokens on Solana via Jupiter aggregator.
+
+    INPUT_TOKEN: Token to sell (symbol like SOL, USDC, or mint address).
+    OUTPUT_TOKEN: Token to buy (symbol or mint address).
+    AMOUNT: Amount of input token to swap.
+
+    Examples:
+        fastbooks swap jupiter SOL USDC 1.5
+        fastbooks swap jupiter USDC BONK 100 --slippage 2.0
+        fastbooks swap jupiter SOL JUP 10 --quote-only
+        fastbooks swap jupiter So111...112 EPjF...1v 5.0
+    """
+    client: FastBooksClient = ctx.obj
+
+    input_mint = SOLANA_TOKENS.get(input_token.upper(), input_token)
+    output_mint = SOLANA_TOKENS.get(output_token.upper(), output_token)
+    slippage_bps = int(slippage * 100)
+
+    try:
+        # Get quote from Jupiter
+        quote = client.jupiter_quote(input_mint, output_mint, amount, slippage_bps)
+
+        if quote_only:
+            emit(ctx, quote, lambda d: _format_jupiter_quote(d, input_token, output_token, amount))
+            return
+
+        if not wallet_key:
+            emit_error(ctx, "Wallet key required for swap execution. Use --wallet-key or set SOLANA_PRIVATE_KEY.")
+            ctx.exit(1)
+            return
+
+        # Confirm before executing
+        if not ctx.find_root().params.get("json_output", False):
+            out_amount = quote.get("outAmount", quote.get("expectedOutput", "?"))
+            click.echo(f"\n  Swap: {amount} {input_token} -> {out_amount} {output_token}")
+            click.echo(f"  Slippage: {slippage}%  Priority fee: {priority_fee} uL")
+            if not click.confirm("\n  Execute swap?"):
+                click.echo("  Cancelled.")
+                return
+
+        result = client.jupiter_swap_execute(
+            input_mint=input_mint,
+            output_mint=output_mint,
+            amount=amount,
+            slippage_bps=slippage_bps,
+            wallet_key=wallet_key,
+            priority_fee=priority_fee,
+        )
+        emit(ctx, result, lambda d: _format_swap_result(d, "Jupiter"))
+
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+def _format_jupiter_quote(data: dict, input_tok: str, output_tok: str, amount: float) -> str:
+    lines = [
+        "",
+        "  Jupiter Quote",
+        "  =============",
+        f"  Input:    {amount} {input_tok}",
+        f"  Output:   {data.get('outAmount', data.get('expectedOutput', '?'))} {output_tok}",
+        f"  Price Impact: {data.get('priceImpactPct', '?')}%",
+    ]
+    routes = data.get("routePlan", [])
+    if routes:
+        lines.append(f"  Route: {' -> '.join(r.get('swapInfo', {}).get('label', '?') for r in routes)}")
+    return "\n".join(lines)
+
+
+# ── Uniswap swap (EVM) ──────────────────────────────────────────────────
+
+@swap.command("uniswap")
+@click.argument("input_token", type=str)
+@click.argument("output_token", type=str)
+@click.argument("amount", type=float)
+@click.option("--chain", type=click.Choice(["ethereum", "arbitrum", "polygon", "base", "optimism"]), default="ethereum", help="EVM chain.")
+@click.option("--slippage", type=float, default=0.5, help="Max slippage in percent.")
+@click.option("--wallet-key", envvar="EVM_PRIVATE_KEY", default=None, help="EVM private key (or set EVM_PRIVATE_KEY).")
+@click.option("--rpc-url", envvar="EVM_RPC_URL", default=None, help="EVM RPC URL.")
+@click.option("--quote-only", is_flag=True, help="Only get quote, don't execute.")
+@click.option("--gas-limit", type=int, default=None, help="Gas limit override.")
+@click.pass_context
+def uniswap_swap(ctx, input_token, output_token, amount, chain, slippage, wallet_key, rpc_url, quote_only, gas_limit):
+    """Swap tokens on EVM chains via Uniswap V3 / compatible routers.
+
+    INPUT_TOKEN: Token to sell (symbol like ETH, USDC, or contract address).
+    OUTPUT_TOKEN: Token to buy (symbol or contract address).
+    AMOUNT: Amount of input token to swap.
+
+    Examples:
+        fastbooks swap uniswap ETH USDC 1.0
+        fastbooks swap uniswap USDC WBTC 5000 --chain ethereum
+        fastbooks swap uniswap ETH ARB 2.0 --chain arbitrum --slippage 1.0
+        fastbooks swap uniswap USDC DAI 1000 --quote-only
+    """
+    client: FastBooksClient = ctx.obj
+
+    input_address = EVM_TOKENS.get(input_token.upper(), input_token)
+    output_address = EVM_TOKENS.get(output_token.upper(), output_token)
+    slippage_bps = int(slippage * 100)
+
+    try:
+        quote = client.uniswap_quote(
+            input_address, output_address, amount,
+            chain=chain, slippage_bps=slippage_bps,
+        )
+
+        if quote_only:
+            emit(ctx, quote, lambda d: _format_uniswap_quote(d, input_token, output_token, amount, chain))
+            return
+
+        if not wallet_key:
+            emit_error(ctx, "Wallet key required. Use --wallet-key or set EVM_PRIVATE_KEY.")
+            ctx.exit(1)
+            return
+
+        if not ctx.find_root().params.get("json_output", False):
+            out_amount = quote.get("outputAmount", quote.get("expectedOutput", "?"))
+            gas_est = quote.get("gasEstimate", "?")
+            click.echo(f"\n  Swap: {amount} {input_token} -> {out_amount} {output_token}")
+            click.echo(f"  Chain: {chain}  Slippage: {slippage}%  Gas: ~{gas_est}")
+            if not click.confirm("\n  Execute swap?"):
+                click.echo("  Cancelled.")
+                return
+
+        result = client.uniswap_swap_execute(
+            input_address=input_address,
+            output_address=output_address,
+            amount=amount,
+            chain=chain,
+            slippage_bps=slippage_bps,
+            wallet_key=wallet_key,
+            gas_limit=gas_limit,
+        )
+        emit(ctx, result, lambda d: _format_swap_result(d, "Uniswap"))
+
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+def _format_uniswap_quote(data: dict, input_tok: str, output_tok: str, amount: float, chain: str) -> str:
+    return "\n".join([
+        "",
+        f"  Uniswap Quote ({chain})",
+        "  ====================",
+        f"  Input:    {amount} {input_tok}",
+        f"  Output:   {data.get('outputAmount', '?')} {output_tok}",
+        f"  Gas Est:  {data.get('gasEstimate', '?')} gwei",
+        f"  Price Impact: {data.get('priceImpact', '?')}%",
+        f"  Route: {data.get('route', '?')}",
+    ])
+
+
+# ── SushiSwap (multi-chain) ─────────────────────────────────────────────
+
+@swap.command("sushi")
+@click.argument("input_token", type=str)
+@click.argument("output_token", type=str)
+@click.argument("amount", type=float)
+@click.option("--chain", type=click.Choice(["ethereum", "arbitrum", "polygon", "avalanche"]), default="ethereum")
+@click.option("--slippage", type=float, default=0.5, help="Max slippage percent.")
+@click.option("--wallet-key", envvar="EVM_PRIVATE_KEY", default=None)
+@click.option("--quote-only", is_flag=True)
+@click.pass_context
+def sushi_swap(ctx, input_token, output_token, amount, chain, slippage, wallet_key, quote_only):
+    """Swap tokens via SushiSwap on multiple EVM chains.
+
+    Examples:
+        fastbooks swap sushi ETH USDC 1.0 --chain ethereum
+        fastbooks swap sushi USDC WBTC 5000 --chain arbitrum
+        fastbooks swap sushi ETH DAI 2.0 --quote-only
+    """
+    client: FastBooksClient = ctx.obj
+
+    input_address = EVM_TOKENS.get(input_token.upper(), input_token)
+    output_address = EVM_TOKENS.get(output_token.upper(), output_token)
+
+    try:
+        result = client.sushi_swap(
+            input_address=input_address,
+            output_address=output_address,
+            amount=amount,
+            chain=chain,
+            slippage_bps=int(slippage * 100),
+            wallet_key=wallet_key,
+            quote_only=quote_only,
+        )
+        emit(ctx, result, lambda d: _format_swap_result(d, f"SushiSwap ({chain})"))
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── token lookup ─────────────────────────────────────────────────────────
+
+@swap.command("tokens")
+@click.option("--chain", type=click.Choice(CHAINS), default="solana", help="Chain to list tokens for.")
+@click.pass_context
+def list_tokens(ctx, chain):
+    """List known token addresses for a chain.
+
+    Examples:
+        fastbooks swap tokens --chain solana
+        fastbooks swap tokens --chain ethereum
+    """
+    if chain == "solana":
+        data = {"chain": chain, "tokens": [{"symbol": k, "address": v} for k, v in SOLANA_TOKENS.items()]}
+    else:
+        data = {"chain": chain, "tokens": [{"symbol": k, "address": v} for k, v in EVM_TOKENS.items()]}
+
+    emit(ctx, data, lambda d: _format_token_list(d))
+
+
+def _format_token_list(data: dict) -> str:
+    lines = [f"\n  Known Tokens ({data['chain']})", "  " + "=" * 30]
+    for t in data["tokens"]:
+        lines.append(f"  {t['symbol']:>10}  {t['address']}")
+    return "\n".join(lines)
+
+
+# ── shared formatters ────────────────────────────────────────────────────
+
+def _format_swap_result(data: dict, dex_name: str) -> str:
+    lines = [
+        "",
+        f"  {dex_name} Swap Result",
+        "  " + "=" * 25,
+    ]
+    if data.get("txHash") or data.get("signature"):
+        tx = data.get("txHash") or data.get("signature")
+        lines.append(f"  Tx: {tx}")
+    if data.get("inputAmount"):
+        lines.append(f"  Input:  {data['inputAmount']} {data.get('inputToken', '')}")
+    if data.get("outputAmount"):
+        lines.append(f"  Output: {data['outputAmount']} {data.get('outputToken', '')}")
+    if data.get("gasUsed"):
+        lines.append(f"  Gas:    {data['gasUsed']}")
+    status = data.get("status", "unknown")
+    lines.append(f"  Status: {status}")
+    return "\n".join(lines)