"""Run only ball/player tracking and soccer-state extraction.

Examples:
    python preprocess.py --init-config configs/tracking_states.json
    python preprocess.py --config configs/tracking_states.json
"""
from __future__ import annotations

import argparse
import json

from soccer_ad.preprocessing.pipeline import (
    load_preprocessing_config,
    run_preprocessing,
    write_preprocessing_config,
)
from soccer_ad.utils.logging import setup_logging


def main() -> int:
    parser = argparse.ArgumentParser(description="Soccer ball/player tracking and state extraction")
    choice = parser.add_mutually_exclusive_group(required=True)
    choice.add_argument("--config", help="JSON preprocessing configuration")
    choice.add_argument("--init-config", metavar="PATH", help="write an editable JSON template and exit")
    parser.add_argument("--log-level", default="INFO")
    args = parser.parse_args()
    setup_logging(args.log_level)

    if args.init_config:
        path = write_preprocessing_config(args.init_config)
        print(f"wrote {path}")
        return 0

    try:
        report = run_preprocessing(load_preprocessing_config(args.config))
    except Exception as exc:
        parser.error(str(exc))
    print(json.dumps(report, indent=2))
    return 0 if report["summary"]["failed"] == 0 else 1


if __name__ == "__main__":
    raise SystemExit(main())