"""用户列表 — 对接 MOM sys_user"""
from fastapi import APIRouter, Query, HTTPException, status
from pydantic import BaseModel
from app.core.mom_database import MomSessionLocal
from sqlalchemy import text

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


class UserOption(BaseModel):
    id: str
    username: str
    full_name: str
    department: str = ""

    model_config = {"from_attributes": True}


@router.get("/", response_model=list[UserOption])
def list_users(
    keyword: str = Query("", description="按用户名/姓名模糊搜索"),
    limit: int = Query(100, ge=1, le=500),
    dept: str = Query("IRIS", description="部门过滤"),
):
    """获取 MOM 系统用户列表，默认只返回 IRIS 部门"""
    db = MomSessionLocal()
    try:
        # 尝试按部门过滤；若 sys_user 无 department 列则降级全量查询
        try:
            base_sql = """
                SELECT id, username,
                       SPLIT_PART(username, '/', 1) AS full_name,
                       COALESCE(department, '') AS department
                FROM sys_user
                WHERE department = :dept
            """
            params = {"dept": dept, "lim": limit}
            if keyword.strip():
                sql = text(base_sql + " AND username ILIKE :kw ORDER BY username LIMIT :lim")
                params["kw"] = f"%{keyword.strip()}%"
            else:
                sql = text(base_sql + " ORDER BY username LIMIT :lim")
            rows = db.execute(sql, params).fetchall()
        except Exception:
            # 降级：不使用 department 列过滤
            fallback_sql = """
                SELECT id, username,
                       SPLIT_PART(username, '/', 1) AS full_name,
                       '' AS department
                FROM sys_user
            """
            params = {"lim": limit}
            if keyword.strip():
                sql = text(fallback_sql + " WHERE username ILIKE :kw ORDER BY username LIMIT :lim")
                params["kw"] = f"%{keyword.strip()}%"
            else:
                sql = text(fallback_sql + " ORDER BY username LIMIT :lim")
            rows = db.execute(sql, params).fetchall()

        return [
            UserOption(
                id=str(row.id),
                username=row.username.split("/")[-1] if "/" in row.username else row.username,
                full_name=row.full_name,
                department=row.department or "",
            )
            for row in rows
        ]
    except Exception as e:
        raise HTTPException(
            status_code=status.HTTP_502_BAD_GATEWAY,
            detail=f"MOM 用户查询失败: {str(e)}",
        )
    finally:
        db.close()