TechbloxModdingAPI/TechbloxModdingAPI/Common/EcsObjectBaseEngine.cs

using System;
using HarmonyLib;
using Svelto.ECS;
using Svelto.ECS.Hybrid;
using TechbloxModdingAPI.Blocks.Engines;
using TechbloxModdingAPI.Common.Engines;
using TechbloxModdingAPI.Utility;
using TechbloxModdingAPI.Utility.ECS;

namespace TechbloxModdingAPI.Common;

public class EcsObjectBaseEngine : IApiEngine
{
    public void Ready()
    {
    }

    public EntitiesDB entitiesDB { get; set; }

    public void Dispose()
    {
    }

    public EntityReference GetEntityReference(EGID egid)
    {
        return entitiesDB is not null && egid != default ? egid.ToEntityReference(entitiesDB) : EntityReference.Invalid;
    }

    public EGID GetEgid(EntityReference reference)
    {
        return entitiesDB is not null && reference.ToEGID(entitiesDB, out var egid) ? egid : default;
    }

    public OptionalRef<T> GetComponentOptional<T>(EcsObjectBase obj) where T : unmanaged, IEntityComponent
    {
        return entitiesDB.QueryEntityOptional<T>(obj);
    }

    public ref T GetComponent<T>(EcsObjectBase obj) where T : unmanaged, IEntityComponent
    {
#if DEBUG
        if (entitiesDB.Exists<T>(obj.Id) && obj.InitData.Valid)
            throw new ArgumentException("The block exists but the init data has not been removed!");
#endif
        return ref entitiesDB.QueryEntityOrDefault<T>(obj);
    }

    public ref T GetViewComponent<T>(EcsObjectBase obj) where T : struct, IEntityViewComponent
    {
        return ref entitiesDB.QueryEntityOrDefault<T>(obj);
    }

    public object GetComponent(EcsObjectBase obj, Type type, string name)
    {
        var opt = AccessTools.Method(typeof(NativeApiExtensions), "QueryEntityOptional",
                new[] { typeof(EntitiesDB), typeof(EcsObjectBase), typeof(ExclusiveGroupStruct) }, new[] { type })
            .Invoke(null, new object[] { entitiesDB, obj, null });
        var str = AccessTools.Property(opt.GetType(), "Value").GetValue(opt);
        return AccessTools.Field(str.GetType(), name).GetValue(str);
    }

    public void SetComponent(EcsObjectBase obj, Type type, string name, object value)
    {
        var opt = AccessTools.Method(typeof(BlockEngine), "GetBlockInfoOptional", generics: new[] { type })
            .Invoke(this, new object[] { obj });
        var prop = AccessTools.Property(opt.GetType(), "Value");
        var str = prop.GetValue(opt);
        AccessTools.Field(str.GetType(), name).SetValue(str, value);
        prop.SetValue(opt, str);
    }
}