"""Small CLI that keeps optional model dependencies out of the game path."""

import argparse
import json
from pathlib import Path


def main() -> None:
    parser = argparse.ArgumentParser(description="Stackcraft: play, replay, and measure decisions.")
    parser.add_argument("--version", action="version", version="stackcraft 0.1.0")
    commands = parser.add_subparsers(dest="command", required=True)
    serve = commands.add_parser("serve", help="Start the local game and replay service")
    serve.add_argument("--host", default="127.0.0.1")
    serve.add_argument("--port", type=int, default=8080)
    tournament = commands.add_parser("tournament", help="Run paired offline development baselines")
    tournament.add_argument("--seed-start", type=int, default=1000)
    tournament.add_argument("--episodes", type=int, default=20)
    tournament.add_argument("--max-pieces", type=int, default=100)
    tournament.add_argument("--output", type=Path, required=True)
    generate = commands.add_parser("generate-data", help="Generate and audit expert-labelled data")
    generate.add_argument("--output", type=Path, required=True)
    args = parser.parse_args()
    if args.command == "serve":
        import uvicorn

        uvicorn.run("stackcraft.server:create_app", factory=True, host=args.host, port=args.port)
    elif args.command == "tournament":
        from stackcraft.players import HeuristicPlayer, RandomPlayer
        from stackcraft.tournament import run_tournament

        if args.episodes < 1 or args.max_pieces < 1:
            parser.error("--episodes and --max-pieces must be positive")
        if args.output.exists():
            parser.error(f"output already exists: {args.output}; choose a new path")
        seeds = list(range(args.seed_start, args.seed_start + args.episodes))
        result = run_tournament(
            {"random": RandomPlayer, "heuristic": HeuristicPlayer}, seeds, args.max_pieces
        )
        args.output.parent.mkdir(parents=True, exist_ok=True)
        with args.output.open("x") as output:
            json.dump(result, output, indent=2, allow_nan=False)
            output.write("\n")
        print(f"Saved {args.episodes} paired development episodes to {args.output}")
    elif args.command == "generate-data":
        from stackcraft.data import DatasetConfig, generate_dataset, write_dataset
        from stackcraft.provenance import source_identity

        if args.output.exists():
            parser.error(f"output already exists: {args.output}; choose a new path")
        try:
            identity = source_identity(Path(__file__).resolve().parents[2])
        except ValueError as error:
            parser.error(str(error))
        commit = identity["source_commit"] + ("+working-tree" if identity["source_dirty"] else "")
        bundle = generate_dataset(DatasetConfig(), source_commit=commit)
        write_dataset(bundle, args.output)
        print(f"Saved expert dataset and manifest to {args.output}")


if __name__ == "__main__":
    main()