package com.chinaztt.mes.warehouse.state.joinstockorder;

import cn.hutool.core.collection.CollectionUtil;
import com.chinaztt.mes.common.handler.StateMachineHandler;
import com.chinaztt.mes.warehouse.dto.JoinStockOrderDTO;
import com.chinaztt.mes.warehouse.entity.JoinStockOrder;
import com.chinaztt.mes.warehouse.entity.Stock;
import com.chinaztt.mes.warehouse.mapper.JoinStockOrderMapper;
import com.chinaztt.mes.warehouse.mapper.StockMapper;
import com.chinaztt.mes.warehouse.state.joinstockorder.constant.JoinStockOrderEvents;
import com.chinaztt.mes.warehouse.state.joinstockorder.constant.JoinStockOrderStateStringValues;
import com.chinaztt.mes.warehouse.util.StockUtils;
import com.chinaztt.mes.warehouse.util.TransactionType;
import lombok.AllArgsConstructor;
import org.springframework.messaging.Message;
import org.springframework.statemachine.StateMachine;
import org.springframework.statemachine.annotation.OnTransition;
import org.springframework.statemachine.annotation.WithStateMachine;
import org.springframework.stereotype.Component;

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


/**
 * @Author: cxf
 * @Date: 2020-07-24 9:51
 */
@AllArgsConstructor
@Component
@WithStateMachine(id = "joinStockOrderStateMachine")
public class JoinStockOrderStateListener {
	private StockUtils stockUtils;
	private JoinStockOrderMapper joinStockOrderMapper;

	private StockMapper stockMapper;

	/**
	 * 确认
	 */
	@OnTransition(source = JoinStockOrderStateStringValues.UNCONFIRMED, target = JoinStockOrderStateStringValues.CONFIRMED)
	public boolean confirm(StateMachine stateMachine, Message<JoinStockOrderEvents> msg) {
		JoinStockOrder joinStockOrder = (JoinStockOrder) msg.getHeaders().get("joinStockOrder");
		try {
			// 1.更新状态
			joinStockOrder.setStatus(JoinStockOrderStateStringValues.CONFIRMED);
			joinStockOrderMapper.updateById(joinStockOrder);
			// 2.更新实时库存锁定数量
			stockUtils.updateById(joinStockOrder.getStockId(), BigDecimal.ZERO, joinStockOrder.getReservedQuantity(), BigDecimal.ZERO, BigDecimal.ZERO, null, TransactionType.CONFIRM_RESERVED.getValue());
			// 3.更新库存的客户订单号
			JoinStockOrderDTO order = joinStockOrderMapper.getStockOrderById(joinStockOrder.getId());
			Stock update = new Stock();
			update.setId(joinStockOrder.getStockId());
			update.setCustomerOrderNo(order.getCustomerOrderNo());
			stockMapper.updateById(update);
		} catch (RuntimeException e) {
			stateMachine.getExtendedState().getVariables().put(StateMachineHandler.ERROR_KEY, joinStockOrder.getPartBatchNo() + "：" + e.getMessage());
			return false;
		}
		return true;
	}

	/**
	 * 取消
	 *
	 * @param stateMachine
	 * @param msg
	 * @return
	 */
	@OnTransition(source = JoinStockOrderStateStringValues.CONFIRMED, target = JoinStockOrderStateStringValues.UNCONFIRMED)
	public boolean cancel(StateMachine stateMachine, Message<JoinStockOrderEvents> msg) {
		JoinStockOrder joinStockOrder = (JoinStockOrder) msg.getHeaders().get("joinStockOrder");
		try {
			// 1.更新状态
			joinStockOrder.setStatus(JoinStockOrderStateStringValues.UNCONFIRMED);
			joinStockOrderMapper.updateById(joinStockOrder);
			// 2.更新实时库存锁定数量
			stockUtils.updateById(joinStockOrder.getStockId(), BigDecimal.ZERO, joinStockOrder.getReservedQuantity().negate(), BigDecimal.ZERO, BigDecimal.ZERO, null, TransactionType.CANCEL_RESERVED.getValue());
		} catch (RuntimeException e) {
			stateMachine.getExtendedState().getVariables().put(StateMachineHandler.ERROR_KEY, joinStockOrder.getPartBatchNo() + "：" + e.getMessage());
			return false;
		}
		return true;
	}


}