package cn.iocoder.yudao.module.hrm.controller.admin.performanceappraise;

import cn.hutool.core.collection.CollUtil;
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.hrm.controller.admin.performanceappraise.vo.*;
import cn.iocoder.yudao.module.hrm.dal.dataobject.employee.HrmEmployeeDO;
import cn.iocoder.yudao.module.hrm.dal.dataobject.performanceappraise.*;
import cn.iocoder.yudao.module.hrm.enums.HrmAuditStatusEnum;
import cn.iocoder.yudao.module.hrm.service.employee.HrmEmployeeService;
import cn.iocoder.yudao.module.hrm.service.performanceappraise.HrmPerformanceAppraiseService;
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.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertMap;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;
import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;

@Tag(name = "管理后台 - HRM 绩效考核方案")
@RestController
@RequestMapping("/hrm/performance-appraise")
@Validated
public class HrmPerformanceAppraiseController {

    @Resource
    private HrmPerformanceAppraiseService appraiseService;
    @Resource
    private HrmEmployeeService employeeService;

    @PostMapping("/create")
    @Operation(summary = "创建绩效考核方案")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:create')")
    public CommonResult<Long> createAppraise(@Valid @RequestBody HrmPerformanceAppraiseSaveReqVO createReqVO) {
        return success(appraiseService.createAppraise(createReqVO));
    }

    @PutMapping("/update")
    @Operation(summary = "更新绩效考核方案")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:update')")
    public CommonResult<Boolean> updateAppraise(@Valid @RequestBody HrmPerformanceAppraiseSaveReqVO updateReqVO) {
        appraiseService.updateAppraise(updateReqVO);
        return success(true);
    }

    @DeleteMapping("/delete")
    @Operation(summary = "删除绩效考核方案")
    @Parameter(name = "ids", description = "编号数组", required = true)
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:delete')")
    public CommonResult<Boolean> deleteAppraise(@RequestParam("ids") List<Long> ids) {
        appraiseService.deleteAppraise(ids);
        return success(true);
    }

    @GetMapping("/get")
    @Operation(summary = "获得绩效考核方案")
    @Parameter(name = "id", description = "编号", required = true, example = "1024")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:query')")
    public CommonResult<HrmPerformanceAppraiseRespVO> getAppraise(@RequestParam("id") Long id) {
        HrmPerformanceAppraiseDO appraise = appraiseService.getAppraise(id);
        if (appraise == null) {
            return success(null);
        }
        return success(buildAppraiseRespVO(appraise));
    }

    @GetMapping("/page")
    @Operation(summary = "获得绩效考核方案分页")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:query')")
    public CommonResult<PageResult<HrmPerformanceAppraiseRespVO>> getAppraisePage(@Valid HrmPerformanceAppraisePageReqVO pageReqVO) {
        PageResult<HrmPerformanceAppraiseDO> pageResult = appraiseService.getAppraisePage(pageReqVO);
        if (CollUtil.isEmpty(pageResult.getList())) {
            return success(PageResult.empty(pageResult.getTotal()));
        }
        List<HrmPerformanceAppraiseRespVO> list = pageResult.getList().stream()
                .map(this::buildAppraiseRespVO).collect(Collectors.toList());
        return success(new PageResult<>(list, pageResult.getTotal()));
    }

    // ========== 考核指标 ==========

    @GetMapping("/item-list")
    @Operation(summary = "获得绩效考核指标列表")
    @Parameter(name = "appraiseId", description = "方案编号", required = true, example = "1")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:query')")
    public CommonResult<List<HrmPerformanceAppraiseItemRespVO>> getAppraiseItemList(@RequestParam("appraiseId") Long appraiseId) {
        return success(BeanUtils.toBean(appraiseService.getAppraiseItemList(appraiseId), HrmPerformanceAppraiseItemRespVO.class));
    }

    @PostMapping("/item-save")
    @Operation(summary = "保存绩效考核指标（整表替换）")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:update')")
    public CommonResult<Boolean> saveAppraiseItems(@Valid @RequestBody HrmPerformanceAppraiseItemSaveReqVO saveReqVO) {
        appraiseService.saveAppraiseItems(saveReqVO);
        return success(true);
    }

    // ========== 审批 ==========

    @PostMapping("/submit")
    @Operation(summary = "提交绩效考核方案审批")
    @Parameter(name = "id", description = "方案编号", required = true)
    @Parameter(name = "processDefinitionKey", description = "流程定义Key", required = true)
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:submit')")
    public CommonResult<Boolean> submitAppraise(@RequestParam("id") Long id,
                                                @RequestParam("processDefinitionKey") String processDefinitionKey) {
        appraiseService.submitAppraise(id, processDefinitionKey, getLoginUserId());
        return success(true);
    }

    @PutMapping("/audit")
    @Operation(summary = "审批绩效考核方案（通过/驳回）")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:audit')")
    public CommonResult<Boolean> auditAppraise(@Valid @RequestBody HrmPerformanceAppraiseAuditReqVO auditReqVO) {
        appraiseService.auditAppraise(auditReqVO);
        return success(true);
    }

