#!/usr/bin/env python
"""COBOL 迁移验证平台全流程入口：先黑盒 LLM 数据生成，再白盒+黑盒运行比对。

用法:
  python run.py --design <詳細設計書.md> --source <程序.cbl> \
      --file-db-md <COPY句定義書.md> --cpy <COPYBOOK目录> \
      --db-md <DB定義書.md> --output <输出目录>

步骤:
  1) black-box-data-create (DeepSeek LLM 数据生成) → 透传全部参数给 black-box-data-create/main.py
  2) cobol_testgen  (白盒静态分析 + 黑盒数据比对) → python -m cobol_testgen --gcov --black-box <output>/<program_id> <source> <output>

任一步失败即停止，返回该步退出码。
"""
import argparse
import os
import subprocess
import sys

ROOT = os.path.dirname(os.path.abspath(__file__))
BLACKBOX_MAIN = os.path.join(ROOT, "black-box-data-create", "main.py")


def build_parser():
    p = argparse.ArgumentParser(
        description="COBOL 迁移验证平台：先跑黑盒 LLM 数据生成，再跑白盒+黑盒 cobol_testgen")
    p.add_argument("--design", required=True, help="詳細設計書 .md のパス")
    p.add_argument("--source", required=True, help="COBOL ソース .cbl のパス")
    p.add_argument("--file-db-md", required=True, help="ファイル/DB 構造 .md のパス")
    p.add_argument("--cpy", required=True, help="COPYBOOK 格納ディレクトリ")
    p.add_argument("--db-md", required=True, help="DB 定義書 .md のパス")
    p.add_argument("--output", default="output", help="出力ディレクトリ")
    p.add_argument("--api-key", help="DeepSeek API Key（透传给黑盒）")
    p.add_argument("--model", help="API モデル名（透传给黑盒）")
    p.add_argument("--rules", help="ルール格納ディレクトリ（透传给黑盒）")
    p.add_argument("--max-tokens", type=int, help="API 生成トークン上限（透传给黑盒）")
    p.add_argument("--no-black-box", action="store_true",
                   help="跳过黑盒数据生成，仅跑白盒 cobol_testgen")
    p.add_argument("--dry-run", action="store_true", help="只打印要执行的命令，不真正执行")
    return p


def _run(cmd, cwd, label, dry_run=False):
    print(f"\n== {label} ==")
    print(f"  $ {' '.join(cmd)}")
    if dry_run:
        print("  [dry-run] 跳过执行")
        return 0
    r = subprocess.run(cmd, cwd=cwd)
    return r.returncode


def main():
    args = build_parser().parse_args()
    program_id = os.path.splitext(os.path.basename(args.source))[0]
    program_dir = os.path.join(args.output, program_id)

    if not args.no_black_box:
        bb_cmd = [
            sys.executable, BLACKBOX_MAIN,
            "--design", args.design,
            "--source", args.source,
            "--file-db-md", args.file_db_md,
            "--cpy", args.cpy,
            "--db-md", args.db_md,
            "--output", args.output,
        ]
        for opt in ("--api-key", "--model", "--rules", "--max-tokens"):
            v = getattr(args, opt.lstrip("-").replace("-", "_"))
            if v is not None:
                bb_cmd.extend([opt, str(v)])

        rc = _run(
            bb_cmd,
            cwd=ROOT,
            label="步骤1: black-box-data-create LLM 数据生成",
            dry_run=args.dry_run,
        )
        if rc != 0:
            return rc

    ct_cmd = [sys.executable, "-m", "cobol_testgen", "--gcov"]
    if not args.no_black_box:
        ct_cmd += ["--black-box", program_dir]
    ct_cmd += [args.source, args.output]

    return _run(
        ct_cmd,
        cwd=ROOT,
        label="步骤2: cobol_testgen 白盒+黑盒 运行比对",
        dry_run=args.dry_run,
    )


if __name__ == "__main__":
    sys.exit(main())