Make energy swords use RgbLightController (#7344)

Content.Client/Toggleable/ToggleableLightVisualsSystem.cs

@@ -0,0 +1,113 @@
+using Content.Client.Clothing;
+using Content.Client.Items.Systems;
+using Content.Shared.Clothing;
+using Content.Shared.Hands;
+using Content.Shared.Item;
+using Content.Shared.Toggleable;
+using Robust.Client.GameObjects;
+using Robust.Shared.Utility;
+using System.Linq;
+
+namespace Content.Client.Toggleable;
+
+public sealed class ToggleableLightVisualsSystem : VisualizerSystem<ToggleableLightVisualsComponent>
+{
+    [Dependency] private readonly SharedItemSystem _itemSys = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+        SubscribeLocalEvent<ToggleableLightVisualsComponent, GetInhandVisualsEvent>(OnGetHeldVisuals, after: new[] { typeof(ItemSystem) });
+        SubscribeLocalEvent<ToggleableLightVisualsComponent, GetEquipmentVisualsEvent>(OnGetEquipmentVisuals, after: new[] { typeof(ClothingSystem) });
+    }
+
+    protected override void OnAppearanceChange(EntityUid uid, ToggleableLightVisualsComponent component, ref AppearanceChangeEvent args)
+    {
+        if (!args.Component.TryGetData(ToggleableLightVisuals.Enabled, out bool enabled))
+            return;
+
+        var modulate = args.Component.TryGetData(ToggleableLightVisuals.Color, out Color color);
+
+        // Update the item's sprite
+        if (TryComp(uid, out SpriteComponent? sprite) && sprite.LayerMapTryGet(component.SpriteLayer, out var layer))
+        {
+            sprite.LayerSetVisible(layer, enabled);
+            if (modulate)
+                sprite.LayerSetColor(layer, color);
+        }
+
+        // Update any point-lights
+        if (TryComp(uid, out PointLightComponent? light))
+        {
+            DebugTools.Assert(!light.NetSyncEnabled, "light visualizers require point lights without net-sync");
+            light.Enabled = enabled;
+            if (enabled && modulate)
+                light.Color = color;
+        }
+
+        // update clothing & in-hand visuals.
+        _itemSys.VisualsChanged(uid);
+    }
+
+    /// <summary>
+    ///     Add the unshaded light overlays to any clothing sprites.
+    /// </summary>
+    private void OnGetEquipmentVisuals(EntityUid uid, ToggleableLightVisualsComponent component, GetEquipmentVisualsEvent args)
+    {
+        if (!TryComp(uid, out AppearanceComponent? appearance)
+            || !appearance.TryGetData(ToggleableLightVisuals.Enabled, out bool enabled)
+            || !enabled)
+            return;
+
+        if (!component.ClothingVisuals.TryGetValue(args.Slot, out var layers))
+            return;
+
+        var modulate = appearance.TryGetData(ToggleableLightVisuals.Color, out Color color);
+
+        var i = 0;
+        foreach (var layer in layers)
+        {
+            var key = layer.MapKeys?.FirstOrDefault();
+            if (key == null)
+            {
+                key = i == 0 ? $"{args.Slot}-toggle" : $"{args.Slot}-toggle-{i}";
+                i++;
+            }
+
+            if (modulate)
+                layer.Color = color;
+
+            args.Layers.Add((key, layer));
+        }
+    }
+
+    private void OnGetHeldVisuals(EntityUid uid, ToggleableLightVisualsComponent component, GetInhandVisualsEvent args)
+    {
+        if (!TryComp(uid, out AppearanceComponent? appearance)
+            || !appearance.TryGetData(ToggleableLightVisuals.Enabled, out bool enabled)
+            || !enabled)
+            return;
+
+        if (!component.InhandVisuals.TryGetValue(args.Location, out var layers))
+            return;
+
+        var modulate = appearance.TryGetData(ToggleableLightVisuals.Color, out Color color);
+
+        var i = 0;
+        var defaultKey = $"inhand-{args.Location.ToString().ToLowerInvariant()}-toggle";
+        foreach (var layer in layers)
+        {
+            var key = layer.MapKeys?.FirstOrDefault();
+            if (key == null)
+            {
+                key = i == 0 ? defaultKey : $"{defaultKey}-{i}";
+                i++;
+            }
+
+            if (modulate)
+                layer.Color = color;
+
+            args.Layers.Add((key, layer));
+        }
+    }
+}