"""Seed script: Create test users (admin and tm_manager)."""

import asyncio
import sys
from pathlib import Path

# Add backend to path so we can import app modules
_seed_dir = Path(__file__).resolve().parent
_backend_dir = _seed_dir.parent / "backend"
if _backend_dir.is_dir():
    sys.path.insert(0, str(_backend_dir))
else:
    sys.path.insert(0, str(_seed_dir.parent))

from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine

from app.auth.providers.jwt_provider import JWTAuthProvider
from app.config import settings
from app.models.client import Client
from app.models.user import User, UserClient, UserRole, UserStatus


TEST_USERS = [
    {
        "email": "admin@amazon.com",
        "name": "Admin User",
        "password": "admin123!",
        "role": UserRole.admin,
    },
    {
        "email": "manager@amazon.com",
        "name": "TM Manager",
        "password": "manager123!",
        "role": UserRole.tm_manager,
    },
    {
        "email": "reviewer@amazon.com",
        "name": "Reviewer User",
        "password": "reviewer123!",
        "role": UserRole.reviewer,
    },
]


async def seed_test_users() -> None:
    """Create test users if they don't exist."""
    engine = create_async_engine(settings.DATABASE_URL, pool_pre_ping=True)
    factory = async_sessionmaker(bind=engine, class_=AsyncSession, expire_on_commit=False)
    auth = JWTAuthProvider()

    async with factory() as db:
        # Get the Amazon client for association
        client_result = await db.execute(select(Client).where(Client.name == "Amazon"))
        amazon_client = client_result.scalar_one_or_none()

        for user_data in TEST_USERS:
            # Check if user exists
            result = await db.execute(
                select(User).where(User.email == user_data["email"])
            )
            existing = result.scalar_one_or_none()

            if existing:
                print(f"User '{user_data['email']}' already exists (id={existing.id})")
                continue

            user = User(
                email=user_data["email"],
                name=user_data["name"],
                password_hash=auth.hash_password(user_data["password"]),
                role=user_data["role"],
                status=UserStatus.active,
            )
            db.add(user)
            await db.flush()

            # Associate with Amazon client
            if amazon_client:
                uc = UserClient(user_id=user.id, client_id=amazon_client.id)
                db.add(uc)

            print(
                f"Created user '{user_data['email']}' "
                f"(role={user_data['role'].value}, id={user.id})"
            )

        await db.commit()

    await engine.dispose()
    print("\nTest credentials:")
    for u in TEST_USERS:
        print(f"  {u['email']} / {u['password']} ({u['role'].value})")


if __name__ == "__main__":
    asyncio.run(seed_test_users())