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

import com.chinaztt.mes.plan.entity.ManufacturingOrder;
import com.chinaztt.mes.plan.entity.MpsRequirements;
import com.chinaztt.mes.plan.state.manufacturing.constant.ManufacturingOrderEvents;
import com.chinaztt.mes.plan.state.manufacturing.constant.ManufacturingOrderStates;
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: 2020-09-19 17:21
 */
@Configuration
@EnableStateMachineFactory(name = "manufacturingOrderStateMachineFactory")
public class ManufacturingOrderStateMachineConfig extends EnumStateMachineConfigurerAdapter<ManufacturingOrderStates, ManufacturingOrderEvents> {

    public final static String MACHINE_ID = "manufacturingOrderStateMachine";

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

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

				.and()
				.withExternal()
				.source(ManufacturingOrderStates.ISSUED).target(ManufacturingOrderStates.COMPLETED)
				.event(ManufacturingOrderEvents.COMPLETE)
				.and()
				.withExternal()
				.source(ManufacturingOrderStates.ISSUED).target(ManufacturingOrderStates.CANCELED)
				.event(ManufacturingOrderEvents.CANCEL)

				.and()
				.withExternal()
				.source(ManufacturingOrderStates.COMPLETED).target(ManufacturingOrderStates.CANCELED)
				.event(ManufacturingOrderEvents.CANCEL)

				.and()
				.withExternal()
				.source(ManufacturingOrderStates.CANCELED).target(ManufacturingOrderStates.COMPLETED)
				.event(ManufacturingOrderEvents.COMPLETE)
				.and()
				.withExternal()
				.source(ManufacturingOrderStates.CANCELED).target(ManufacturingOrderStates.ISSUED)
				.event(ManufacturingOrderEvents.COMMAND)
				.and()
				.withExternal()
				.source(ManufacturingOrderStates.CANCELED).target(ManufacturingOrderStates.PLANNED)
				.event(ManufacturingOrderEvents.PLAN)
				.and()
				.withExternal()
				.source(ManufacturingOrderStates.COMPLETED).target(ManufacturingOrderStates.ISSUED)
				.event(ManufacturingOrderEvents.COMMAND)
				.and()
				.withExternal()
				.source(ManufacturingOrderStates.COMPLETED).target(ManufacturingOrderStates.PLANNED)
				.event(ManufacturingOrderEvents.PLAN)
				.and()
				.withExternal()
				.source(ManufacturingOrderStates.ISSUED).target(ManufacturingOrderStates.PLANNED)
				.event(ManufacturingOrderEvents.PLAN);
    }

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

    /**
     * 持久化配置
     *
     * @return
     */
    @Bean(name = "manufacturingOrderPersister")
    public StateMachinePersister<ManufacturingOrderStates, ManufacturingOrderEvents, ManufacturingOrder> persister() {
        return new DefaultStateMachinePersister<>(new StateMachinePersist<ManufacturingOrderStates, ManufacturingOrderEvents, ManufacturingOrder>() {
            @Override
            public void write(StateMachineContext<ManufacturingOrderStates, ManufacturingOrderEvents> stateMachineContext, ManufacturingOrder manufacturingOrder) throws Exception {

            }

            @Override
            public StateMachineContext<ManufacturingOrderStates, ManufacturingOrderEvents> read(ManufacturingOrder manufacturingOrder) throws Exception {
                StateMachineContext<ManufacturingOrderStates, ManufacturingOrderEvents> result = new DefaultStateMachineContext<ManufacturingOrderStates, ManufacturingOrderEvents>(ManufacturingOrderStates.getEnum(manufacturingOrder.getState()),
                        null, null, null, null, MACHINE_ID);
                return result;
            }
        });
    }

}