"""Cross-device handoff: device_tokens + session platform

Revision ID: 002
Revises: 001
Create Date: 2026-03-28
"""

from alembic import op

revision = "002"
down_revision = "001"
branch_labels = None
depends_on = None


def upgrade():
    op.execute("""
    ALTER TABLE users ADD COLUMN IF NOT EXISTS device_tokens JSONB DEFAULT '[]';

    ALTER TABLE sessions ADD COLUMN IF NOT EXISTS platform TEXT DEFAULT 'mac';
    ALTER TABLE sessions ALTER COLUMN platform SET NOT NULL;
    """)


def downgrade():
    op.execute("""
    ALTER TABLE sessions DROP COLUMN IF EXISTS platform;
    ALTER TABLE users DROP COLUMN IF EXISTS device_tokens;
    """)