from __future__ import annotations
import logging
from datetime import datetime, timedelta, timezone
from typing import TYPE_CHECKING

from sqlalchemy import delete

from roundtable.models.monitors import DnsResult, PingResult
from roundtable.models.metrics import PluginMetric
from roundtable.models.ansible import AnsibleRun

if TYPE_CHECKING:
    from quart import Quart

logger = logging.getLogger(__name__)


async def run_cleanup(app: "Quart") -> None:
    """Delete rows older than DATA_RETENTION_DAYS from time-series tables."""
    retention_days: int = app.config.get("DATA_RETENTION_DAYS", 90)
    cutoff = datetime.now(timezone.utc) - timedelta(days=retention_days)

    async with app.db_sessionmaker() as session:
        async with session.begin():
            for model, ts_col in [
                (PingResult, PingResult.probed_at),
                (DnsResult, DnsResult.resolved_at),
                (PluginMetric, PluginMetric.recorded_at),
                (AnsibleRun, AnsibleRun.started_at),
            ]:
                result = await session.execute(
                    delete(model).where(ts_col < cutoff)
                )
                if result.rowcount:
                    logger.info(f"Pruned {result.rowcount} rows from {model.__tablename__}")