// MIT License - Copyright (c) 2025 wallstop
// Full license text: https://github.com/wallstop/unity-helpers/blob/main/LICENSE

namespace WallstopStudios.UnityHelpers.Editor.CustomDrawers
{
#if UNITY_EDITOR
    using System;
    using System.Collections.Generic;
    using UnityEditor;
    using UnityEditor.UIElements;
    using UnityEngine;
    using UnityEngine.UIElements;
    using WallstopStudios.UnityHelpers.Core.DataStructure.Adapters;

    /// <summary>
    /// Thin wrapper that forwards SerializableType editing to the underlying StringInList-enabled field.
    /// </summary>
    [CustomPropertyDrawer(typeof(SerializableType))]
    public sealed class SerializableTypeDrawer : PropertyDrawer
    {
        private sealed class CachedProperty
        {
            public SerializedProperty typeNameProperty;
            public int lastCacheFrame = -1;
        }

        private static readonly Dictionary<string, CachedProperty> PropertyCache = new(
            StringComparer.Ordinal
        );

        public override float GetPropertyHeight(SerializedProperty property, GUIContent label)
        {
            SerializedProperty typeNameProperty = GetCachedTypeNameProperty(property);
            if (typeNameProperty == null)
            {
                return EditorGUIUtility.singleLineHeight;
            }
            return EditorGUI.GetPropertyHeight(typeNameProperty, label, true);
        }

        public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
        {
            SerializedProperty typeNameProperty = GetCachedTypeNameProperty(property);
            if (typeNameProperty == null)
            {
                EditorGUI.PropertyField(position, property, label, true);
                return;
            }

            EditorGUI.PropertyField(position, typeNameProperty, label, true);
        }

        public override VisualElement CreatePropertyGUI(SerializedProperty property)
        {
            SerializedProperty typeNameProperty = property.FindPropertyRelative(
                SerializableType.SerializedPropertyNames.AssemblyQualifiedName
            );
            if (typeNameProperty == null)
            {
                return new PropertyField(property);
            }

            return new PropertyField(typeNameProperty, property.displayName);
        }

        private static SerializedProperty GetCachedTypeNameProperty(SerializedProperty property)
        {
            string key = property.propertyPath;
            int currentFrame = Time.frameCount;

            if (PropertyCache.TryGetValue(key, out CachedProperty cached))
            {
                if (cached.lastCacheFrame == currentFrame && cached.typeNameProperty != null)
                {
                    return cached.typeNameProperty;
                }
            }
            else
            {
                cached = new CachedProperty();
                PropertyCache[key] = cached;
            }

            cached.typeNameProperty = property.FindPropertyRelative(
                SerializableType.SerializedPropertyNames.AssemblyQualifiedName
            );
            cached.lastCacheFrame = currentFrame;
            return cached.typeNameProperty;
        }
    }
#endif
}