CatStation/Content.Shared/Disposal/Unit/SharedDisposalUnitSystem.VisualsAndUi.cs
āda 86e633be6e
Disposal Units tag their contents + UI comp state (#45613)
* commit

* also flipped

* oops

* fuck it added to the UI

* fix micro annoyance

* Unify components also doesn't work yet

* reuse the existing UI also clean up the UI also applying state headache

* cheat code

* flippy

* More interface changes

* yaml ;p

* private tagger

* betterer tagger

* private unit window

* limit tagger window

* access blocked

* typo

* validate

---------

Co-authored-by: iaada <iaada@users.noreply.github.com>
2026-09-22 08:55:54 +00:00

79 lines
3 KiB
C#

using Content.Shared.Construction.EntitySystems;
using Content.Shared.Database;
using Content.Shared.Disposal.Components;
namespace Content.Shared.Disposal.Unit;
public abstract partial class SharedDisposalUnitSystem
{
private void OnUiButtonPressed(Entity<DisposalUnitComponent> ent, ref DisposalUnitUiButtonPressedMessage args)
{
if (args.Actor is not { Valid: true } player)
return;
switch (args.Button)
{
case DisposalUnitUiButton.Eject:
EjectContents(ent);
_adminLog.Add(LogType.Action, LogImpact.Low, $"{ToPrettyString(player):player} hit eject button on {ToPrettyString(ent)}");
break;
case DisposalUnitUiButton.Engage:
ToggleEngage(ent);
_adminLog.Add(LogType.Action, LogImpact.Low, $"{ToPrettyString(player):player} hit flush button on {ToPrettyString(ent)}, it's now {(ent.Comp.Engaged ? "on" : "off")}");
break;
case DisposalUnitUiButton.Power:
_power.TogglePower(ent.Owner, user: args.Actor);
break;
default:
throw new ArgumentOutOfRangeException($"{ToPrettyString(player):player} attempted to hit a nonexistant button on {ToPrettyString(ent)}");
}
}
/// <summary>
/// Returns the estimated time when the disposal unit will be back to full pressure.
/// </summary>
/// <param name="ent">The disposal unit.</param>
/// <returns>The estimated time.</returns>
public TimeSpan EstimatedFullPressure(Entity<DisposalUnitComponent> ent)
{
if (ent.Comp.NextPressurized < _timing.CurTime)
return TimeSpan.Zero;
return ent.Comp.NextPressurized;
}
protected void UpdateUI(Entity<DisposalUnitComponent> entity)
{
if (_ui.TryGetOpenUi(entity.Owner, DisposalUnitUiKey.Key, out var bui))
{
bui.Update();
}
}
/// <summary>
/// Updates the appearance data of a disposal unit.
/// </summary>
/// <param name="ent">The disposal unit.</param>
protected void UpdateVisualState(Entity<DisposalUnitComponent> ent)
{
if (!TryComp(ent, out AppearanceComponent? appearance))
return;
var isAnchored = Transform(ent).Anchored;
_appearance.SetData(ent, AnchorVisuals.Anchored, isAnchored, appearance);
if (!isAnchored)
return;
var state = GetState(ent);
_appearance.SetData(ent, DisposalUnitVisuals.IsReady, state == DisposalsPressureState.Ready, appearance);
_appearance.SetData(ent, DisposalUnitVisuals.IsFlushing, state == DisposalsPressureState.Flushed, appearance);
_appearance.SetData(ent, DisposalUnitVisuals.IsEngaged, ent.Comp.Engaged, appearance);
if (!_power.IsPowered(ent.Owner))
return;
_appearance.SetData(ent, DisposalUnitVisuals.IsFull, GetContainedEntityCount(ent) > 0, appearance);
_appearance.SetData(ent, DisposalUnitVisuals.IsCharging, state != DisposalsPressureState.Ready, appearance);
}
}