LuaCsForBarotraumaEP/Barotrauma/BarotraumaShared/SharedSource/LuaCs/Compatibility/LuaCsConfig.cs

using Barotrauma;
using System;
using System.Collections;
using System.Collections.Generic;
using System.Collections.Immutable;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Runtime.Loader;
using System.Xml.Linq;

namespace Barotrauma;

class LuaCsConfig
{
    private enum ValueType
    {
        None,
        Text,
        Integer,
        Decimal,
        Boolean,
        Collection,
        Object,
        Enum
    }

    private static Type[] LoadDocTypes(XElement typesElem)
    {
        var result = new List<Type>();
        var loadedTypes = AssemblyLoadContext.All
            .Where(alc => alc != AssemblyLoadContext.Default)
            .SelectMany(alc => alc.Assemblies)
            .SelectMany(asm => asm.GetTypes())
            .ToImmutableArray();

        foreach (var elem in typesElem.Elements())
        {
            var typesFound = loadedTypes.Where(t => t.FullName?.EndsWith(elem.Value) ?? false).ToImmutableList();
            if (!typesFound.Any())
            {
                ModUtils.Logging.PrintError(
                    $"{nameof(LuaCsConfig)}::{nameof(LoadDocTypes)}() | Unable to find a matching type for {elem.Value}");
                continue;
            }
            result.AddRange(typesFound);
        }

        return result.ToArray();
    }

    private static IEnumerable<XElement> SaveDocTypes(IEnumerable<Type> types)
    {
        return types.Select(t => new XElement("Type", t.ToString()));
    }

    private static Type GetTypeAttr(Type[] types, XElement elem)
    {
        var idx = elem.GetAttributeInt("Type", -1);
        if (idx < 0 || idx >= types.Length) throw new Exception($"Type index '{idx}' is outside of saved types bounds");
        return types[idx];
    }
    private static ValueType GetValueType(XElement elem)
    {
        Enum.TryParse(typeof(ValueType), elem.Attribute("Value")?.Value, out object result);
        if (result != null) return (ValueType)result;
        else return ValueType.None;
    }
    private static object ParseValue(Type[] types, XElement elem)
    {
        var type = GetValueType(elem);

        if (elem.IsEmpty) return null;
        if (type == ValueType.Enum)
        {
            var tType = GetTypeAttr(types, elem);
            if (tType == null || !tType.IsSubclassOf(typeof(Enum))) return null;
            if (Enum.TryParse(tType, elem.Value, out object result)) return result;
            else return null;
        }
        if (type == ValueType.Collection)
        {
            var tType = GetTypeAttr(types, elem);
            var tInt = tType.GetInterfaces().FirstOrDefault(i => i.IsGenericType && i.GetGenericTypeDefinition() == typeof(IEnumerable<>));
            var gArg = tInt.GetGenericArguments()[0];
            if (tType == null || !tType.GetInterfaces().Any(i => i.IsGenericType && i.GetGenericTypeDefinition() == typeof(IEnumerable<>))) return null;

            object result = null;

            if (result == null)
            {
                var ctor = tType.GetConstructors(BindingFlags.Public | BindingFlags.Instance).FirstOrDefault(c =>
                {
                    var param = c.GetParameters();
                    return param.Count() == 1 && param.Any(p => p.ParameterType.IsGenericType && p.ParameterType.GetGenericTypeDefinition() == typeof(IEnumerable<>));
                });
                if (ctor != null)
                {
                    var elements = elem.Elements().Select(x => ParseValue(types, x));
                    var castElems = typeof(Enumerable).GetMethod("Cast").MakeGenericMethod(gArg).Invoke(elements, new object[] { elements });
                    result = ctor.Invoke(new object[] { castElems });
                }
            }

            if (result == null)
            {
                var ctor = tType.GetConstructors(BindingFlags.Public | BindingFlags.Instance).FirstOrDefault(c => c.GetParameters().Count() == 0);
                var addMethod = tType.GetMethods(BindingFlags.Instance | BindingFlags.Public).FirstOrDefault(m =>
                {
                    if (m.Name != "Add") return false;
                    var param = m.GetParameters();
                    return param.Count() == 1 && param[0].ParameterType == gArg;
                });
                if (ctor != null && addMethod != null)
                {
                    var elements = elem.Elements().Select(x => ParseValue(types, x));
                    result = ctor.Invoke(null);
                    foreach (var el in elements) addMethod.Invoke(result, new object[] { el });
                }
            }

            if (result == null)
            {
                var ctor = tType.GetConstructors(BindingFlags.Public | BindingFlags.Instance).FirstOrDefault();
                var setMethod = tType.GetMethods(BindingFlags.Instance | BindingFlags.Public).FirstOrDefault(m =>
                {
                    if (m.Name != "Set") return false;
                    var param = m.GetParameters();
                    return param.Count() == 2 && param[0].ParameterType == typeof(int) && param[1].ParameterType == gArg;
                });
                if (ctor != null || setMethod != null)
                {
                    var elements = elem.Elements().Select(x => ParseValue(types, x));
                    result = ctor.Invoke(new object[] { elements.Count() });
                    int i = 0;
                    foreach (var el in elements)
                    {
                        setMethod.Invoke(result, new object[] { i, el });
                        i++;
                    }
                }
            }

            return result;
        }
        else if (type == ValueType.Text) return elem.Value;
        else if (type == ValueType.Integer)
        {
            int.TryParse(elem.Value, out var num);
            return num;
        }
        else if (type == ValueType.Decimal)
        {
            float.TryParse(elem.Value, out var num);
            return num;
        }
        else if (type == ValueType.Boolean)
        {
            bool.TryParse(elem.Value, out var boolean);
            return boolean;
        }
        else if (type == ValueType.Object)
        {
            var tType = GetTypeAttr(types, elem);
            if (tType == null) return null;

            IEnumerable<FieldInfo> fields = tType.GetFields(BindingFlags.Instance | BindingFlags.Public)
                .Concat(tType.GetFields(BindingFlags.Instance | BindingFlags.NonPublic));
            IEnumerable<PropertyInfo> properties = tType.GetProperties(BindingFlags.Instance | BindingFlags.Public).Where(p => p.GetSetMethod() != null)
                .Concat(tType.GetProperties(BindingFlags.Instance | BindingFlags.NonPublic).Where(p => p.GetSetMethod() != null));

            object result = null;
            var ctor = tType.GetConstructors(BindingFlags.Public | BindingFlags.Instance).FirstOrDefault(c => c.GetParameters().Count() == 0);
            if (ctor == null)
            {
                if (!tType.IsValueType) return null;
                result = Activator.CreateInstance(tType);
            }
            else result = ctor.Invoke(null);

            foreach (var el in elem.Elements())
            {
                var value = ParseValue(types, el);

                var field = fields.FirstOrDefault(f => f.Name == el.Name.LocalName);
                if (field != null) field.SetValue(result, value);
                var property = properties.FirstOrDefault(p => p.Name == el.Name.LocalName);
                if (property != null) property.SetValue(result, value);
            }
            return result;
        }
        else return elem.Value;

    }

