98 lines
2.9 KiB
C#
98 lines
2.9 KiB
C#
using Content.Server.Bed.Sleep;
|
|
using Content.Server.Popups;
|
|
using Content.Server.Revenant.Components;
|
|
using Content.Shared.Bed.Sleep;
|
|
using Content.Shared.Mobs;
|
|
using Content.Shared.Mobs.Systems;
|
|
using Robust.Shared.Random;
|
|
|
|
namespace Content.Server.Revenant.EntitySystems;
|
|
|
|
public sealed class BlightSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly SleepingSystem _sleeping = default!;
|
|
[Dependency] private readonly PopupSystem _popup = default!;
|
|
[Dependency] private readonly MobStateSystem _mobState = default!;
|
|
[Dependency] private readonly IRobustRandom _random = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<BlightComponent, MobStateChangedEvent>(OnMobStateChanged);
|
|
SubscribeLocalEvent<BlightComponent, ComponentStartup>(OnStartup);
|
|
SubscribeLocalEvent<BlightComponent, ComponentShutdown>(OnShutdown);
|
|
}
|
|
|
|
private void OnShutdown(Entity<BlightComponent> ent, ref ComponentShutdown args)
|
|
{
|
|
if (!Deleted(ent) && !EntityManager.IsQueuedForDeletion(ent) && _mobState.IsAlive(ent))
|
|
_popup.PopupEntity("Вы вновь чувствуете себя здоровым.", ent, ent);
|
|
}
|
|
|
|
private void OnStartup(Entity<BlightComponent> ent, ref ComponentStartup args)
|
|
{
|
|
SetDelay(ent.Comp);
|
|
SetDuration(ent.Comp);
|
|
}
|
|
|
|
private void OnMobStateChanged(Entity<BlightComponent> ent, ref MobStateChangedEvent args)
|
|
{
|
|
RemCompDeferred<BlightComponent>(ent);
|
|
}
|
|
|
|
public override void Update(float frameTime)
|
|
{
|
|
base.Update(frameTime);
|
|
|
|
var query = EntityQueryEnumerator<BlightComponent>();
|
|
var sleepingQuery = GetEntityQuery<SleepingComponent>();
|
|
|
|
while (query.MoveNext(out var ent, out var blight))
|
|
{
|
|
blight.Duration += frameTime;
|
|
|
|
if (blight.Duration >= blight.MaxDuration.TotalSeconds)
|
|
{
|
|
RemCompDeferred<BlightComponent>(ent);
|
|
continue;
|
|
}
|
|
|
|
if (sleepingQuery.HasComponent(ent))
|
|
{
|
|
if (blight.BedSleep)
|
|
{
|
|
blight.SleepDelay += frameTime;
|
|
if (blight.SleepDelay >= blight.SleepingCureTime.TotalSeconds)
|
|
RemCompDeferred<BlightComponent>(ent);
|
|
}
|
|
|
|
continue;
|
|
}
|
|
|
|
blight.BedSleep = false;
|
|
blight.SleepDelay = 0f;
|
|
|
|
blight.Delay += frameTime;
|
|
|
|
if (blight.Delay < blight.MaxDelay.TotalSeconds)
|
|
continue;
|
|
|
|
_sleeping.TrySleeping(ent);
|
|
|
|
blight.Delay = 0f;
|
|
SetDelay(blight);
|
|
}
|
|
}
|
|
|
|
private void SetDuration(BlightComponent comp)
|
|
{
|
|
comp.MaxDuration = TimeSpan.FromSeconds(_random.Next(300, 420));
|
|
}
|
|
|
|
private void SetDelay(BlightComponent comp)
|
|
{
|
|
comp.MaxDelay = TimeSpan.FromSeconds(_random.Next(20, 50));
|
|
}
|
|
}
|