TechbloxModdingAPI/TechbloxModdingAPI/Utility/MenuEngineManager.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

using Svelto.ECS;
using TechbloxModdingAPI.Engines;

namespace TechbloxModdingAPI.Utility
{
    /// <summary>
    /// Keeps track of custom menu-modifying engines
    /// </summary>
    public static class MenuEngineManager
    {
        private static Dictionary<string, IApiEngine> _menuEngines = new Dictionary<string, IApiEngine>();

		private static EnginesRoot _lastEngineRoot;

        // menu engine management
        public static void AddMenuEngine(IApiEngine engine)
        {
            _menuEngines[engine.Name] = engine;
			if (_lastEngineRoot != null)
            {
				Logging.MetaDebugLog($"Registering Menu IApiEngine {engine.Name}");
                _lastEngineRoot.AddEngine(engine);
				if (engine is IFactoryEngine factoryEngine)
                    factoryEngine.Factory = _lastEngineRoot.GenerateEntityFactory();
                if (engine is IFunEngine funEngine)
                    funEngine.Functions = _lastEngineRoot.GenerateEntityFunctions();
            }
        }

        public static bool ExistsMenuEngine(string name)
        {
            return _menuEngines.ContainsKey(name);
        }

        public static bool ExistsMenuEngine(IApiEngine engine)
        {
            return ExistsMenuEngine(engine.Name);
        }

        public static IApiEngine GetMenuEngine(string name)
        {
            return _menuEngines[name];
        }

        public static string[] GetMenuEngineNames()
        {
            return _menuEngines.Keys.ToArray();
        }

        public static void RemoveMenuEngine(string name)
        {
			if (_menuEngines[name].isRemovable)
			{
				_menuEngines.Remove(name);
			}
        }

        public static void RegisterEngines(EnginesRoot enginesRoot)
        {
			_lastEngineRoot = enginesRoot;
			IEntityFactory factory = enginesRoot.GenerateEntityFactory();
            IEntityFunctions functions = enginesRoot.GenerateEntityFunctions();
            foreach (var key in _menuEngines.Keys)
            {
                Logging.MetaDebugLog($"Registering Menu IApiEngine {_menuEngines[key].Name}");
                enginesRoot.AddEngine(_menuEngines[key]);
				if (_menuEngines[key] is IFactoryEngine factEngine) factEngine.Factory = factory;
                if(_menuEngines[key] is IFunEngine funEngine) funEngine.Functions = functions;
            }
        }
    }
}