Tri-Inspector/Editor/TriPropertyDefinition.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using JetBrains.Annotations;
using TriInspector.Resolvers;
using TriInspector.Utilities;
using UnityEngine;

namespace TriInspector
{
    public class TriPropertyDefinition
    {
        private readonly ValueGetterDelegate _valueGetter;
        [CanBeNull] private readonly ValueSetterDelegate _valueSetter;

        private readonly List<string> _extensionErrors = new List<string>();
        private readonly MemberInfo _memberInfo;
        private readonly List<Attribute> _attributes;
        private readonly bool _skipNullValuesFix;

        private TriPropertyDefinition _arrayElementDefinitionBackingField;

        private IReadOnlyList<TriCustomDrawer> _drawersBackingField;
        private IReadOnlyList<TriValidator> _validatorsBackingField;
        private IReadOnlyList<TriPropertyHideProcessor> _hideProcessorsBackingField;
        private IReadOnlyList<TriPropertyDisableProcessor> _disableProcessorsBackingField;

        public static TriPropertyDefinition CreateForFieldInfo(int order, FieldInfo fi)
        {
            return CreateForMemberInfo(fi, order, fi.Name, fi.FieldType, MakeGetter(fi), MakeSetter(fi));
        }

        public static TriPropertyDefinition CreateForPropertyInfo(int order, PropertyInfo pi)
        {
            return CreateForMemberInfo(pi, order, pi.Name, pi.PropertyType, MakeGetter(pi), MakeSetter(pi));
        }

        public static TriPropertyDefinition CreateForMethodInfo(int order, MethodInfo mi)
        {
            return CreateForMemberInfo(mi, order, mi.Name, typeof(MethodInfo), MakeGetter(mi), MakeSetter(mi));
        }

        private static TriPropertyDefinition CreateForMemberInfo(
            MemberInfo memberInfo, int order, string propertyName, Type propertyType,
            ValueGetterDelegate valueGetter, ValueSetterDelegate valueSetter)
        {
            var attributes = memberInfo?.GetCustomAttributes().ToList();
            var ownerType = memberInfo?.DeclaringType ?? typeof(object);

            return new TriPropertyDefinition(
                memberInfo, ownerType, order, propertyName, propertyType, valueGetter, valueSetter, attributes, false);
        }

        internal static TriPropertyDefinition CreateForGetterSetter(
            int order, string name, Type fieldType,
            ValueGetterDelegate valueGetter, ValueSetterDelegate valueSetter)
        {
            return new TriPropertyDefinition(
                null, null, order, name, fieldType, valueGetter, valueSetter, null, false);
        }

        internal TriPropertyDefinition(
            MemberInfo memberInfo,
            Type ownerType,
            int order,
            string fieldName,
            Type fieldType,
            ValueGetterDelegate valueGetter,
            ValueSetterDelegate valueSetter,
            List<Attribute> attributes,
            bool isArrayElement)
        {
            OwnerType = ownerType;
            Name = fieldName;
            FieldType = fieldType;
            IsArrayElement = isArrayElement;

            _attributes = attributes ?? new List<Attribute>();
            _memberInfo = memberInfo;
            _valueGetter = valueGetter;
            _valueSetter = valueSetter;

            _skipNullValuesFix = memberInfo != null && memberInfo.GetCustomAttribute<SerializeReference>() != null;

            Order = order;
            IsReadOnly = _valueSetter == null || Attributes.TryGet(out ReadOnlyAttribute _);

            if (TriReflectionUtilities.IsArrayOrList(FieldType, out var elementType))
            {
                IsArray = true;
                ArrayElementType = elementType;
            }

            if (Attributes.TryGet(out LabelTextAttribute labelTextAttribute))
            {
                CustomLabel = ValueResolver.ResolveString(this, labelTextAttribute.Text);
            }

            if (Attributes.TryGet(out PropertyTooltipAttribute tooltipAttribute))
            {
                CustomTooltip = ValueResolver.ResolveString(this, tooltipAttribute.Tooltip);
            }
            else if (Attributes.TryGet(out TooltipAttribute unityTooltipAttribute))
            {
                CustomTooltip = new ConstantValueResolver<string>(unityTooltipAttribute.tooltip);
            }
        }

