"""profile history reads

Revision ID: e8b3f1a6c2d4
Revises: d8e3b5a1c7f4
Create Date: 2026-10-19 03:00:00.000000

"""

from collections.abc import Sequence

from alembic import op

revision: str = "e8b3f1a6c2d4"
down_revision: str | None = "d8e3b5a1c7f4"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None

_PEER_STATE = (
    "first_name, last_name, username, phone, photo_unique_id, is_deleted_account"
)


def upgrade() -> None:
    op.execute(
        "ALTER TABLE peer_history DROP CONSTRAINT peer_history_pkey, "
        "ADD CONSTRAINT peer_history_pkey "
        f"PRIMARY KEY (account_id, peer_id, observed_at) INCLUDE ({_PEER_STATE})"
    )
    op.execute(
        "ALTER TABLE chat_history DROP CONSTRAINT chat_history_pkey, "
        "ADD CONSTRAINT chat_history_pkey "
        "PRIMARY KEY (account_id, chat_id, message_id, ts)"
    )
    op.execute(
        "CREATE INDEX ix_memberships_user ON memberships (account_id, user_id, ts DESC)"
    )
    op.execute(
        "CREATE INDEX ix_jobs_kind_created ON jobs (account_id, kind, created_at DESC)"
    )
    with op.get_context().autocommit_block():
        op.execute("VACUUM (ANALYZE) peer_history")


def downgrade() -> None:
    op.execute("DROP INDEX IF EXISTS ix_jobs_kind_created")
    op.execute("DROP INDEX IF EXISTS ix_memberships_user")
    op.execute(
        "ALTER TABLE chat_history DROP CONSTRAINT chat_history_pkey, "
        "ADD CONSTRAINT chat_history_pkey PRIMARY KEY (account_id, chat_id, message_id)"
    )
    op.execute(
        "ALTER TABLE peer_history DROP CONSTRAINT peer_history_pkey, "
        "ADD CONSTRAINT peer_history_pkey PRIMARY KEY (account_id, peer_id, observed_at)"
    )