40 lines
1.5 KiB
C#
40 lines
1.5 KiB
C#
#nullable enable
|
|
using System;
|
|
using JetBrains.Annotations;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Shared.Utility
|
|
{
|
|
[UsedImplicitly]
|
|
public static class EntityPrototypeHelpers
|
|
{
|
|
public static bool HasComponent<T>(this EntityPrototype prototype, IComponentFactory? componentFactory = null) where T : IComponent
|
|
{
|
|
return prototype.HasComponent(typeof(T), componentFactory);
|
|
}
|
|
|
|
public static bool HasComponent(this EntityPrototype prototype, Type component, IComponentFactory? componentFactory = null)
|
|
{
|
|
componentFactory ??= IoCManager.Resolve<IComponentFactory>();
|
|
|
|
var registration = componentFactory.GetRegistration(component);
|
|
|
|
return prototype.Components.ContainsKey(registration.Name);
|
|
}
|
|
|
|
public static bool HasComponent<T>(string prototype, IPrototypeManager? prototypeManager = null, IComponentFactory? componentFactory = null) where T : IComponent
|
|
{
|
|
return HasComponent(prototype, typeof(T), prototypeManager, componentFactory);
|
|
}
|
|
|
|
public static bool HasComponent(string prototype, Type component, IPrototypeManager? prototypeManager = null, IComponentFactory? componentFactory = null)
|
|
{
|
|
prototypeManager ??= IoCManager.Resolve<IPrototypeManager>();
|
|
|
|
return prototypeManager.TryIndex(prototype, out EntityPrototype? proto) && proto.HasComponent(component, componentFactory);
|
|
}
|
|
}
|
|
}
|