Re-organize all projects (#4166)

Content.Shared/Whitelist/EntityWhitelist.cs

@@ -0,0 +1,85 @@
+using System.Collections.Generic;
+using Content.Shared.Tag;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+using Robust.Shared.Log;
+using Robust.Shared.Serialization;
+using Robust.Shared.Serialization.Manager.Attributes;
+
+namespace Content.Shared.Whitelist
+{
+    /// <summary>
+    ///     Used to determine whether an entity fits a certain whitelist.
+    ///     Does not whitelist by prototypes, since that is undesirable; you're better off just adding a tag to all
+    ///     entity prototypes that need to be whitelisted, and checking for that.
+    /// </summary>
+    /// <code>
+    /// whitelist:
+    ///   tags:
+    ///     - Cigarette
+    ///     - FirelockElectronics
+    ///   components:
+    ///     - Buckle
+    ///     - AsteroidRock
+    /// </code>
+    [DataDefinition]
+    public class EntityWhitelist : ISerializationHooks
+    {
+        /// <summary>
+        ///     Component names that are allowed in the whitelist.
+        /// </summary>
+        [DataField("components")] public string[]? Components = null;
+
+        private List<IComponentRegistration>? _registrations = null;
+
+        /// <summary>
+        ///     Tags that are allowed in the whitelist.
+        /// </summary>
+        [DataField("tags")] public string[]? Tags = null;
+
+        void ISerializationHooks.AfterDeserialization()
+        {
+            UpdateRegistrations();
+        }
+
+        public void UpdateRegistrations()
+        {
+            if (Components == null) return;
+
+            var compfact = IoCManager.Resolve<IComponentFactory>();
+            _registrations = new List<IComponentRegistration>();
+            foreach (var name in Components)
+            {
+                if (!compfact.TryGetRegistration(name, out var registration))
+                {
+                    Logger.Warning($"Invalid component name {name} passed to EntityWhitelist!");
+                    continue;
+                }
+
+                _registrations.Add(registration);
+            }
+        }
+
+        /// <summary>
+        ///     Returns whether a given entity fits the whitelist.
+        /// </summary>
+        public bool IsValid(IEntity entity)
+        {
+            if (Tags != null)
+            {
+                if (entity.HasAnyTag(Tags))
+                        return true;
+            }
+
+            if (_registrations != null)
+            {
+                foreach (var reg in _registrations)
+                {
+                    if (entity.TryGetComponent(reg.Type, out _))
+                        return true;
+                }
+            }
+            return false;
+        }
+    }
+}