59 lines
2.2 KiB
C#
59 lines
2.2 KiB
C#
using Content.Shared._DV.Abilities;
|
|
using Content.Shared.Damage;
|
|
using Content.Shared.Mobs.Components;
|
|
using Content.Shared.Mobs.Systems;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Timing;
|
|
|
|
namespace Content.Server._DV.Abilities.Chitinid;
|
|
|
|
public sealed partial class ChitinidSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly IGameTiming _timing = default!;
|
|
[Dependency] private readonly IPrototypeManager _proto = default!;
|
|
[Dependency] private readonly ItemCougherSystem _cougher = default!;
|
|
[Dependency] private readonly DamageableSystem _damageable = default!;
|
|
[Dependency] private readonly MobStateSystem _mobState = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
SubscribeLocalEvent<ChitinidComponent, MapInitEvent>(OnMapInit);
|
|
SubscribeLocalEvent<ChitinidComponent, ItemCoughedUpEvent>(OnItemCoughedUp);
|
|
}
|
|
|
|
public override void Update(float frameTime)
|
|
{
|
|
base.Update(frameTime);
|
|
var query = EntityQueryEnumerator<ChitinidComponent, DamageableComponent, MobStateComponent, ItemCougherComponent>();
|
|
while (query.MoveNext(out var uid, out var comp, out var damageable, out var mobState, out var itemCougher))
|
|
{
|
|
if (_timing.CurTime < comp.NextUpdate)
|
|
continue;
|
|
|
|
comp.NextUpdate += comp.UpdateInterval;
|
|
|
|
if (comp.AmountAbsorbed >= comp.MaximumAbsorbed || _mobState.IsDead(uid, mobState))
|
|
continue;
|
|
|
|
if (_damageable.TryChangeDamage(uid, comp.Healing, damageable: damageable) is not {} delta)
|
|
continue;
|
|
|
|
// damage healed is subtracted, so the delta is negative.
|
|
comp.AmountAbsorbed -= delta.GetTotal();
|
|
if (comp.AmountAbsorbed >= comp.MaximumAbsorbed)
|
|
_cougher.EnableAction((uid, itemCougher));
|
|
}
|
|
}
|
|
|
|
private void OnMapInit(Entity<ChitinidComponent> ent, ref MapInitEvent args)
|
|
{
|
|
ent.Comp.NextUpdate = _timing.CurTime + ent.Comp.UpdateInterval;
|
|
}
|
|
|
|
private void OnItemCoughedUp(Entity<ChitinidComponent> ent, ref ItemCoughedUpEvent args)
|
|
{
|
|
// start healing radiation again
|
|
ent.Comp.AmountAbsorbed = 0f;
|
|
}
|
|
}
|