87 lines
2.8 KiB
C#
87 lines
2.8 KiB
C#
#nullable enable
|
|
using Content.Server.Observer;
|
|
using Content.Shared.GameObjects.Components.Body;
|
|
using Content.Shared.GameObjects.Components.Body.Part;
|
|
using Content.Shared.GameObjects.Components.Damage;
|
|
using Content.Shared.GameObjects.Components.Movement;
|
|
using Robust.Server.GameObjects.Components.Container;
|
|
using Robust.Server.Interfaces.Player;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.Log;
|
|
using Robust.Shared.Players;
|
|
|
|
namespace Content.Server.GameObjects.Components.Body
|
|
{
|
|
[RegisterComponent]
|
|
[ComponentReference(typeof(SharedBodyComponent))]
|
|
[ComponentReference(typeof(IBody))]
|
|
public class BodyComponent : SharedBodyComponent, IRelayMoveInput
|
|
{
|
|
private Container _partContainer = default!;
|
|
|
|
protected override bool CanAddPart(string slot, IBodyPart part)
|
|
{
|
|
return base.CanAddPart(slot, part) &&
|
|
_partContainer.CanInsert(part.Owner);
|
|
}
|
|
|
|
protected override void OnAddPart(string slot, IBodyPart part)
|
|
{
|
|
base.OnAddPart(slot, part);
|
|
|
|
_partContainer.Insert(part.Owner);
|
|
}
|
|
|
|
protected override void OnRemovePart(string slot, IBodyPart part)
|
|
{
|
|
base.OnRemovePart(slot, part);
|
|
|
|
_partContainer.ForceRemove(part.Owner);
|
|
}
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
_partContainer = ContainerManagerComponent.Ensure<Container>($"{Name}-{nameof(BodyComponent)}", Owner);
|
|
|
|
foreach (var (slot, partId) in PartIds)
|
|
{
|
|
// Using MapPosition instead of Coordinates here prevents
|
|
// a crash within the character preview menu in the lobby
|
|
var entity = Owner.EntityManager.SpawnEntity(partId, Owner.Transform.MapPosition);
|
|
|
|
if (!entity.TryGetComponent(out IBodyPart? part))
|
|
{
|
|
Logger.Error($"Entity {partId} does not have a {nameof(IBodyPart)} component.");
|
|
continue;
|
|
}
|
|
|
|
TryAddPart(slot, part, true);
|
|
}
|
|
}
|
|
|
|
protected override void Startup()
|
|
{
|
|
base.Startup();
|
|
|
|
// This is ran in Startup as entities spawned in Initialize
|
|
// are not synced to the client since they are assumed to be
|
|
// identical on it
|
|
foreach (var part in Parts.Values)
|
|
{
|
|
part.Dirty();
|
|
}
|
|
}
|
|
|
|
void IRelayMoveInput.MoveInputPressed(ICommonSession session)
|
|
{
|
|
if (Owner.TryGetComponent(out IDamageableComponent? damageable) &&
|
|
damageable.CurrentState == DamageState.Dead)
|
|
{
|
|
new Ghost().Execute(null, (IPlayerSession) session, null);
|
|
}
|
|
}
|
|
}
|
|
}
|