from __future__ import annotations

import argparse, json
from pathlib import Path
from .pipeline import load_samples, run_stage, scan, select_samples
from .reports import generate_reports, read_json

DEFAULT_INPUT = Path("data/cadfs-sample/CADFS_test")
DEFAULT_OUTPUT = Path("cadfs_to_cdsl/output")


def _parser() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(description="Convert CADFS FeatureScript to CDSL and validate against STEP")
    commands = parser.add_subparsers(dest="command", required=True)
    for name in ("scan", "convert", "rebuild", "compare", "report", "pipeline"):
        command = commands.add_parser(name)
        command.add_argument("--input", type=Path, default=DEFAULT_INPUT)
        command.add_argument("--output", type=Path, default=DEFAULT_OUTPUT)
        if name != "scan":
            command.add_argument("--sample-id", action="append")
            command.add_argument("--offset", type=int, default=0)
            command.add_argument("--limit", type=int)
            command.add_argument("--seed", type=int)
            command.add_argument("--workers", type=int, default=1)
            command.add_argument("--compare-mode", choices=("rp", "strict"), default="rp")
            command.add_argument("--force", action="store_true")
            command.add_argument("--timeout-seconds", type=float, default=30.0, help="per-model OCC timeout (default: 30)")
    return parser


def main(argv: list[str] | None = None) -> int:
    args = _parser().parse_args(argv); args.output.mkdir(parents=True, exist_ok=True)
    if args.command == "scan":
        records = scan(args.input, args.output); result = {"sample_count": len(records), "output": str(args.output / "dataset_index.json")}
    elif args.command == "report":
        manifest = args.output / "manifest.jsonl"; records = [json.loads(line) for line in manifest.read_text().splitlines() if line.strip()]
        result = generate_reports(args.output, records)
    else:
        if not 1 <= args.workers <= 8: raise ValueError("--workers must be between 1 and 8")
        samples = select_samples(load_samples(args.input, args.output), sample_ids=args.sample_id, offset=args.offset, limit=args.limit, seed=args.seed)
        if args.timeout_seconds <= 0: raise ValueError("--timeout-seconds must be positive")
        records = run_stage(args.command, samples, args.output, force=args.force, compare_mode=args.compare_mode, timeout_seconds=args.timeout_seconds, workers=args.workers)
        counts: dict[str, int] = {}
        for record in records: counts[record["status"]] = counts.get(record["status"], 0) + 1
        result = {"sample_count": len(records), "statuses": counts, "summary": str(args.output / "summary.json")}
    print(json.dumps(result, ensure_ascii=True, indent=2, sort_keys=True)); return 0