# plugins/traefik/models.py
from __future__ import annotations
import uuid
from datetime import datetime, timezone
from sqlalchemy import DateTime, Float, Integer, String, Text
from sqlalchemy.orm import Mapped, mapped_column
from fabledscryer.models.base import Base


class TraefikMetric(Base):
    __tablename__ = "traefik_metrics"

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    router_name: Mapped[str] = mapped_column(String(255), nullable=False)
    scraped_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, default=lambda: datetime.now(timezone.utc)
    )
    request_rate: Mapped[float] = mapped_column(Float, nullable=False)
    error_rate_4xx_pct: Mapped[float] = mapped_column(Float, nullable=False)
    error_rate_5xx_pct: Mapped[float] = mapped_column(Float, nullable=False)
    latency_p50_ms: Mapped[float] = mapped_column(Float, nullable=False)
    latency_p95_ms: Mapped[float] = mapped_column(Float, nullable=False)
    latency_p99_ms: Mapped[float] = mapped_column(Float, nullable=False)
    response_bytes_rate: Mapped[float] = mapped_column(Float, nullable=False, default=0.0)


class TraefikCert(Base):
    """TLS certificate expiry tracking — upserted by serial each scrape."""
    __tablename__ = "traefik_certs"

    serial: Mapped[str] = mapped_column(String(255), primary_key=True)
    cn: Mapped[str | None] = mapped_column(String(512), nullable=True)
    sans: Mapped[str | None] = mapped_column(Text, nullable=True)
    not_after: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    scraped_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, default=lambda: datetime.now(timezone.utc)
    )


class TraefikAccessSummary(Base):
    """Aggregated traffic-origin summary — one row per scheduler tick."""
    __tablename__ = "traefik_access_summaries"

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    period_start: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    period_end: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    total_requests: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    internal_requests: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    external_requests: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    total_bytes: Mapped[float] = mapped_column(Float, nullable=False, default=0.0)
    top_ips_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
    top_countries_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
    top_routers_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")


class TraefikGlobalStat(Base):
    """One row per scrape — process-level and config stats."""
    __tablename__ = "traefik_global_stats"

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    scraped_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, default=lambda: datetime.now(timezone.utc)
    )
    open_conns_total: Mapped[float | None] = mapped_column(Float, nullable=True)
    config_reloads_total: Mapped[float | None] = mapped_column(Float, nullable=True)
    config_last_reload_success: Mapped[float | None] = mapped_column(Float, nullable=True)
    process_memory_bytes: Mapped[float | None] = mapped_column(Float, nullable=True)