tbd-station-14/Content.Server/Commands/AttachBodyPartCommand.cs

#nullable enable
using Content.Server.GameObjects.Components.Body.Part;
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
{
    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);
        }
    }
}