53 lines
1.7 KiB
C#
53 lines
1.7 KiB
C#
using System.Linq;
|
|
using Content.Server.Storage.Components;
|
|
using Content.Shared.Audio;
|
|
using Content.Shared.Storage.Components;
|
|
using Robust.Shared.Audio;
|
|
using Robust.Shared.Player;
|
|
using Robust.Shared.Random;
|
|
|
|
namespace Content.Server.Storage.EntitySystems;
|
|
|
|
public sealed class CursedEntityStorageSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly IRobustRandom _random = default!;
|
|
[Dependency] private readonly EntityStorageSystem _entityStorage = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<CursedEntityStorageComponent, StorageAfterCloseEvent>(OnClose);
|
|
}
|
|
|
|
private void OnClose(EntityUid uid, CursedEntityStorageComponent component, ref StorageAfterCloseEvent args)
|
|
{
|
|
if (!TryComp<EntityStorageComponent>(uid, out var storage))
|
|
return;
|
|
|
|
if (storage.Open || storage.Contents.ContainedEntities.Count <= 0)
|
|
return;
|
|
|
|
var lockers = new List<Entity<EntityStorageComponent>>();
|
|
var query = EntityQueryEnumerator<EntityStorageComponent>();
|
|
while (query.MoveNext(out var storageUid, out var storageComp))
|
|
{
|
|
lockers.Add((storageUid, storageComp));
|
|
}
|
|
|
|
lockers.RemoveAll(e => e.Owner == uid);
|
|
|
|
if (lockers.Count == 0)
|
|
return;
|
|
|
|
var lockerEnt = _random.Pick(lockers).Owner;
|
|
|
|
foreach (var entity in storage.Contents.ContainedEntities.ToArray())
|
|
{
|
|
storage.Contents.Remove(entity);
|
|
_entityStorage.AddToContents(entity, lockerEnt);
|
|
}
|
|
SoundSystem.Play(component.CursedSound.GetSound(), Filter.Pvs(uid), uid, AudioHelpers.WithVariation(0.125f, _random));
|
|
}
|
|
}
|