        public Type OwnerType { get; }

        public Type FieldType { get; }

        public string Name { get; }

        public int Order { get; internal set; }

        public IReadOnlyList<Attribute> Attributes => _attributes;

        public bool IsReadOnly { get; }

        public bool IsArrayElement { get; }
        public Type ArrayElementType { get; }

        public bool IsArray { get; }

        [CanBeNull] public ValueResolver<string> CustomLabel { get; }
        [CanBeNull] public ValueResolver<string> CustomTooltip { get; }

        public IReadOnlyList<TriPropertyHideProcessor> HideProcessors => PopulateHideProcessor();
        public IReadOnlyList<TriPropertyDisableProcessor> DisableProcessors => PopulateDisableProcessors();
        public IReadOnlyList<TriCustomDrawer> Drawers => PopulateDrawers();
        public IReadOnlyList<TriValidator> Validators => PopulateValidators();

        internal IReadOnlyList<string> ExtensionErrors
        {
            get
            {
                PopulateHideProcessor();
                PopulateDisableProcessors();
                PopulateDrawers();
                PopulateValidators();
                return _extensionErrors;
            }
        }

        public List<Attribute> GetEditableAttributes()
        {
            return _attributes;
        }

        public bool TryGetMemberInfo(out MemberInfo memberInfo)
        {
            memberInfo = _memberInfo;
            return memberInfo != null;
        }

        public object GetValue(TriProperty property, int targetIndex)
        {
            var value = _valueGetter(property, targetIndex);

            if (value == null && !_skipNullValuesFix)
            {
                value = TriUnitySerializationUtilities.PopulateUnityDefaultValueForType(FieldType);

                if (value != null)
                {
                    _valueSetter?.Invoke(property, targetIndex, value);
                }
            }

            return value;
        }

        public bool SetValue(TriProperty property, object value, int targetIndex, out object parentValue)
        {
            if (IsReadOnly)
            {
                Debug.LogError("Cannot set value for readonly property");
                parentValue = default;
                return false;
            }

            parentValue = _valueSetter?.Invoke(property, targetIndex, value);
            return true;
        }

        public TriPropertyDefinition ArrayElementDefinition
        {
            get
            {
                if (_arrayElementDefinitionBackingField == null)
                {
                    if (!IsArray)
                    {
                        throw new InvalidOperationException(
                            $"Cannot get array element definition for non array property: {FieldType}");
                    }

                    var elementGetter = new ValueGetterDelegate((self, targetIndex) =>
                    {
                        var parentValue = (IList) self.Parent.GetValue(targetIndex);
                        return parentValue[self.IndexInArray];
                    });
                    var elementSetter = new ValueSetterDelegate((self, targetIndex, value) =>
                    {
                        var parentValue = (IList) self.Parent.GetValue(targetIndex);
                        parentValue[self.IndexInArray] = value;
                        return parentValue;
                    });

                    _arrayElementDefinitionBackingField = new TriPropertyDefinition(_memberInfo, OwnerType, 0,
                        "Element", ArrayElementType, elementGetter, elementSetter, _attributes, true);
                }

                return _arrayElementDefinitionBackingField;
            }
        }

        private IReadOnlyList<TriPropertyHideProcessor> PopulateHideProcessor()
        {
            if (_hideProcessorsBackingField != null)
            {
                return _hideProcessorsBackingField;
            }

            return _hideProcessorsBackingField = TriDrawersUtilities
                .CreateHideProcessorsFor(FieldType, Attributes)
                .Where(CanApplyExtensionOnSelf)
                .ToList();
        }