    @GetMapping("/approve-process-list")
    @Operation(summary = "获得绩效考核方案审批流程定义列表")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:query')")
    public CommonResult<List<Map<String, Object>>> getAppraiseApproveProcessDefinitionList() {
        return success(appraiseService.getAppraiseApproveProcessDefinitionList());
    }

    // ========== 评分/结果 ==========

    @GetMapping("/result-page")
    @Operation(summary = "获得绩效考核结果分页")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:query')")
    public CommonResult<PageResult<HrmPerformanceAppraiseResultRespVO>> getAppraiseResultPage(
            @Valid HrmPerformanceAppraiseResultPageReqVO pageReqVO) {
        PageResult<HrmPerformanceAppraiseResultDO> pageResult = appraiseService.getAppraiseResultPage(pageReqVO);
        if (CollUtil.isEmpty(pageResult.getList())) {
            return success(PageResult.empty(pageResult.getTotal()));
        }
        Map<Long, HrmEmployeeDO> employeeMap = employeeService.getEmployeeList(
                convertSet(pageResult.getList(), HrmPerformanceAppraiseResultDO::getEmployeeId)).stream()
                .collect(Collectors.toMap(HrmEmployeeDO::getId, e -> e, (a, b) -> a));
        List<HrmPerformanceAppraiseResultRespVO> list = pageResult.getList().stream()
                .map(result -> buildResultRespVO(result, employeeMap.get(result.getEmployeeId())))
                .collect(Collectors.toList());
        return success(new PageResult<>(list, pageResult.getTotal()));
    }

    @PostMapping("/result-save")
    @Operation(summary = "保存员工考核结果（自动计算综合得分与档级）")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:update')")
    public CommonResult<Boolean> saveAppraiseResult(@Valid @RequestBody HrmPerformanceAppraiseResultSaveReqVO saveReqVO) {
        appraiseService.saveAppraiseResult(saveReqVO);
        return success(true);
    }

    @GetMapping("/result")
    @Operation(summary = "获得员工考核结果（含指标明细）")
    @Parameter(name = "appraiseId", description = "方案编号", required = true, example = "1")
    @Parameter(name = "employeeId", description = "员工编号", required = true, example = "100")
    @PreAuthorize("@ss.hasPermission('hrm:performance-appraise:query')")
    public CommonResult<HrmPerformanceAppraiseResultRespVO> getAppraiseResult(
            @RequestParam("appraiseId") Long appraiseId,
            @RequestParam("employeeId") Long employeeId) {
        HrmPerformanceAppraiseResultDO result = appraiseService.getAppraiseResult(appraiseId, employeeId);
        if (result == null) {
            return success(null);
        }
        return success(buildResultRespVO(result, employeeService.getEmployee(employeeId)));
    }

    // ========== 内部方法：拼装 VO ==========

    private HrmPerformanceAppraiseRespVO buildAppraiseRespVO(HrmPerformanceAppraiseDO appraise) {
        HrmPerformanceAppraiseRespVO vo = BeanUtils.toBean(appraise, HrmPerformanceAppraiseRespVO.class);
        vo.setStatusName(convertAppraiseStatusName(appraise.getStatus()));
        vo.setAuditStatusName(HrmAuditStatusEnum.getNameByStatus(appraise.getAuditStatus()));
        vo.setItemCount(appraiseService.getAppraiseItemList(appraise.getId()).size());
        HrmPerformanceAppraiseResultPageReqVO reqVO = new HrmPerformanceAppraiseResultPageReqVO();
        reqVO.setAppraiseId(appraise.getId());
        reqVO.setPageSize(1);
        reqVO.setPageNo(1);
        vo.setResultCount(Math.toIntExact(appraiseService.getAppraiseResultPage(reqVO).getTotal()));
        return vo;
    }

    private HrmPerformanceAppraiseResultRespVO buildResultRespVO(HrmPerformanceAppraiseResultDO result, HrmEmployeeDO employee) {
        HrmPerformanceAppraiseResultRespVO vo = BeanUtils.toBean(result, HrmPerformanceAppraiseResultRespVO.class);
        if (employee != null) {
            vo.setEmployeeName(employee.getName());
        }
        List<HrmPerformanceAppraiseResultItemDO> itemDOs = appraiseService.getAppraiseResultItems(result.getId());
        if (CollUtil.isNotEmpty(itemDOs)) {
            vo.setItems(BeanUtils.toBean(itemDOs, HrmPerformanceAppraiseResultRespVO.Item.class));
        }
        return vo;
    }

    private String convertAppraiseStatusName(Integer status) {
        return switch (status == null ? -1 : status) {
            case 0 -> "草稿";
            case 10 -> "已发布";
            case 20 -> "进行中";
            case 30 -> "已完成";
            default -> "";
        };
    }

}