using System;

using System.Collections;

using System.Collections.Generic;

using System.Linq;

using System.Text;

using System.Windows.Threading;

​

public class Program

{

    public static void Main()

    {

        var sd = new StateDiagram();

        var hungry = new State("Hungry");

        var satisfied = new State("Satisfied");

        var sick = new State("Sick");

        sd.AddState(hungry);

        sd.AddState(satisfied);

        sd.AddState(sick);

        sd.ConnectStates(hungry, satisfied, "eat");

        sd.ConnectStates(satisfied, sick, "eat too much");

        sd.ConnectStates(sick, hungry, "wait");

        Console.WriteLine(sd.ExportToStateMachineCat());

    }

​

    public class State

    {

        public string Name { get; set; }

​

        public List<Transition> Transitions { get; set; }

​

        public bool IsInitial { get; set; }

​

        public bool IsFinal { get; set; }

​

        public State(string name, bool isInitial = false, bool isFinal = false)

        {

            Name = name;

            Transitions = new List<Transition>();

            IsInitial = isInitial;

            IsFinal = isFinal;

        }

​

        public void AddTransition(State toState, string label = null, Func<bool> guard = null)

        {

            Transitions.Add(new Transition(toState, label, guard));

        }

    }

​

    public class Transition

    {

        public State ToState { get; set; }

​

        public string Label { get; set; }

​

        public Func<bool> Guard { get; set; }

​

        public Transition(State toState, string label = null, Func<bool> guard = null)

        {

            ToState = toState;

            Label = label;

            Guard = guard;

        }

    }

​

    public class LabeledChoice

    {

        public string Label { get; set; }

​

        public bool Value { get; set; }

​

        public LabeledChoice(string label, bool value)

        {

            Label = label;

            Value = value;

        }

    }

​

    public class StateDiagram

    {

        public List<State> States { get; set; }

​

        public List<LabeledChoice> LabeledChoices { get; set; }

​

        public StateDiagram()

        {

            States = new List<State>();

            LabeledChoices = new List<LabeledChoice>();

        }

​

        public void AddState(State state)

        {

            States.Add(state);

        }

​

        public void AddLabeledChoice(LabeledChoice labeledChoice)

        {

            LabeledChoices.Add(labeledChoice);

        }

​

        public void ConnectStates(State fromState, State toState, string label = null, Func<bool> guard = null)

        {

            fromState.AddTransition(toState, label, guard);

        }

​

        public string ExportToStateMachineCat()

        {

            var builder = new StringBuilder();

            // Add state declarations

            builder.AppendLine(string.Join(",", States.Select(s => s.Name)) + ";");

            // Add transition declarations

            foreach (var state in States)

            {

                foreach (var transition in state.Transitions)

                {

                    builder.Append(state.Name);

                    builder.Append(" => ");

                    builder.Append(transition.ToState.Name);

                    if (!string.IsNullOrEmpty(transition.Label))

                    {

                        builder.Append(" : ");

                        builder.Append(transition.Label);

                    }

​

                    if (transition.Guard != null)

                    {

                        builder.Append(" [");

                        builder.Append(transition.Guard);

                        builder.Append("]");

                    }

​

                    builder.AppendLine(";");

                }

            }

​

            // Add initial state declaration

            var initialState = States.FirstOrDefault(s => s.IsInitial);

            if (initialState != null)

            {

                builder.AppendLine("initial => " + initialState.Name);

            }

​

            return builder.ToString();

        }

    }

}