"""add_user_invitation_fields

Revision ID: 003_add_user_invitation_fields
Revises: 002_add_missing_user_fields
Create Date: 2026-01-04

Adds optional pre-filled fields to user_invitations table:
- first_name
- last_name
- phone
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = '003_add_user_invitation_fields'
down_revision: Union[str, None] = '002_add_missing_user_fields'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """Add optional pre-filled information fields to user_invitations"""

    op.add_column('user_invitations', sa.Column('first_name', sa.String(), nullable=True))
    op.add_column('user_invitations', sa.Column('last_name', sa.String(), nullable=True))
    op.add_column('user_invitations', sa.Column('phone', sa.String(), nullable=True))


def downgrade() -> None:
    """Remove added fields (rollback)"""

    op.drop_column('user_invitations', 'phone')
    op.drop_column('user_invitations', 'last_name')
    op.drop_column('user_invitations', 'first_name')