TechbloxModdingAPI/TechbloxModdingAPI/Blocks/Engines/BlockCloneEngine.cs

using System;
using System.Collections.Generic;
using System.Reflection;
using Gamecraft.Wires;
using HarmonyLib;
using RobocraftX.Blocks;
using RobocraftX.Character;
using Svelto.DataStructures;
using Svelto.ECS;
using TechbloxModdingAPI.Common.Engines;

namespace TechbloxModdingAPI.Blocks.Engines
{
    public class BlockCloneEngine : IApiEngine
    {
        private static Type copyEngineType =
            AccessTools.TypeByName("Gamecraft.GUI.Tweaks.Engines.CopyTweaksOnPickEngine");
        private static Type copyWireEngineType =
            AccessTools.TypeByName("Gamecraft.Wires.WireConnectionCopyOnPickEngine");
        private static Type createWireEngineType =
            AccessTools.TypeByName("RobocraftX.GUI.Wires.WireConnectionCreateOnPlaceEngine");

        private MethodBase copyFromBlock = AccessTools.Method(copyEngineType, "CopyTweaksFromBlock");
        private MethodBase copyToBlock = AccessTools.Method(copyEngineType, "ApplyTweaksToPlacedBlock");
        private MethodBase copyWireFromBlock = AccessTools.Method(copyWireEngineType, "CopyWireInputsAndOutputs");
        private MethodBase copyWireToBlock = AccessTools.Method(createWireEngineType, "PlaceWiresOnPlaceNewCube");

        public void Ready()
        {
        }

        public EntitiesDB entitiesDB { get; set; }

        public void Dispose()
        {
        }

        public void CopyBlockStats(EGID sourceID, EGID targetID)
        {
            var allCharacters = (LocalFasterReadOnlyList<ExclusiveGroupStruct>) CharacterExclusiveGroups.AllCharacters;
            foreach (var ((pickedBlockColl, count), _) in entitiesDB.QueryEntities<PickedBlockExtraDataStruct>(allCharacters))
            {
                for (int i = 0; i < count; ++i)
                {
                    ref PickedBlockExtraDataStruct pickedBlock = ref pickedBlockColl[i];
                    var oldStruct = pickedBlock;
                    pickedBlock.pickedBlockEntityID = sourceID;
                    pickedBlock.placedBlockEntityID = targetID;
                    pickedBlock.placedBlockTweaksMustCopy = true;
                    if (entitiesDB.Exists<BlockTagEntityStruct>(pickedBlock.pickedBlockEntityID)
                        && entitiesDB.Exists<BlockTagEntityStruct>(pickedBlock.placedBlockEntityID))
                    {
                        copyFromBlock.Invoke(Patch.copyEngine, new object[] {pickedBlock.ID, pickedBlock});

                        uint playerID = Player.LocalPlayer.Id;
                        var parameters = new object[] {playerID, pickedBlock};
                        copyWireFromBlock.Invoke(Patch.copyWireEngine, parameters);
                        pickedBlock = (PickedBlockExtraDataStruct) parameters[1]; //ref arg
                        
                        copyToBlock.Invoke(Patch.copyEngine, new object[] {pickedBlock.ID, pickedBlock});

                        ExclusiveGroupStruct group = BuildModeWiresGroups.WIRES_COPY_GROUP + playerID;
                        copyWireToBlock.Invoke(Patch.createWireEngine, new object[] {group, pickedBlock.ID});
                        
                        pickedBlock.placedBlockTweaksMustCopy = false;
                    }

                    pickedBlock = oldStruct; //Make sure to not interfere with the game - Although that might not be the case with the wire copying
                }
            }
        }

        [HarmonyPatch]
        private static class Patch
        {
            public static object copyEngine;
            public static object copyWireEngine;
            public static object createWireEngine;

            public static void Postfix(object __instance)
            {
                if (__instance.GetType() == copyEngineType)
                    copyEngine = __instance;
                else if (__instance.GetType() == copyWireEngineType)
                    copyWireEngine = __instance;
                else if (__instance.GetType() == createWireEngineType)
                    createWireEngine = __instance;
            }

            public static IEnumerable<MethodBase> TargetMethods()
            {
                return new[]
                {
                    AccessTools.GetDeclaredConstructors(copyEngineType)[0],
                    AccessTools.GetDeclaredConstructors(copyWireEngineType)[0],
                    AccessTools.GetDeclaredConstructors(createWireEngineType)[0]
                };
            }
        }
    }
}