72 lines
2.7 KiB
C#
72 lines
2.7 KiB
C#
#nullable enable
|
|
using Content.Client.GameObjects.Components.Disposal;
|
|
using Content.Client.Interfaces.GameObjects.Components.Interaction;
|
|
using Content.Shared.GameObjects.Components.Body;
|
|
using Content.Shared.GameObjects.Components.Damage;
|
|
using Robust.Client.Interfaces.GameObjects.Components;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.Interfaces.GameObjects;
|
|
using Robust.Shared.Interfaces.Network;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Players;
|
|
|
|
namespace Content.Client.GameObjects.Components.Body
|
|
{
|
|
[RegisterComponent]
|
|
[ComponentReference(typeof(IDamageableComponent))]
|
|
[ComponentReference(typeof(IBodyManagerComponent))]
|
|
public class BodyManagerComponent : SharedBodyManagerComponent, IClientDraggable
|
|
{
|
|
#pragma warning disable 649
|
|
[Dependency] private readonly IEntityManager _entityManager = default!;
|
|
#pragma warning restore 649
|
|
|
|
public bool ClientCanDropOn(CanDropEventArgs eventArgs)
|
|
{
|
|
return eventArgs.Target.HasComponent<DisposalUnitComponent>();
|
|
}
|
|
|
|
public bool ClientCanDrag(CanDragEventArgs eventArgs)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
public override void HandleNetworkMessage(ComponentMessage message, INetChannel netChannel, ICommonSession? session = null)
|
|
{
|
|
if (!Owner.TryGetComponent(out ISpriteComponent? sprite))
|
|
{
|
|
return;
|
|
}
|
|
|
|
switch (message)
|
|
{
|
|
case BodyPartAddedMessage partAdded:
|
|
sprite.LayerSetVisible(partAdded.RSIMap, true);
|
|
sprite.LayerSetRSI(partAdded.RSIMap, partAdded.RSIPath);
|
|
sprite.LayerSetState(partAdded.RSIMap, partAdded.RSIState);
|
|
break;
|
|
case BodyPartRemovedMessage partRemoved:
|
|
sprite.LayerSetVisible(partRemoved.RSIMap, false);
|
|
|
|
if (!partRemoved.Dropped.HasValue ||
|
|
!_entityManager.TryGetEntity(partRemoved.Dropped.Value, out var entity) ||
|
|
!entity.TryGetComponent(out ISpriteComponent? droppedSprite))
|
|
{
|
|
break;
|
|
}
|
|
|
|
var color = sprite[partRemoved.RSIMap].Color;
|
|
|
|
droppedSprite.LayerSetColor(0, color);
|
|
break;
|
|
case MechanismSpriteAddedMessage mechanismAdded:
|
|
sprite.LayerSetVisible(mechanismAdded.RSIMap, true);
|
|
break;
|
|
case MechanismSpriteRemovedMessage mechanismRemoved:
|
|
sprite.LayerSetVisible(mechanismRemoved.RSIMap, false);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|