#!/usr/bin/env node
// Open Broker CLI - Hyperliquid trading toolkit

import { spawn } from 'child_process';
import { fileURLToPath } from 'url';
import path from 'path';

const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);
const scriptsDir = path.resolve(__dirname, '../scripts');

const commands: Record<string, { script: string; description: string }> = {
  // Setup
  'install': { script: 'setup/install.ts', description: 'Install companion packages or an agent harness' },
  'setup': { script: 'setup/onboard.ts', description: 'Interactive setup wizard' },
  'onboard': { script: 'setup/onboard.ts', description: 'Interactive setup wizard' },
  'approve-builder': { script: 'setup/approve-builder.ts', description: 'Approve builder fee' },

  // Info
  'account': { script: 'info/account.ts', description: 'View account balance and equity' },
  'positions': { script: 'info/positions.ts', description: 'View open positions' },
  'funding': { script: 'info/funding.ts', description: 'View funding rates' },
  'markets': { script: 'info/markets.ts', description: 'View market data' },
  'all-markets': { script: 'info/all-markets.ts', description: 'View all markets (perps, HIP-3, spot, HIP-4)' },
  'search': { script: 'info/search-markets.ts', description: 'Search for assets across providers' },
  'spot': { script: 'info/spot.ts', description: 'View spot markets and balances' },
  'fills': { script: 'info/fills.ts', description: 'View trade fill history' },
  'orders': { script: 'info/orders.ts', description: 'View order history' },
  'order-status': { script: 'info/order-status.ts', description: 'Check status of a specific order' },
  'fees': { script: 'info/fees.ts', description: 'View fee schedule and volume' },
  'candles': { script: 'info/candles.ts', description: 'View OHLCV candle data' },
  'funding-history': { script: 'info/funding-history.ts', description: 'View historical funding rates' },
  'trades': { script: 'info/trades.ts', description: 'View recent trades for an asset' },
  'rate-limit': { script: 'info/rate-limit.ts', description: 'View API rate limit status' },
  'funding-scan': { script: 'info/funding-scan.ts', description: 'Scan funding rates across all dexes' },
  'outcomes': { script: 'info/outcomes.ts', description: 'Search and inspect HIP-4 outcome markets' },

  // Operations
  'buy': { script: 'operations/market-order.ts', description: 'Market buy order' },
  'sell': { script: 'operations/market-order.ts', description: 'Market sell order' },
  'market': { script: 'operations/market-order.ts', description: 'Market order' },
  'limit': { script: 'operations/limit-order.ts', description: 'Limit order' },
  'trigger': { script: 'operations/trigger-order.ts', description: 'Trigger order (TP/SL)' },
  'tpsl': { script: 'operations/set-tpsl.ts', description: 'Set TP/SL on position' },
  'cancel': { script: 'operations/cancel.ts', description: 'Cancel orders' },
  'twap': { script: 'operations/twap.ts', description: 'Native TWAP order' },
  'twap-cancel': { script: 'operations/twap-cancel.ts', description: 'Cancel a TWAP order' },
  'twap-status': { script: 'operations/twap-status.ts', description: 'View TWAP order status' },
  'scale': { script: 'operations/scale.ts', description: 'Scale in/out orders' },
  'bracket': { script: 'operations/bracket.ts', description: 'Bracket order (entry + TP + SL)' },
  'chase': { script: 'operations/chase.ts', description: 'Chase order with ALO' },
  'spot-buy': { script: 'operations/spot-order.ts', description: 'Spot buy order' },
  'spot-sell': { script: 'operations/spot-order.ts', description: 'Spot sell order' },
  'spot-order': { script: 'operations/spot-order.ts', description: 'Spot order (market or limit)' },
  'outcome-buy': { script: 'operations/outcome-order.ts', description: 'Buy a HIP-4 outcome token' },
  'outcome-sell': { script: 'operations/outcome-order.ts', description: 'Sell a HIP-4 outcome token' },
  'outcome-open': { script: 'operations/outcome-order.ts', description: 'Open a HIP-4 outcome position' },
  'outcome-close': { script: 'operations/outcome-order.ts', description: 'Close a HIP-4 outcome position' },
  'outcome-order': { script: 'operations/outcome-order.ts', description: 'HIP-4 outcome order (market or limit)' },

  // Automations
  'auto': { script: 'auto/cli.ts', description: 'Run/manage trading automations' },

  // Guardian (position risk monitoring)
  'guardian': { script: 'guardian/cli.ts', description: 'Watch positions for risk, alert via Telegram' },
};

