from __future__ import annotations

import argparse
import json
import sys
from dataclasses import asdict

from .config import settings
from .database import SessionLocal
from .importer import import_records
from .community_importer import stage_observations
from .retention import RetentionPolicy, apply_retention
from .storage import delete_screenshot
from .catalog_audit import audit_catalog


def main() -> int:
    parser = argparse.ArgumentParser(prog="python -m app.cli")
    sub = parser.add_subparsers(dest="command", required=True)
    command = sub.add_parser("import-records")
    command.add_argument("--url", default="https://rf4game.de/records/region/RU/")
    command.add_argument("--region", default="RU")
    command.add_argument("--category", default="records")
    community = sub.add_parser("stage-community-json")
    community.add_argument("--input", default="-", help="JSON array path or - for stdin")
    community.add_argument("--limit", type=int, default=500)
    cleanup = sub.add_parser("cleanup-retention")
    cleanup.add_argument("--apply", action="store_true", help="apply changes; default is dry-run")
    sub.add_parser("audit-catalog")
    args = parser.parse_args()
    with SessionLocal() as session:
        if args.command == "import-records":
            run = import_records(session, url=args.url, region=args.region, category=args.category)
            print(f"import {run.status.value}: seen={run.rows_seen} created={run.rows_created} updated={run.rows_updated}")
        elif args.command == "stage-community-json":
            if not 1 <= args.limit <= 5000:
                parser.error("--limit must be between 1 and 5000")
            stream = sys.stdin if args.input == "-" else open(args.input, encoding="utf-8")
            try:
                payload = json.load(stream)
            finally:
                if stream is not sys.stdin:
                    stream.close()
            if not isinstance(payload, list):
                parser.error("input must be a JSON array")
            created, updated = stage_observations(session, payload[:args.limit])
            print(f"staged: created={created} updated={updated}")
        elif args.command == "cleanup-retention":
            policy = RetentionPolicy(
                submission_days=settings.retention_submission_days,
                unreviewed_days=settings.retention_unreviewed_days,
                approved_personal_days=settings.retention_approved_personal_days,
                staging_days=settings.retention_staging_days,
                audit_days=settings.retention_audit_days,
                published_payload_days=settings.retention_published_payload_days,
            )
            counts = apply_retention(session, policy=policy, dry_run=not args.apply, delete_object=delete_screenshot)
            print(json.dumps({"mode": "apply" if args.apply else "dry-run", "policy": asdict(policy), "counts": counts}, ensure_ascii=False))
        else:
            result = audit_catalog(session)
            print(json.dumps(result, ensure_ascii=False))
            return 1 if result["failures"] else 0
    return 0


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