package cn.iocoder.yudao.module.mes.service.wm.equipmentreceipt;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
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.wm.equipmentreceipt.vo.MesWmEquipmentReceiptPageReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.wm.equipmentreceipt.vo.MesWmEquipmentReceiptSaveReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.equipmentreceipt.MesWmEquipmentReceiptDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.equipmentreceipt.MesWmEquipmentReceiptLineDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseAreaDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseLocationDO;
import cn.iocoder.yudao.module.mes.dal.mysql.wm.equipmentreceipt.MesWmEquipmentReceiptMapper;
import cn.iocoder.yudao.module.mes.enums.MesBizTypeConstants;
import cn.iocoder.yudao.module.mes.enums.wm.MesWmEquipmentReceiptStatusEnum;
import cn.iocoder.yudao.module.mes.enums.wm.MesWmTransactionTypeEnum;
import cn.iocoder.yudao.module.mes.service.wm.transaction.MesWmTransactionService;
import cn.iocoder.yudao.module.mes.service.wm.transaction.dto.MesWmTransactionSaveReqDTO;
import cn.iocoder.yudao.module.mes.service.wm.warehouse.MesWmWarehouseAreaService;
import cn.iocoder.yudao.module.mes.service.wm.warehouse.MesWmWarehouseLocationService;
import cn.iocoder.yudao.module.mes.service.wm.warehouse.MesWmWarehouseService;
import cn.iocoder.yudao.module.system.api.approval.ApprovalConfigApi;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.annotation.Validated;

import java.math.BigDecimal;
import java.util.List;
import java.util.Set;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList;
import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.*;

/**
 * MES 设备入库单 Service 实现类
 */
@Service
@Validated
@Slf4j
public class MesWmEquipmentReceiptServiceImpl implements MesWmEquipmentReceiptService {

    /**
     * 设备入库单审批的业务类型编码
     *
     * 与 system_approval_config.biz_type 对应，审核人在「系统管理 - 审批配置」中维护。
     */
    private static final String EQUIPMENT_RECEIPT_APPROVE_BIZ_TYPE = "equipment_receipt_approve";

    @Resource
    private MesWmEquipmentReceiptMapper equipmentReceiptMapper;
    @Resource
    @Lazy
    private MesWmEquipmentReceiptLineService equipmentReceiptLineService;
    @Resource
    private MesWmTransactionService wmTransactionService;
    @Resource
    private MesWmWarehouseService wmWarehouseService;
    @Resource
    private MesWmWarehouseLocationService wmLocationService;
    @Resource
    private MesWmWarehouseAreaService wmAreaService;
    @Resource
    private AdminUserApi adminUserApi;
    @Resource
    private ApprovalConfigApi approvalConfigApi;

    @Override
    public Long createEquipmentReceipt(MesWmEquipmentReceiptSaveReqVO createReqVO) {
        // 校验编码唯一
        validateCodeUnique(null, createReqVO.getCode());

        MesWmEquipmentReceiptDO receipt = BeanUtils.toBean(createReqVO, MesWmEquipmentReceiptDO.class);
        receipt.setStatus(MesWmEquipmentReceiptStatusEnum.PREPARE.getStatus());
        equipmentReceiptMapper.insert(receipt);
        return receipt.getId();
    }

