from __future__ import annotations
import time, json, argparse
from pathlib import Path
from dir2md.core import Config, generate_markdown_report

def run_case(root: Path, preset: str, mode: str | None, budget: int, file_budget: int) -> dict:
    cfg = Config(
        root=root, output=root/"_BENCH.md", include_globs=[], exclude_globs=[], omit_globs=[],
        respect_gitignore=True, follow_symlinks=False, max_bytes=200_000, max_lines=2000,
        include_contents=True, only_ext=None, add_stats=True, add_toc=False,
        llm_mode=(mode or "ref"), budget_tokens=budget, max_file_tokens=file_budget,
        dedup_bits=16, sample_head=120, sample_tail=40, strip_comments=False,
        emit_manifest=False, preset=preset, explain_capsule=True,
    )
    t0 = time.perf_counter()
    md = generate_markdown_report(cfg)
    dt = time.perf_counter() - t0
    est = md.split("Estimated tokens (prompt): `")[-1].split("`")[0]
    return {"preset": preset, "mode": cfg.llm_mode, "elapsed_sec": round(dt,3), "est_tokens": int(est)}


def main():
    ap = argparse.ArgumentParser()
    ap.add_argument("path", nargs="?", default=".")
    ns = ap.parse_args()
    root = Path(ns.path).resolve()
    cases = [
        ("iceberg", None, 6000, 1000),
        ("pro", "summary", 6000, 1000),
        ("pro", "ref", 4000, 1000),
        ("pro", "inline", 8000, 1200),
    ]
    rows = [run_case(root, *c) for c in cases]
    print(json.dumps(rows, indent=2))

if __name__ == "__main__":
    main()