107 lines
3.9 KiB
C#
107 lines
3.9 KiB
C#
#nullable enable
|
|
using Content.Server.Administration;
|
|
using Content.Server.GameObjects.Components.Body.Part;
|
|
using Content.Shared.Administration;
|
|
using Content.Shared.GameObjects.Components.Body;
|
|
using Content.Shared.GameObjects.Components.Body.Part;
|
|
using Robust.Server.Interfaces.Console;
|
|
using Robust.Server.Interfaces.Player;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.Interfaces.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
|
|
namespace Content.Server.Commands
|
|
{
|
|
[AdminCommand(AdminFlags.Fun)]
|
|
public class AttachBodyPartCommand : IClientCommand
|
|
{
|
|
public string Command => "attachbodypart";
|
|
public string Description => "Attaches a body part to you or someone else.";
|
|
public string Help => $"{Command} <partEntityUid> / {Command} <entityUid> <partEntityUid>";
|
|
|
|
public void Execute(IConsoleShell shell, IPlayerSession? player, string[] args)
|
|
{
|
|
var entityManager = IoCManager.Resolve<IEntityManager>();
|
|
|
|
IEntity entity;
|
|
EntityUid partUid;
|
|
|
|
switch (args.Length)
|
|
{
|
|
case 1:
|
|
if (player == null)
|
|
{
|
|
shell.SendText(player, $"You need to specify an entity to attach the part to if you aren't a player.\n{Help}");
|
|
return;
|
|
}
|
|
|
|
if (player.AttachedEntity == null)
|
|
{
|
|
shell.SendText(player, $"You need to specify an entity to attach the part to if you aren't attached to an entity.\n{Help}");
|
|
return;
|
|
}
|
|
|
|
if (!EntityUid.TryParse(args[0], out partUid))
|
|
{
|
|
shell.SendText(player, $"{args[0]} is not a valid entity uid.");
|
|
return;
|
|
}
|
|
|
|
entity = player.AttachedEntity;
|
|
|
|
break;
|
|
case 2:
|
|
if (!EntityUid.TryParse(args[0], out var entityUid))
|
|
{
|
|
shell.SendText(player, $"{args[0]} is not a valid entity uid.");
|
|
return;
|
|
}
|
|
|
|
if (!EntityUid.TryParse(args[1], out partUid))
|
|
{
|
|
shell.SendText(player, $"{args[1]} is not a valid entity uid.");
|
|
return;
|
|
}
|
|
|
|
if (!entityManager.TryGetEntity(entityUid, out var tempEntity))
|
|
{
|
|
shell.SendText(player, $"{entityUid} is not a valid entity.");
|
|
return;
|
|
}
|
|
|
|
entity = tempEntity;
|
|
break;
|
|
default:
|
|
shell.SendText(player, Help);
|
|
return;
|
|
}
|
|
|
|
if (!entity.TryGetComponent(out IBody? body))
|
|
{
|
|
shell.SendText(player, $"Entity {entity.Name} with uid {entity.Uid} does not have a {nameof(IBody)} component.");
|
|
return;
|
|
}
|
|
|
|
if (!entityManager.TryGetEntity(partUid, out var partEntity))
|
|
{
|
|
shell.SendText(player, $"{partUid} is not a valid entity.");
|
|
return;
|
|
}
|
|
|
|
if (!partEntity.TryGetComponent(out IBodyPart? part))
|
|
{
|
|
shell.SendText(player, $"Entity {partEntity.Name} with uid {args[0]} does not have a {nameof(IBodyPart)} component.");
|
|
return;
|
|
}
|
|
|
|
if (body.HasPart(part))
|
|
{
|
|
shell.SendText(player, $"Body part {partEntity.Name} with uid {partEntity.Uid} is already attached to entity {entity.Name} with uid {entity.Uid}");
|
|
return;
|
|
}
|
|
|
|
body.TryAddPart($"{nameof(BodyPartComponent.AttachBodyPartVerb)}-{partEntity.Uid}", part, true);
|
|
}
|
|
}
|
|
}
|