import subprocess
subprocess.check_call(['pip', 'install', 'pymysql', '-q'])

import pymysql, os

conn = pymysql.connect(
    host=os.environ['MYSQL_HOST'],
    port=int(os.environ.get('MYSQL_PORT', 3306)),
    user=os.environ['MYSQL_USER'],
    password=os.environ['MYSQL_PASSWORD'],
    database=os.environ['MYSQL_DATABASE'],
)

with conn.cursor() as cur:
    cur.execute("SHOW COLUMNS FROM portal_users LIKE 'role'")
    row = cur.fetchone()
    print(f'[v0] Definição da coluna role: {row}')

    # Garante que tenant_admin está no ENUM; se não estiver, altera a coluna
    col_def = str(row[1]) if row else ''
    print(f'[v0] Tipo atual: {col_def}')

    if 'tenant_admin' not in col_def:
        print('[v0] tenant_admin NAO esta no ENUM. Corrigindo...')
        cur.execute(
            """ALTER TABLE portal_users
               MODIFY COLUMN role ENUM('admin_global','tenant_admin','support','viewer') NOT NULL DEFAULT 'tenant_admin'"""
        )
        conn.commit()
        print('[v0] ENUM corrigido com sucesso.')
    else:
        print('[v0] tenant_admin ja esta no ENUM. OK.')

conn.close()