namespace UnityHelpers.Editor
{
#if UNITY_EDITOR
    using System.Reflection;
    using Core.Attributes;
    using Core.Extension;
    using UnityEngine;
    using UnityEditor;

    [CustomPropertyDrawer(typeof(WShowIfAttribute))]
    public sealed class WShowIfPropertyDrawer : PropertyDrawer
    {
        public override float GetPropertyHeight(SerializedProperty property, GUIContent label)
        {
            return !ShouldShow(property) ? 0f : EditorGUI.GetPropertyHeight(property, label, true);
        }

        public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
        {
            if (ShouldShow(property))
            {
                EditorGUI.PropertyField(position, property, label, true);
            }
        }

        private bool ShouldShow(SerializedProperty property)
        {
            if (attribute is not WShowIfAttribute showIf)
            {
                return true;
            }

            SerializedProperty conditionProperty = property.serializedObject.FindProperty(
                showIf.conditionField
            );
            if (conditionProperty is not { propertyType: SerializedPropertyType.Boolean })
            {
                if (conditionProperty != null)
                {
                    return true;
                }

                // This might not be a unity object, so fall back to reflection
                object enclosingObject = property.GetEnclosingObject(out _);
                FieldInfo conditionField = enclosingObject
                    ?.GetType()
                    .GetField(
                        showIf.conditionField,
                        BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic
                    );
                if (conditionField?.GetValue(enclosingObject) is bool maybeCondition)
                {
                    return showIf.inverse ? !maybeCondition : maybeCondition;
                }
                return true;
            }

            bool condition = conditionProperty.boolValue;
            return showIf.inverse ? !condition : condition;
        }
    }
#endif
}