76 lines
3.1 KiB
C#
76 lines
3.1 KiB
C#
using Content.Shared.Damage;
|
|
using Content.Shared.Electrocution;
|
|
using Content.Shared.Explosion;
|
|
using Content.Shared.IdentityManagement.Components;
|
|
using Content.Shared.Movement.Systems;
|
|
using Content.Shared.Radio;
|
|
using Content.Shared.Slippery;
|
|
using Content.Shared.Strip.Components;
|
|
using Content.Shared.Temperature;
|
|
|
|
namespace Content.Shared.Inventory;
|
|
|
|
public partial class InventorySystem
|
|
{
|
|
public void InitializeRelay()
|
|
{
|
|
SubscribeLocalEvent<InventoryComponent, DamageModifyEvent>(RelayInventoryEvent);
|
|
SubscribeLocalEvent<InventoryComponent, ElectrocutionAttemptEvent>(RelayInventoryEvent);
|
|
SubscribeLocalEvent<InventoryComponent, SlipAttemptEvent>(RelayInventoryEvent);
|
|
SubscribeLocalEvent<InventoryComponent, RefreshMovementSpeedModifiersEvent>(RelayInventoryEvent);
|
|
SubscribeLocalEvent<InventoryComponent, GetExplosionResistanceEvent>(RelayInventoryEvent);
|
|
SubscribeLocalEvent<InventoryComponent, BeforeStripEvent>(RelayInventoryEvent);
|
|
SubscribeLocalEvent<InventoryComponent, SeeIdentityAttemptEvent>(RelayInventoryEvent);
|
|
SubscribeLocalEvent<InventoryComponent, ModifyChangedTemperatureEvent>(RelayInventoryEvent);
|
|
SubscribeLocalEvent<InventoryComponent, GetDefaultRadioChannelEvent>(RelayInventoryEvent);
|
|
}
|
|
|
|
protected void RelayInventoryEvent<T>(EntityUid uid, InventoryComponent component, T args) where T : EntityEventArgs, IInventoryRelayEvent
|
|
{
|
|
if (args.TargetSlots == SlotFlags.NONE)
|
|
return;
|
|
|
|
var containerEnumerator = new ContainerSlotEnumerator(uid, component.TemplateId, _prototypeManager, this, args.TargetSlots);
|
|
var ev = new InventoryRelayedEvent<T>(args);
|
|
while (containerEnumerator.MoveNext(out var container))
|
|
{
|
|
if (!container.ContainedEntity.HasValue) continue;
|
|
RaiseLocalEvent(container.ContainedEntity.Value, ev, false);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Event wrapper for relayed events.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// This avoids nested inventory relays, and makes it easy to have certain events only handled by the initial
|
|
/// target entity. E.g. health based movement speed modifiers should not be handled by a hat, even if that hat
|
|
/// happens to be a dead mouse. Clothing that wishes to modify movement speed must subscribe to
|
|
/// InventoryRelayedEvent<RefreshMovementSpeedModifiersEvent>
|
|
/// </remarks>
|
|
public sealed class InventoryRelayedEvent<TEvent> : EntityEventArgs where TEvent : EntityEventArgs, IInventoryRelayEvent
|
|
{
|
|
public readonly TEvent Args;
|
|
|
|
public InventoryRelayedEvent(TEvent args)
|
|
{
|
|
Args = args;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Events that should be relayed to inventory slots should implement this interface.
|
|
/// </summary>
|
|
public interface IInventoryRelayEvent
|
|
{
|
|
/// <summary>
|
|
/// What inventory slots should this event be relayed to, if any?
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// In general you may want to exclude <see cref="SlotFlags.POCKET"/>, given that those items are not truly
|
|
/// "equipped" by the user.
|
|
/// </remarks>
|
|
public SlotFlags TargetSlots { get; }
|
|
}
|