    private static void AddTypeAttr(List<Type> types, Type type, XElement elem)
    {
        if (!types.Contains(type)) types.Add(type);
        elem.SetAttributeValue("Type", types.IndexOf(type));
    }

    private static XElement ParseObject(List<Type> types, string name, object value)
    {
        XElement result = new XElement(name);

        if (value != null)
        {
            var tType = value.GetType();

            if (tType.IsEnum)
            {
                result.SetAttributeValue("Value", ValueType.Enum);
                AddTypeAttr(types, tType, result);

                result.Value = Enum.GetName(tType, value) ?? "";
            }
            else if (value is string str)
            {
                result.SetAttributeValue("Value", ValueType.Text);
                result.Value = str;
            }
            else if (value is int integer)
            {
                result.SetAttributeValue("Value", ValueType.Integer);
                result.Value = integer.ToString();
            }
            else if (value is float || value is double)
            {
                result.SetAttributeValue("Value", ValueType.Decimal);
                result.Value = value.ToString();
            }
            else if (value is bool boolean)
            {
                result.SetAttributeValue("Value", ValueType.Boolean);
                result.Value = boolean.ToString();
            }
            else if (tType.GetInterfaces().Any(i => i.IsGenericType && i.GetGenericTypeDefinition() == typeof(IEnumerable<>)))
            {
                result.SetAttributeValue("Value", ValueType.Collection);
                AddTypeAttr(types, tType, result);

                var enumerator = (IEnumerator)tType.GetMethod("GetEnumerator").Invoke(value, null);
                while (enumerator.MoveNext())
                {
                    var elVal = ParseObject(types, "Item", enumerator.Current);
                    result.Add(elVal);
                }
            }
            else if (tType.IsClass || tType.IsValueType)
            {
                result.SetAttributeValue("Value", ValueType.Object);
                AddTypeAttr(types, tType, result);

                IEnumerable<FieldInfo> fields = tType.GetFields(BindingFlags.Instance | BindingFlags.Public)
                    .Concat(tType.GetFields(BindingFlags.Instance | BindingFlags.NonPublic));
                IEnumerable<PropertyInfo> properties = tType.GetProperties(BindingFlags.Instance | BindingFlags.Public).Where(p => p.GetSetMethod() != null)
                    .Concat(tType.GetProperties(BindingFlags.Instance | BindingFlags.NonPublic).Where(p => p.GetSetMethod() != null));

                foreach (var field in fields) result.Add(ParseObject(types, field.Name, field.GetValue(value)));
                foreach (var property in properties) result.Add(ParseObject(types, property.Name, property.GetValue(value)));
            }
            else
            {
                result.SetAttributeValue("Value", ValueType.None);
                result.Value = value.ToString();
            }
        }

        return result;
    }


    public static T Load<T>(FileStream file)
    {
        var doc = XDocument.Load(file);

        var rootElems = doc.Root.Elements().ToArray();
        var types = rootElems[0];
        var elem = rootElems[1];

        var dict = ParseValue(LoadDocTypes(types), elem);
        if (dict.GetType() == typeof(T)) return (T)dict;
        else throw new Exception($"Loaded configuration is not of the type '{typeof(T).Name}'");
    }

    public static void Save(FileStream file, object obj)
    {
        var types = new List<Type>();
        var elem = ParseObject(types, "Root", obj);
        var root = new XElement("Configuration", new XElement("Types", SaveDocTypes(types)), elem);

        var doc = new XDocument(root);
        doc.Save(file);
    }

    public static T Load<T>(string path)
    {
        using (var file = LuaCsFile.OpenRead(path)) return Load<T>(file);
    }

    public static void Save(string path, object obj)
    {
        using (var file = LuaCsFile.OpenWrite(path)) Save(file, obj);
    }
}