HideDress/SkinHide/Utils/RefHelp.cs

using Aki.Reflection.Utils;
using HarmonyLib;
using System;
using System.Linq;
using System.Reflection;
using System.Reflection.Emit;

namespace SkinHide.Utils
{
    public class RefHelp
    {
        public static DelegateType ObjectMethodDelegate<DelegateType>(MethodInfo method, bool virtualCall = true) where DelegateType : Delegate
        {
            if (method is null)
            {
                throw new ArgumentNullException(nameof(method));
            }

            var delegateType = typeof(DelegateType);

            var declaringType = method.DeclaringType;

            var DelegateParameters = delegateType.GetMethod("Invoke").GetParameters();
            var DelegateparameterTypes = DelegateParameters.Select(x => x.ParameterType).ToArray();

            var dmd = new DynamicMethod("OpenInstanceDelegate_" + method.Name, method.ReturnType, DelegateparameterTypes);

            var ilGen = dmd.GetILGenerator();

            if (!method.IsStatic)
            {
                var parameters = method.GetParameters();
                var numParameters = parameters.Length;
                var parameterTypes = new Type[numParameters + 1];
                parameterTypes[0] = typeof(object);
                for (var i = 0; i < numParameters; i++)
                {
                    parameterTypes[i + 1] = parameters[i].ParameterType;
                }

                if (declaringType != null && declaringType.IsValueType)
                {
                    ilGen.Emit(OpCodes.Ldarga_S, 0);
                }
                else
                {
                    ilGen.Emit(OpCodes.Ldarg_0);
                }

                ilGen.Emit(OpCodes.Castclass, declaringType);

                for (var i = 1; i < parameterTypes.Length; i++)
                {
                    ilGen.Emit(OpCodes.Ldarg, i);
                    ilGen.Emit(OpCodes.Castclass, parameterTypes[i]);
                }
            }
            else
            {
                var parameterTypes = method.GetParameters().Select(x => x.ParameterType).ToArray();

                for (var i = 0; i < parameterTypes.Length; i++)
                {
                    ilGen.Emit(OpCodes.Ldarg, i);
                    ilGen.Emit(OpCodes.Castclass, parameterTypes[i]);
                }
            }


            if (method.IsStatic || !virtualCall)
            {
                ilGen.Emit(OpCodes.Call, method);
            }
            else
            {
                ilGen.Emit(OpCodes.Callvirt, method);
            }

            ilGen.Emit(OpCodes.Ret);

            return (DelegateType)dmd.CreateDelegate(delegateType);
        }

        public class PropertyRef<T, F> where T : class
        {
            private Func<T, F> RefGetValue;

            private Action<T, F> RefSetValue;

            private PropertyInfo PropertyInfo;

            private MethodInfo GetMethodInfo;

            private MethodInfo SetMethodInfo;

            private Type TType;

            private T Instance;

            public Type InType;

            public Type ReturnType;

            public PropertyRef(PropertyInfo propertyinfo, object instance = null)
            {
                Init(propertyinfo, instance);
            }

            public PropertyRef(Type type, string propertyname, object instance = null)
            {
                Init(type.GetProperty(propertyname, AccessTools.all), instance);
            }

            private void Init(PropertyInfo propertyinfo, object instance)
            {
                PropertyInfo = propertyinfo;

                TType = PropertyInfo.DeclaringType;

                Instance = (T)instance;

                GetMethodInfo = PropertyInfo.GetGetMethod(true);

                if (GetMethodInfo != null)
                {
                    RefGetValue = ObjectMethodDelegate<Func<T, F>>(GetMethodInfo);
                }

                SetMethodInfo = PropertyInfo.GetSetMethod(true);

                if (SetMethodInfo != null)
                {
                    RefSetValue = ObjectMethodDelegate<Action<T, F>>(SetMethodInfo);
                }

                if (GetMethodInfo != null && SetMethodInfo != null)
                {
                    ReturnType = GetMethodInfo.ReturnType;
                }
                else if (GetMethodInfo != null || SetMethodInfo != null)
                {
                    ReturnType = GetMethodInfo != null ? GetMethodInfo.ReturnType : SetMethodInfo.ReturnType;
                }
            }

