"""add messages.restore_mailbox_id

Records the folder a message was in before it was moved to Archive/Trash/Junk, so
the Restore action can return it to where it came from (e.g. Sent) instead of
always dumping it in the Inbox.

Revision ID: 0010
Revises: 0009
Create Date: 2026-07-20
"""

from __future__ import annotations

from collections.abc import Sequence

from alembic import op

revision: str = "0010"
down_revision: str | None = "0009"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    op.execute(
        "ALTER TABLE messages ADD COLUMN restore_mailbox_id uuid "
        "REFERENCES mailboxes(id) ON DELETE SET NULL"
    )


def downgrade() -> None:
    op.execute("ALTER TABLE messages DROP COLUMN IF EXISTS restore_mailbox_id")