OP Ballades (#1372)

Content.Shared/_CP14/MagicSpell/Spells/CP14SpellArea.cs

@@ -0,0 +1,64 @@
+using Content.Shared.Whitelist;
+using Robust.Shared.Map;
+
+namespace Content.Shared._CP14.MagicSpell.Spells;
+
+public sealed partial class CP14SpellArea : CP14SpellEffect
+{
+    [DataField(required: true)]
+    public List<CP14SpellEffect> Effects { get; set; } = new();
+
+    [DataField]
+    public EntityWhitelist? Whitelist;
+
+    /// <summary>
+    /// How many entities can be subject to EntityEffect? Leave 0 to remove the restriction.
+    /// </summary>
+    [DataField]
+    public int MaxTargets = 0;
+
+    [DataField(required: true)]
+    public float Range = 1f;
+
+    [DataField]
+    public bool AffectCaster = false;
+
+    public override void Effect(EntityManager entManager, CP14SpellEffectBaseArgs args)
+    {
+        EntityCoordinates? targetPoint = null;
+
+        if (args.Target is not null &&
+            entManager.TryGetComponent<TransformComponent>(args.Target.Value, out var transformComponent))
+            targetPoint = transformComponent.Coordinates;
+        else if (args.Position is not null)
+            targetPoint = args.Position;
+
+        if (targetPoint is null)
+            return;
+
+        var lookup = entManager.System<EntityLookupSystem>();
+        var whitelist = entManager.System<EntityWhitelistSystem>();
+
+        var entitiesAround = lookup.GetEntitiesInRange(targetPoint.Value, Range, LookupFlags.Uncontained);
+
+        var count = 0;
+        foreach (var entity in entitiesAround)
+        {
+            if (entity == args.User && !AffectCaster)
+                continue;
+
+            if (Whitelist is not null && !whitelist.IsValid(Whitelist, entity))
+                continue;
+
+            foreach (var effect in Effects)
+            {
+                effect.Effect(entManager, new CP14SpellEffectBaseArgs(args.User, null, entity,  targetPoint));
+            }
+
+            count++;
+
+            if (MaxTargets > 0 && count >= MaxTargets)
+                break;
+        }
+    }
+}