"""create scan ranges table

Revision ID: 0003_scan_ranges
Revises: 0002_audit_logs
Create Date: 2026-07-08
"""

from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql


revision: str = "0003_scan_ranges"
down_revision: Union[str, None] = "0002_audit_logs"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.create_table(
        "scan_ranges",
        sa.Column("id", postgresql.UUID(as_uuid=True), primary_key=True, server_default=sa.text("gen_random_uuid()")),
        sa.Column("room_id", postgresql.UUID(as_uuid=True), sa.ForeignKey("rooms.id", ondelete="SET NULL"), nullable=True),
        sa.Column("name", sa.Text(), nullable=False),
        sa.Column("range_type", sa.Text(), nullable=False, server_default=sa.text("'cidr'")),
        sa.Column("cidr", sa.Text(), nullable=True),
        sa.Column("start_ip", postgresql.INET(), nullable=True),
        sa.Column("end_ip", postgresql.INET(), nullable=True),
        sa.Column("enabled", sa.Boolean(), nullable=False, server_default=sa.text("true")),
        sa.Column("scan_interval_seconds", sa.Integer(), nullable=False, server_default=sa.text("900")),
        sa.Column("timeout_ms", sa.Integer(), nullable=False, server_default=sa.text("3000")),
        sa.Column("retry_limit", sa.Integer(), nullable=False, server_default=sa.text("1")),
        sa.Column("parallelism_limit", sa.Integer(), nullable=False, server_default=sa.text("10")),
        sa.Column("stagger_ms", sa.Integer(), nullable=False, server_default=sa.text("250")),
        sa.Column("api_port", sa.Integer(), nullable=False, server_default=sa.text("4028")),
        sa.Column("last_scan_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("next_scan_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("created_by_user_id", postgresql.UUID(as_uuid=True), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
        sa.CheckConstraint(
            """
            (range_type = 'cidr' AND cidr IS NOT NULL)
            OR
            (range_type = 'start_end' AND start_ip IS NOT NULL AND end_ip IS NOT NULL)
            """,
            name="scan_range_has_valid_range",
        ),
    )

    op.create_index("idx_scan_ranges_room_id", "scan_ranges", ["room_id"])
    op.create_index("idx_scan_ranges_enabled", "scan_ranges", ["enabled"])
    op.create_index("idx_scan_ranges_next_scan_at", "scan_ranges", ["next_scan_at"])


def downgrade() -> None:
    op.drop_index("idx_scan_ranges_next_scan_at", table_name="scan_ranges")
    op.drop_index("idx_scan_ranges_enabled", table_name="scan_ranges")
    op.drop_index("idx_scan_ranges_room_id", table_name="scan_ranges")
    op.drop_table("scan_ranges")