package cn.iocoder.yudao.module.mes.controller.admin.safety;

import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.statistics.MesSafetyHazardExcelVO;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.statistics.MesSafetyStatisticsReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.statistics.MesSafetyStatisticsRespVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkshopDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.safety.MesSafetyHazardDO;
import cn.iocoder.yudao.module.mes.dal.mysql.md.workstation.MesMdWorkshopMapper;
import cn.iocoder.yudao.module.mes.dal.mysql.safety.MesSafetyHazardMapper;
import cn.iocoder.yudao.module.mes.service.safety.MesSafetyStatisticsService;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletResponse;
import jakarta.validation.Valid;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.io.IOException;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.*;
import java.util.stream.Collectors;

import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;

/**
 * 安全生产 - 统计分析 Controller
 *
 * @author 超级管理员
 */
@Tag(name = "管理后台 - 安全生产-统计分析")
@RestController
@RequestMapping("/mes/safety/statistics")
@Validated
public class MesSafetyStatisticsController {

    @Resource
    private MesSafetyStatisticsService statisticsService;

    @Resource
    private MesSafetyHazardMapper hazardMapper;

    @Resource
    private MesMdWorkshopMapper workshopMapper;

    @Resource
    private AdminUserApi adminUserApi;

    /** 隐患等级文本 */
    private static final String[] HAZARD_LEVEL_NAMES = {null, "一般隐患", "重大隐患"};
    /** 隐患来源文本 */
    private static final String[] HAZARD_SOURCE_NAMES = {null, "日常排查", "专项检查", "上级检查", "员工上报", "其他"};

    @GetMapping("/report")
    @Operation(summary = "获得安全生产统计报表")
    @PreAuthorize("@ss.hasPermission('mes:safety-statistics:query')")
    public CommonResult<MesSafetyStatisticsRespVO> getReport(@Valid MesSafetyStatisticsReqVO reqVO) {
        return success(statisticsService.getStatistics(reqVO));
    }

    @GetMapping("/export-hazard")
    @Operation(summary = "导出隐患明细报表")
    @PreAuthorize("@ss.hasPermission('mes:safety-statistics:export')")
    public void exportHazard(@Valid MesSafetyStatisticsReqVO reqVO,
                             HttpServletResponse response) throws IOException {
        List<MesSafetyHazardDO> list = hazardMapper.selectAllList().stream()
                .filter(h -> inRange(h.getCreateTime(), reqVO))
                .filter(h -> reqVO.getWorkshopId() == null
                        || Objects.equals(h.getWorkshopId(), reqVO.getWorkshopId()))
                .collect(Collectors.toList());

        // 字典与关联名称在导出前转为中文，业务人员可直接查看
        Map<Long, String> workshopNameMap = workshopMapper.selectList().stream()
                .collect(Collectors.toMap(MesMdWorkshopDO::getId, MesMdWorkshopDO::getName, (a, b) -> a));
        Set<Long> userIds = list.stream().map(MesSafetyHazardDO::getChargeUserId)
                .filter(Objects::nonNull).collect(Collectors.toSet());
        Map<Long, AdminUserRespDTO> userMap = userIds.isEmpty()
                ? Collections.emptyMap() : adminUserApi.getUserMap(userIds);

        LocalDate today = LocalDate.now();
        List<MesSafetyHazardExcelVO> excelList = new ArrayList<>(list.size());
        for (MesSafetyHazardDO hazard : list) {
            MesSafetyHazardExcelVO vo = BeanUtils.toBean(hazard, MesSafetyHazardExcelVO.class);
            vo.setHazardLevelName(dictName(HAZARD_LEVEL_NAMES, hazard.getHazardLevel()));
            vo.setSourceName(dictName(HAZARD_SOURCE_NAMES, hazard.getSource()));
            vo.setWorkshopName(workshopNameMap.get(hazard.getWorkshopId()));
            vo.setStatusName(hazardStatusName(hazard.getStatus()));
            AdminUserRespDTO charge = hazard.getChargeUserId() == null
                    ? null : userMap.get(hazard.getChargeUserId());
            vo.setChargeUserName(charge == null ? null : charge.getNickname());
            boolean overdue = hazard.getDeadline() != null
                    && !MesSafetyHazardDO.STATUS_CLOSED.equals(hazard.getStatus())
                    && hazard.getDeadline().isBefore(today);
            vo.setOverdueText(overdue ? "是" : "否");
            excelList.add(vo);
        }

        ExcelUtils.write(response, "隐患明细.xlsx", "隐患明细", MesSafetyHazardExcelVO.class, excelList);
    }

    // ==================== 工具 ====================

    private String dictName(String[] names, Integer value) {
        if (value == null || value < 0 || value >= names.length) {
            return null;
        }
        return names[value];
    }

    private String hazardStatusName(Integer status) {
        if (status == null) {
            return null;
        }
        return switch (status) {
            case 0 -> "待整改";
            case 10 -> "整改中";
            case 20 -> "待验收";
            case 30 -> "已闭环";
            default -> "未知";
        };
    }

    private boolean inRange(LocalDateTime createTime, MesSafetyStatisticsReqVO reqVO) {
        if (reqVO.getStartDate() == null && reqVO.getEndDate() == null) {
            return true;
        }
        if (createTime == null) {
            return false;
        }
        if (reqVO.getStartDate() != null && createTime.isBefore(reqVO.getStartDate().atStartOfDay())) {
            return false;
        }
        return reqVO.getEndDate() == null || !createTime.isAfter(reqVO.getEndDate().atTime(23, 59, 59));
    }

}