TechbloxModdingAPI/TechbloxModdingAPI/Client/Editor/Engines/PlacementEngine.cs

using System.Reflection;

using DataLoader;
using Gamecraft.Blocks.BlockGroups;
using Gamecraft.Wires;
using HarmonyLib;
using RobocraftX.Blocks;
using RobocraftX.Character;
using RobocraftX.Common;
using RobocraftX.CR.MachineEditing.BoxSelect;
using RobocraftX.Rendering;
using RobocraftX.Rendering.GPUI;
using Svelto.ECS;
using Svelto.ECS.EntityStructs;
using TechbloxModdingAPI.Common.Engines;
using Unity.Mathematics;
using TechbloxModdingAPI.Utility;
using TechbloxModdingAPI.Utility.ECS;

namespace TechbloxModdingAPI.Blocks.Engines
{
    /// <summary>
    /// Engine which executes block placement actions
    /// </summary>
    public class PlacementEngine : IApiEngine
    {
        public bool IsInGame;

        public void Dispose()
        {
            IsInGame = false;
        }

        public void Ready()
        {
            IsInGame = true;
        }

        public EntitiesDB entitiesDB { get; set; }
        private static BlockEntityFactory _blockEntityFactory; //Injected from PlaceSingleBlockEngine
        private static IEntityFactory _entityFactory;

        public EntityInitializer PlaceBlock(BlockIDs block, float3 position, Player player, bool autoWire)
        {
            return BuildBlock((ushort) block, position, autoWire, (player ?? Player.LocalPlayer).Id);
        }

        private EntityInitializer BuildBlock(ushort block, float3 position, bool autoWire, uint playerId)
        {
            if (_blockEntityFactory == null)
                throw new BlockException("The factory is null.");
            if(!FullGameFields._dataDb.ContainsKey<CubeListData>(block))
                throw new BlockException("Block with ID " + block + " not found!");
            //RobocraftX.CR.MachineEditing.PlaceSingleBlockEngine

            var structInitializer = _blockEntityFactory.Build(CommonExclusiveGroups.blockIDGeneratorClient.Next(), block); //The ghost block index is only used for triggers
            
            // Use the default steel material for the block until it's changed and set up the graphics for it
            uint prefabAssetID = structInitializer.Has<PrefabAssetIDComponent>()
                ? structInitializer.Get<PrefabAssetIDComponent>().prefabAssetID
                : throw new BlockException("Prefab asset ID not found!"); //Set by the game
            uint prefabId = PrefabsID.GetOrAddPrefabID((ushort) prefabAssetID, (byte) BlockMaterial.SteelBodywork, 1, false);
            structInitializer.Init(new GFXPrefabEntityStructGPUI(prefabId));
            structInitializer.Get<CubeMaterialStruct>().materialId = (byte) BlockMaterial.SteelBodywork;
            
            // The DBID is the block's type (cube, servo, seat etc.)
            structInitializer.Init(new DBEntityStruct {DBID = block});
            structInitializer.Init(new PositionEntityStruct {position = position});
            structInitializer.Init(new RotationEntityStruct {rotation = quaternion.identity});
            structInitializer.Init(new ScalingEntityStruct {scale = new float3(1, 1, 1)});
            structInitializer.Init(new GridRotationStruct
            {
                position = position,
                rotation = quaternion.identity
            });
            structInitializer.Init(new UniformBlockScaleEntityStruct {scaleFactor = 1});
            var bssesopt = entitiesDB.QueryEntityOptional<BoxSelectStateEntityStruct>(new EGID(playerId,
                BoxSelectExclusiveGroups.BoxSelectVolumeExclusiveGroup));
            if (!bssesopt)
                throw new BlockException("Invalid player ID specified for block placement");
            structInitializer.Init(new BlockPlacementInfoStruct
            {
                loadedFromDisk = false,
                placedByBuildingDrone = bssesopt.Get().buildingDroneReference,
                triggerAutoWiring = autoWire && structInitializer.Has<BlockPortsStruct>()
            });
            
            int nextFilterId = BlockGroupUtility.NextFilterId;
            structInitializer.Init(new BlockGroupEntityComponent
            {
                currentBlockGroup = nextFilterId
            });
            _entityFactory.BuildEntity<BlockGroupEntityDescriptor>((uint) nextFilterId,
                    BlockGroupExclusiveGroups.BlockGroupEntityGroup)
                .Init(new BlockGroupTransformEntityComponent
                {
                    blockGroupGridRotation = quaternion.identity,
                    blockGroupGridPosition = position
                });

            foreach (var group in CharacterExclusiveGroups.AllCharacters)
            {
                EGID playerEGID = new EGID(playerId, group);
                if (!entitiesDB.TryQueryEntitiesAndIndex<PickedBlockExtraDataStruct>(playerEGID, out uint index,
                    out var array)) continue;
                ref PickedBlockExtraDataStruct pickedBlock = ref array[index];
                pickedBlock.placedBlockEntityID = structInitializer.EGID;
                pickedBlock.placedBlockWasAPickedBlock = false;
            }
            return structInitializer;
        }

        [HarmonyPatch]
        class FactoryObtainerPatch
        {
            static void Postfix(BlockEntityFactory blockEntityFactory, IEntityFactory entityFactory)
            {
                _blockEntityFactory = blockEntityFactory;
                _entityFactory = entityFactory;
                Logging.MetaDebugLog("Block entity factory injected.");
            }

            static MethodBase TargetMethod(Harmony instance)
            {
                return AccessTools.TypeByName("RobocraftX.CR.MachineEditing.PlaceSingleBlockEngine").GetConstructors()[0];
            }
        }
    }
}