package com.chinaztt.mes.plan.state.outsourcing;

import com.chinaztt.mes.plan.entity.OutsourcingOrder;
import com.chinaztt.mes.plan.state.outsourcing.constant.OutsourcingOrderEvents;
import com.chinaztt.mes.plan.state.outsourcing.constant.OutsourcingOrderStates;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.statemachine.StateMachineContext;
import org.springframework.statemachine.StateMachinePersist;
import org.springframework.statemachine.config.EnableStateMachineFactory;
import org.springframework.statemachine.config.EnumStateMachineConfigurerAdapter;
import org.springframework.statemachine.config.builders.StateMachineConfigurationConfigurer;
import org.springframework.statemachine.config.builders.StateMachineStateConfigurer;
import org.springframework.statemachine.config.builders.StateMachineTransitionConfigurer;
import org.springframework.statemachine.persist.DefaultStateMachinePersister;
import org.springframework.statemachine.persist.StateMachinePersister;
import org.springframework.statemachine.support.DefaultStateMachineContext;

import java.util.EnumSet;

/**
 * @Author: cxf
 * @Date: 2021-08-17 17:21
 */
@Configuration
@EnableStateMachineFactory(name = "outsourcingOrderStateMachineFactory")
public class OutsourcingOrderStateMachineConfig extends EnumStateMachineConfigurerAdapter<OutsourcingOrderStates, OutsourcingOrderEvents> {

	public final static String MACHINE_ID = "outsourcingOrderStateMachine";

	@Override
	public void configure(StateMachineStateConfigurer<OutsourcingOrderStates, OutsourcingOrderEvents> states) throws Exception {
		states.withStates()
				.initial(OutsourcingOrderStates.PLANNED)
				.states(EnumSet.allOf(OutsourcingOrderStates.class));
	}

	@Override
	public void configure(StateMachineTransitionConfigurer<OutsourcingOrderStates, OutsourcingOrderEvents> transitions) throws Exception {
		transitions
				.withExternal()
				.source(OutsourcingOrderStates.PLANNED).target(OutsourcingOrderStates.ISSUED)
				.event(OutsourcingOrderEvents.COMMAND)
				.and()
				.withExternal()
				.source(OutsourcingOrderStates.PLANNED).target(OutsourcingOrderStates.COMPLETED)
				.event(OutsourcingOrderEvents.COMPLETE)
				.and()
				.withExternal()
				.source(OutsourcingOrderStates.PLANNED).target(OutsourcingOrderStates.CANCELED)
				.event(OutsourcingOrderEvents.CANCEL)

				.and()
				.withExternal()
				.source(OutsourcingOrderStates.ISSUED).target(OutsourcingOrderStates.COMPLETED)
				.event(OutsourcingOrderEvents.COMPLETE)
				.and()
				.withExternal()
				.source(OutsourcingOrderStates.ISSUED).target(OutsourcingOrderStates.CANCELED)
				.event(OutsourcingOrderEvents.CANCEL)
				.and()
				.withExternal()
				.source(OutsourcingOrderStates.ISSUED).target(OutsourcingOrderStates.PLANNED)
				.event(OutsourcingOrderEvents.PLAN)

				.and()
				.withExternal()
				.source(OutsourcingOrderStates.CANCELED).target(OutsourcingOrderStates.COMPLETED)
				.event(OutsourcingOrderEvents.COMPLETE)
				.and()
				.withExternal()
				.source(OutsourcingOrderStates.CANCELED).target(OutsourcingOrderStates.ISSUED)
				.event(OutsourcingOrderEvents.COMMAND)
				.and()
				.withExternal()
				.source(OutsourcingOrderStates.CANCELED).target(OutsourcingOrderStates.PLANNED)
				.event(OutsourcingOrderEvents.PLAN)
				.and()
				.withExternal()
				.source(OutsourcingOrderStates.COMPLETED).target(OutsourcingOrderStates.ISSUED)
				.event(OutsourcingOrderEvents.COMMAND)
				.and()
				.withExternal()
				.source(OutsourcingOrderStates.COMPLETED).target(OutsourcingOrderStates.PLANNED)
				.event(OutsourcingOrderEvents.PLAN)
				.and()
				.withExternal()
				.source(OutsourcingOrderStates.COMPLETED).target(OutsourcingOrderStates.CANCELED)
				.event(OutsourcingOrderEvents.CANCEL);

	}

	@Override
	public void configure(StateMachineConfigurationConfigurer<OutsourcingOrderStates, OutsourcingOrderEvents> config)
			throws Exception {
		config.withConfiguration().machineId(MACHINE_ID);
	}

	/**
	 * 持久化配置
	 *
	 * @return
	 */
	@Bean(name = "outsourcingOrderPersister")
	public StateMachinePersister<OutsourcingOrderStates, OutsourcingOrderEvents, OutsourcingOrder> persister() {
		return new DefaultStateMachinePersister<>(new StateMachinePersist<OutsourcingOrderStates, OutsourcingOrderEvents, OutsourcingOrder>() {
			@Override
			public void write(StateMachineContext<OutsourcingOrderStates, OutsourcingOrderEvents> stateMachineContext, OutsourcingOrder outsourcingOrder) throws Exception {

			}

			@Override
			public StateMachineContext<OutsourcingOrderStates, OutsourcingOrderEvents> read(OutsourcingOrder outsourcingOrder) throws Exception {
				StateMachineContext<OutsourcingOrderStates, OutsourcingOrderEvents> result = new DefaultStateMachineContext<OutsourcingOrderStates, OutsourcingOrderEvents>(OutsourcingOrderStates.getEnum(outsourcingOrder.getState()),
						null, null, null, null, MACHINE_ID);
				return result;
			}
		});
	}

}