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 RobocraftX.StateSync;
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 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++;
                        }
                    }
                }
            }
            Debug.Log($"Added {engineCount} custom command engines");
        }

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

        private delegate void ComposeAction(UnityContext<FullGameCompositionRoot> c, EnginesRoot r, World w, Action a,
            MultiplayerInitParameters mip, ref StateSyncRegistrationHelper ssrh); //Ref params can't be used with Actions
        private static MethodInfo _ComposeMethodInfo(ComposeAction a)
        {
            return a.Method;
        }
    }
}