X-Financial/server/src/app/services/expense_claim_approval_flow.py

from __future__ import annotations

import uuid
from datetime import UTC, datetime
from typing import Any

from app.api.deps import CurrentUserContext
from app.services.expense_claim_workflow_constants import (
    APPROVAL_DONE_STAGE,
    BUDGET_MANAGER_APPROVAL_STAGE,
    DIRECT_MANAGER_APPROVAL_STAGE,
    FINANCE_APPROVAL_STAGE,
    PAYMENT_PAID_STAGE,
    PAYMENT_PAID_STATUS,
    PAYMENT_PENDING_STAGE,
    PAYMENT_PENDING_STATUS,
)


class ExpenseClaimApprovalFlowMixin:
    def approve_claim(
        self,
        claim_id: str,
        current_user: CurrentUserContext,
        *,
        opinion: str | None = None,
    ):
        claim = self.get_claim(claim_id, current_user)
        if claim is None:
            return None

        normalized_status = str(claim.status or "").strip().lower()
        if normalized_status != "submitted":
            raise ValueError("只有审批中的单据可以审批通过。")

        previous_stage = str(claim.approval_stage or "").strip()
        is_application_claim = self._is_expense_application_claim(claim)
        next_budget_manager = None
        if previous_stage == DIRECT_MANAGER_APPROVAL_STAGE:
            if not self._access_policy.can_approve_claim(current_user, claim):
                raise ValueError("只有当前直属领导审批人可以审批通过该单据。")
            approval_source = "manual_approval"
            event_type = "expense_application_approval" if is_application_claim else "expense_claim_approval"
            label = "领导审批通过"
            if is_application_claim:
                next_budget_manager = self._access_policy.resolve_department_budget_manager(claim)
                next_status = "submitted"
                next_stage = BUDGET_MANAGER_APPROVAL_STAGE
                default_message = "{operator} 已确认直属领导审核，流转至预算管理者审批。"
            else:
                next_status = "submitted"
                next_stage = FINANCE_APPROVAL_STAGE
                default_message = "{operator} 已审批通过，流转至{next_stage}。"
        elif previous_stage == BUDGET_MANAGER_APPROVAL_STAGE:
            if not is_application_claim:
                raise ValueError("只有费用申请需要预算管理者审批。")
            if not self._access_policy.can_approve_claim(current_user, claim):
                raise ValueError("只有当前预算管理者可以审批通过该费用申请。")
            approval_source = "budget_approval"
            event_type = "expense_application_budget_approval"
            label = "预算管理者审核通过"
            next_status = "approved"
            next_stage = APPROVAL_DONE_STAGE
            default_message = "{operator} 已完成预算审核，申请流程完成并生成报销草稿。"
        elif previous_stage == FINANCE_APPROVAL_STAGE:
            if is_application_claim:
                raise ValueError("费用申请需先完成预算管理者审批。")
            if not self._access_policy.can_approve_claim(current_user, claim):
                raise ValueError("只有财务人员可以完成财务终审。")
            approval_source = "finance_approval"
            event_type = "expense_claim_finance_approval"
            label = "财务审核通过"
            next_status = PAYMENT_PENDING_STATUS
            next_stage = PAYMENT_PENDING_STAGE
            default_message = "{operator} 已完成财务审核，进入待付款。"
        else:
            raise ValueError("当前节点不支持审批通过。")

        approval_opinion = str(opinion or "").strip()
        if previous_stage in {DIRECT_MANAGER_APPROVAL_STAGE, BUDGET_MANAGER_APPROVAL_STAGE} and not approval_opinion:
            approval_opinion = "同意"

        before_json = self._serialize_claim(claim)
        operator = self._access_policy.resolve_current_user_display_name(current_user)
        budget_flags: list[dict[str, Any]] = []
        if approval_source == "finance_approval" and not is_application_claim:
            consumed_budget_flag = self._consume_budget_for_finance_approval(claim, current_user)
            if consumed_budget_flag is not None:
                budget_flags.append(consumed_budget_flag)
        approval_flag = {
            "source": approval_source,
            "event_type": event_type,
            "approval_event_id": str(uuid.uuid4()),
            "severity": "info",
            "label": label,
            "message": approval_opinion or default_message.format(operator=operator, next_stage=next_stage),
            "opinion": approval_opinion,
            "operator": operator,
            "operator_username": current_user.username,
            "operator_role_codes": [
                str(item).strip().lower()
                for item in current_user.role_codes
                if str(item).strip()
            ],
            "previous_status": str(claim.status or "").strip(),
            "previous_approval_stage": previous_stage,
            "next_status": next_status,
            "next_approval_stage": next_stage,
            "created_at": datetime.now(UTC).isoformat(),
        }
        if next_budget_manager is not None:
            approval_flag.update(
                {
                    "next_approver_name": str(next_budget_manager.name or "").strip(),
                    "next_approver_employee_id": next_budget_manager.id,
                    "next_approver_grade": str(next_budget_manager.grade or "").strip(),
                    "next_approver_role_code": "budget_monitor",
                }
            )

        claim.status = next_status
        claim.approval_stage = next_stage
        if claim.submitted_at is None:
            claim.submitted_at = datetime.now(UTC)
        if is_application_claim and previous_stage == BUDGET_MANAGER_APPROVAL_STAGE:
            approval_flag["leader_opinion"] = self._resolve_latest_approval_opinion(
                claim,
                source="manual_approval",
            )
            approval_flag["budget_opinion"] = approval_opinion
            generated_draft = self._create_reimbursement_draft_from_application(
                application_claim=claim,
                approval_flag=approval_flag,
                operator=operator,
            )
            transferred_budget_flag = self._transfer_application_budget_to_reimbursement(
                application_claim=claim,
                draft_claim=generated_draft,
                current_user=current_user,
            )
            if transferred_budget_flag is not None:
                budget_flags.append(transferred_budget_flag)
                generated_draft.risk_flags_json = self._append_budget_flags(
                    generated_draft.risk_flags_json,
                    transferred_budget_flag,
                )
        claim.risk_flags_json = self._append_budget_flags(
            [*list(claim.risk_flags_json or []), approval_flag],
            budget_flags,
        )

        self.db.commit()
        self.db.refresh(claim)

        self.audit_service.log_action(
            actor=operator,
            action="expense_claim.approve",
            resource_type="expense_claim",
            resource_id=claim.id,
            before_json=before_json,
            after_json=self._serialize_claim(claim),
        )

        return claim

    def mark_claim_paid(
        self,
        claim_id: str,
        current_user: CurrentUserContext,
    ):
        claim = self.get_claim(claim_id, current_user)
        if claim is None:
            return None

        normalized_status = str(claim.status or "").strip().lower()
        if normalized_status == PAYMENT_PAID_STATUS:
            raise ValueError("该报销单已付款，无需重复确认。")
        if normalized_status != PAYMENT_PENDING_STATUS:
            raise ValueError("只有待付款状态的报销单可以确认已付款。")
        if not self._access_policy.can_mark_claim_paid(current_user, claim):
            raise ValueError("只有财务人员或高级财务人员可以确认付款，且不能处理本人单据。")

        before_json = self._serialize_claim(claim)
        operator = self._access_policy.resolve_current_user_display_name(current_user)
        previous_stage = str(claim.approval_stage or "").strip()
        payment_flag = {
            "source": "payment",
            "event_type": "expense_claim_payment_completed",
            "payment_event_id": str(uuid.uuid4()),
            "severity": "info",
            "label": "付款完成",
            "message": f"{operator} 已确认付款，报销单进入已付款。",
            "operator": operator,
            "operator_username": current_user.username,
            "operator_role_codes": [
                str(item).strip().lower()
                for item in current_user.role_codes
                if str(item).strip()
            ],
            "previous_status": str(claim.status or "").strip(),
            "previous_approval_stage": previous_stage,
            "next_status": PAYMENT_PAID_STATUS,
            "next_approval_stage": PAYMENT_PAID_STAGE,
            "created_at": datetime.now(UTC).isoformat(),
        }

        claim.status = PAYMENT_PAID_STATUS
        claim.approval_stage = PAYMENT_PAID_STAGE
        claim.risk_flags_json = [*list(claim.risk_flags_json or []), payment_flag]

        self.db.commit()
        self.db.refresh(claim)

        self.audit_service.log_action(
            actor=operator,
            action="expense_claim.mark_paid",
            resource_type="expense_claim",
            resource_id=claim.id,
            before_json=before_json,
            after_json=self._serialize_claim(claim),
        )

        return claim

    @staticmethod
    def _resolve_latest_approval_opinion(claim, *, source: str) -> str:
        for flag in reversed(list(claim.risk_flags_json or [])):
            if not isinstance(flag, dict):
                continue
            if str(flag.get("source") or "").strip() != source:
                continue
            opinion = str(flag.get("opinion") or flag.get("message") or "").strip()
            if opinion:
                return opinion
        return ""