"""
Script to create an admin user account.
Usage: python backend/scripts/create_admin.py
"""
import sys
import os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

from db.db import users_db
from models.user import User
from werkzeug.security import generate_password_hash
from tinydb import Query
import uuid

def create_admin_user(email: str, password: str, first_name: str, last_name: str):
    """Create an admin user account."""
    
    # Check if user already exists
    Query_ = Query()
    existing_user = users_db.get(Query_.email == email)
    
    if existing_user:
        print(f"Error: User with email {email} already exists")
        return False
    
    admin = User(
        id=str(uuid.uuid4()),
        email=email,
        first_name=first_name,
        last_name=last_name,
        password=generate_password_hash(password),
        verified=True,
        role='admin'
    )
    
    users_db.insert(admin.to_dict())
    print(f"✓ Admin user created successfully!")
    print(f"  Email: {email}")
    print(f"  Name: {first_name} {last_name}")
    print(f"  Role: admin")
    return True

if __name__ == '__main__':
    print("=== Create Admin User ===\n")

    env_email = os.environ.get('ADMIN_EMAIL')
    env_password = os.environ.get('ADMIN_PASSWORD')
    env_first_name = os.environ.get('ADMIN_FIRST_NAME')
    env_last_name = os.environ.get('ADMIN_LAST_NAME')

    if env_email and env_password:
        email = env_email
        password = env_password
        first_name = env_first_name or 'First Name'
        last_name = env_last_name or 'Last Name'
        print(f"Using environment variables for admin user '{email}'")
    else:
        email = input("Email: ").strip()
        password = input("Password: ").strip()
        first_name = input("First name: ").strip()
        last_name = input("Last name: ").strip()

        if not all([email, password, first_name, last_name]):
            print("Error: All fields are required")
            sys.exit(1)

        confirm = input(f"\nCreate admin user '{email}'? (yes/no): ").strip().lower()
        if confirm != 'yes':
            print("Cancelled")
            sys.exit(0)

    if len(password) < 8:
        print("Error: Password must be at least 8 characters")
        sys.exit(1)

    create_admin_user(email, password, first_name, last_name)