BeatSaber-IPA-Reloaded/IPA.Loader/Loader/Features/DefineFeature.cs

using IPA.Logging;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using System;
using System.IO;

namespace IPA.Loader.Features
{
    internal class DefineFeature : Feature
    {
        public static bool NewFeature = true;

        private class DataModel
        {
            [JsonProperty("type", Required = Required.Always)]
            public string TypeName;
            [JsonProperty("name", Required = Required.DisallowNull)]
            public string ActualName = null;

            public string Name => ActualName ?? TypeName;
        }

        private DataModel data;

        protected override bool Initialize(PluginMetadata meta, JObject featureData)
        {
            Logger.features.Debug("Executing DefineFeature Init");

            try
            {
                data = featureData.ToObject<DataModel>();
            }
            catch (Exception e)
            {
                InvalidMessage = $"Invalid data: {e}";
                return false;
            }

            InvalidMessage = $"Feature {data.Name} already exists";
            return PreregisterFeature(meta, data.Name);
        }

        public override void AfterInit(PluginMetadata meta, object pluginInstance)
        {
            Logger.features.Debug("Executing DefineFeature AfterInit");

            Type type;
            try
            {
                type = meta.Assembly.GetType(data.TypeName);
            }
            catch (ArgumentException)
            {
                Logger.features.Error($"Invalid type name {data.TypeName}");
                return;
            }
            catch (Exception e) when (e is FileNotFoundException || e is FileLoadException || e is BadImageFormatException)
            {
                var filename = "";

                switch (e)
                {
                    case FileNotFoundException fn:
                        filename = fn.FileName;
                        break;
                    case FileLoadException fl:
                        filename = fl.FileName;
                        break;
                    case BadImageFormatException bi:
                        filename = bi.FileName;
                        break;
                }

                Logger.features.Error($"Could not find {filename} while loading type");
                return;
            }

            if (type == null)
            {
                Logger.features.Error($"Invalid type name {data.TypeName}");
                return;
            }

            try
            {
                if (RegisterFeature(meta, data.Name, type))
                {
                    NewFeature = true;
                    return;
                }

                Logger.features.Error($"Feature with name {data.Name} already exists");
                return;
            }
            catch (ArgumentException)
            {
                Logger.features.Error($"{type.FullName} not a subclass of {nameof(Feature)}");
                return;
            }
        }
    }
}