44 lines
1.6 KiB
C#
44 lines
1.6 KiB
C#
using Content.Shared.Hands.Components;
|
|
using Content.Shared.Inventory.Events;
|
|
|
|
namespace Content.Shared.Hands.EntitySystems;
|
|
|
|
public sealed class ExtraHandsEquipmentSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly SharedHandsSystem _hands = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<ExtraHandsEquipmentComponent, GotEquippedEvent>(OnEquipped);
|
|
SubscribeLocalEvent<ExtraHandsEquipmentComponent, GotUnequippedEvent>(OnUnequipped);
|
|
}
|
|
|
|
private void OnEquipped(Entity<ExtraHandsEquipmentComponent> ent, ref GotEquippedEvent args)
|
|
{
|
|
if (!TryComp<HandsComponent>(args.Equipee, out var handsComp))
|
|
return;
|
|
|
|
foreach (var (handName, hand) in ent.Comp.Hands)
|
|
{
|
|
// add the NetEntity id to the container name to prevent multiple items with this component from conflicting
|
|
var handId = $"{GetNetEntity(ent.Owner).Id}-{handName}";
|
|
_hands.AddHand((args.Equipee, handsComp), handId, hand.Location);
|
|
}
|
|
}
|
|
|
|
private void OnUnequipped(Entity<ExtraHandsEquipmentComponent> ent, ref GotUnequippedEvent args)
|
|
{
|
|
if (!TryComp<HandsComponent>(args.Equipee, out var handsComp))
|
|
return;
|
|
|
|
foreach (var handName in ent.Comp.Hands.Keys)
|
|
{
|
|
// add the NetEntity id to the container name to prevent multiple items with this component from conflicting
|
|
var handId = $"{GetNetEntity(ent.Owner).Id}-{handName}";
|
|
_hands.RemoveHand((args.Equipee, handsComp), handId);
|
|
}
|
|
}
|
|
}
|