"""Traefik access log traffic summaries

Revision ID: traefik_003_access_log
Revises: traefik_002_stats
Create Date: 2026-03-20
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa

revision: str = "traefik_003_access_log"
down_revision: Union[str, None] = "traefik_002_stats"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.create_table(
        "traefik_access_summaries",
        sa.Column("id", sa.String(36), primary_key=True),
        sa.Column("period_start", sa.DateTime(timezone=True), nullable=False),
        sa.Column("period_end", sa.DateTime(timezone=True), nullable=False),
        sa.Column("total_requests", sa.Integer, nullable=False, server_default="0"),
        sa.Column("internal_requests", sa.Integer, nullable=False, server_default="0"),
        sa.Column("external_requests", sa.Integer, nullable=False, server_default="0"),
        sa.Column("total_bytes", sa.Float, nullable=False, server_default="0"),
        sa.Column("top_ips_json", sa.Text, nullable=False, server_default="[]"),
        sa.Column("top_countries_json", sa.Text, nullable=False, server_default="[]"),
        sa.Column("top_routers_json", sa.Text, nullable=False, server_default="[]"),
    )
    op.create_index(
        "ix_traefik_access_summaries_period_end",
        "traefik_access_summaries",
        ["period_end"],
    )


def downgrade() -> None:
    op.drop_index("ix_traefik_access_summaries_period_end", "traefik_access_summaries")
    op.drop_table("traefik_access_summaries")