#define FFSM2_ENABLE_PLANS

#include <ffsm2/machine_dev.hpp>

struct Context {
    bool powerOn;
};

using Config = ffsm2::Config::ContextT<Context>;

using M = ffsm2::MachineT<Config>;

#define S(s) struct s
using FSM = M::Root<S(On),
                S(Off),     // initial state
                S(Red),
                S(Yellow),
                S(Green),
                S(Done)
            >;
#undef S

struct Event {};

struct Off
    : FSM::State
{
    // called before state activation, use to re-route transitions
    void entryGuard(FullControl& control) {
        // access shared data
        if (control.context().powerOn)
            // initiate a transition into 'On' region
            control.changeTo<Red>();
    }
    // called on state deactivation
    void exit(PlanControl& /*control*/) {}
};
struct On
    : FSM::State
{
    // called on state activation
    void enter(PlanControl& control) {
        // access the plan for the region
        auto plan = control.plan();
        // sequence plan steps, executed when the previous state succeeds
        plan.change<Red, Yellow>();
        plan.change<Yellow, Green>();
        plan.change<Green, Yellow>();
        plan.change<Yellow, Red>();
    }
    // called on state deactivation
    void exit(PlanControl& /*control*/) {}
    // called on the successful completion of all plan steps
    void planSucceeded(FullControl& control) {
        control.changeTo<Done>();
    }
    // called if any of the plan steps fails
    void planFailed(FullControl& /*control*/) {}
};
struct Red
    : FSM::State
{
    // called on periodic state machine updates
    void update(FullControl& control) {
        // notify successful completion of the plan step
        control.succeed();
        // plan will advance to the 'Yellow' state
    }
};
struct Yellow
    : FSM::State
{
    void update(FullControl& control) {
        // plan will advance to the 'Green' state on the first entry
        // and 'Red' state on the second one
        control.succeed();
    }
};
struct Green
    : FSM::State
{
    // called on external events
    void react(const Event&, FullControl& control) {
        // advance to the next plan step
        control.succeed();
    }
};
struct Done
    : FSM::State
{};

TEST_CASE("Wiki.Tutorial") {
    // Create context and state machine instances:
    Context context;
    context.powerOn = true;
    FSM::Instance fsm{context};
    REQUIRE(fsm.isActive<Red>());     // On's initial sub-state
    fsm.update();
    REQUIRE(fsm.isActive<Yellow>());  // 1st setp of On's plan
    fsm.update();
    REQUIRE(fsm.isActive<Green>());   // 2nd setp of On's plan
    fsm.react(Event{});
    REQUIRE(fsm.isActive<Yellow>());  // 3rd setp of On's plan
    fsm.update();
    REQUIRE(fsm.isActive<Red>());     // 4th setp of On's plan
    fsm.update();
    REQUIRE(fsm.isActive<Done>());    // activated by On::planSucceeded()
}