e8c8cae31e
* fix: don't retroactively drain disabled batteries If something that used PowerCellDraw temporarily disabled said draw, once it became re-enabled the system would play catch-up trying to drain the battery for all the time since the component was disabled. * fixup! fix: don't retroactively drain disabled batteries
49 lines
1.7 KiB
C#
49 lines
1.7 KiB
C#
using Content.Shared.Item.ItemToggle;
|
|
using Content.Shared.Item.ItemToggle.Components;
|
|
using Content.Shared.PowerCell.Components;
|
|
|
|
namespace Content.Shared.PowerCell;
|
|
|
|
/// <summary>
|
|
/// Handles events to integrate PowerCellDraw with ItemToggle
|
|
/// </summary>
|
|
public sealed class ToggleCellDrawSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly ItemToggleSystem _toggle = default!;
|
|
[Dependency] private readonly SharedPowerCellSystem _cell = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<ToggleCellDrawComponent, MapInitEvent>(OnMapInit);
|
|
SubscribeLocalEvent<ToggleCellDrawComponent, ItemToggleActivateAttemptEvent>(OnActivateAttempt);
|
|
SubscribeLocalEvent<ToggleCellDrawComponent, ItemToggledEvent>(OnToggled);
|
|
SubscribeLocalEvent<ToggleCellDrawComponent, PowerCellSlotEmptyEvent>(OnEmpty);
|
|
}
|
|
|
|
private void OnMapInit(Entity<ToggleCellDrawComponent> ent, ref MapInitEvent args)
|
|
{
|
|
_cell.SetDrawEnabled(ent.Owner, _toggle.IsActivated(ent.Owner));
|
|
}
|
|
|
|
private void OnActivateAttempt(Entity<ToggleCellDrawComponent> ent, ref ItemToggleActivateAttemptEvent args)
|
|
{
|
|
if (!_cell.HasDrawCharge(ent, user: args.User)
|
|
|| !_cell.HasActivatableCharge(ent, user: args.User))
|
|
args.Cancelled = true;
|
|
}
|
|
|
|
private void OnToggled(Entity<ToggleCellDrawComponent> ent, ref ItemToggledEvent args)
|
|
{
|
|
var uid = ent.Owner;
|
|
var draw = Comp<PowerCellDrawComponent>(uid);
|
|
_cell.SetDrawEnabled((uid, draw), args.Activated);
|
|
}
|
|
|
|
private void OnEmpty(Entity<ToggleCellDrawComponent> ent, ref PowerCellSlotEmptyEvent args)
|
|
{
|
|
_toggle.TryDeactivate(ent.Owner);
|
|
}
|
|
}
|