X-Financial/server/src/app/api/v1/endpoints/employee_profiles.py

from __future__ import annotations

from typing import Annotated

from fastapi import APIRouter, Depends, Query
from sqlalchemy import func, or_, select
from sqlalchemy.orm import Session

from app.api.deps import CurrentUserContext, get_current_user, get_db
from app.models.employee import Employee
from app.schemas.employee_profile import EmployeeProfileLatestRead
from app.services.employee_behavior_profile_service import EmployeeBehaviorProfileService

router = APIRouter(prefix="/employee-profiles")
DbSession = Annotated[Session, Depends(get_db)]
CurrentUser = Annotated[CurrentUserContext, Depends(get_current_user)]


@router.get(
    "/me/latest",
    response_model=EmployeeProfileLatestRead,
    summary="读取当前登录人的最新用户画像",
    description="按当前登录用户的邮箱或姓名匹配员工目录，返回个人工作台使用的综合用户画像。",
)
def get_current_employee_latest_profile(
    db: DbSession,
    current_user: CurrentUser,
    scene: Annotated[str, Query(max_length=50)] = "operations",
    window_days: Annotated[int, Query(ge=1, le=365)] = 90,
    expense_type_scope: Annotated[str, Query(max_length=50)] = "overall",
) -> EmployeeProfileLatestRead:
    employee = _resolve_current_employee(db, current_user)
    if employee is None:
        return EmployeeProfileLatestRead(
            employee_id=current_user.username,
            employee_name=current_user.name,
            scene=scene,
            window_days=window_days,
            expense_type_scope=expense_type_scope,
            empty_reason="当前登录用户未匹配到员工目录，暂无法形成用户画像。",
        )

    service = EmployeeBehaviorProfileService(db)
    latest = service.get_latest_profile(
        employee_id=employee.id,
        scene=scene,
        window_days=window_days,
        expense_type_scope=expense_type_scope,
    )
    if latest.empty_reason:
        service.refresh_employee_profiles(
            employee_id=employee.id,
            window_days=(window_days,),
            expense_type_scope=expense_type_scope,
            source_task_type="workbench_on_demand",
        )
        latest = service.get_latest_profile(
            employee_id=employee.id,
            scene=scene,
            window_days=window_days,
            expense_type_scope=expense_type_scope,
        )
    return latest


@router.get(
    "/{employee_id}/latest",
    response_model=EmployeeProfileLatestRead,
    summary="读取员工最新业务行为画像",
    description="返回员工在指定场景下的最新画像快照，审批场景默认只展示费用支出和流程质量画像。",
)
def get_employee_latest_profile(
    employee_id: str,
    db: DbSession,
    current_user: CurrentUser,
    scene: Annotated[str, Query(max_length=50)] = "approval",
    claim_id: Annotated[str | None, Query(max_length=80)] = None,
    window_days: Annotated[int, Query(ge=1, le=365)] = 90,
    expense_type_scope: Annotated[str, Query(max_length=50)] = "overall",
) -> EmployeeProfileLatestRead:
    del current_user
    return EmployeeBehaviorProfileService(db).get_latest_profile(
        employee_id=employee_id,
        scene=scene,
        claim_id=claim_id,
        window_days=window_days,
        expense_type_scope=expense_type_scope,
    )


def _resolve_current_employee(
    db: Session,
    current_user: CurrentUserContext,
) -> Employee | None:
    identities = [
        str(current_user.username or "").strip(),
        str(current_user.name or "").strip(),
    ]
    normalized = [item for item in dict.fromkeys(identities) if item]
    if not normalized:
        return None

    email_values = [item.lower() for item in normalized if "@" in item]
    exact_values = [item for item in normalized if "@" not in item]
    conditions = []

    if email_values:
        conditions.append(func.lower(Employee.email).in_(email_values))
    if exact_values:
        conditions.append(Employee.name.in_(exact_values))
        conditions.append(Employee.employee_no.in_(exact_values))

    if not conditions:
        return None

    stmt = select(Employee).where(or_(*conditions)).order_by(Employee.created_at.asc()).limit(1)
    return db.scalars(stmt).first()