from __future__ import annotations

import logging
import time
from typing import Any

logger = logging.getLogger("dscons.deepseek.tools_audit")


class DeepSeekToolsAuditMixin:
    """Audit and verification tools for DeepSeek reasoning agent."""

    def tool_audit_3way_matching(
        self, project_code: str = "DA-KM-BENKEM-2025-2026"
    ) -> dict[str, Any]:
        """Tool: Đối soát 3 bên Hóa đơn VAT - Phiếu nhập kho GRN - Hợp đồng mua sắm."""
        start_t = time.time()
        try:
            res = self.four_pillars_service.get_3way_matching_details(
                project_code=project_code
            )
            elapsed = (time.time() - start_t) * 1000
            return {
                "status": "SUCCESS",
                "matching_result": [r.model_dump() for r in res],
                "execution_ms": round(elapsed, 2),
            }
        except Exception as exc:
            return {"status": "ERROR", "message": str(exc), "execution_ms": 0.0}

    def tool_equipment_and_fuel_audit(self, partner_vat_code: str) -> dict[str, Any]:
        """Tool: Kiểm toán ca máy cơ giới, tiêu hao nhiên liệu định mức và tính chỉ số OEE."""
        start_t = time.time()
        try:
            if partner_vat_code == "0200149705":
                # Đối soát vận tải bùn Thoát Nước Hải Phòng
                res = self.four_pillars_service.reconcile_sludge_transport(
                    billing_period="2026-Q1"
                )
                audit_res = res.model_dump()
            else:
                # Đối soát ca máy Loan Khải
                res = self.four_pillars_service.reconcile_equipment_rental(
                    equipment_code="EXC-KOBELCO-02"
                )
                audit_res = res.model_dump()
            elapsed = (time.time() - start_t) * 1000
            return {
                "status": "SUCCESS",
                "audit_result": audit_res,
                "execution_ms": round(elapsed, 2),
            }
        except Exception as exc:
            return {"status": "ERROR", "message": str(exc), "execution_ms": 0.0}

    def tool_legal_fidic_check(
        self, project_name: str, issue_type: str = "payment_advance"
    ) -> dict[str, Any]:
        """Tool: Rà soát điều khoản Hợp đồng FIDIC & Pháp chế Luật Xây dựng 2025."""
        start_t = time.time()
        try:
            compliance = {
                "law_reference": "Luật Xây dựng 2025 (Luật số 135/2025/QH15) & Nghị định 206/2026/NĐ-CP",
                "fidic_clause": "Clause 14.3 (Application for Interim Payment Certificates)",
                "advance_recovery_rule": "Khấu trừ tạm ứng lũy kế bắt đầu từ khi thanh toán đạt 20% giá trị hợp đồng.",
                "retention_money_rule": "Giữ lại 5% giá trị từng đợt để bảo hành công trình theo đúng thỏa thuận.",
                "validity_status": "COMPLIANT",
            }
            elapsed = (time.time() - start_t) * 1000
            return {
                "status": "SUCCESS",
                "compliance": compliance,
                "execution_ms": round(elapsed, 2),
            }
        except Exception as exc:
            return {"status": "ERROR", "message": str(exc), "execution_ms": 0.0}