42 lines
1.4 KiB
C#
42 lines
1.4 KiB
C#
using Content.Server.Power.Components;
|
|
using Content.Server.Power.EntitySystems;
|
|
using Content.Server.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)
|
|
{
|
|
// Tool is a battery use its power
|
|
if (_battery.TryUseCharge(ent, ent.Comp.PowerCost))
|
|
return;
|
|
|
|
// Tool has a battery in a power cell slot use that power
|
|
if (_cell.TryGetBatteryFromSlot(ent, out var batteryEnt, out var batteryComp)
|
|
&& batteryEnt is { } batteryUid)
|
|
{
|
|
_battery.TryUseCharge(batteryUid, ent.Comp.PowerCost, batteryComp);
|
|
return;
|
|
}
|
|
}
|
|
|
|
private void OnBeforePry(Entity<PryingRequiresPowerComponent> ent, ref BeforePryEvent args)
|
|
{
|
|
if (!_cell.HasCharge(ent, ent.Comp.PowerCost, null, args.User))
|
|
args.Cancelled = true;
|
|
}
|
|
}
|