rjw-quirks/RJW-Quirks/Modules/Shared/Selectors/Pawn/HasTrait.cs

27 lines
618 B
C#

using RimWorld;
using System.Collections.Generic;
using Verse;
namespace rjwquirks.Modules.Shared.PawnSelectors
{
public class HasTrait : PawnSelector
{
public TraitDef trait;
public override bool PawnSatisfies(Pawn pawn) => pawn.story?.traits?.HasTrait(trait) == true;
public override IEnumerable<string> ConfigErrors()
{
foreach (string error in base.ConfigErrors())
{
yield return error;
}
if (trait == null)
{
yield return "<trait> is empty";
}
}
}
}