DragonECS-Unity/src/Connectors/Editor/EcsEntityConnectEditor.cs

#if UNITY_EDITOR
using DCFApixels.DragonECS.Unity.Internal;
using UnityEditor;
using UnityEngine;

namespace DCFApixels.DragonECS.Unity.Editors
{
    [CustomEditor(typeof(EcsEntityConnect))]
    [CanEditMultipleObjects]
    internal class EcsEntityConnectEditor : Editor
    {
        private bool _isInit = false;
        private EcsEntityConnect Target => (EcsEntityConnect)target;
        private bool IsMultipleTargets => targets.Length > 1;

        private void Init()
        {
            if (_isInit)
            {
                return;
            }
            _isInit = true;
        }

        public override void OnInspectorGUI()
        {
            Init();
            EcsEntityConnect[] targets = new EcsEntityConnect[this.targets.Length];
            for (int i = 0; i < targets.Length; i++)
            {
                targets[i] = (EcsEntityConnect)this.targets[i];
            }
            DrawEntityInfo(targets);

            DrawTemplates();

            DrawControlButtons(targets);
            DrawComponents(targets);
        }

        private void DrawEntityInfo(EcsEntityConnect[] targets)
        {
            bool isConnected = Target.Entity.TryUnpack(out int id, out short gen, out EcsWorld world);
            EcsGUI.EntityStatus status = IsMultipleTargets ? EcsGUI.EntityStatus.Undefined : isConnected ? EcsGUI.EntityStatus.Alive : EcsGUI.EntityStatus.NotAlive;
            EcsGUI.Layout.EntityBar(status, id, gen, world.id);
        }

        private void DrawTemplates()
        {
            EditorGUI.BeginChangeCheck();
            var iterator = serializedObject.GetIterator();
            iterator.NextVisible(true);
            while (iterator.NextVisible(false))
            {
                EditorGUILayout.PropertyField(iterator, true);
            }
            if (EditorGUI.EndChangeCheck())
            {
                serializedObject.ApplyModifiedProperties();
            }
        }

        private void DrawControlButtons(EcsEntityConnect[] targets)
        {
            float height = EcsGUI.EntityBarHeight;
            Rect rect = GUILayoutUtility.GetRect(EditorGUIUtility.currentViewWidth, height);
            EditorGUI.DrawRect(rect, new Color(0f, 0f, 0f, 0.1f));
            rect = RectUtility.AddPadding(rect, 2f, 0f);
            var (_, buttonRect) = RectUtility.HorizontalSliceRight(rect, height);
            if (EcsGUI.AutosetCascadeButton(buttonRect))
            {
                foreach (var target in targets)
                {
                    target.AutosetCascade_Editor();
                }
            }
            buttonRect = RectUtility.Move(buttonRect, -height, 0);
            if (EcsGUI.AutosetButton(buttonRect))
            {
                foreach (var target in targets)
                {
                    target.Autoset_Editor();
                }
            }
            using (new EditorGUI.DisabledScope(!Application.isPlaying))
            {
                buttonRect = RectUtility.Move(buttonRect, -height, 0);
                if (EcsGUI.DelEntityButton(buttonRect))
                {
                    foreach (var target in targets)
                    {
                        target.DeleteEntity_Editor();
                    }
                }
                buttonRect = RectUtility.Move(buttonRect, -height, 0);
                if (EcsGUI.UnlinkButton(buttonRect))
                {
                    foreach (var target in targets)
                    {
                        target.UnlinkEntity_Editor();
                    }
                }
            }
        }

        private void DrawComponents(EcsEntityConnect[] targets)
        {
            if (IsMultipleTargets)
            {
                for (int i = 0; i < targets.Length; i++)
                {
                    if (targets[i].IsConected == true)
                    {
                        EditorGUILayout.HelpBox("Multiple component editing is not available.", MessageType.Warning);
                        return;
                    }
                }
            }
            if (Target.Entity.TryUnpack(out int entityID, out EcsWorld world))
            {
                EcsGUI.Layout.DrawRuntimeComponents(entityID, world);
            }
        }
    }
}
#endif