"""Add whitelisted_user_ids column to guild_settings table.

Revision ID: 20260125_add_whitelist
Revises: 20260125_add_in_channel_warnings
Create Date: 2026-01-25 01:00:00.000000
"""

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

# revision identifiers, used by Alembic.
revision = "20260125_add_whitelist"
down_revision = "20260125_add_in_channel_warnings"
branch_labels = None
depends_on = None


def upgrade() -> None:
    """Add whitelisted_user_ids column to guild_settings table."""
    op.add_column(
        "guild_settings",
        sa.Column(
            "whitelisted_user_ids",
            JSONB().with_variant(sa.JSON(), "sqlite"),
            nullable=False,
            server_default="[]",
        ),
    )


def downgrade() -> None:
    """Remove whitelisted_user_ids column from guild_settings table."""
    op.drop_column("guild_settings", "whitelisted_user_ids")