"""create users table

Revision ID: a1b2c3d4e5f6
Revises: 
Create Date: 2023-01-01 12:00:00.000000
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import ENUM
from datetime import datetime
import uuid

# revision identifiers
revision = 'a1b2c3d4e5f6'
down_revision = None
branch_labels = None
depends_on = None

def upgrade():
    user_roles = ENUM('admin', 'lawyer', 'client', name='user_roles')
    user_roles.create(op.get_bind())
    
    op.create_table(
        'users',
        sa.Column('id', sa.String(), primary_key=True, default=lambda: str(uuid.uuid4())),
        sa.Column('first_name', sa.String(100), nullable=False),
        sa.Column('last_name', sa.String(100), nullable=False),
        sa.Column('email', sa.String(255), nullable=False, unique=True),
        sa.Column('password', sa.String(255), nullable=False),
        sa.Column('phone_number', sa.String(20), nullable=False),
        sa.Column('role', user_roles, nullable=False),
        sa.Column('bar_number', sa.String(50), nullable=True),
        sa.Column('is_active', sa.Boolean(), default=True),
        sa.Column('address', sa.Text(), nullable=True),
        sa.Column('profile_picture', sa.String(255), nullable=True),
        sa.Column('last_login', sa.DateTime(), nullable=True),
        sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
        sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now(), onupdate=sa.func.now()),
    )
    
    op.create_index('ix_users_email', 'users', ['email'], unique=True)
    op.create_index('ix_users_bar_number', 'users', ['bar_number'], unique=True)

def downgrade():
    op.drop_index('ix_users_bar_number', 'users')
    op.drop_index('ix_users_email', 'users')
    op.drop_table('users')
    ENUM(name='user_roles').drop(op.get_bind())