41 lines
1.2 KiB
C#
41 lines
1.2 KiB
C#
using System.Threading;
|
|
using Content.Shared.GameTicking;
|
|
using Timer = Robust.Shared.Timing.Timer;
|
|
|
|
namespace Content.Server._Sunrise.Misc.TimedRemoveComponents;
|
|
|
|
public sealed class TimedRemoveComponentsSystem : EntitySystem
|
|
{
|
|
private static CancellationTokenSource _timerDespawnToken = new ();
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<TimedRemoveComponentsComponent, ComponentInit>(OnInit);
|
|
|
|
SubscribeLocalEvent<RoundRestartCleanupEvent>(_ => Clear());
|
|
}
|
|
|
|
private void OnInit(Entity<TimedRemoveComponentsComponent> ent, ref ComponentInit args)
|
|
{
|
|
Timer.Spawn(ent.Comp.RemoveAfter, () => RemoveComponents(ent), _timerDespawnToken.Token);
|
|
}
|
|
|
|
private void RemoveComponents(Entity<TimedRemoveComponentsComponent> ent)
|
|
{
|
|
if (TerminatingOrDeleted(ent))
|
|
return;
|
|
|
|
EntityManager.RemoveComponents(ent, ent.Comp.Components);
|
|
|
|
// блять, я себя захуярил
|
|
RemComp<TimedRemoveComponentsComponent>(ent);
|
|
}
|
|
|
|
private static void Clear()
|
|
{
|
|
_timerDespawnToken.Cancel();
|
|
_timerDespawnToken = new();
|
|
}
|
|
}
|