function printHelp() {
  console.log(`
Open Broker - Hyperliquid Trading CLI

Usage: openbroker <command> [options]

Setup:
  install              Install companion packages or OpenBroker for an agent harness
  setup                One-command setup (wallet + config + builder approval)

Info Commands:
  account              View account balance, equity, and margin
  positions            View open positions with PnL
  fills                View trade fill history
  orders               View order history (all statuses)
  order-status         Check status of a specific order
  fees                 View fee schedule, tiers, and volume
  funding              View funding rates (sorted by annualized rate)
  funding-history      View historical funding rates for an asset
  candles              View OHLCV candle data
  trades               View recent trades (tape) for an asset
  markets              View market data for main perps
  all-markets          View all markets (perps, HIP-3, spot, HIP-4)
  search               Search for assets across all providers
  spot                 View spot markets and balances
  rate-limit           View API rate limit status
  funding-scan         Scan funding rates across all dexes (main + HIP-3)
  outcomes             Search and inspect HIP-4 outcome markets

Trading Commands:
  buy                  Market buy order
  sell                 Market sell order
  market               Market order (specify --side)
  limit                Limit order
  trigger              Trigger order (stop loss / take profit)
  tpsl                 Set TP/SL on existing position
  cancel               Cancel orders

Spot Trading:
  spot-buy             Spot buy order
  spot-sell            Spot sell order
  spot-order           Spot order (market or limit, specify --side)

HIP-4 Outcome Trading:
  outcome-buy          Buy a YES/NO outcome token
  outcome-sell         Sell a YES/NO outcome token
  outcome-open         Alias for outcome-buy
  outcome-close        Alias for outcome-sell
  outcome-order        Outcome order (market or limit, specify --side)

Advanced Execution:
  twap                 Native TWAP order (exchange-managed)
  twap-cancel          Cancel a running TWAP order
  twap-status          View TWAP order history/status
  scale                Scale in/out with multiple orders
  bracket              Entry with TP and SL
  chase                Chase price with ALO orders

Automations:
  auto run <script>    Run a custom trading automation
  auto run --example <name> [--set key=value]  Run a bundled example
  auto examples        List bundled examples (dca, grid, funding-arb, mm-spread, mm-maker)
  auto list            List available automations
  auto status          Show running automations

Guardian (read-only risk monitoring):
  guardian run         Watch positions for liquidation risk, missing TP/SL,
                       funding bleed and more; alerts to console + Telegram
  guardian connect     Link a Telegram chat for alerts
  guardian test        Send a test Telegram message
  guardian status      Show guardian configuration
  guardian rules       List alert rules and thresholds

Options:
  -c, --config <path>  Use a specific .env config file
  --testnet            Use testnet
  --help, -h           Show help for a command
  --dry                Preview without executing
  --verbose            Show debug output

Utility:
  approve-builder      Check or retry builder fee approval

Examples:
  npx openbroker@latest install --codex        # Install skill + CLI + API-wallet setup
  openbroker install monitoring                 # Install or upgrade the automation dashboard
  openbroker install extended                   # Install or upgrade the Extended Exchange CLI
  openbroker setup --api-wallet                 # Recommended restricted API-wallet setup
  openbroker account                            # View account info
  openbroker buy --coin ETH --size 0.1          # Market buy 0.1 ETH
  openbroker limit --coin BTC --side buy --size 0.01 --price 60000
  openbroker search --query GOLD                # Find GOLD across providers
  openbroker outcomes --query BTC               # Find HIP-4 outcome markets
  openbroker outcome-buy --outcome 123 --outcome-side yes --size 10 --dry
  openbroker tpsl --coin HYPE --tp 40 --sl 30   # Set TP/SL on position

Documentation: https://github.com/aurracloud/open-broker
`);
}

