"""用户管理路由"""
from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session
from typing import List
from database import get_db
from models import User, UserRole, PhaseGroup
from schemas import UserCreate, UserUpdate, UserOut
from auth import get_current_user, hash_password, require_role

router = APIRouter(prefix="/api/users", tags=["用户管理"])


def user_to_out(u: User) -> UserOut:
    return UserOut(
        id=u.id, username=u.username, name=u.name,
        phase_group=u.phase_group.value if hasattr(u.phase_group, 'value') else u.phase_group,
        role=u.role.value if hasattr(u.role, 'value') else u.role,
        monthly_salary=u.monthly_salary, daily_cost=u.daily_cost,
        is_active=u.is_active, created_at=u.created_at,
    )


@router.get("/", response_model=List[UserOut])
def list_users(
    db: Session = Depends(get_db),
    current_user: User = Depends(require_role(UserRole.OWNER, UserRole.SUPERVISOR, UserRole.LEADER))
):
    users = db.query(User).order_by(User.created_at.desc()).all()
    return [user_to_out(u) for u in users]


@router.post("/", response_model=UserOut)
def create_user(
    req: UserCreate,
    db: Session = Depends(get_db),
    current_user: User = Depends(require_role(UserRole.OWNER))
):
    if db.query(User).filter(User.username == req.username).first():
        raise HTTPException(status_code=400, detail="用户名已存在")
    user = User(
        username=req.username,
        password_hash=hash_password(req.password),
        name=req.name,
        phase_group=PhaseGroup(req.phase_group),
        role=UserRole(req.role),
        monthly_salary=req.monthly_salary,
    )
    db.add(user)
    db.commit()
    db.refresh(user)
    return user_to_out(user)


@router.put("/{user_id}", response_model=UserOut)
def update_user(
    user_id: int,
    req: UserUpdate,
    db: Session = Depends(get_db),
    current_user: User = Depends(require_role(UserRole.OWNER))
):
    user = db.query(User).filter(User.id == user_id).first()
    if not user:
        raise HTTPException(status_code=404, detail="用户不存在")
    if req.name is not None:
        user.name = req.name
    if req.phase_group is not None:
        user.phase_group = PhaseGroup(req.phase_group)
    if req.role is not None:
        user.role = UserRole(req.role)
    if req.monthly_salary is not None:
        user.monthly_salary = req.monthly_salary
    if req.is_active is not None:
        user.is_active = req.is_active
    db.commit()
    db.refresh(user)
    return user_to_out(user)


@router.get("/{user_id}", response_model=UserOut)
def get_user(
    user_id: int,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    user = db.query(User).filter(User.id == user_id).first()
    if not user:
        raise HTTPException(status_code=404, detail="用户不存在")
    return user_to_out(user)