#!/usr/bin/env python3
"""Wait for the FMA archive to finish, then run post-download readiness."""

from __future__ import annotations

import argparse
import json
import subprocess
import time

PYTHON = "/usr/local/miniconda3/bin/python"
INSPECT = [PYTHON, "scripts/prepare_fma_archive.py", "inspect"]
POST = [PYTHON, "scripts/fma_postdownload_ready.py"]


def inspect() -> dict:
    return json.loads(subprocess.check_output(INSPECT, text=True))


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--interval", type=float, default=30.0)
    parser.add_argument("--max-cycles", type=int, default=3)
    args = parser.parse_args()

    snapshots = []
    for _ in range(args.max_cycles):
        snap = inspect()
        snapshots.append(snap)
        if snap.get("archive_size", 0) >= snap.get("archive_bytes_expected", 0):
            result = json.loads(subprocess.check_output(POST, text=True))
            print(json.dumps({"status": "completed", "snapshots": snapshots, "postdownload": result}, indent=2, ensure_ascii=False))
            return
        time.sleep(args.interval)

    print(json.dumps({"status": "waiting", "snapshots": snapshots}, indent=2, ensure_ascii=False))


if __name__ == "__main__":
    main()