function runScript(scriptPath: string, args: string[]) {
  const fullPath = path.join(scriptsDir, scriptPath);

  // Handle global flags: set env vars and strip from args
  const env = { ...process.env };
  const testnetIdx = args.indexOf('--testnet');
  if (testnetIdx !== -1) {
    env.HYPERLIQUID_NETWORK = 'testnet';
    args = [...args.slice(0, testnetIdx), ...args.slice(testnetIdx + 1)];
  }

  // Handle -c / --config flag: load the specified .env file
  // Resolve relative to the user's original working directory (not the package root)
  let configIdx = args.indexOf('-c');
  if (configIdx === -1) configIdx = args.indexOf('--config');
  if (configIdx !== -1 && args[configIdx + 1]) {
    const originalCwd = process.env.OPENBROKER_CWD || process.cwd();
    const configPath = path.resolve(originalCwd, args[configIdx + 1]);
    env.OPENBROKER_CONFIG = configPath;
    args = [...args.slice(0, configIdx), ...args.slice(configIdx + 2)];
  }

  // Use tsx to run TypeScript directly
  const child = spawn('npx', ['tsx', fullPath, ...args], {
    stdio: 'inherit',
    cwd: path.resolve(__dirname, '..'),
    env,
  });

  child.on('error', (err) => {
    if ((err as NodeJS.ErrnoException).code === 'ENOENT') {
      console.error('Error: npx/tsx not found. Please ensure Node.js is installed.');
    } else {
      console.error('Error:', err.message);
    }
    process.exit(1);
  });

  child.on('exit', (code) => {
    process.exit(code ?? 0);
  });
}

function main() {
  const args = process.argv.slice(2);

  if (args.length === 0 || args[0] === '--help' || args[0] === '-h' || args[0] === 'help') {
    printHelp();
    process.exit(0);
  }

  const command = args[0].toLowerCase();
  const commandArgs = args.slice(1);

  // Handle buy/sell shortcuts
  if (command === 'buy') {
    runScript(commands['market'].script, ['--side', 'buy', ...commandArgs]);
    return;
  }
  if (command === 'sell') {
    runScript(commands['market'].script, ['--side', 'sell', ...commandArgs]);
    return;
  }
  if (command === 'spot-buy') {
    runScript(commands['spot-order'].script, ['--side', 'buy', ...commandArgs]);
    return;
  }
  if (command === 'spot-sell') {
    runScript(commands['spot-order'].script, ['--side', 'sell', ...commandArgs]);
    return;
  }
  if (command === 'outcome-buy' || command === 'outcome-open') {
    runScript(commands['outcome-order'].script, ['--side', 'buy', ...commandArgs]);
    return;
  }
  if (command === 'outcome-sell' || command === 'outcome-close') {
    runScript(commands['outcome-order'].script, ['--side', 'sell', ...commandArgs]);
    return;
  }

  // Handle version
  if (command === '--version' || command === '-v') {
    import('../package.json', { with: { type: 'json' } }).then((pkg) => {
      console.log(`openbroker v${pkg.default.version}`);
      process.exit(0);
    });
    return;
  }

  const cmd = commands[command];
  if (!cmd) {
    console.error(`Unknown command: ${command}`);
    console.log('Run "openbroker --help" for usage information.');
    process.exit(1);
  }

  runScript(cmd.script, commandArgs);
}

main();