#!/usr/bin/env python3
"""
Script to set a user as admin in the database
Usage: python scripts/set_admin.py <user_email>
"""

import sys
import os
import asyncio
from datetime import datetime

# Add the app directory to Python path
backend_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
app_dir = os.path.join(backend_dir, 'app')
sys.path.insert(0, app_dir)

# Change to app directory
os.chdir(app_dir)

from services.auth.auth_service import AuthService
from models.common import UserRole
from core.firebase_config import FirebaseConfig

async def set_user_as_admin(email: str):
    """Set a user as admin in the database"""
    FirebaseConfig.initialize_firebase()
    auth_service = AuthService()
    
    # Get all users
    users = await auth_service.get_all_users()
    
    # Find user by email
    target_user = None
    for user in users:
        if user.email == email:
            target_user = user
            break
    
    if not target_user:
        print(f"User with email {email} not found in database")
        print("Available users:")
        for user in users:
            print(f"  - {user.email} (role: {user.role.value})")
        return
    
    print(f"Found user: {target_user.email}")
    print(f"Current role: {target_user.role.value}")
    
    if target_user.role == UserRole.ADMIN:
        print("User is already an admin!")
        return
    
    # Update user role to admin
    try:
        auth_service.users_collection.document(target_user.uid).update({
            'role': UserRole.ADMIN.value,
            'updated_at': datetime.utcnow()
        })
        print(f"Successfully set {target_user.email} as admin!")
    except Exception as e:
        print(f"Error setting user as admin: {e}")

async def list_users():
    """List all users in the database"""
    FirebaseConfig.initialize_firebase()
    auth_service = AuthService()
    
    users = await auth_service.get_all_users()
    
    print(f"Total users in database: {len(users)}")
    for user in users:
        print(f"  - {user.email} (role: {user.role.value}, active: {user.is_active})")

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("Usage: python scripts/set_admin.py <user_email>")
        print("Or: python scripts/set_admin.py --list to see all users")
        sys.exit(1)
    
    if sys.argv[1] == "--list":
        asyncio.run(list_users())
    else:
        email = sys.argv[1]
        asyncio.run(set_user_as_admin(email))