            public static PropertyRef<T, F> Create(PropertyInfo propertyinfo, object instance = null)
            {
                return new PropertyRef<T, F>(propertyinfo, instance);
            }

            public static PropertyRef<T, F> Create(string propertyname, object instance = null)
            {
                return new PropertyRef<T, F>(typeof(T), propertyname, instance);
            }

            public static PropertyRef<T, F> Create(Type type, string propertyname, object instance = null)
            {
                return new PropertyRef<T, F>(type, propertyname, instance);
            }

            public F GetValue(T instance)
            {
                if (RefGetValue == null)
                {
                    throw new ArgumentNullException(nameof(RefGetValue));
                }

                if (instance != null && TType.IsAssignableFrom(instance.GetType()))
                {
                    return RefGetValue(instance);
                }
                else if (Instance != null && instance == null)
                {
                    return RefGetValue(Instance);
                }
                else
                {
                    return default(F);
                }
            }

            public void SetValue(T instance, F value)
            {
                if (RefSetValue == null)
                {
                    throw new ArgumentNullException(nameof(RefSetValue));
                }

                if (instance != null && TType.IsAssignableFrom(instance.GetType()))
                {
                    RefSetValue(instance, value);
                }
                else if (Instance != null && instance == null)
                {
                    RefSetValue(Instance, value);
                }
            }
        }

        public class FieldRef<T, F>
        {
            private AccessTools.FieldRef<T, F> HarmonyFieldRef;

            private FieldInfo FieldInfo;

            private Type TType;

            private T Instance;

            public Type InType;

            public Type FieldType;

            public FieldRef(FieldInfo fieldinfo, object instance = null)
            {
                Init(fieldinfo, instance);
            }

            public FieldRef(Type type, string fieldname, object instance = null)
            {
                Init(type.GetField(fieldname, AccessTools.all), instance);
            }

            public static FieldRef<T, F> Create(FieldInfo fieldinfo, object instance = null)
            {
                return new FieldRef<T, F>(fieldinfo, instance);
            }

            public static FieldRef<T, F> Create(string fieldname, object instance = null)
            {
                return new FieldRef<T, F>(typeof(T), fieldname, instance);
            }

            public static FieldRef<T, F> Create(Type type, string fieldname, object instance = null)
            {
                return new FieldRef<T, F>(type, fieldname, instance);
            }

            private void Init(FieldInfo fieldinfo, object instance = null)
            {
                FieldInfo = fieldinfo;

                TType = FieldInfo.DeclaringType;

                InType = TType;

                FieldType = FieldInfo.FieldType;

                Instance = (T)instance;

                HarmonyFieldRef = AccessTools.FieldRefAccess<T, F>(FieldInfo);
            }

            public F GetValue(T instance)
            {
                if (HarmonyFieldRef == null)
                {
                    throw new ArgumentNullException(nameof(HarmonyFieldRef));
                }

                if (instance != null && TType.IsAssignableFrom(instance.GetType()))
                {
                    return HarmonyFieldRef(instance);
                }
                else if (Instance != null && instance == null)
                {
                    return HarmonyFieldRef(Instance);
                }
                else
                {
                    return default(F);
                }

            }

            public void SetValue(T instance, F value)
            {
                if (HarmonyFieldRef == null)
                {
                    throw new ArgumentNullException(nameof(HarmonyFieldRef));
                }

                if (instance != null && TType.IsAssignableFrom(instance.GetType()))
                {
                    HarmonyFieldRef(instance) = value;
                }
                else if (Instance != null && instance == null)
                {
                    HarmonyFieldRef(Instance) = value;
                }
            }
        }

        public static Type GetEftType(Func<Type, bool> func)
        {
            return PatchConstants.EftTypes.Single(func);
        }

        public static MethodInfo GetEftMethod(Type type, BindingFlags flags, Func<MethodInfo, bool> func)
        {
            return type.GetMethods(flags).Single(func);
        }

        public static MethodInfo GetEftMethod(Func<Type, bool> func, BindingFlags flags, Func<MethodInfo, bool> func2)
        {
            return GetEftMethod(GetEftType(func), flags, func2);
        }
    }
}