extracommands/extracommands/CommandLineCompositionRootSaveCommandPatch.cs

using System;
using System.Reflection;
using Harmony;
using UnityEngine;
using Unity.Entities;
using RobocraftX;
using RobocraftX.GUI.CommandLine;
using RobocraftX.Multiplayer;
using Svelto.ECS;
using Svelto.Context;

namespace ExtraCommands
{
    [HarmonyPatch]
    class CommandLineCompositionRootSaveCommandPatch
    {
        static void Postfix(UnityContext<FullGameCompositionRoot> contextHolder, EnginesRoot enginesRoot, World physicsWorld, Action reloadGame, MultiplayerInitParameters multiplayerParameters)
        {
            int commandCount = 0;
            int engineCount = 0;
            MethodInfo commandHelp = Harmony.AccessTools.Method(Harmony.AccessTools.TypeByName("RobocraftX.GUI.CommandLine.CommandLineUtility"), "SaveCommandHelp", new Type[] { typeof(string), typeof(string) });
            foreach (Type t in typeof(CustomCommandEngine).Assembly.GetTypes())
            {
                CustomCommandAttribute[] attributes = (CustomCommandAttribute[])t.GetCustomAttributes(typeof(CustomCommandAttribute), false);
                CustomCommandEngine inst = null;
                foreach (CustomCommandAttribute attr in attributes)
                {
                    if (attr != null && t.IsSubclassOf(typeof(CustomCommandEngine)))
                    {
                        if (inst == null)
                        {
                            // create instance by getting the constructor through reflection
                            inst = (CustomCommandEngine)t.GetConstructor(new Type[] { typeof(UnityContext<FullGameCompositionRoot>), typeof(EnginesRoot), typeof(World), typeof(Action), typeof(MultiplayerInitParameters) })
                                    .Invoke(new object[] { contextHolder, enginesRoot, physicsWorld, reloadGame, multiplayerParameters });
                            // add to engineRoot
                            enginesRoot.AddEngine(inst);
                            engineCount++;
                        }
                        // add to Gamecraft help command
                        commandHelp.Invoke(null, new string[] { attr.Name, attr.Description });
                        commandCount++;
                    }
                }
            }
            enginesRoot.AddEngine(new UnregisterCommandEngine(contextHolder, enginesRoot, physicsWorld, reloadGame, multiplayerParameters));
            Debug.Log($"Added {commandCount} custom commands in {engineCount} engines");
        }

        static MethodBase TargetMethod(HarmonyInstance instance)
        {
            return _ComposeMethodInfo(CommandLineCompositionRoot.Compose<CommandLineContext>);
        }

        private static MethodInfo _ComposeMethodInfo(Action<CommandLineContext, EnginesRoot, World, Action, MultiplayerInitParameters> a)
        {
            return a.Method;
        }
    }
}