    @Override
    public void updateEquipmentReceipt(MesWmEquipmentReceiptSaveReqVO updateReqVO) {
        // 校验存在 + 草稿状态
        validateEquipmentReceiptExistsAndPrepare(updateReqVO.getId());
        // 校验编码唯一
        validateCodeUnique(updateReqVO.getId(), updateReqVO.getCode());

        MesWmEquipmentReceiptDO updateObj = BeanUtils.toBean(updateReqVO, MesWmEquipmentReceiptDO.class);
        equipmentReceiptMapper.updateById(updateObj);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteEquipmentReceipt(Long id) {
        // 校验存在 + 草稿状态
        validateEquipmentReceiptExistsAndPrepare(id);

        // 级联删除行
        equipmentReceiptLineService.deleteByReceiptId(id);
        // 删除主表
        equipmentReceiptMapper.deleteById(id);
    }

    @Override
    public MesWmEquipmentReceiptDO getEquipmentReceipt(Long id) {
        return equipmentReceiptMapper.selectById(id);
    }

    @Override
    public PageResult<MesWmEquipmentReceiptDO> getEquipmentReceiptPage(MesWmEquipmentReceiptPageReqVO pageReqVO) {
        return equipmentReceiptMapper.selectPage(pageReqVO);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void submitEquipmentReceiptApproval(Long id) {
        // 1. 校验存在 + 草稿状态
        validateEquipmentReceiptExistsAndPrepare(id);
        // 2. 校验至少有一条行
        List<MesWmEquipmentReceiptLineDO> lines = equipmentReceiptLineService.getEquipmentReceiptLineListByReceiptId(id);
        if (CollUtil.isEmpty(lines)) {
            throw exception(WM_EQUIP_RECEIPT_NO_LINE);
        }
        // 3. 校验审批配置已启用且配置了有效审核人（未配置时抛出带明确提示的业务异常）
        approvalConfigApi.validateApprovalEnabledAndGetApprovers(EQUIPMENT_RECEIPT_APPROVE_BIZ_TYPE);
        // 4. 状态置为审批中，并清空上一轮审核结果
        equipmentReceiptMapper.submitAndResetAuditInfo(id, MesWmEquipmentReceiptStatusEnum.APPROVING.getStatus());
    }

    @Override
    public Set<Long> getEquipmentReceiptApproverUserIds() {
        return approvalConfigApi.getApproverUserIds(EQUIPMENT_RECEIPT_APPROVE_BIZ_TYPE);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void auditEquipmentReceipt(Long id, Boolean pass, String reviewRemark) {
        // 1. 校验存在
        MesWmEquipmentReceiptDO receipt = validateEquipmentReceiptExists(id);
        // 2. 校验处于审批中状态
        if (!MesWmEquipmentReceiptStatusEnum.APPROVING.getStatus().equals(receipt.getStatus())) {
            throw exception(WM_EQUIP_RECEIPT_NOT_APPROVING);
        }
        // 3. 校验当前登录用户是该业务类型的审核人（或签：任一人均可审核）
        Long userId = getLoginUserId();
        approvalConfigApi.validateApprover(EQUIPMENT_RECEIPT_APPROVE_BIZ_TYPE, userId);
        // 4. 审核不通过：必须填写原因，退回草稿可修改后重新提交
        if (!Boolean.TRUE.equals(pass)) {
            if (StrUtil.isBlank(reviewRemark)) {
                throw exception(WM_EQUIP_RECEIPT_REJECT_REASON_REQUIRED);
            }
            equipmentReceiptMapper.auditEquipmentReceipt(id, MesWmEquipmentReceiptStatusEnum.PREPARE.getStatus(),
                    userId, getUserNickname(userId), reviewRemark);
            return;
        }
        // 5. 审核通过：记录审核人与审核意见，状态置为已审批（可执行入库）
        equipmentReceiptMapper.auditEquipmentReceipt(id, MesWmEquipmentReceiptStatusEnum.APPROVED.getStatus(),
                userId, getUserNickname(userId), reviewRemark);
    }

    /**
     * 获取用户昵称
     *
     * @param userId 用户编号
     * @return 昵称，用户不存在时返回 null
     */
    private String getUserNickname(Long userId) {
        if (userId == null) {
            return null;
        }
        AdminUserRespDTO user = adminUserApi.getUser(userId);
        return user == null ? null : user.getNickname();
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void finishEquipmentReceipt(Long id) {
        // 校验存在
        MesWmEquipmentReceiptDO receipt = validateEquipmentReceiptExists(id);
        if (ObjUtil.notEqual(MesWmEquipmentReceiptStatusEnum.APPROVED.getStatus(), receipt.getStatus())) {
            throw exception(WM_EQUIP_RECEIPT_STATUS_NOT_APPROVED);
        }

        // 2. 遍历所有行，固定设备仓三元组，创建 IN 库存事务
        createEquipmentStockTransaction(receipt);

        // 3. 执行入库（已审批 → 已完成）
        equipmentReceiptMapper.updateById(new MesWmEquipmentReceiptDO()
                .setId(id).setStatus(MesWmEquipmentReceiptStatusEnum.FINISHED.getStatus()));
    }

    private void createEquipmentStockTransaction(MesWmEquipmentReceiptDO receipt) {
        // 固定使用设备仓三元组
        MesWmWarehouseDO warehouse = wmWarehouseService.getWarehouseByCode(MesWmWarehouseDO.EQUIPMENT_WAREHOUSE);
        MesWmWarehouseLocationDO location = wmLocationService.getWarehouseLocationByCode(MesWmWarehouseLocationDO.EQUIPMENT_LOCATION);
        MesWmWarehouseAreaDO area = wmAreaService.getWarehouseAreaByCode(MesWmWarehouseAreaDO.EQUIPMENT_AREA);

        List<MesWmEquipmentReceiptLineDO> lines = equipmentReceiptLineService.getEquipmentReceiptLineListByReceiptId(receipt.getId());
        wmTransactionService.createTransactionList(convertList(lines, line -> new MesWmTransactionSaveReqDTO()
                .setType(MesWmTransactionTypeEnum.IN.getType())
                .setItemId(line.getItemId())
                .setQuantity(line.getQuantity() == null ? BigDecimal.ONE : line.getQuantity())
                .setWarehouseId(warehouse.getId()).setLocationId(location.getId()).setAreaId(area.getId())
                .setBizType(MesBizTypeConstants.DV_EQUIPMENT_RECEIPT).setBizId(receipt.getId())
                .setBizCode(receipt.getCode()).setBizLineId(line.getId())));
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void cancelEquipmentReceipt(Long id) {
        // 校验存在
        MesWmEquipmentReceiptDO receipt = validateEquipmentReceiptExists(id);

        // 已完成和已取消状态不允许取消
        if (ObjUtil.equal(MesWmEquipmentReceiptStatusEnum.FINISHED.getStatus(), receipt.getStatus())
                || ObjUtil.equal(MesWmEquipmentReceiptStatusEnum.CANCELED.getStatus(), receipt.getStatus())) {
            throw exception(WM_EQUIP_RECEIPT_CANCEL_NOT_ALLOWED);
        }

        // 取消
        equipmentReceiptMapper.updateById(new MesWmEquipmentReceiptDO()
                .setId(id).setStatus(MesWmEquipmentReceiptStatusEnum.CANCELED.getStatus()));
    }

    @Override
    public MesWmEquipmentReceiptDO validateEquipmentReceiptExists(Long id) {
        MesWmEquipmentReceiptDO receipt = equipmentReceiptMapper.selectById(id);
        if (receipt == null) {
            throw exception(WM_EQUIP_RECEIPT_NOT_EXISTS);
        }
        return receipt;
    }

    private void validateCodeUnique(Long id, String code) {
        MesWmEquipmentReceiptDO receipt = equipmentReceiptMapper.selectByCode(code);
        if (receipt == null) {
            return;
        }
        if (id == null || ObjUtil.notEqual(id, receipt.getId())) {
            throw exception(WM_EQUIP_RECEIPT_CODE_DUPLICATE);
        }
    }

    private MesWmEquipmentReceiptDO validateEquipmentReceiptExistsAndPrepare(Long id) {
        MesWmEquipmentReceiptDO receipt = validateEquipmentReceiptExists(id);
        if (ObjUtil.notEqual(MesWmEquipmentReceiptStatusEnum.PREPARE.getStatus(), receipt.getStatus())) {
            throw exception(WM_EQUIP_RECEIPT_STATUS_NOT_PREPARE);
        }
        return receipt;
    }

    @Override
    public void validateEquipmentReceiptEditable(Long id) {
        validateEquipmentReceiptExistsAndPrepare(id);
    }

}