Suit Storage Whitelist

SlamBamActionman · Jun 22, 2024 · b1f5035 · b1f5035
1 parent 7252a92
commit b1f5035
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/Content.Shared/Armor/AllowSuitStorageComponent.cs b/Content.Shared/Armor/AllowSuitStorageComponent.cs
@@ -1,3 +1,5 @@
+using Content.Shared.Whitelist;
+
 namespace Content.Shared.Armor;
 
 /// <summary>
@@ -7,4 +9,13 @@ namespace Content.Shared.Armor;
 public sealed partial class AllowSuitStorageComponent : Component
 {
 
+    [DataField]
+    public EntityWhitelist Whitelist = new()
+    {
+        Components = new[]
+        {
+            "Item"
+        }
+    };
+
 }
diff --git a/Content.Shared/Inventory/InventorySystem.Equip.cs b/Content.Shared/Inventory/InventorySystem.Equip.cs
@@ -248,9 +248,17 @@ public bool CanEquip(EntityUid actor, EntityUid target, EntityUid itemUid, strin
                 return false;
 
             if (slotDefinition.DependsOnComponents is { } componentRegistry)
+            {
                 foreach (var (_, entry) in componentRegistry)
+                {
                     if (!HasComp(slotEntity, entry.Component.GetType()))
                         return false;
+
+                    if (TryComp<AllowSuitStorageComponent>(slotEntity, out var comp) &&
+                        !_whitelistSystem.IsWhitelistPass(comp.Whitelist, itemUid))
+                        return false;
+                }
+            }
         }
 
         var fittingInPocket = slotDefinition.SlotFlags.HasFlag(SlotFlags.POCKET) &&