package cn.iocoder.yudao.module.mes.controller.admin.pro.maintenanceplan;

import cn.hutool.core.collection.CollUtil;
import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.common.util.collection.MapUtils;
import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenanceplan.vo.MesProMaintenancePlanPageReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenanceplan.vo.MesProMaintenancePlanRespVO;
import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenanceplan.vo.MesProMaintenancePlanSaveReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.md.item.MesMdItemDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.md.unitmeasure.MesMdUnitMeasureDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.maintenanceplan.MesProMaintenancePlanDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.maintenanceplan.MesProMaintenancePlanItemDO;
import cn.iocoder.yudao.module.mes.service.md.item.MesMdItemService;
import cn.iocoder.yudao.module.mes.service.md.unitmeasure.MesMdUnitMeasureService;
import cn.iocoder.yudao.module.mes.service.pro.maintenanceplan.MesProMaintenancePlanService;
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.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.*;

import java.io.IOException;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertMultiMap;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;

@Tag(name = "管理后台 - MES 电力作业计划")
@RestController
@RequestMapping("/mes/pro/maintenance-plan")
@Validated
public class MesProMaintenancePlanController {

    @Resource
    private MesProMaintenancePlanService planService;
    @Resource
    private MesMdItemService itemService;
    @Resource
    private MesMdUnitMeasureService unitMeasureService;

    @PostMapping("/create")
    @Operation(summary = "创建电力作业计划")
    @PreAuthorize("@ss.hasPermission('mes:pro-maintenance-plan:create')")
    public CommonResult<Long> createPlan(@Valid @RequestBody MesProMaintenancePlanSaveReqVO createReqVO) {
        return success(planService.createPlan(createReqVO));
    }

    @PutMapping("/update")
    @Operation(summary = "更新电力作业计划")
    @PreAuthorize("@ss.hasPermission('mes:pro-maintenance-plan:update')")
    public CommonResult<Boolean> updatePlan(@Valid @RequestBody MesProMaintenancePlanSaveReqVO updateReqVO) {
        planService.updatePlan(updateReqVO);
        return success(true);
    }

    @DeleteMapping("/delete")
    @Operation(summary = "删除电力作业计划")
    @Parameter(name = "ids", description = "编号数组", required = true)
    @PreAuthorize("@ss.hasPermission('mes:pro-maintenance-plan:delete')")
    public CommonResult<Boolean> deletePlan(@RequestParam("ids") List<Long> ids) {
        planService.deletePlan(ids);
        return success(true);
    }

    @GetMapping("/get")
    @Operation(summary = "获得电力作业计划")
    @Parameter(name = "id", description = "编号", required = true, example = "1024")
    @PreAuthorize("@ss.hasPermission('mes:pro-maintenance-plan:query')")
    public CommonResult<MesProMaintenancePlanRespVO> getPlan(@RequestParam("id") Long id) {
        MesProMaintenancePlanDO plan = planService.getPlan(id);
        if (plan == null) {
            return success(null);
        }
        List<MesProMaintenancePlanItemDO> planItemList = planService.getPlanItemListByPlanId(id);
        return success(buildPlanRespVO(plan, planItemList));
    }

    @GetMapping("/page")
    @Operation(summary = "获得电力作业计划分页")
    @PreAuthorize("@ss.hasPermission('mes:pro-maintenance-plan:query')")
    public CommonResult<PageResult<MesProMaintenancePlanRespVO>> getPlanPage(@Valid MesProMaintenancePlanPageReqVO pageReqVO) {
        PageResult<MesProMaintenancePlanDO> pageResult = planService.getPlanPage(pageReqVO);
        if (CollUtil.isEmpty(pageResult.getList())) {
            return success(PageResult.empty(pageResult.getTotal()));
        }
        List<MesProMaintenancePlanItemDO> planItemList = planService.getPlanItemListByPlanIds(
                convertSet(pageResult.getList(), MesProMaintenancePlanDO::getId));
        Map<Long, List<MesProMaintenancePlanItemDO>> planItemMap = convertMultiMap(planItemList, MesProMaintenancePlanItemDO::getPlanId);
        return success(BeanUtils.toBean(pageResult, MesProMaintenancePlanRespVO.class,
                plan -> buildPlanRespVOItems(plan, planItemMap.get(plan.getId()))));
    }

