79 lines
2.4 KiB
C#
79 lines
2.4 KiB
C#
using System.Linq;
|
|
using Content.Shared.EntityEffects;
|
|
using Content.Shared.Nutrition.Components;
|
|
using Content.Shared.Nutrition.EntitySystems;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Localization;
|
|
using Content.Shared.Body.Part;
|
|
using Content.Shared.Body.Systems;
|
|
|
|
namespace Content.Shared._DV.EntityEffects.EffectConditions;
|
|
|
|
/// <summary>
|
|
/// Reagent effect condition that depends on if the entity has a given component(s), potentially on a body part
|
|
/// </summary>
|
|
public sealed partial class HasComponent : EntityEffectCondition
|
|
{
|
|
/// <summary>
|
|
/// The set of components that this condition cares about
|
|
/// </summary>
|
|
[DataField(required: true)]
|
|
public ComponentRegistry Components;
|
|
|
|
/// <summary>
|
|
/// Whether or not the given components should be present
|
|
/// </summary>
|
|
[DataField]
|
|
public bool ShouldHave = true;
|
|
|
|
/// <summary>
|
|
/// Whether the check is an existential or universal check
|
|
/// </summary>
|
|
[DataField]
|
|
public bool ConsiderAll;
|
|
|
|
/// <summary>
|
|
/// The explanation displayed in the guidebook for this condition
|
|
/// </summary>
|
|
[DataField(required: true)]
|
|
public LocId Explanation;
|
|
|
|
/// <summary>
|
|
/// The body part of the entity to test for the components
|
|
/// </summary>
|
|
[DataField]
|
|
public BodyPartType? BodyPart;
|
|
|
|
/// <summary>
|
|
/// The side of the entity's body to test for the components
|
|
/// </summary>
|
|
[DataField]
|
|
public BodyPartSymmetry? BodyPartSymmetry;
|
|
|
|
public override bool Condition(EntityEffectBaseArgs args)
|
|
{
|
|
var _body = args.EntityManager.System<SharedBodySystem>();
|
|
var entity =
|
|
BodyPart is {} bodyPart
|
|
? _body.GetBodyChildrenOfType(args.TargetEntity, bodyPart, symmetry: BodyPartSymmetry).Select(it => it.Id).FirstOrDefault()
|
|
: args.TargetEntity;
|
|
|
|
if (!entity.IsValid())
|
|
{
|
|
return !ShouldHave;
|
|
}
|
|
|
|
var tested =
|
|
ConsiderAll
|
|
? Components.Values.All(c => args.EntityManager.HasComponent(entity, c.Component.GetType()))
|
|
: Components.Values.Any(c => args.EntityManager.HasComponent(entity, c.Component.GetType()));
|
|
|
|
return tested ^ !ShouldHave;
|
|
}
|
|
|
|
public override string GuidebookExplanation(IPrototypeManager prototype)
|
|
{
|
|
return Loc.GetString(Explanation);
|
|
}
|
|
}
|