36 lines
1.1 KiB
C#
36 lines
1.1 KiB
C#
using Content.Server.Power.EntitySystems;
|
|
using Content.Shared.PowerCell;
|
|
using Content.Shared.Prying.Components;
|
|
|
|
namespace Content.Server._DV.Tools;
|
|
|
|
public sealed class PryingRequiresPowerSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly PowerCellSystem _cell = default!;
|
|
[Dependency] private readonly BatterySystem _battery = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<PryingRequiresPowerComponent, BeforePryEvent>(OnBeforePry);
|
|
SubscribeLocalEvent<PryingRequiresPowerComponent, PriedEvent>(OnPried);
|
|
}
|
|
|
|
private void OnPried(Entity<PryingRequiresPowerComponent> ent, ref PriedEvent args)
|
|
{
|
|
// Entity has a PowerCellSlot, try that first
|
|
if (_cell.TryUseCharge(ent.Owner, ent.Comp.PowerCost))
|
|
return;
|
|
|
|
// The entity itself is a battery
|
|
_battery.TryUseCharge(ent.Owner, ent.Comp.PowerCost);
|
|
}
|
|
|
|
private void OnBeforePry(Entity<PryingRequiresPowerComponent> ent, ref BeforePryEvent args)
|
|
{
|
|
if (!_cell.HasCharge(ent.Owner, ent.Comp.PowerCost, args.User))
|
|
args.Cancelled = true;
|
|
}
|
|
}
|