        private IReadOnlyList<TriPropertyDisableProcessor> PopulateDisableProcessors()
        {
            if (_disableProcessorsBackingField != null)
            {
                return _disableProcessorsBackingField;
            }

            return _disableProcessorsBackingField = TriDrawersUtilities
                .CreateDisableProcessorsFor(FieldType, Attributes)
                .Where(CanApplyExtensionOnSelf)
                .ToList();
        }

        private IReadOnlyList<TriValidator> PopulateValidators()
        {
            if (_validatorsBackingField != null)
            {
                return _validatorsBackingField;
            }

            return _validatorsBackingField = Enumerable.Empty<TriValidator>()
                .Concat(TriDrawersUtilities.CreateValueValidatorsFor(FieldType))
                .Concat(TriDrawersUtilities.CreateAttributeValidatorsFor(FieldType, Attributes))
                .Where(CanApplyExtensionOnSelf)
                .ToList();
        }

        private IReadOnlyList<TriCustomDrawer> PopulateDrawers()
        {
            if (_drawersBackingField != null)
            {
                return _drawersBackingField;
            }

            return _drawersBackingField = Enumerable.Empty<TriCustomDrawer>()
                .Concat(TriDrawersUtilities.CreateValueDrawersFor(FieldType))
                .Concat(TriDrawersUtilities.CreateAttributeDrawersFor(FieldType, Attributes))
                .Concat(new[]
                {
                    new ValidatorsDrawer {Order = TriDrawerOrder.Validator,},
                })
                .Where(CanApplyExtensionOnSelf)
                .OrderBy(it => it.Order)
                .ToList();
        }

        private static ValueGetterDelegate MakeGetter(FieldInfo fi)
        {
            return (self, targetIndex) =>
            {
                var parentValue = self.Parent.GetValue(targetIndex);
                return fi.GetValue(parentValue);
            };
        }

        private static ValueSetterDelegate MakeSetter(FieldInfo fi)
        {
            return (self, targetIndex, value) =>
            {
                var parentValue = self.Parent.GetValue(targetIndex);
                fi.SetValue(parentValue, value);
                return parentValue;
            };
        }

        private static ValueGetterDelegate MakeGetter(PropertyInfo pi)
        {
            var method = pi.GetMethod;
            return (self, targetIndex) =>
            {
                var parentValue = self.Parent.GetValue(targetIndex);
                return method.Invoke(parentValue, null);
            };
        }

        private static ValueSetterDelegate MakeSetter(PropertyInfo pi)
        {
            var method = pi.SetMethod;
            if (method == null)
            {
                return null;
            }

            return (self, targetIndex, value) =>
            {
                var parentValue = self.Parent.GetValue(targetIndex);
                method.Invoke(parentValue, new[] {value,});
                return parentValue;
            };
        }

        private static ValueGetterDelegate MakeGetter(MethodInfo mi)
        {
            return (self, targetIndex) => mi;
        }

        private static ValueSetterDelegate MakeSetter(MethodInfo mi)
        {
            return (self, targetIndex, value) =>
            {
                var parentValue = self.Parent.GetValue(targetIndex);
                return parentValue;
            };
        }

        private bool CanApplyExtensionOnSelf(TriPropertyExtension propertyExtension)
        {
            if (propertyExtension.ApplyOnArrayElement.HasValue)
            {
                if (IsArrayElement && !propertyExtension.ApplyOnArrayElement.Value ||
                    IsArray && propertyExtension.ApplyOnArrayElement.Value)
                {
                    return false;
                }
            }

            var result = propertyExtension.Initialize(this);
            if (result.IsError)
            {
                _extensionErrors.Add(result.ErrorMessage);
            }

            return result.ShouldApply;
        }

        public delegate object ValueGetterDelegate(TriProperty self, int targetIndex);

        public delegate object ValueSetterDelegate(TriProperty self, int targetIndex, object value);
    }
}