"""Add employee_id to users table for watermark feature

Revision ID: 010
Revises: 009
Create Date: 2026-01-04

MED-009: Add employee_id field to support dynamic watermark with user identification.
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa

revision: str = '010'
down_revision: Union[str, None] = '009'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # Add employee_id column to pjctrl_users table
    op.add_column(
        'pjctrl_users',
        sa.Column('employee_id', sa.String(50), nullable=True, unique=True)
    )

    # Create index for employee_id lookups
    op.create_index('ix_pjctrl_users_employee_id', 'pjctrl_users', ['employee_id'])


def downgrade() -> None:
    op.drop_index('ix_pjctrl_users_employee_id', table_name='pjctrl_users')
    op.drop_column('pjctrl_users', 'employee_id')