"""Docker collection goes per-host: add host_id, re-key by (host_id, name)

Collection moved from the central single-socket scrape to the host agent, so
containers are now scoped to the host that reported them. docker_containers is
re-keyed (host_id, name) and docker_metrics gains host_id.

Dev-only posture (rule 122): the old tables only ever held the Steward box's
own containers (a single global namespace), which are disposable — so this
DROP+recreates rather than backfilling a host_id onto orphan rows.

Revision ID: docker_002_host_scoped
Revises: docker_001_initial
Create Date: 2026-06-18
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa

revision: str = "docker_002_host_scoped"
down_revision: Union[str, None] = "docker_001_initial"
branch_labels: Union[str, Sequence[str], None] = None
# FK targets hosts.id (created in 0002_core_monitors) — make the edge explicit.
depends_on: Union[str, Sequence[str], None] = ("0002_core_monitors",)


def upgrade() -> None:
    op.drop_table("docker_metrics")
    op.drop_table("docker_containers")

    op.create_table(
        "docker_containers",
        sa.Column("host_id", sa.String(36),
                  sa.ForeignKey("hosts.id", ondelete="CASCADE"), primary_key=True),
        sa.Column("name", sa.String(255), primary_key=True),
        sa.Column("container_id", sa.String(64), nullable=False, server_default=""),
        sa.Column("image", sa.String(512), nullable=False, server_default=""),
        sa.Column("status", sa.String(32), nullable=False, server_default="unknown"),
        sa.Column("cpu_pct", sa.Float, nullable=True),
        sa.Column("mem_usage_bytes", sa.Integer, nullable=True),
        sa.Column("mem_limit_bytes", sa.Integer, nullable=True),
        sa.Column("mem_pct", sa.Float, nullable=True),
        sa.Column("restart_count", sa.Integer, nullable=False, server_default="0"),
        sa.Column("ports_json", sa.Text, nullable=False, server_default="[]"),
        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("scraped_at", sa.DateTime(timezone=True), nullable=False),
    )

    op.create_table(
        "docker_metrics",
        sa.Column("id", sa.String(36), primary_key=True),
        sa.Column("host_id", sa.String(36),
                  sa.ForeignKey("hosts.id", ondelete="CASCADE"), nullable=False),
        sa.Column("container_name", sa.String(255), nullable=False),
        sa.Column("scraped_at", sa.DateTime(timezone=True), nullable=False),
        sa.Column("cpu_pct", sa.Float, nullable=False, server_default="0"),
        sa.Column("mem_pct", sa.Float, nullable=False, server_default="0"),
        sa.Column("mem_usage_bytes", sa.Integer, nullable=False, server_default="0"),
    )
    op.create_index("ix_docker_metrics_host_id", "docker_metrics", ["host_id"])
    op.create_index("ix_docker_metrics_container_name", "docker_metrics",
                    ["container_name"])
    op.create_index("ix_docker_metrics_scraped_at", "docker_metrics", ["scraped_at"])
    op.create_index("ix_docker_metrics_host_container_time", "docker_metrics",
                    ["host_id", "container_name", "scraped_at"])


def downgrade() -> None:
    op.drop_table("docker_metrics")
    op.drop_table("docker_containers")

    op.create_table(
        "docker_containers",
        sa.Column("name", sa.String(255), primary_key=True),
        sa.Column("container_id", sa.String(64), nullable=False, server_default=""),
        sa.Column("image", sa.String(512), nullable=False, server_default=""),
        sa.Column("status", sa.String(32), nullable=False, server_default="unknown"),
        sa.Column("cpu_pct", sa.Float, nullable=True),
        sa.Column("mem_usage_bytes", sa.Integer, nullable=True),
        sa.Column("mem_limit_bytes", sa.Integer, nullable=True),
        sa.Column("mem_pct", sa.Float, nullable=True),
        sa.Column("restart_count", sa.Integer, nullable=False, server_default="0"),
        sa.Column("ports_json", sa.Text, nullable=False, server_default="[]"),
        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("scraped_at", sa.DateTime(timezone=True), nullable=False),
    )
    op.create_table(
        "docker_metrics",
        sa.Column("id", sa.String(36), primary_key=True),
        sa.Column("container_name", sa.String(255), nullable=False, index=True),
        sa.Column("scraped_at", sa.DateTime(timezone=True), nullable=False, index=True),
        sa.Column("cpu_pct", sa.Float, nullable=False, server_default="0"),
        sa.Column("mem_pct", sa.Float, nullable=False, server_default="0"),
        sa.Column("mem_usage_bytes", sa.Integer, nullable=False, server_default="0"),
    )