"""add placement policies

Revision ID: 20260414_0004
Revises: 20260413_0003
Create Date: 2026-04-14 10:00:00.000000
"""

from alembic import op
import sqlalchemy as sa


revision = "20260414_0004"
down_revision = "20260413_0003"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "placement_policies",
        sa.Column("id", sa.String(), nullable=False),
        sa.Column("file_class", sa.String(), nullable=False),
        sa.Column("require_local", sa.Boolean(), server_default="1", nullable=False),
        sa.Column("stable_replica_count", sa.Integer(), server_default="1", nullable=False),
        sa.Column("opportunistic_replica_count", sa.Integer(), server_default="0", nullable=False),
        sa.Column("preferred_backend_ids_json", sa.Text(), server_default="[]", nullable=False),
        sa.Column("excluded_backend_ids_json", sa.Text(), server_default="[]", nullable=False),
        sa.Column("max_non_local_size_bytes", sa.Integer(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index(
        "idx_placement_policies_file_class",
        "placement_policies",
        ["file_class"],
        unique=True,
    )


def downgrade() -> None:
    op.drop_index("idx_placement_policies_file_class", table_name="placement_policies")
    op.drop_table("placement_policies")