#!/usr/bin/env python3
"""
Comprehensive seed script for PMFlow/Neural Sync.
Creates users, domains, and initial data.
"""
import asyncio
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.future import select
from app.database import AsyncSessionLocal
from app.models import User, ProjectDomain, Project
from app.services.auth import AuthService


async def seed_users():
    """Create default users with known passwords."""
    # Create initial users (with simple password hashes for demo)
    # In production, these should be created with proper password hashing
    # admin_hash = AuthService.hash_password("admin123")
    # boss_hash = AuthService.hash_password("boss123")
    # ai_hash = AuthService.hash_password("ai123")
    # be_hash = AuthService.hash_password("be123")
    # fe_hash = AuthService.hash_password("fe123")

    async with AsyncSessionLocal() as session:
        # Check if users already exist
        result = await session.execute(select(User).where(User.username == "admin"))
        admin = result.scalar_one_or_none()

        if not admin:
            # Create admin user
            admin = User(
                username="admin",
                password_hash=AuthService.hash_password("admin123"),
                name="系统管理员",
                email="admin@pmflow.ai",
                role="BOSS",
                assigned_domains=[],
                workload=0,
                status="active",
                is_active=True,
                tenant_id=1,
            )
            session.add(admin)
            print("✅ Created admin user (username: admin, password: admin123)")

        # Create boss user
        result = await session.execute(select(User).where(User.username == "boss"))
        boss = result.scalar_one_or_none()
        if not boss:
            boss = User(
                username="boss",
                password_hash=AuthService.hash_password("boss123"),
                name="老板",
                email="boss@pmflow.ai",
                role="BOSS",
                assigned_domains=[],
                workload=0,
                status="active",
                is_active=True,
                tenant_id=1,
            )
            session.add(boss)
            print("✅ Created boss user (username: boss, password: boss123)")

        # Create AI engineer user
        result = await session.execute(select(User).where(User.username == "ai_engineer"))
        ai_user = result.scalar_one_or_none()
        if not ai_user:
            ai_user = User(
                username="ai_engineer",
                password_hash=AuthService.hash_password("ai123"),
                name="AI工程师",
                email="ai@pmflow.ai",
                role="AI",
                assigned_domains=[],
                workload=0,
                status="active",
                is_active=True,
                tenant_id=1,
            )
            session.add(ai_user)
            print("✅ Created ai_engineer user (username: ai_engineer, password: ai123)")

        # Create backend user
        result = await session.execute(select(User).where(User.username == "backend"))
        backend_user = result.scalar_one_or_none()
        if not backend_user:
            backend_user = User(
                username="backend",
                password_hash=AuthService.hash_password("be123"),
                name="后端开发",
                email="backend@pmflow.ai",
                role="BE",
                assigned_domains=[],
                workload=0,
                status="active",
                is_active=True,
                tenant_id=1,
            )
            session.add(backend_user)
            print("✅ Created backend user (username: backend, password: be123)")

        # Create frontend user
        result = await session.execute(select(User).where(User.username == "frontend"))
        frontend_user = result.scalar_one_or_none()
        if not frontend_user:
            frontend_user = User(
                username="frontend",
                password_hash=AuthService.hash_password("fe123"),
                name="前端开发",
                email="frontend@pmflow.ai",
                role="FE",
                assigned_domains=[],
                workload=0,
                status="active",
                is_active=True,
                tenant_id=1,
            )
            session.add(frontend_user)
            print("✅ Created frontend user (username: frontend, password: fe123)")

        await session.commit()


async def seed_domains():
    """Create default project domains."""
    async with AsyncSessionLocal() as session:
        # Check if domains already exist
        result = await session.execute(select(ProjectDomain))
        domains = result.scalars().all()

        if not domains:
            default_domains = [
                ProjectDomain(
                    name="AI模型工程",
                    slug="ai-engineering",
                    description="AI模型训练、评估和部署相关项目",
                    is_default=True,
                    signal_count=0,
                    context_node_count=0,
                    experiment_count=0,
                    tenant_id=1,
                ),
                ProjectDomain(
                    name="后端服务",
                    slug="backend-services",
                    description="后端API、数据库、微服务等项目",
                    is_default=False,
                    signal_count=0,
                    context_node_count=0,
                    experiment_count=0,
                    tenant_id=1,
                ),
                ProjectDomain(
                    name="前端应用",
                    slug="frontend-apps",
                    description="Vue/React前端应用、UI组件等",
                    is_default=False,
                    signal_count=0,
                    context_node_count=0,
                    experiment_count=0,
                    tenant_id=1,
                ),
            ]

            for domain in default_domains:
                session.add(domain)

            await session.commit()
            print("✅ Created 3 default project domains: ai-engineering, backend-services, frontend-apps")
        else:
            print(f"ℹ️  Project domains already exist ({len(domains)} found)")


async def main():
    print("🌱 Seeding PMFlow database...")
    print("=" * 50)

    await seed_users()
    await seed_domains()

    print("=" * 50)
    print("✅ Seeding completed!")
    print()
    print("📝 Default users created:")
    print("  - admin / admin123 (BOSS)")
    print("  - boss / boss123 (BOSS)")
    print("  - ai_engineer / ai123 (AI)")
    print("  - backend / be123 (BE)")
    print("  - frontend / fe123 (FE)")
    print()
    print("🔐 Use these credentials to login at: http://127.0.0.1:5174/login")


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