"""Add composite indexes for calendar_members and ntfy_sent

Revision ID: 053
Revises: 052
"""
from alembic import op

revision = "053"
down_revision = "052"


def upgrade():
    # AW-1: Hot query polled every 5s uses (user_id, status) together
    op.create_index(
        "ix_calendar_members_user_id_status",
        "calendar_members",
        ["user_id", "status"],
    )
    # AS-6: Dedup lookup in notification dispatch uses (user_id, sent_at)
    op.create_index(
        "ix_ntfy_sent_user_id_sent_at",
        "ntfy_sent",
        ["user_id", "sent_at"],
    )


def downgrade():
    op.drop_index("ix_ntfy_sent_user_id_sent_at", table_name="ntfy_sent")
    op.drop_index("ix_calendar_members_user_id_status", table_name="calendar_members")