"""Add permission enhancements - manager flag and project members table

Revision ID: 014
Revises: a0a0f2710e01
Create Date: 2026-01-10

Add is_department_manager flag to users and create project_members table
for cross-department collaboration support.
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa

revision: str = '014'
down_revision: Union[str, None] = 'a0a0f2710e01'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # Add is_department_manager column to pjctrl_users table
    op.add_column(
        'pjctrl_users',
        sa.Column('is_department_manager', sa.Boolean(), nullable=False, server_default='0')
    )

    # Create project_members table for cross-department collaboration
    op.create_table(
        'pjctrl_project_members',
        sa.Column('id', sa.String(36), primary_key=True),
        sa.Column('project_id', sa.String(36), sa.ForeignKey('pjctrl_projects.id', ondelete='CASCADE'), nullable=False),
        sa.Column('user_id', sa.String(36), sa.ForeignKey('pjctrl_users.id', ondelete='CASCADE'), nullable=False),
        sa.Column('role', sa.String(50), nullable=False, server_default='member'),
        sa.Column('added_by', sa.String(36), sa.ForeignKey('pjctrl_users.id'), nullable=False),
        sa.Column('created_at', sa.DateTime(), server_default=sa.func.now(), nullable=False),
        # Ensure a user can only be added once per project
        sa.UniqueConstraint('project_id', 'user_id', name='uq_project_member'),
    )

    # Create indexes for efficient lookups
    op.create_index('ix_pjctrl_project_members_project_id', 'pjctrl_project_members', ['project_id'])
    op.create_index('ix_pjctrl_project_members_user_id', 'pjctrl_project_members', ['user_id'])


def downgrade() -> None:
    op.drop_index('ix_pjctrl_project_members_user_id', table_name='pjctrl_project_members')
    op.drop_index('ix_pjctrl_project_members_project_id', table_name='pjctrl_project_members')
    op.drop_table('pjctrl_project_members')
    op.drop_column('pjctrl_users', 'is_department_manager')