CatStation/Content.Client/PowerCell/PowerChargerVisualizerSystem.cs

using Content.Shared.Power.Components;
using Robust.Client.GameObjects;

namespace Content.Client.PowerCell;

public sealed partial class PowerChargerVisualizerSystem : VisualizerSystem<PowerChargerVisualsComponent>
{
    protected override void OnAppearanceChange(EntityUid uid, PowerChargerVisualsComponent comp, ref AppearanceChangeEvent args)
    {
        if (args.Sprite == null)
            return;

        // Update base item
        if (args.TryGetData<bool>(CellVisual.Occupied, out var occupied) && occupied)
        {
            // TODO: don't throw if it doesn't have a full state
            SpriteSystem.LayerSetRsiState((uid, args.Sprite), PowerChargerVisualLayers.Base, comp.OccupiedState);
        }
        else
        {
            SpriteSystem.LayerSetRsiState((uid, args.Sprite), PowerChargerVisualLayers.Base, comp.EmptyState);
        }

        // Update lighting
        if (args.TryGetData<CellChargerStatus>(CellVisual.Light, out var status)
            && comp.LightStates.TryGetValue(status, out var lightState))
        {
            SpriteSystem.LayerSetRsiState((uid, args.Sprite), PowerChargerVisualLayers.Light, lightState);
            SpriteSystem.LayerSetVisible((uid, args.Sprite), PowerChargerVisualLayers.Light, true);
        }
        else
            SpriteSystem.LayerSetVisible((uid, args.Sprite), PowerChargerVisualLayers.Light, false);
    }
}

public enum PowerChargerVisualLayers : byte
{
    Base,
    Light,
}