from __future__ import annotations

from typing import Any

from fastapi import APIRouter, Depends, HTTPException

from app.modules.auth.presentation.auth import get_current_user
from app.core.postgres.project_crud_repository import ProjectCrudRepository
from app.modules.financial.application.evm_calculation_service import EvmCalculationService

router = APIRouter()


@router.get("/projects/{project_id_or_code}/bim-elements")
def get_project_bim_elements(
    project_id_or_code: str,
    current_user: dict[str, Any] = Depends(get_current_user),
) -> list[dict[str, Any]]:
    """Lấy danh sách cấu kiện 3D BIM của dự án gắn liền với mã WBS và trạng thái thi công."""
    repo = ProjectCrudRepository()
    detail = repo.get_project_detail(project_id_or_code)
    if not detail:
        raise HTTPException(status_code=404, detail="Không tìm thấy dự án")

    project_id = detail["id"]
    elements = repo.get_project_bim_elements(str(project_id))

    if not elements:
        # Fallback generated BIM structure if not populated yet
        elements = [
            {
                "element_guid": "BIM-FND-001",
                "element_name": "Đài Móng Cụm Trạm Bơm ĐM1",
                "category": "foundation",
                "level_floor": "Móng ngầm",
                "material_type": "Bê tông C30 cốt thép CB400",
                "volume_m3": 45.5,
                "unit_cost_vnd": 2450000.0,
                "total_cost_vnd": 111475000.0,
                "installation_day": 20,
                "status": "completed",
                "wbs_code": "2.1",
            },
            {
                "element_guid": "BIM-FND-002",
                "element_name": "Bể Chứa Nước Sạch 500m3 (Đáy bể)",
                "category": "foundation",
                "level_floor": "Móng ngầm",
                "material_type": "Bê tông chống thấm B8",
                "volume_m3": 120.0,
                "unit_cost_vnd": 2850000.0,
                "total_cost_vnd": 342000000.0,
                "installation_day": 35,
                "status": "completed",
                "wbs_code": "2.1",
            },
            {
                "element_guid": "BIM-COL-001",
                "element_name": "Cột Khung Nhà Điều Hành C1-C8",
                "category": "column",
                "level_floor": "Tầng 1",
                "material_type": "Bê tông C30",
                "volume_m3": 28.0,
                "unit_cost_vnd": 3100000.0,
                "total_cost_vnd": 86800000.0,
                "installation_day": 60,
                "status": "in_progress",
                "wbs_code": "3.1",
            },
            {
                "element_guid": "BIM-SLB-001",
                "element_name": "Bản Sàn Mái Bể Chứa Nước Sạch",
                "category": "slab",
                "level_floor": "Tầng 1",
                "material_type": "Bê tông C30",
                "volume_m3": 65.0,
                "unit_cost_vnd": 2900000.0,
                "total_cost_vnd": 188500000.0,
                "installation_day": 75,
                "status": "in_progress",
                "wbs_code": "3.1",
            },
            {
                "element_guid": "BIM-PIP-001",
                "element_name": "Tuyến Ống Hút & Đẩy Trạm Bơm HDPE D250",
                "category": "pipe",
                "level_floor": "Tuyến ngoài trời",
                "material_type": "HDPE Tiền Phong PN16",
                "volume_m3": 350.0,
                "unit_cost_vnd": 1250000.0,
                "total_cost_vnd": 437500000.0,
                "installation_day": 90,
                "status": "planned",
                "wbs_code": "3.2",
            },
        ]
    return elements


@router.get("/projects/{project_id_or_code}/evm-analytics")
def get_project_evm_analytics(
    project_id_or_code: str,
    current_user: dict[str, Any] = Depends(get_current_user),
) -> dict[str, Any]:
    """Lấy dữ liệu phân tích chi phí EVM và đường cong S-Curve thời gian thực."""
    service = EvmCalculationService()
    res = service.get_project_evm_analytics(project_id_or_code)
    if "error" in res:
        raise HTTPException(status_code=404, detail="Không tìm thấy dự án")
    return res