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

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.excel.core.util.ExcelUtils;
import cn.iocoder.yudao.module.mes.controller.admin.pro.workrecord.vo.MesProWorkRecordLogPageReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.pro.workrecord.vo.MesProWorkRecordLogRespVO;
import cn.iocoder.yudao.module.mes.controller.admin.pro.workrecord.vo.MesProWorkRecordRespVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkstationDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.workrecord.MesProWorkRecordDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.workrecord.MesProWorkRecordLogDO;
import cn.iocoder.yudao.module.mes.service.md.workstation.MesMdWorkstationService;
import cn.iocoder.yudao.module.mes.service.pro.workrecord.MesProWorkRecordService;
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.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.convertSet;
import static cn.iocoder.yudao.framework.common.util.collection.MapUtils.findAndThen;
import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;

@Tag(name = "管理后台 - MES 工作记录")
@RestController
@RequestMapping("/mes/pro/workrecord")
@Validated
public class MesProWorkRecordController {

    @Resource
    private MesProWorkRecordService workRecordService;
    @Resource
    private MesMdWorkstationService workstationService;

    @Resource
    private AdminUserApi adminUserApi;

    @GetMapping("/log/page")
    @Operation(summary = "获得工作记录流水分页")
    @PreAuthorize("@ss.hasPermission('mes:pro-workrecord:query')")
    public CommonResult<PageResult<MesProWorkRecordLogRespVO>> getWorkRecordLogPage(
            @Valid MesProWorkRecordLogPageReqVO pageReqVO) {
        PageResult<MesProWorkRecordLogDO> pageResult = workRecordService.getWorkRecordLogPage(pageReqVO);
        return success(new PageResult<>(buildWorkRecordLogRespVOList(pageResult.getList()), pageResult.getTotal()));
    }

    @GetMapping("/log/get")
    @Operation(summary = "获得工作记录流水")
    @Parameter(name = "id", description = "编号", required = true, example = "1024")
    @PreAuthorize("@ss.hasPermission('mes:pro-workrecord:query')")
    public CommonResult<MesProWorkRecordLogRespVO> getWorkRecordLog(@RequestParam("id") Long id) {
        MesProWorkRecordLogDO log = workRecordService.getWorkRecordLog(id);
        return success(buildWorkRecordLogRespVO(log));
    }

    @GetMapping("/log/export-excel")
    @Operation(summary = "导出工作记录 Excel")
    @PreAuthorize("@ss.hasPermission('mes:pro-workrecord:export')")
    @ApiAccessLog(operateType = EXPORT)
    public void exportWorkRecordLogExcel(@Valid MesProWorkRecordLogPageReqVO pageReqVO,
                                       HttpServletResponse response) throws IOException {
        pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
        List<MesProWorkRecordLogDO> list = workRecordService.getWorkRecordLogPage(pageReqVO).getList();
        List<MesProWorkRecordLogRespVO> voList = buildWorkRecordLogRespVOList(list);
        ExcelUtils.write(response, "工作记录.xls", "数据", MesProWorkRecordLogRespVO.class, voList);
    }

    @PutMapping("/clock-in")
    @Operation(summary = "上线（绑定工作站）")
    @Parameter(name = "workstationId", description = "工作站编号", required = true)
    @PreAuthorize("@ss.hasPermission('mes:pro-workrecord:clock')")
    public CommonResult<Long> clockInWorkRecord(@RequestParam("workstationId") Long workstationId) {
        return success(workRecordService.clockInWorkRecord(getLoginUserId(), workstationId));
    }

    @PutMapping("/clock-out")
    @Operation(summary = "下线（解绑工作站）")
    @PreAuthorize("@ss.hasPermission('mes:pro-workrecord:clock')")
    public CommonResult<Long> clockOutWorkRecord() {
        return success(workRecordService.clockOutWorkRecord(getLoginUserId()));
    }

    @GetMapping("/get-my")
    @Operation(summary = "获取当前用户绑定的工作站状态")
    @PreAuthorize("@ss.hasPermission('mes:pro-workrecord:query')")
    public CommonResult<MesProWorkRecordRespVO> getMyWorkRecord() {
        MesProWorkRecordDO record = workRecordService.getWorkRecord(getLoginUserId());
        return success(buildWorkRecordRespVO(record));
    }

    // ==================== 拼接 VO ====================

    private MesProWorkRecordLogRespVO buildWorkRecordLogRespVO(MesProWorkRecordLogDO log) {
        if (log == null) {
            return null;
        }
        return buildWorkRecordLogRespVOList(Collections.singletonList(log)).get(0);
    }

    private List<MesProWorkRecordLogRespVO> buildWorkRecordLogRespVOList(List<MesProWorkRecordLogDO> list) {
        if (CollUtil.isEmpty(list)) {
            return Collections.emptyList();
        }
        // 1. 查询相关数据
        Map<Long, AdminUserRespDTO> userMap = adminUserApi.getUserMap(
                convertSet(list, MesProWorkRecordLogDO::getUserId));
        Map<Long, MesMdWorkstationDO> workstationMap = workstationService.getWorkstationMap(
                convertSet(list, MesProWorkRecordLogDO::getWorkstationId));
        // 2. 拼接 VO
        return BeanUtils.toBean(list, MesProWorkRecordLogRespVO.class, vo -> {
            findAndThen(userMap, vo.getUserId(), user ->
                    vo.setUserNickname(user.getNickname()));
            findAndThen(workstationMap, vo.getWorkstationId(), ws ->
                    vo.setWorkstationCode(ws.getCode()).setWorkstationName(ws.getName()));
        });
    }

    private MesProWorkRecordRespVO buildWorkRecordRespVO(MesProWorkRecordDO record) {
        if (record == null) {
            return null;
        }
        MesProWorkRecordRespVO vo = BeanUtils.toBean(record, MesProWorkRecordRespVO.class);
        // 拼接工作站信息
        MesMdWorkstationDO ws = workstationService.getWorkstation(record.getWorkstationId());
        if (ws != null) {
            vo.setWorkstationCode(ws.getCode()).setWorkstationName(ws.getName());
        }
        // 拼接用户信息
        AdminUserRespDTO user = adminUserApi.getUser(record.getUserId());
        if (user != null) {
            vo.setUserNickname(user.getNickname());
        }
        return vo;
    }

}