#!/usr/bin/env bash
set -euo pipefail

usage() {
  cat <<'EOF'
Usage:
  ./scripts/cir/run_cir_replay_sequence_rpki_client.sh \
    --sequence-root <path> \
    [--build-dir <path> | --rpki-client-bin <path>] \
    [--real-rsync-bin <path>]
EOF
}

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
SEQUENCE_ROOT=""
BUILD_DIR=""
RPKI_CLIENT_BIN=""
REAL_RSYNC_BIN="${REAL_RSYNC_BIN:-/usr/bin/rsync}"
STEP_SCRIPT="$ROOT_DIR/scripts/cir/run_cir_replay_rpki_client.sh"

while [[ $# -gt 0 ]]; do
  case "$1" in
    --sequence-root) SEQUENCE_ROOT="$2"; shift 2 ;;
    --build-dir) BUILD_DIR="$2"; shift 2 ;;
    --rpki-client-bin) RPKI_CLIENT_BIN="$2"; shift 2 ;;
    --real-rsync-bin) REAL_RSYNC_BIN="$2"; shift 2 ;;
    -h|--help) usage; exit 0 ;;
    *) echo "unknown argument: $1" >&2; usage; exit 2 ;;
  esac
done

[[ -n "$SEQUENCE_ROOT" && ( -n "$BUILD_DIR" || -n "$RPKI_CLIENT_BIN" ) ]] || { usage >&2; exit 2; }

SEQUENCE_ROOT="$(python3 - <<'PY' "$SEQUENCE_ROOT"
from pathlib import Path
import sys
print(Path(sys.argv[1]).resolve())
PY
)"

SUMMARY_JSON="$SEQUENCE_ROOT/sequence-summary-rpki-client.json"
SUMMARY_MD="$SEQUENCE_ROOT/sequence-summary-rpki-client.md"

python3 - <<'PY' "$SEQUENCE_ROOT" "$SUMMARY_JSON" "$SUMMARY_MD" "$STEP_SCRIPT" "$BUILD_DIR" "$RPKI_CLIENT_BIN" "$REAL_RSYNC_BIN"
import json
import subprocess
import sys
from pathlib import Path

sequence_root = Path(sys.argv[1])
summary_json = Path(sys.argv[2])
summary_md = Path(sys.argv[3])
step_script = Path(sys.argv[4])
build_dir = sys.argv[5]
rpki_client_bin = sys.argv[6]
real_rsync_bin = sys.argv[7]

sequence = json.loads((sequence_root / "sequence.json").read_text(encoding="utf-8"))
repo_bytes_db = sequence_root / sequence["repoBytesDbPath"]
steps = sequence["steps"]
results = []
all_match = True

for step in steps:
    step_id = step["stepId"]
    out_dir = sequence_root / "replay-rpki-client" / step_id
    out_dir.parent.mkdir(parents=True, exist_ok=True)
    cmd = [
        str(step_script),
        "--cir",
        str(sequence_root / step["cirPath"]),
        "--out-dir",
        str(out_dir),
        "--reference-ccr",
        str(sequence_root / step["ccrPath"]),
        "--real-rsync-bin",
        real_rsync_bin,
    ]
    if rpki_client_bin:
        cmd.extend(["--rpki-client-bin", rpki_client_bin])
    else:
        cmd.extend(["--build-dir", build_dir])
    cmd.extend(["--repo-bytes-db", str(repo_bytes_db)])
    proc = subprocess.run(cmd, capture_output=True, text=True)
    if proc.returncode != 0:
        raise SystemExit(
            f"rpki-client sequence replay failed for {step_id}: stdout={proc.stdout} stderr={proc.stderr}"
        )
    compare = json.loads((out_dir / "compare-summary.json").read_text(encoding="utf-8"))
    match = bool(compare["vrps"]["match"]) and bool(compare["vaps"]["match"])
    all_match = all_match and match
    results.append(
        {
            "stepId": step_id,
            "kind": step["kind"],
            "validationTime": step["validationTime"],
            "outDir": str(out_dir),
            "comparePath": str(out_dir / "compare-summary.json"),
            "match": match,
            "compare": compare,
        }
    )

summary = {
    "version": 1,
    "participant": "rpki-client",
    "sequenceRoot": str(sequence_root),
    "stepCount": len(results),
    "allMatch": all_match,
    "steps": results,
}
summary_json.write_text(json.dumps(summary, indent=2), encoding="utf-8")
lines = [
    "# rpki-client CIR Sequence Replay Summary",
    "",
    f"- `sequence_root`: `{sequence_root}`",
    f"- `step_count`: `{len(results)}`",
    f"- `all_match`: `{all_match}`",
    "",
]
summary_md.write_text("\n".join(lines), encoding="utf-8")
PY

echo "done: $SEQUENCE_ROOT"