SpaceshipDemo/LocalPackages/net.peeweek.gameplay-ingredients/Editor/PropertyDrawers/StateMachineStatePropertyDrawer.cs

using UnityEngine.SceneManagement;
using UnityEngine;
using UnityEditor;
using GameplayIngredients.StateMachines;

namespace GameplayIngredients.Editor
{
    [CustomPropertyDrawer(typeof(StateMachineStateAttribute))]
    public class StateMachineStatePropertyDrawer : PropertyDrawer
    {
        public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
        {
            string propertyName = (attribute as StateMachineStateAttribute).PropertyName;
            StateMachine stateMachine;

            if (propertyName == "")
                stateMachine = (StateMachine)property.serializedObject.targetObject;
            else
                stateMachine = (StateMachine)property.serializedObject.FindProperty(propertyName).objectReferenceValue;

            if (stateMachine == null)
            {
                EditorGUI.HelpBox(position, "StateMachineStateAttribute property parameter references a null or not a GameplayIngredients.StateMachines.StateMachine type", MessageType.Error);
            }
            else
            {
                if(stateMachine != null && stateMachine.States != null)
                {
                    string currentValue = property.stringValue;
                    int count = stateMachine.States.Length;
                    int[] indices = new int[count];
                    GUIContent[] labels = new GUIContent[count];
                    int selected = -1;
                    for (int i = 0; i < count; i++)
                    {
                        indices[i] = i;
                        if (stateMachine.States[i] != null)
                        {
                            string stateName = stateMachine.States[i].StateName;

                            if (stateName == currentValue)
                                selected = i;
                            
                            labels[i] = new GUIContent(stateName);
                        }
                        else
                            labels[i] = new GUIContent("(Null State, please fix)");
                    }

                    int newIdx = EditorGUI.IntPopup(position, new GUIContent(property.displayName), selected, labels, indices);
                    if (GUI.changed && newIdx >= 0 && stateMachine.States[newIdx] != null)
                    {
                        property.stringValue = stateMachine.States[newIdx].StateName;
                    }
                }
                else
                {
                    EditorGUI.HelpBox(position, "State machine is null or has no states", MessageType.Error);
                }

            }
        }

        static class Styles
        {
            public static GUIStyle errorfield;

            static Styles()
            {
                errorfield = new GUIStyle(EditorStyles.objectField);
            }
        }
    }
}