Adds option to whitelist or blacklist store entries based on buyer objectives (#41493)

* BuyerObjectiveWhitelistCondition done

* fix to check every buyer objective for a blacklisted objective before checking whitelisted objectives

* resolved requested changes

* forgot a line

* Update Content.Server/Store/Conditions/BuyerObjectiveWhitelistCondition.cs

* Replaced IsBlacklistPass with IsWhitelistPass

* cleaning up

* placed whitelist == null check above objective loop

* Moved empty whitelist check below foreach loop

* final cleanup i pray

---------

Co-authored-by: slarticodefast <161409025+slarticodefast@users.noreply.github.com>

diff --git a/Content.Server/Store/Conditions/BuyerObjectiveWhitelistCondition.cs b/Content.Server/Store/Conditions/BuyerObjectiveWhitelistCondition.cs
new file mode 100644 (file)
index 0000000..b6e7496
--- /dev/null
+++ b/Content.Server/Store/Conditions/BuyerObjectiveWhitelistCondition.cs
@@ -0,0 +1,46 @@
+using Content.Shared.Mind;
+using Content.Shared.Store;
+using Content.Shared.Whitelist;
+
+namespace Content.Server.Store.Conditions;
+
+/// <summary>
+/// Filters out an entry based on the objectives of an entity.
+/// </summary>
+public sealed partial class BuyerObjectiveWhitelistCondition : ListingCondition
+{
+    /// <summary>
+    /// A whitelist of objective types.
+    /// If there is no whitelist, the object will appear by default.
+    /// </summary>
+    [DataField]
+    public EntityWhitelist? Whitelist;
+
+    /// <summary>
+    /// A blacklist of objective types.
+    /// If an objective is both whitelisted and blacklisted the blacklist will take priority and the entry will not appear.
+    /// </summary>
+    [DataField]
+    public EntityWhitelist? Blacklist;
+
+    public override bool Condition(ListingConditionArgs args)
+    {
+        var ent = args.EntityManager;
+        var whitelistSystem = ent.System<EntityWhitelistSystem>();
+
+        if (!args.EntityManager.TryGetComponent<MindComponent>(args.Buyer, out var mindComp))
+            return true; // inanimate objects don't have minds
+
+        var whitelisted = false;
+
+        foreach (var objective in mindComp.Objectives)
+        {
+            if (whitelistSystem.IsWhitelistPass(Blacklist, objective))
+                return false;
+            if (whitelistSystem.IsWhitelistPass(Whitelist, objective))
+                whitelisted = true;
+        }
+
+        return Whitelist == null || whitelisted;
+    }
+}