Entity whitelist/blacklist for traits (#12348)

Content.Server/Traits/TraitSystem.cs

@@ -1,5 +1,6 @@
 using Content.Server.GameTicking;
 using Content.Shared.Traits;
+using Content.Shared.Whitelist;
 using Robust.Shared.Prototypes;
 using Robust.Shared.Serialization.Manager;
 
@@ -28,6 +29,12 @@ public sealed class TraitSystem : EntitySystem
                 return;
             }
 
+            if (traitPrototype.Whitelist != null && !traitPrototype.Whitelist.IsValid(args.Mob))
+                continue;
+
+            if (traitPrototype.Blacklist != null && traitPrototype.Blacklist.IsValid(args.Mob))
+                continue;
+
             // Add all components required by the prototype
             foreach (var entry in traitPrototype.Components.Values)
             {

Content.Shared/Traits/TraitPrototype.cs

@@ -1,3 +1,4 @@
+using Content.Shared.Whitelist;
 using Robust.Shared.Prototypes;
 using static Robust.Shared.Prototypes.EntityPrototype; // don't worry about it
 
@@ -25,6 +26,18 @@ namespace Content.Shared.Traits
         [DataField("description")]
         public string? Description { get; }
 
+        /// <summary>
+        ///     Don't apply this trait to entities this whitelist IS NOT valid for.
+        /// </summary>
+        [DataField("whitelist")]
+        public EntityWhitelist? Whitelist;
+
+        /// <summary>
+        ///     Don't apply this trait to entities this whitelist IS valid for. (hence, a blacklist)
+        /// </summary>
+        [DataField("blacklist")]
+        public EntityWhitelist? Blacklist;
+
         /// <summary>
         ///     The components that get added to the player, when they pick this trait.
         /// </summary>

Resources/Prototypes/Traits/disabilities.yml

@@ -2,6 +2,9 @@
   id: Blindness
   name: Blindness
   description: You lack vision
+  whitelist:
+    components:
+      - Blindable
   components:
     - type: PermanentBlindness
 

Resources/Prototypes/Traits/inconveniences.yml

@@ -2,6 +2,9 @@
   id: UncontrollableSneezing
   name: Runny nose
   description: You sneeze and cough uncontrollably
+  whitelist:
+    components:
+      - DiseaseCarrier
   components:
     - type: UncontrollableSnough
       snoughSound: