"""Remove Arena check records older than the retention window."""
from __future__ import annotations

import os
import sqlite3
import sys
import time
from datetime import datetime, timedelta, timezone

DB_PATH = os.environ.get("DB_PATH", "/data/arena.db")
RETENTION_DAYS = int(os.environ.get("CHECK_RETENTION_DAYS", "30"))


def cleanup(db_path: str = DB_PATH, retention_days: int = RETENTION_DAYS) -> int:
    if retention_days < 1:
        raise ValueError("CHECK_RETENTION_DAYS must be >= 1")
    cutoff = (datetime.now(timezone.utc) - timedelta(days=retention_days)).isoformat()
    db = sqlite3.connect(db_path, timeout=30)
    try:
        db.execute("PRAGMA busy_timeout=30000")
        before = db.total_changes
        db.execute("BEGIN IMMEDIATE")
        db.execute("DELETE FROM checks WHERE checked_at < ?", (cutoff,))
        removed = db.total_changes - before
        db.commit()
        # Keep the persistent WAL bounded without a full VACUUM/long lock.
        db.execute("PRAGMA wal_checkpoint(PASSIVE)")
        print(f"[cleanup] removed={removed} cutoff={cutoff} retention_days={retention_days}")
        return removed
    except Exception:
        db.rollback()
        raise
    finally:
        db.close()


if __name__ == "__main__":
    try:
        cleanup()
    except Exception as exc:
        print(f"[cleanup] ERROR: {exc}", file=sys.stderr)
        raise