    @GetMapping("/export-excel")
    @Operation(summary = "导出电力作业计划 Excel")
    @PreAuthorize("@ss.hasPermission('mes:pro-maintenance-plan:export')")
    @ApiAccessLog(operateType = EXPORT)
    public void exportPlanExcel(@Valid MesProMaintenancePlanPageReqVO pageReqVO, HttpServletResponse response) throws IOException {
        pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
        List<MesProMaintenancePlanRespVO> list = planService.getPlanPage(pageReqVO).getList().stream()
                .map(plan -> buildPlanRespVO(plan, planService.getPlanItemListByPlanId(plan.getId())))
                .collect(java.util.stream.Collectors.toList());
        ExcelUtils.write(response, "电力作业计划.xls", "数据", MesProMaintenancePlanRespVO.class, list);
    }

    @PutMapping("/issue")
    @Operation(summary = "下发电力作业计划")
    @Parameter(name = "id", description = "编号", required = true)
    @PreAuthorize("@ss.hasPermission('mes:pro-maintenance-plan:update')")
    public CommonResult<Boolean> issuePlan(@RequestParam("id") Long id) {
        planService.issuePlan(id);
        return success(true);
    }

    @PutMapping("/finish")
    @Operation(summary = "完成电力作业计划")
    @Parameter(name = "id", description = "编号", required = true)
    @PreAuthorize("@ss.hasPermission('mes:pro-maintenance-plan:update')")
    public CommonResult<Boolean> finishPlan(@RequestParam("id") Long id) {
        planService.finishPlan(id);
        return success(true);
    }

    @PostMapping("/generate-purchase-plan")
    @Operation(summary = "根据备件需求生成采购计划")
    @Parameter(name = "id", description = "编号", required = true)
    @PreAuthorize("@ss.hasPermission('mes:pro-maintenance-plan:generate-purchase-plan')")
    public CommonResult<Long> generatePurchasePlan(@RequestParam("id") Long id) {
        return success(planService.generatePurchasePlan(id));
    }

    @PostMapping("/generate-work-order")
    @Operation(summary = "生成运维/发电/检修工单")
    @Parameter(name = "id", description = "编号", required = true)
    @PreAuthorize("@ss.hasPermission('mes:pro-maintenance-plan:generate-work-order')")
    public CommonResult<Long> generateWorkOrder(@RequestParam("id") Long id) {
        return success(planService.generateWorkOrder(id));
    }

    private MesProMaintenancePlanRespVO buildPlanRespVO(MesProMaintenancePlanDO plan,
                                                        List<MesProMaintenancePlanItemDO> planItems) {
        MesProMaintenancePlanRespVO vo = BeanUtils.toBean(plan, MesProMaintenancePlanRespVO.class);
        buildPlanRespVOItems(vo, planItems);
        return vo;
    }

    private void buildPlanRespVOItems(MesProMaintenancePlanRespVO vo, List<MesProMaintenancePlanItemDO> planItems) {
        if (CollUtil.isEmpty(planItems)) {
            vo.setItems(Collections.emptyList());
            return;
        }
        Map<Long, MesMdItemDO> itemMap = itemService.getItemMap(convertSet(planItems, MesProMaintenancePlanItemDO::getItemId));
        Map<Long, MesMdUnitMeasureDO> unitMeasureMap = unitMeasureService.getUnitMeasureMap(
                convertSet(itemMap.values(), MesMdItemDO::getUnitMeasureId));
        vo.setItems(BeanUtils.toBean(planItems, MesProMaintenancePlanRespVO.Item.class, item -> {
            MapUtils.findAndThen(itemMap, item.getItemId(), mesItem -> item.setItemCode(mesItem.getCode())
                    .setItemName(mesItem.getName())
                    .setUnitMeasureName(unitMeasureMap.get(mesItem.getUnitMeasureId()) != null
                            ? unitMeasureMap.get(mesItem.getUnitMeasureId()).getName() : null));
        }));
    }

}