import type { ExtensionAPI } from "pi-agent";
import { existsSync } from "fs";
import { fileURLToPath } from "url";

export default (pi: ExtensionAPI) => {
  // Resolve the plugin root robustly: __dirname under jiti/CJS-style loaders,
  // else import.meta.url under native ESM (where __dirname is undefined).
  const baseDir =
    typeof __dirname !== "undefined"
      ? `${__dirname}/..`
      : fileURLToPath(new URL("..", import.meta.url));
  const toolsDir = `${baseDir}/tools`;
  const isWin = process.platform === "win32";
  // Windows venvs ship Scripts/python.exe (there is no python3); POSIX venvs ship bin/python3.
  const venvPython = `${baseDir}/.venv/${isWin ? "Scripts/python.exe" : "bin/python3"}`;
  // Fallback: on Windows "python3" is usually the Microsoft Store stub — prefer "python".
  const python = existsSync(venvPython) ? venvPython : isWin ? "python" : "python3";
  const py = (script: string, args: string[]) =>
    pi.exec(python, [`${toolsDir}/${script}`, ...args]);
  const asText = (text: string) => ({
    content: [{ type: "text" as const, text }],
    details: {},
  });

  // --- Data Tools ---

  pi.registerTool({
    name: "get_kline",
    description:
      "获取股票K线数据（OHLCV）。支持A股（如600519）、港股（如00700.HK）、美股（如AAPL）、日股（如7203.T）、韩股（如005930.KS）、台股（如2330.TW）及A股ETF。需要原始历史价格自行计算或画图时用；只问指标用 get_technical_analysis，只问现价用 get_quote",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码，如 600519（A股）、00700.HK（港股）、AAPL（美股）、7203.T（日股）、005930.KS（韩股）、2330.TW（台股）",
        },
        period: {
          type: "string",
          enum: ["daily", "weekly", "monthly"],
          description: "K线周期，默认 daily",
        },
        count: {
          type: "number",
          description: "返回数据条数，默认 60",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol, period = "daily", count = 60 }) {
      const result = await py("stock_data.py", [
        "kline",
        symbol,
        "--period",
        period,
        "--count",
        String(count),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_quote",
    description:
      "获取股票实时行情报价（现价、涨跌幅、量比等）。支持A股、港股、美股、日股、韩股、台股；非交易时段返回最近交易日收盘价并以 as_of/stale 标注",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol }) {
      const result = await py("stock_data.py", ["quote", symbol]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_capital_flow",
    description:
      "获取A股资金流向（主力/超大单/大单/中单/小单净流入）。detail=个股每日明细，summary=多日汇总+趋势，sector_flow=板块资金流排行。可配合 get_chip_distribution 验证主力行为",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "A股股票代码（detail/summary模式必填），如 600519",
        },
        mode: {
          type: "string",
          enum: ["detail", "summary", "sector_flow"],
          description: "模式：detail=每日明细（默认），summary=多日汇总，sector_flow=板块排行",
        },
      },
    },
    async execute(_id, { symbol = "", mode = "detail" }) {
      const result = await py("stock_data.py", [
        "capital_flow",
        ...(symbol ? [symbol] : []),
        "--mode",
        mode,
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_news",
    description: "获取个股最近N天财经新闻快讯（轻量、无需配置）。需要深度全网情报用 search_comprehensive_intel，按主题搜索用 search_stock_news，读文章全文用 extract_article",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
        days: {
          type: "number",
          description: "获取最近几天的新闻，默认 3",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol, days = 3 }) {
      const result = await py("stock_data.py", ["news", symbol, "--days", String(days)]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_financials",
    description:
      "获取股票财务摘要。A股返回最新报告期 ROE/毛利率/净利率/负债率/流动比率；其他市场返回 PE/PB/市值/营收/净利润等。A股估值、成长和分红用 get_fundamental_context",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol }) {
      const result = await py("stock_data.py", ["financials", symbol]);
      return asText(result.stdout);
    },
  });

  // --- Analysis Tools ---

  pi.registerTool({
    name: "get_technical_analysis",
    description:
      "获取股票技术面分析（MA/MACD/RSI/BOLL/KDJ/成交量等指标 + 100分综合评分 + 6级买卖信号 + 趋势/偏离度/支撑压力位）。个股技术面综合判断与买卖时机分析的首选；只要均线数值或自定义周期用 calculate_ma，专问量价用 get_volume_analysis，扫当日异动用 detect_anomaly",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
        period: {
          type: "string",
          enum: ["daily", "weekly", "monthly"],
          description: "分析周期，默认 daily",
        },
        count: {
          type: "number",
          description: "用于计算指标的K线条数，默认 120",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol, period = "daily", count = 120 }) {
      const result = await py("technical.py", [
        "analyze",
        symbol,
        "--period",
        period,
        "--count",
        String(count),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "analyze_pattern",
    description:
      "K线形态识别 — 检测十字星、锤子线、吞没、启明星、黄昏星、双底、20日突破等12+种经典形态。综合技术面判断用 get_technical_analysis",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
        period: {
          type: "string",
          enum: ["daily", "weekly", "monthly"],
          description: "K线周期，默认 daily",
        },
        days: {
          type: "number",
          description: "分析的K线天数，默认 60",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol, period = "daily", days = 60 }) {
      const result = await py("pattern.py", [
        "analyze",
        symbol,
        "--period",
        period,
        "--days",
        String(days),
      ]);
      return asText(result.stdout);
    },
  });

  // --- Market Tools ---

  pi.registerTool({
    name: "get_market_indices",
    description:
      "获取主要市场指数行情。CN: 上证/深证/创业板/科创50/沪深300；HK: 恒生/国企/科技；US: 道琼斯/纳斯达克/标普500；JP: 日经225/东证；KR: KOSPI/KOSDAQ；TW: 台湾加权。大盘复盘用 get_market_review 一站式获取",
    parameters: {
      type: "object",
      properties: {
        region: {
          type: "string",
          enum: ["cn", "hk", "us", "jp", "kr", "tw"],
          description: "市场区域，默认 cn",
        },
      },
    },
    async execute(_id, { region = "cn" }) {
      const result = await py("stock_data.py", [
        "market_indices",
        "--region",
        region,
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_sector_rankings",
    description: "获取A股板块涨跌幅排行（含领涨股、涨跌家数等）。board_type=industry（默认）为行业板块，board_type=concept 时为概念板块排行。支持查看涨幅榜/跌幅榜/双向",
    parameters: {
      type: "object",
      properties: {
        top: {
          type: "number",
          description: "返回排名前N的板块，默认 10",
        },
        direction: {
          type: "string",
          enum: ["top", "bottom", "both"],
          description: "top=涨幅榜（默认），bottom=跌幅榜，both=双向",
        },
        board_type: {
          type: "string",
          enum: ["industry", "concept"],
          description: "板块类型：industry=行业（默认），concept=概念",
        },
      },
    },
    async execute(_id, { top = 10, direction = "top", board_type = "industry" }) {
      const result = await py("stock_data.py", [
        "sector_rankings",
        "--top",
        String(top),
        "--direction",
        direction,
        "--board-type",
        board_type,
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_sector_constituents",
    description:
      "查询A股板块成分股（正向映射：板块→股票列表）。板块名支持模糊匹配（如「创新药」自动解析到精确板块名），东财行业/概念 + 新浪多源 failover，结果缓存24小时。仅支持A股；反向查个股所属板块用 resolve_stock_sectors",
    parameters: {
      type: "object",
      properties: {
        sector: {
          type: "string",
          description: "板块名称（可模糊，如 创新药、半导体）",
        },
        board_type: {
          type: "string",
          enum: ["industry", "concept", "auto"],
          description: "板块类型：industry=行业，concept=概念，auto=自动（默认）",
        },
      },
      required: ["sector"],
    },
    async execute(_id, { sector, board_type = "auto" }) {
      const result = await py("stock_data.py", [
        "sector_constituents",
        sector,
        "--board-type",
        board_type,
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "resolve_stock_sectors",
    description:
      "查询个股所属板块（反向映射：个股→板块）。A股返回东财行业 + efinance 概念/板块（雪球行业兜底，需 XUEQIU_TOKEN），港股返回 yfinance GICS 英文 sector/industry 口径。板块级情报（如「创新药回调」）映射到池内个股时调本工具；正向查板块成分股用 get_sector_constituents",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如 600519，港股如 00700.HK）",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol }) {
      const result = await py("stock_data.py", ["resolve_stock_sectors", symbol]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_stock_info",
    description:
      "获取股票基本信息（行业、板块、上市日期、总股本等）。A股返回板块/行业，其他市场返回行业/公司简介",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol }) {
      const result = await py("stock_data.py", ["stock_info", symbol]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_chip_distribution",
    description:
      "获取A股筹码分布数据（获利比例、平均成本、90%/70%成本集中度）。仅支持A股。与 get_capital_flow 配合验证主力行为",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "A股股票代码，如 600519",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol }) {
      const result = await py("stock_data.py", ["chip_distribution", symbol]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_market_stats",
    description:
      "获取A股市场整体统计（涨跌家数、涨停跌停数、平均涨幅、涨跌Top5、总成交额）。用于衡量市场整体情绪与温度",
    parameters: {
      type: "object",
      properties: {},
    },
    async execute() {
      const result = await py("stock_data.py", ["market_stats"]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_limit_up_pool",
    description:
      "获取A股涨停池/涨停板复盘——当日涨停个股，含连板数、封板资金、炸板次数、首末次封板时间、所属行业。短线情绪与龙头战法核心数据",
    parameters: {
      type: "object",
      properties: {
        date: {
          type: "string",
          description: "日期（YYYYMMDD），默认当日；非交易日或未来日期自动回退到最近交易日（返回含 requested_date 与 stale 标注）",
        },
      },
    },
    async execute(_id, { date }) {
      const result = await py("stock_data.py", [
        "limit_up_pool",
        ...(date ? ["--date", date] : []),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_dragon_tiger",
    description:
      "获取A股龙虎榜——上榜个股净买额/买入额/卖出额/上榜原因/解读，可按个股过滤",
    parameters: {
      type: "object",
      properties: {
        date: {
          type: "string",
          description: "日期（YYYY-MM-DD），默认当日；非交易日或未来日期自动回退到最近交易日（返回含 requested_date 与 stale 标注）",
        },
        symbol: {
          type: "string",
          description: "A股股票代码（可选，按个股过滤），如 600519",
        },
        top: {
          type: "number",
          description: "返回前N条，默认 20",
        },
      },
    },
    async execute(_id, { date, symbol, top = 20 }) {
      const result = await py("stock_data.py", [
        "dragon_tiger",
        ...(date ? ["--date", date] : []),
        ...(symbol ? ["--symbol", symbol] : []),
        "--top",
        String(top),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_hot_stocks",
    description: "获取A股全市场人气热搜榜（东方财富人气榜）",
    parameters: {
      type: "object",
      properties: {
        top: {
          type: "number",
          description: "返回前N只，默认 20",
        },
      },
    },
    async execute(_id, { top = 20 }) {
      const result = await py("stock_data.py", ["hot_stocks", "--top", String(top)]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_fundamental_context",
    description:
      "获取A股基本面上下文（估值 PE/PB/市值 + 营收/净利增速 + ROE/毛利率/净利率 + 分红历史）。用于评估公司质地与长期持有价值",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "A股股票代码，如 600519",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol }) {
      const result = await py("stock_data.py", ["fundamental_context", symbol]);
      return asText(result.stdout);
    },
  });

  // --- Screener & Backtest ---

  pi.registerTool({
    name: "screen_stocks",
    description:
      "全市场股票筛选（AlphaSift L1 多因子硬筛 + 市场情绪调节）。按PE/PB/市值/换手率/涨跌幅/量比等因子过滤和评分，可用 l2 开启质量/成长/动量/波动率/资金流增强",
    parameters: {
      type: "object",
      properties: {
        market: {
          type: "string",
          enum: ["A", "HK", "US"],
          description: "市场，默认 A",
        },
        top: {
          type: "number",
          description: "返回排名前N的股票，默认 20",
        },
        config: {
          type: "string",
          description: "自定义筛选配置YAML文件路径，不填用默认配置",
        },
        l2: {
          type: "boolean",
          description: "开启L2量化增强（质量/成长/真实动量/波动率/资金流因子，逐股抓取数据较慢），默认关闭",
        },
      },
    },
    async execute(_id, { market = "A", top = 20, config, l2 = false }) {
      const result = await py("screener.py", [
        "screen",
        "--market",
        market,
        "--top",
        String(top),
        ...(config ? ["--config", config] : []),
        ...(l2 ? ["--l2"] : []),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "run_backtest",
    description:
      "策略回测（AlphaEvo）。读取YAML策略定义，在历史数据上模拟交易，输出收益率/回撤/胜率等指标。只想知道单个技术信号的历史胜率用更轻量的 evaluate_signal",
    parameters: {
      type: "object",
      properties: {
        strategy: {
          type: "string",
          description: "策略YAML文件路径",
        },
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
        start: {
          type: "string",
          description: "回测起始日期，格式 YYYY-MM-DD",
        },
        end: {
          type: "string",
          description: "回测结束日期，格式 YYYY-MM-DD",
        },
        capital: {
          type: "number",
          description: "初始资金，默认 1000000",
        },
      },
      required: ["strategy", "symbol"],
    },
    async execute(_id, { strategy, symbol, start, end, capital = 1000000 }) {
      const result = await py("backtest.py", [
        "run",
        strategy,
        symbol,
        ...(start ? ["--start", start] : []),
        ...(end ? ["--end", end] : []),
        "--capital",
        String(capital),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "evaluate_signal",
    description:
      "技术信号历史准确率评估 — 回溯历史数据，统计某个技术信号触发后N日的胜率和平均收益。支持9种信号：macd_golden_cross/macd_death_cross/rsi_oversold/rsi_overbought/breakout_20d/breakdown_20d/volume_surge/ma_golden_cross/ma_death_cross。要完整模拟交易过程用 run_backtest",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
        signal: {
          type: "string",
          enum: [
            "macd_golden_cross",
            "macd_death_cross",
            "rsi_oversold",
            "rsi_overbought",
            "breakout_20d",
            "breakdown_20d",
            "volume_surge",
            "ma_golden_cross",
            "ma_death_cross",
          ],
          description: "信号名称",
        },
        forward_days: {
          type: "string",
          description: "逗号分隔的前瞻天数，默认 3,5,10",
        },
        lookback: {
          type: "number",
          description: "回溯K线条数，默认 250（约1年）",
        },
      },
      required: ["symbol", "signal"],
    },
    async execute(_id, { symbol, signal, forward_days = "3,5,10", lookback = 250 }) {
      const result = await py("backtest.py", [
        "evaluate_signal",
        symbol,
        signal,
        "--forward",
        String(forward_days),
        "--lookback",
        String(lookback),
      ]);
      return asText(result.stdout);
    },
  });

  // --- Name Resolution ---

  pi.registerTool({
    name: "resolve_stock_name",
    description:
      "股票名称智能解析 — 输入中文名（贵州茅台）、拼音（guizhou maotai/gzmt）、部分代码，返回匹配的股票代码。仅支持A股。用户给出中文名/拼音/部分代码时先调本工具换成代码再调其他工具",
    parameters: {
      type: "object",
      properties: {
        query: {
          type: "string",
          description: "股票名称、拼音、拼音首字母或部分代码",
        },
        top: {
          type: "number",
          description: "返回匹配数量，默认 5",
        },
      },
      required: ["query"],
    },
    async execute(_id, { query, top = 5 }) {
      const result = await py("name_resolver.py", [
        "resolve",
        query,
        "--top",
        String(top),
      ]);
      return asText(result.stdout);
    },
  });

  // --- Trading Calendar ---

  pi.registerTool({
    name: "check_trading_day",
    description:
      "查询某日是否为交易日。支持CN（A股）、HK（港股）、US（美股）、JP（日股）、KR（韩股）、TW（台股）",
    parameters: {
      type: "object",
      properties: {
        market: {
          type: "string",
          enum: ["CN", "HK", "US", "JP", "KR", "TW"],
          description: "市场",
        },
        date: {
          type: "string",
          description: "日期（YYYY-MM-DD），不填则查今天",
        },
      },
      required: ["market"],
    },
    async execute(_id, { market, date }) {
      const result = await py("trading_calendar.py", [
        "check",
        market,
        ...(date ? ["--date", date] : []),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_trading_days",
    description:
      "获取最近/未来N个交易日列表。支持CN/HK/US/JP/KR/TW",
    parameters: {
      type: "object",
      properties: {
        market: {
          type: "string",
          enum: ["CN", "HK", "US", "JP", "KR", "TW"],
          description: "市场",
        },
        direction: {
          type: "string",
          enum: ["next", "prev"],
          description: "next=未来交易日, prev=过去交易日",
        },
        count: {
          type: "number",
          description: "返回天数，默认 5",
        },
        date: {
          type: "string",
          description: "起始日期（YYYY-MM-DD），默认今天",
        },
      },
      required: ["market"],
    },
    async execute(_id, { market, direction = "next", count = 5, date }) {
      const result = await py("trading_calendar.py", [
        direction,
        market,
        "--count",
        String(count),
        ...(date ? ["--date", date] : []),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_trading_phase",
    description:
      "查询某市场当前交易时段（盘前/早盘/午间休市/午盘/交易中/盘后/休市），用于正确解读实时行情的盘中语义。支持CN/HK/US/JP/KR/TW",
    parameters: {
      type: "object",
      properties: {
        market: {
          type: "string",
          enum: ["CN", "HK", "US", "JP", "KR", "TW"],
          description: "市场",
        },
      },
      required: ["market"],
    },
    async execute(_id, { market }) {
      const result = await py("trading_calendar.py", ["phase", market]);
      return asText(result.stdout);
    },
  });

  // --- Standalone MA Calculator ---

  pi.registerTool({
    name: "calculate_ma",
    description:
      "独立均线计算器 — 支持任意周期MA（5/10/20/30/60/120/250或自定义）+ 偏离度 + 均线排列 + 金叉死叉检测。只要均线数值或需要30/120/250等非默认周期时用本工具；综合技术面分析用 get_technical_analysis",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
        periods: {
          type: "string",
          description: "逗号分隔的MA周期列表，如 5,10,20,60,120,250",
        },
        kline_period: {
          type: "string",
          enum: ["daily", "weekly", "monthly"],
          description: "K线周期，默认 daily",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol, periods = "5,10,20,30,60,120,250", kline_period = "daily" }) {
      const result = await py("technical.py", [
        "calculate_ma",
        symbol,
        "--periods",
        periods,
        "--period",
        kline_period,
        "--count",
        "300",
      ]);
      return asText(result.stdout);
    },
  });

  // --- Volume-Price Analysis ---

  pi.registerTool({
    name: "get_volume_analysis",
    description:
      "独立量价分析 — 量价相关性、上涨/下跌日成交量对比、量能趋势、量价模式解读（放量上涨/缩量回调等）。综合技术面判断用 get_technical_analysis，扫当日异动用 detect_anomaly",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
        period: {
          type: "string",
          enum: ["daily", "weekly", "monthly"],
          description: "K线周期，默认 daily",
        },
        count: {
          type: "number",
          description: "分析K线条数，默认 60",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol, period = "daily", count = 60 }) {
      const result = await py("volume_analysis.py", [
        "analyze",
        symbol,
        "--period",
        period,
        "--count",
        String(count),
      ]);
      return asText(result.stdout);
    },
  });

  // --- Search Intelligence ---

  pi.registerTool({
    name: "search_stock_news",
    description:
      "多引擎股票新闻搜索（Tavily/Brave/SerpAPI/Bocha/SearXNG）。需配置至少一个搜索源的 Key 或 SearXNG URL。get_news 快讯不够或要按主题搜索时用；深度6维情报用 search_comprehensive_intel",
    parameters: {
      type: "object",
      properties: {
        query: {
          type: "string",
          description: "搜索关键词",
        },
        count: {
          type: "number",
          description: "返回结果数量，默认 10",
        },
      },
      required: ["query"],
    },
    async execute(_id, { query, count = 10 }) {
      const result = await py("search_intel.py", [
        "search",
        query,
        "--count",
        String(count),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "search_comprehensive_intel",
    description:
      "股票综合情报搜索 — 从6个维度（新闻/公告/行情分析/风险/业绩/行业）搜索综合信息。用于个股的深入研究/全面调研；快速看新闻用 get_news",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
        name: {
          type: "string",
          description: "股票名称（可选，提升搜索准确度）",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol, name }) {
      const result = await py("search_intel.py", [
        "comprehensive",
        symbol,
        ...(name ? ["--name", name] : []),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_social_sentiment",
    description:
      "获取股票社交媒体情绪数据。A股：东方财富股吧热度+雪球讨论热度（无需配置）；美股/港股：Reddit/X/Polymarket情绪（需 SENTIMENT_API_KEY）。自动按市场选数据源。面向个股维度；全市场热门讨论用 get_trending_sentiment",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol }) {
      const result = await py("search_intel.py", ["sentiment", symbol]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_trending_sentiment",
    description:
      "获取社交媒体热门趋势（Reddit/X/Polymarket热门股票讨论）。适用于发现市场热点；查个股情绪用 get_social_sentiment",
    parameters: {
      type: "object",
      properties: {},
    },
    async execute() {
      const result = await py("search_intel.py", ["trending"]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "extract_article",
    description:
      "网页文章全文提取 — 输入URL，提取文章标题、正文（最多3000字）、作者、发布日期等。配合 get_news/search_stock_news 的搜索结果做深度阅读",
    parameters: {
      type: "object",
      properties: {
        url: {
          type: "string",
          description: "文章URL",
        },
      },
      required: ["url"],
    },
    async execute(_id, { url }) {
      const result = await py("search_intel.py", ["extract", url]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "screen_risk",
    description:
      "A股风险专项筛查 — 检查估值、技术预警、解禁、减持、业绩预警、监管和行业政策，返回风险评级和 veto_buy。新闻搜索命中是待核实线索，不是已证实风险",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "A股股票代码，如 600519",
        },
        name: {
          type: "string",
          description: "股票名称（可选，提升新闻搜索准确度）",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol, name }) {
      const result = await py("risk_screening.py", [
        "screen",
        symbol,
        ...(name ? ["--name", name] : []),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "detect_market_regime",
    description:
      "市场状态检测 — 分析大盘指数判断当前市场阶段（上涨趋势/下跌趋势/横盘震荡/高波动/板块热点），并推荐适合的分析策略",
    parameters: {
      type: "object",
      properties: {
        market: {
          type: "string",
          enum: ["A", "HK", "US"],
          description: "市场代码，默认 A",
        },
      },
    },
    async execute(_id, { market }) {
      const m = market || "A";
      const result = await py("market_regime.py", ["detect", m]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_market_review",
    description:
      "大盘复盘 — 获取市场日度复盘数据，包含指数、涨跌统计、板块排名、新闻、市场温度与策略建议。复盘类需求的首选，无需再分别调指数/统计/板块工具",
    parameters: {
      type: "object",
      properties: {
        market: {
          type: "string",
          enum: ["A", "HK", "US", "all"],
          description: "市场代码，默认 A。all 表示所有市场",
        },
      },
    },
    async execute(_id, { market }) {
      const m = market || "A";
      const result = await py("market_review.py", ["review", "--market", m]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "run_watchlist_analysis",
    description:
      "批量自选股分析 — 对多只股票并行采集行情/技术/资金/风险等数据，返回汇总结果。适用于每日定时分析自选股列表",
    parameters: {
      type: "object",
      properties: {
        symbols: {
          type: "string",
          description: "逗号分隔的股票代码列表，如 600519,000001,300750",
        },
        workers: {
          type: "number",
          description: "并发数，默认 3（建议不超过5，避免API限流）",
        },
      },
      required: ["symbols"],
    },
    async execute(_id, { symbols, workers = 3 }) {
      const result = await py("watchlist.py", [
        "analyze",
        symbols,
        "--workers",
        String(workers),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "detect_anomaly",
    description:
      "异常/事件检测 — 一键扫描股票当前所有异动信号（MACD金叉死叉、RSI超买超卖、20日突破、放量异动、涨跌停、布林突破、KDJ极值、资金异动等），返回结构化异常列表。当日异动归因的首选；综合技术面判断用 get_technical_analysis",
    parameters: {
      type: "object",
      properties: {
        symbol: {
          type: "string",
          description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）",
        },
      },
      required: ["symbol"],
    },
    async execute(_id, { symbol }) {
      const result = await py("anomaly_detect.py", ["detect", symbol]);
      return asText(result.stdout);
    },
  });

  // --- Diagnostics & Capabilities ---

  pi.registerTool({
    name: "diagnose_data_sources",
    description:
      "数据源诊断 — 检查当前环境各数据 provider 的包与凭据就绪情况（不探测网络可达性），输出每个市场的可用链路、缺失 env、warnings。工具拿不到数据或报错时调用排查",
    parameters: {
      type: "object",
      properties: {
        market: {
          type: "string",
          enum: ["A", "HK", "US", "JP", "KR", "TW", "all"],
          description: "市场，默认 all",
        },
      },
    },
    async execute(_id, { market = "all" }) {
      const result = await py("diagnostics.py", ["check", "--market", market]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_market_capabilities",
    description:
      "市场能力边界 — 返回指定市场支持/不支持的工具列表。不确定某市场能否用某工具（如港股的筹码分布、美股的资金流）时先调本工具，避免调用不支持的工具后编造数据",
    parameters: {
      type: "object",
      properties: {
        market: { type: "string", enum: ["A", "HK", "US", "JP", "KR", "TW"], description: "市场代码" },
        symbol: { type: "string", description: "股票代码（自动识别市场，与 market 二选一）" },
      },
    },
    async execute(_id, { market, symbol }) {
      const args = symbol
        ? ["get", "--symbol", symbol]
        : ["get", "--market", market || "A"];
      const result = await py("capabilities.py", args);
      return asText(result.stdout);
    },
  });

  // --- Report Rendering ---

  pi.registerTool({
    name: "render_stock_report",
    description:
      "股票分析报告渲染 — 将结构化报告 JSON 通过 j2 模板渲染为 Markdown。report 字段以 skill 提供的 report_schema 为准，不要自造字段。template: brief|full。全部分析完成后的最后一步调用。仅渲染，不保存不推送",
    parameters: {
      type: "object",
      properties: {
        report: {
          type: "object",
          description: "结构化股票报告 JSON",
        },
        template: {
          type: "string",
          enum: ["brief", "full"],
          description: "模板类型，默认 full",
        },
      },
      required: ["report"],
    },
    async execute(_id, { report, template = "full" }) {
      const b64 = Buffer.from(JSON.stringify(report), "utf-8").toString("base64");
      const result = await py("report_renderer.py", [
        "stock",
        "--template",
        template,
        "--input-b64",
        b64,
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "render_market_report",
    description:
      "大盘复盘报告渲染 — 将结构化报告 JSON 通过 j2 模板渲染为 Markdown。report 字段以 schemas/market_review_schema.json 为准。仅渲染，不保存不推送",
    parameters: {
      type: "object",
      properties: {
        report: { type: "object", description: "结构化市场复盘" },
      },
      required: ["report"],
    },
    async execute(_id, { report }) {
      const b64 = Buffer.from(JSON.stringify(report), "utf-8").toString("base64");
      const result = await py("report_renderer.py", ["market", "--input-b64", b64]);
      return asText(result.stdout);
    },
  });

  // --- Watchlist / Position / Alert Context ---

  pi.registerTool({
    name: "build_watchlist_context",
    description:
      "自选股上下文包 — 对多只股票输出评分/趋势/异常/风险/建议 next_tools 的 agent 友好摘要。宿主 agent 决定如何写日报或深入分析",
    parameters: {
      type: "object",
      properties: {
        symbols: { type: "string", description: "逗号分隔的股票代码列表" },
        include_market_review: {
          type: "boolean",
          description: "是否附带各市场复盘，默认 false",
        },
        workers: { type: "number", description: "并发数，默认 3" },
      },
      required: ["symbols"],
    },
    async execute(_id, { symbols, include_market_review = false, workers = 3 }) {
      const result = await py("watchlist_context.py", [
        "build",
        symbols,
        "--workers",
        String(workers),
        ...(include_market_review ? ["--include-market-review"] : []),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "analyze_position_context",
    description:
      "持仓上下文分析 — 输入成本/仓位/止损止盈，结合现价和技术位输出浮盈亏、离止损距离、风险级别、操作建议。无状态、不存账户",
    parameters: {
      type: "object",
      properties: {
        symbol: { type: "string", description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）" },
        cost: { type: "number", description: "成本价" },
        quantity: { type: "number", description: "持仓数量" },
        stop_loss: { type: "number", description: "止损价" },
        take_profit: { type: "number", description: "止盈价" },
      },
      required: ["symbol", "cost", "quantity"],
    },
    async execute(_id, { symbol, cost, quantity, stop_loss, take_profit }) {
      const result = await py("position_context.py", [
        "analyze",
        symbol,
        "--cost",
        String(cost),
        "--quantity",
        String(quantity),
        ...(stop_loss !== undefined ? ["--stop-loss", String(stop_loss)] : []),
        ...(take_profit !== undefined ? ["--take-profit", String(take_profit)] : []),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "check_alert_rules",
    description:
      "无状态告警规则检查 — 传入规则数组，返回当前是否触发。规则类型：price_below/price_above/change_pct_above/change_pct_below/volume_ratio_above/anomaly/risk_veto/risk_level_at_least。不做调度不存历史",
    parameters: {
      type: "object",
      properties: {
        symbol: { type: "string", description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）" },
        rules: {
          type: "array",
          description: "规则列表，每项 { type, value }",
          items: {
            type: "object",
            properties: {
              type: { type: "string" },
              value: {},
            },
          },
        },
      },
      required: ["symbol", "rules"],
    },
    async execute(_id, { symbol, rules }) {
      const b64 = Buffer.from(JSON.stringify(rules), "utf-8").toString("base64");
      const result = await py("alert_rules.py", ["check", symbol, "--rules-b64", b64]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "record_signal",
    description:
      "记录一条 AI 分析建议信号（方向/入场价/目标价/止损价/持有期限），供后续胜率评估。信号持久化在本地 JSONL（$STOCK_SIGNAL_STORE 或 ~/.stock-analysis/signals.jsonl），宿主 Agent 自行决定何时 record/evaluate",
    parameters: {
      type: "object",
      properties: {
        symbol: { type: "string", description: "股票代码（A股如600519，美股如AAPL，港股如00700.HK）" },
        direction: {
          type: "string",
          enum: ["buy", "sell"],
          description: "建议方向",
        },
        entry_price: { type: "number", description: "入场价（可选）" },
        target_price: { type: "number", description: "目标价（可选）" },
        stop_price: { type: "number", description: "止损价（可选）" },
        horizon_days: { type: "number", description: "持有期限（交易日），默认 10" },
        source: { type: "string", description: "信号来源（如触发该建议的 skill 名）" },
        note: { type: "string", description: "备注（可选）" },
      },
      required: ["symbol", "direction"],
    },
    async execute(_id, { symbol, direction, entry_price, target_price, stop_price, horizon_days = 10, source, note }) {
      const result = await py("signal_tracker.py", [
        "record",
        symbol,
        "--direction",
        direction,
        ...(entry_price !== undefined ? ["--entry-price", String(entry_price)] : []),
        ...(target_price !== undefined ? ["--target-price", String(target_price)] : []),
        ...(stop_price !== undefined ? ["--stop-price", String(stop_price)] : []),
        "--horizon-days",
        String(horizon_days),
        ...(source ? ["--source", source] : []),
        ...(note ? ["--note", note] : []),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "evaluate_signals",
    description:
      "评估到期信号：拉取记录日之后的日K，判定 target_hit/stop_hit/timeout 并计算收益，回写存储。记录建议用 record_signal，胜率复盘用 get_signal_summary",
    parameters: {
      type: "object",
      properties: {
        symbol: { type: "string", description: "股票代码（可选，只评估该股的信号）" },
      },
    },
    async execute(_id, { symbol }) {
      const result = await py("signal_tracker.py", [
        "evaluate",
        ...(symbol ? ["--symbol", symbol] : []),
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "get_signal_summary",
    description:
      "信号胜率汇总（可按 source/symbol/status 过滤）：胜率、平均收益、结果分布与明细",
    parameters: {
      type: "object",
      properties: {
        source: { type: "string", description: "按信号来源过滤（可选）" },
        symbol: { type: "string", description: "按股票代码过滤（可选）" },
        status: {
          type: "string",
          enum: ["open", "closed", "all"],
          description: "open=未结算，closed=已结算，all=全部（默认）",
        },
      },
    },
    async execute(_id, { source, symbol, status = "all" }) {
      const result = await py("signal_tracker.py", [
        "summary",
        ...(source ? ["--source", source] : []),
        ...(symbol ? ["--symbol", symbol] : []),
        "--status",
        status,
      ]);
      return asText(result.stdout);
    },
  });

  pi.registerTool({
    name: "parse_stock_list",
    description:
      "自选股/文本导入解析 — 从自然语言、CSV、Markdown 表格提取股票，自动识别 A 股 6 位代码、港股 xxxxx.HK、美股 ticker、日股 xxxx.T、韩股 xxxxxx.KS/KQ、台股 xxxx.TW，并调用 name_resolver 处理中文股票名",
    parameters: {
      type: "object",
      properties: {
        text: { type: "string", description: "待解析的文本" },
      },
      required: ["text"],
    },
    async execute(_id, { text }) {
      const b64 = Buffer.from(String(text), "utf-8").toString("base64");
      const result = await py("import_parser.py", ["parse", "--text-b64", b64]);
      return asText(result.stdout);
    },
  });

  // --- Skill Discovery ---

  pi.on("resources_discover", () => ({
    skillPaths: [
      `${baseDir}/skills/stock-analysis/SKILL.md`,
      `${baseDir}/skills/stock-screener/SKILL.md`,
      `${baseDir}/skills/strategy-backtest/SKILL.md`,
      `${baseDir}/skills/bull-trend/SKILL.md`,
      `${baseDir}/skills/shrink-pullback/SKILL.md`,
      `${baseDir}/skills/ma-crossover/SKILL.md`,
      `${baseDir}/skills/volume-breakout/SKILL.md`,
      `${baseDir}/skills/bottom-volume/SKILL.md`,
      `${baseDir}/skills/dragon-head/SKILL.md`,
      `${baseDir}/skills/chan-theory/SKILL.md`,
      `${baseDir}/skills/wave-theory/SKILL.md`,
      `${baseDir}/skills/box-oscillation/SKILL.md`,
      `${baseDir}/skills/emotion-cycle/SKILL.md`,
      `${baseDir}/skills/one-yang-three-yin/SKILL.md`,
      `${baseDir}/skills/wisburg-research/SKILL.md`,
      `${baseDir}/skills/market-review/SKILL.md`,
      `${baseDir}/skills/event-driven/SKILL.md`,
      `${baseDir}/skills/expectation-repricing/SKILL.md`,
      `${baseDir}/skills/growth-quality/SKILL.md`,
      `${baseDir}/skills/hot-theme/SKILL.md`,
    ],
  }));
};