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

import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.certificate.MesSafetyCertificatePageReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.certificate.MesSafetyCertificateRespVO;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.certificate.MesSafetyCertificateSaveReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.safety.MesSafetyCertificateDO;
import cn.iocoder.yudao.module.mes.service.safety.MesSafetyCertificateService;
import cn.iocoder.yudao.module.system.api.dept.DeptApi;
import cn.iocoder.yudao.module.system.api.dept.dto.DeptRespDTO;
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.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import jakarta.validation.Valid;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

import java.time.LocalDate;
import java.time.temporal.ChronoUnit;
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/certificate")
@Validated
public class MesSafetyCertificateController {

    /** 即将到期阈值（天） */
    private static final long EXPIRING_SOON_DAYS = 30L;

    @Resource
    private MesSafetyCertificateService certificateService;

    @Resource
    private AdminUserApi adminUserApi;

    @Resource
    private DeptApi deptApi;

    @PostMapping("/create")
    @Operation(summary = "创建资格证书")
    @PreAuthorize("@ss.hasPermission('mes:safety-certificate:create')")
    public CommonResult<Long> createCertificate(@Valid @RequestBody MesSafetyCertificateSaveReqVO createReqVO) {
        return success(certificateService.createCertificate(createReqVO));
    }

    @PutMapping("/update")
    @Operation(summary = "更新资格证书")
    @PreAuthorize("@ss.hasPermission('mes:safety-certificate:update')")
    public CommonResult<Boolean> updateCertificate(@Valid @RequestBody MesSafetyCertificateSaveReqVO updateReqVO) {
        certificateService.updateCertificate(updateReqVO);
        return success(true);
    }

    @DeleteMapping("/delete")
    @Operation(summary = "删除资格证书")
    @Parameter(name = "id", description = "编号", required = true)
    @PreAuthorize("@ss.hasPermission('mes:safety-certificate:delete')")
    public CommonResult<Boolean> deleteCertificate(@RequestParam("id") Long id) {
        certificateService.deleteCertificate(id);
        return success(true);
    }

    @GetMapping("/get")
    @Operation(summary = "获得资格证书")
    @Parameter(name = "id", description = "编号", required = true, example = "1024")
    @PreAuthorize("@ss.hasPermission('mes:safety-certificate:query')")
    public CommonResult<MesSafetyCertificateRespVO> getCertificate(@RequestParam("id") Long id) {
        MesSafetyCertificateDO certificate = certificateService.getCertificate(id);
        if (certificate == null) {
            return success(null);
        }
        List<MesSafetyCertificateRespVO> list = buildRespList(Collections.singletonList(certificate));
        return success(list.isEmpty() ? null : list.get(0));
    }

    @GetMapping("/page")
    @Operation(summary = "获得资格证书分页")
    @PreAuthorize("@ss.hasPermission('mes:safety-certificate:query')")
    public CommonResult<PageResult<MesSafetyCertificateRespVO>> getCertificatePage(@Valid MesSafetyCertificatePageReqVO pageReqVO) {
        PageResult<MesSafetyCertificateDO> pageResult = certificateService.getCertificatePage(pageReqVO);
        List<MesSafetyCertificateRespVO> list = buildRespList(pageResult.getList());
        return success(new PageResult<>(list, pageResult.getTotal()));
    }

    /**
     * 转换为 VO 并填充持证人姓名、部门名称、到期天数与到期标记
     */
    private List<MesSafetyCertificateRespVO> buildRespList(List<MesSafetyCertificateDO> list) {
        if (list == null || list.isEmpty()) {
            return Collections.emptyList();
        }
        List<MesSafetyCertificateRespVO> respList = BeanUtils.toBean(list, MesSafetyCertificateRespVO.class);

        Set<Long> userIds = list.stream().map(MesSafetyCertificateDO::getUserId)
                .filter(Objects::nonNull).collect(Collectors.toSet());
        Map<Long, AdminUserRespDTO> userMap = userIds.isEmpty()
                ? Collections.emptyMap() : adminUserApi.getUserMap(userIds);

        Set<Long> deptIds = list.stream().map(MesSafetyCertificateDO::getDeptId)
                .filter(Objects::nonNull).collect(Collectors.toSet());
        Map<Long, String> deptNameMap = new HashMap<>();
        if (!deptIds.isEmpty()) {
            List<DeptRespDTO> depts = deptApi.getDeptList(deptIds);
            if (depts != null) {
                depts.forEach(d -> deptNameMap.put(d.getId(), d.getName()));
            }
        }

        LocalDate today = LocalDate.now();
        for (MesSafetyCertificateRespVO vo : respList) {
            AdminUserRespDTO user = userMap.get(vo.getUserId());
            if (user != null) {
                vo.setUserName(user.getNickname());
            }
            vo.setDeptName(deptNameMap.get(vo.getDeptId()));
            if (vo.getExpireDate() != null) {
                long days = ChronoUnit.DAYS.between(today, vo.getExpireDate());
                vo.setDaysToExpire(days);
                vo.setExpired(days < 0);
                vo.setExpiringSoon(days >= 0 && days <= EXPIRING_SOON_DAYS);
            } else {
                vo.setExpired(false);
                vo.setExpiringSoon(false);
            }
        }
        return respList;
    }

}