"""Add roles system

Revision ID: 001_add_roles
Revises:
Create Date: 2024-12-14

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision: str = '001_add_roles'
down_revision: Union[str, None] = None
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # Add role column to users table
    op.add_column('users', sa.Column('role', sa.String(20), nullable=False, server_default='user'))

    # Add role column to participants table
    op.add_column('participants', sa.Column('role', sa.String(20), nullable=False, server_default='participant'))

    # Rename organizer_id to creator_id in marathons table
    op.alter_column('marathons', 'organizer_id', new_column_name='creator_id')

    # Update existing participants: set role='organizer' for marathon creators
    op.execute("""
        UPDATE participants p
        SET role = 'organizer'
        FROM marathons m
        WHERE p.marathon_id = m.id AND p.user_id = m.creator_id
    """)

    # Add status column to games table
    op.add_column('games', sa.Column('status', sa.String(20), nullable=False, server_default='approved'))

    # Rename added_by_id to proposed_by_id in games table
    op.alter_column('games', 'added_by_id', new_column_name='proposed_by_id')

    # Add approved_by_id column to games table
    op.add_column('games', sa.Column('approved_by_id', sa.Integer(), nullable=True))
    op.create_foreign_key(
        'fk_games_approved_by_id',
        'games', 'users',
        ['approved_by_id'], ['id'],
        ondelete='SET NULL'
    )


def downgrade() -> None:
    # Remove approved_by_id from games
    op.drop_constraint('fk_games_approved_by_id', 'games', type_='foreignkey')
    op.drop_column('games', 'approved_by_id')

    # Rename proposed_by_id back to added_by_id
    op.alter_column('games', 'proposed_by_id', new_column_name='added_by_id')

    # Remove status from games
    op.drop_column('games', 'status')

    # Rename creator_id back to organizer_id
    op.alter_column('marathons', 'creator_id', new_column_name='organizer_id')

    # Remove role from participants
    op.drop_column('participants', 'role')

    # Remove role from users
    op.drop_column('users', 'role')