EndpointAttributeDrawer.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Linq;
using UnityEditor;
using UnityEngine;
 
namespace PixoVR.Apex
{
    public class EndpointDisplayAttribute : PropertyAttribute
    {
        public List<string> enumDisplayList = new List<string>();
        public List<PlatformServer> enumList = new List<PlatformServer>();
 
        public EndpointDisplayAttribute()
        {
            enumList = Enum.GetValues(typeof(PlatformServer)).Cast<PlatformServer>().ToList();
 
            foreach(PlatformServer enumItem in enumList)
            {
                enumDisplayList.Add(enumItem.ToDisplayString());
            }
        }
    }
#if UNITY_EDITOR
    [CustomPropertyDrawer(typeof(EndpointDisplayAttribute))]
    public class EndpointDisplayDrawer : PropertyDrawer
    {
        [SerializeField]
        int selectedIndex = -1;
        
        public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
        {
            EndpointDisplayAttribute displayAttribute = attribute as EndpointDisplayAttribute;
 
            if (displayAttribute.enumDisplayList.Count > 0)
            {
                int newIndex = EditorGUI.Popup(position, property.name, selectedIndex, displayAttribute.enumDisplayList.ToArray());
                if (newIndex < 0)
                {
                    newIndex = property.enumValueIndex;
                }
 
                if (newIndex != selectedIndex)
                {
                    selectedIndex = newIndex;
                    property.enumValueIndex = selectedIndex;
                    UnityEngine.Object dirtyObject = property.serializedObject.targetObject;
 
                    if (dirtyObject != null)
                    {
                        Debug.Log("Selected Index: " + selectedIndex);
                        EditorUtility.SetDirty(dirtyObject);
                    }
                }
            }
            else
            {
                Debug.Log("Reset");
                EditorGUI.PropertyField(position, property, label);
            }
        }
    }
#endif
}