tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Defibrillator (#15922)

Browse Source 
Co-authored-by: metalgearsloth <comedian_vs_clown@hotmail.com>
This commit is contained in:
Nemanja
2023-05-02 20:10:19 -04:00
committed by GitHub
parent 32b29fb05a
commit 0604c93d50
38 changed files with 796 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
Content.Shared/Mobs/Systems/MobThresholdSystem.cs
View File

@@ -4,7 +4,7 @@ using Content.Shared.Alert;
using Content.Shared.Damage;
using Content.Shared.FixedPoint;
using Content.Shared.Mobs.Components;
using Robust.Shared.GameStates;
namespace Content.Shared.Mobs.Systems;
public sealed class MobThresholdSystem : EntitySystem
@@ -17,8 +17,6 @@ public sealed class MobThresholdSystem : EntitySystem
SubscribeLocalEvent<MobThresholdsComponent, ComponentShutdown>(MobThresholdShutdown);
SubscribeLocalEvent<MobThresholdsComponent, ComponentStartup>(MobThresholdStartup);
SubscribeLocalEvent<MobThresholdsComponent, DamageChangedEvent>(OnDamaged);
SubscribeLocalEvent<MobThresholdsComponent, ComponentGetState>(OnGetComponentState);
SubscribeLocalEvent<MobThresholdsComponent, ComponentHandleState>(OnHandleComponentState);
SubscribeLocalEvent<MobThresholdsComponent, UpdateMobStateEvent>(OnUpdateMobState);
}
@@ -249,6 +247,14 @@ public sealed class MobThresholdSystem : EntitySystem
UpdateAlerts(target, mobState.CurrentState, threshold, damageable);
}
public void SetAllowRevives(EntityUid uid, bool val, MobThresholdsComponent? component = null)
{
if (!Resolve(uid, ref component, false))
return;
component.AllowRevives = val;
Dirty(component);
}
#endregion
#region Private Implementation
@@ -278,7 +284,8 @@ public sealed class MobThresholdSystem : EntitySystem
return;
}
thresholds.CurrentThresholdState = newState;
if (mobState.CurrentState != MobState.Dead || thresholds.AllowRevives)
thresholds.CurrentThresholdState = newState;
_mobStateSystem.UpdateMobState(target, mobState);
Dirty(target);
@@ -334,32 +341,6 @@ public sealed class MobThresholdSystem : EntitySystem
UpdateAlerts(target, mobState.CurrentState, thresholds, args.Damageable);
}
private void OnHandleComponentState(EntityUid target, MobThresholdsComponent component,
ref ComponentHandleState args)
{
if (args.Current is not MobThresholdComponentState state)
return;
if (component.Thresholds.Count != state.Thresholds.Count ||
!component.Thresholds.SequenceEqual(state.Thresholds))
{
component.Thresholds.Clear();
foreach (var threshold in state.Thresholds)
{
component.Thresholds.Add(threshold.Key, threshold.Value);
}
}
component.CurrentThresholdState = state.CurrentThresholdState;
}
private void OnGetComponentState(EntityUid target, MobThresholdsComponent component, ref ComponentGetState args)
{
args.State = new MobThresholdComponentState(component.CurrentThresholdState,
new Dictionary<FixedPoint2, MobState>(component.Thresholds));
}
private void MobThresholdStartup(EntityUid target, MobThresholdsComponent thresholds, ComponentStartup args)
{
if (!TryComp<MobStateComponent>(target, out var mobState) || !TryComp<DamageableComponent>(target, out var damageable))
@@ -378,8 +359,14 @@ public sealed class MobThresholdSystem : EntitySystem
private void OnUpdateMobState(EntityUid target, MobThresholdsComponent component, ref UpdateMobStateEvent args)
{
if (component.CurrentThresholdState != MobState.Invalid)
if (!component.AllowRevives && component.CurrentThresholdState == MobState.Dead)
{
args.State = MobState.Dead;
}
else if (component.CurrentThresholdState != MobState.Invalid)
{
args.State = component.CurrentThresholdState;
}
}
#endregion
@@ -394,6 +381,4 @@ public sealed class MobThresholdSystem : EntitySystem
/// <param name="Damageable">Damageable Component owned by the Target</param>
[ByRefEvent]
public readonly record struct MobThresholdChecked(EntityUid Target, MobStateComponent MobState,
MobThresholdsComponent Threshold, DamageableComponent Damageable)
{
}
MobThresholdsComponent Threshold, DamageableComponent Damageable);