"""folders and capture policy

Revision ID: b2f7c1a9d3e4
Revises: 77df960a31de
Create Date: 2026-05-29 17:30:00.000000

"""

from collections.abc import Sequence

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

revision: str = "b2f7c1a9d3e4"
down_revision: str | None = "77df960a31de"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None

_TOGGLES = (
    "messages",
    "media",
    "self_destruct_media",
    "stt",
    "reactions",
    "track_edits_deletes",
    "profile_history",
    "stories",
    "presence",
    "backfill",
)


def _toggle_columns() -> list[sa.Column]:
    return [
        sa.Column(name, sa.Boolean(), nullable=False, server_default=sa.false())
        for name in _TOGGLES
    ]


def upgrade() -> None:
    op.create_table(
        "folders",
        sa.Column("account_id", sa.Integer(), nullable=False),
        sa.Column("folder_id", sa.Integer(), nullable=False),
        sa.Column("title", sa.String(), nullable=False),
        sa.Column("order_index", sa.Integer(), nullable=False),
        sa.Column("is_chatlist", sa.Boolean(), nullable=False),
        sa.Column("raw", postgresql.JSONB(astext_type=sa.Text()), nullable=False),
        sa.Column(
            "updated_at",
            sa.DateTime(timezone=True),
            server_default=sa.text("now()"),
            nullable=False,
        ),
        sa.PrimaryKeyConstraint("account_id", "folder_id"),
    )
    op.create_table(
        "capture_policy",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("account_id", sa.Integer(), nullable=True),
        sa.Column("scope_type", sa.String(), nullable=False),
        sa.Column("scope_id", sa.BigInteger(), nullable=True),
        *_toggle_columns(),
        sa.Column(
            "created_at",
            sa.DateTime(timezone=True),
            server_default=sa.text("now()"),
            nullable=False,
        ),
        sa.Column(
            "updated_at",
            sa.DateTime(timezone=True),
            server_default=sa.text("now()"),
            nullable=False,
        ),
        sa.PrimaryKeyConstraint("id"),
        sa.UniqueConstraint("account_id", "scope_type", "scope_id"),
    )
    op.execute(
        "CREATE UNIQUE INDEX ix_capture_policy_default "
        "ON capture_policy (scope_type) WHERE scope_type LIKE 'default_%'"
    )

    op.execute(
        "INSERT INTO capture_policy "
        "(scope_type, messages, media, self_destruct_media, stt, reactions, "
        "track_edits_deletes, profile_history, stories, presence, backfill) VALUES "
        "('default_channel', FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE), "
        "('default_group', TRUE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE), "
        "('default_dm', TRUE, TRUE, TRUE, TRUE, TRUE, TRUE, TRUE, TRUE, TRUE, TRUE)"
    )


def downgrade() -> None:
    op.drop_table("capture_policy")
    op.drop_table("folders")