decouple ItemToggle from PowerCellDraw (#31392)
* remove ItemToggle from PowerCellDraw query * add EntityQuery for resolves, make them all optional * move integration to ToggleCellDraw * add ToggleCellDraw to almost every PowerCellDraw prototype * :trollface: * :trollface: * :trollface: * let it disable on mapinit * set update time on mapinit, make borg power logic consistent now * :trollface: --------- Co-authored-by: deltanedas <@deltanedas:kde.org>
This commit is contained in:
49
Content.Shared/PowerCell/ToggleCellDrawSystem.cs
Normal file
49
Content.Shared/PowerCell/ToggleCellDrawSystem.cs
Normal file
@@ -0,0 +1,49 @@
|
||||
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.QueueUpdate((uid, draw));
|
||||
_cell.SetDrawEnabled((uid, draw), args.Activated);
|
||||
}
|
||||
|
||||
private void OnEmpty(Entity<ToggleCellDrawComponent> ent, ref PowerCellSlotEmptyEvent args)
|
||||
{
|
||||
_toggle.TryDeactivate(ent.Owner);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user