CatStation/Content.Client/Light/Controls/LightReplacerStatusControl.cs

using Content.Client.Message;
using Content.Client.Stylesheets;
using Content.Shared.Light.Components;
using Robust.Client.UserInterface;
using Robust.Client.UserInterface.Controls;
using Robust.Shared.Prototypes;
using Robust.Shared.Timing;

namespace Content.Client.Light.Controls;

/// <summary>
/// Handles the label on the light replacer
/// </summary>
public sealed partial class LightReplacerStatusControl : Control
{
    [Dependency] private IPrototypeManager _prototype = default!;
    [Dependency] private IEntityManager _entity = default!;

    private readonly Entity<LightReplacerComponent> _parent;
    private readonly RichTextLabel _label;

    private EntProtoId? _prevActiveLightTube;
    private EntProtoId? _prevActiveLightBulb;
    private string? _labelTube;
    private string? _labelBulb;

    public LightReplacerStatusControl(Entity<LightReplacerComponent> parent)
    {
        IoCManager.InjectDependencies(this);

        _parent = parent;
        _label = new RichTextLabel { StyleClasses = { StyleClass.ItemStatus } };
        AddChild(_label);
    }

    protected override void FrameUpdate(FrameEventArgs args)
    {
        base.FrameUpdate(args);

        // only updates the UI if any of the details are different than they previously were
        if (_prevActiveLightTube == _parent.Comp.ActiveLightTube
            && _prevActiveLightBulb == _parent.Comp.ActiveLightBulb)
            return;

        _prevActiveLightTube = _parent.Comp.ActiveLightTube;
        _prevActiveLightBulb = _parent.Comp.ActiveLightBulb;

        if (!_prototype.Resolve(_prevActiveLightTube, out var tube)
            || !tube.Components.TryGetComponent<LightBulbComponent>(_entity.ComponentFactory, out var lightTube)
            || !_prototype.Resolve(_prevActiveLightBulb, out var bulb)
            || !bulb.Components.TryGetComponent<LightBulbComponent>(_entity.ComponentFactory, out var lightBulb))
            return;

        _labelTube = lightTube.ShortName is { } tubeName ? Loc.GetString(tubeName) : tube.Name;
        _labelBulb = lightBulb.ShortName is { } bulbName ? Loc.GetString(bulbName) : bulb.Name;

        // Update current active lights
        _label.SetMarkup(Loc.GetString("comp-light-replacer-label",
            ("tube", _labelTube),
            ("bulb", _labelBulb)));
    }
}