/*
 *    Copyright (c) 2018-2025, ztt All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * Neither the name of the pig4cloud.com developer nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * Author: ztt
 */
package com.chinaztt.mes.warehouse.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.chinaztt.mes.basic.entity.Location;
import com.chinaztt.mes.basic.entity.Part;
import com.chinaztt.mes.basic.mapper.LocationMapper;
import com.chinaztt.mes.basic.mapper.PartMapper;
import com.chinaztt.mes.common.handler.StateMachineHandler;
import com.chinaztt.mes.common.util.StateResult;
import com.chinaztt.mes.warehouse.dto.PrepareDetailDTO;
import com.chinaztt.mes.warehouse.dto.ReserveDetailDTO;
import com.chinaztt.mes.warehouse.dto.StockAddDTO;
import com.chinaztt.mes.warehouse.entity.*;
import com.chinaztt.mes.warehouse.mapper.*;
import com.chinaztt.mes.warehouse.service.ReserveDetailService;
import com.chinaztt.mes.warehouse.state.ReserveStateMachineConfig;
import com.chinaztt.mes.warehouse.state.constant.ReserveEvents;
import com.chinaztt.mes.warehouse.state.constant.ReserveStates;
import com.chinaztt.mes.warehouse.util.StockUtils;
import com.chinaztt.ztt.common.core.util.R;
import lombok.AllArgsConstructor;
import org.springframework.messaging.Message;
import org.springframework.messaging.support.MessageBuilder;
import org.springframework.statemachine.config.StateMachineFactory;
import org.springframework.statemachine.persist.StateMachinePersister;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 领料明细表
 *
 * @author sunxl
 * @date 2020-10-13 11:12:46
 */
@AllArgsConstructor
@Transactional(rollbackFor = Exception.class)
@Service
public class ReserveDetailServiceImpl extends ServiceImpl<ReserveDetailMapper, ReserveDetail> implements ReserveDetailService {
	private PrepareDetailMapper prepareDetailMapper;
	private ReserveMainMapper reserveMainMapper;
	private StateMachineFactory<ReserveStates, ReserveEvents> reserveStateMachineFactory;
	private StateMachinePersister<ReserveStates, ReserveEvents, ReserveMain> persister;
	private StockMapper stockMapper;
	private StockUtils stockUtils;
	private PartMapper partMapper;
	private LocationMapper locationMapper;
	private StockInCodeMapper stockInCodeMapper;

	@Override
	public IPage<List<ReserveDetailDTO>> getReservePage(Page page, QueryWrapper<ReserveDetailDTO> gen) {
		return baseMapper.getReservePage(page, gen);
	}

	@Override
	public ReserveDetailDTO getByReserveId(Long id) {
		return baseMapper.getByReserveId(id);
	}


	@Override
	public R<PrepareDetailDTO> getByReserveDetail(PrepareDetailDTO prepareDetailDTO) {
		Part part = partMapper.selectOne(Wrappers.<Part>lambdaQuery().eq(Part::getPartNo, prepareDetailDTO.getPartNo()));
		PrepareDetail prepareDetail = prepareDetailMapper.selectByPrepare(prepareDetailDTO.getId(), part.getId(), prepareDetailDTO.getPartBatchNo());
		if (prepareDetail.getWarehouseReserveDetailId() == null) {
			return R.failed("请确认输入信息，没有找到对应的备料明细");
		}
		ReserveDetail reserveDetail = baseMapper.selectById(prepareDetail.getWarehouseReserveDetailId());
		ReserveMain reserveMain = reserveMainMapper.selectById(prepareDetailDTO.getId());
		StockInCode stockInCode = stockInCodeMapper.selectById(reserveMain.getResType());
		Location location = locationMapper.selectOne(Wrappers.<Location>lambdaQuery().eq(Location::getLocNo, prepareDetailDTO.getLocNo()));
		BigDecimal reserveQty = reserveDetail.getRecvQty();
		if (reserveQty.add(prepareDetailDTO.getRecvQty()).compareTo(reserveDetail.getOutQty()) == 1) {
			return R.failed("交接数量超过了出库数量，请检查交接数量");
		}
		if (location.getId() != null) {
			Stock stock = stockMapper.selectOne(Wrappers.<Stock>lambdaQuery().eq(Stock::getPartId, part.getId())
					.eq(Stock::getLocationId, location.getId()).eq(Stock::getPartBatchNo, prepareDetailDTO.getPartBatchNo()).eq(Stock::getSystemNo, prepareDetailDTO.getSystemNo()));
			if (stock != null) {
				stockUtils.updateById(stock.getId(), prepareDetailDTO.getRecvQty(), BigDecimal.ZERO,  BigDecimal.ZERO, BigDecimal.ZERO,reserveMain.getResNo(), stockInCode.getStockCodeName());
			}
			if (stock == null) {
				StockAddDTO stockAdd = new StockAddDTO();
				stockAdd.setPartsId(part.getId());
				stockAdd.setNewLocationId(location.getId());
				stockAdd.setNewPartBatchNo(prepareDetailDTO.getPartBatchNo());

				Stock stock1 = stockUtils.query(stockAdd);
				stockUtils.updateById(stock1.getId(), prepareDetailDTO.getRecvQty(), BigDecimal.ZERO,  BigDecimal.ZERO, BigDecimal.ZERO,reserveMain.getResNo(), stockInCode.getStockCodeName());
			}
		}
		reserveDetail.setRecvQty(reserveQty.add(prepareDetailDTO.getRecvQty()));
		reserveDetail.setAutoReceiveTime(LocalDateTime.now());
		if (reserveQty.add(prepareDetailDTO.getRecvQty()).compareTo(reserveDetail.getOutQty()) == 0) {
			reserveDetail.setStatus("已交接");
		}
		baseMapper.updateById(reserveDetail);
		List<ReserveDetail> reserveDetails = baseMapper.selectList(Wrappers.<ReserveDetail>lambdaQuery().eq(ReserveDetail::getWarehouseMainId, reserveMain.getId()));
		List<ReserveDetail> statusList = reserveDetails.stream().filter(r -> !"已交接".equals(r.getStatus())).collect(Collectors.toList());
		if (statusList.size() == 0) {
			Message<ReserveEvents> message = MessageBuilder.withPayload(ReserveEvents.valueOf("handle")).setHeader("reserveMain", reserveMain).build();
			StateMachineHandler handler = new StateMachineHandler(reserveStateMachineFactory, persister, ReserveStateMachineConfig.MACHINE_ID, reserveMain);
			StateResult res = handler.sendEvent(message, reserveMain.getId());
			if (!res.isSuccess()) {
				return R.failed(res.getMsg());
			}
		}
		return R.ok(baseMapper.getByReserveDetail(prepareDetailDTO));
	}
}