31 lines
930 B
C#
31 lines
930 B
C#
using Content.Server.Power.EntitySystems;
|
|
using Content.Server.Tesla.Components;
|
|
using Content.Server.Lightning;
|
|
using Content.Shared.Power.Components;
|
|
|
|
namespace Content.Server.Tesla.EntitySystems;
|
|
|
|
/// <summary>
|
|
/// Generates electricity from lightning bolts
|
|
/// </summary>
|
|
public sealed class TeslaCoilSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly BatterySystem _battery = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<TeslaCoilComponent, HitByLightningEvent>(OnHitByLightning);
|
|
}
|
|
|
|
//When struck by lightning, charge the internal battery
|
|
private void OnHitByLightning(Entity<TeslaCoilComponent> coil, ref HitByLightningEvent args)
|
|
{
|
|
if (TryComp<BatteryComponent>(coil, out var batteryComponent))
|
|
{
|
|
_battery.SetCharge(coil, batteryComponent.CurrentCharge + coil.Comp.ChargeFromLightning);
|
|
}
|
|
}
|
|
}
|