@@ -3,52 +3,41 @@ using Content.Shared.Administration;
|
||||
using Content.Shared.Body.Part;
|
||||
using Robust.Shared.Console;
|
||||
|
||||
namespace Content.Server.Administration.Commands
|
||||
{
|
||||
namespace Content.Server.Administration.Commands;
|
||||
|
||||
[AdminCommand(AdminFlags.Admin)]
|
||||
public sealed class AddBodyPartCommand : IConsoleCommand
|
||||
public sealed class AddBodyPartCommand : LocalizedEntityCommands
|
||||
{
|
||||
[Dependency] private readonly IEntityManager _entManager = default!;
|
||||
[Dependency] private readonly BodySystem _bodySystem = default!;
|
||||
|
||||
public string Command => "addbodypart";
|
||||
public string Description => "Adds a given entity to a containing body.";
|
||||
public string Help => "Usage: addbodypart <entity uid> <body uid> <part slot> <part type>";
|
||||
public override string Command => "addbodypart";
|
||||
|
||||
public void Execute(IConsoleShell shell, string argStr, string[] args)
|
||||
public override void Execute(IConsoleShell shell, string argStr, string[] args)
|
||||
{
|
||||
if (args.Length != 3)
|
||||
if (args.Length != 4)
|
||||
{
|
||||
shell.WriteError(Loc.GetString("shell-wrong-arguments-number"));
|
||||
return;
|
||||
}
|
||||
|
||||
if (!NetEntity.TryParse(args[0], out var childNetId))
|
||||
if (!NetEntity.TryParse(args[0], out var childNetId) || !EntityManager.TryGetEntity(childNetId, out var childId))
|
||||
{
|
||||
shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
|
||||
shell.WriteError(Loc.GetString("shell-invalid-entity-uid", ("uid", args[0])));
|
||||
return;
|
||||
}
|
||||
|
||||
if (!NetEntity.TryParse(args[1], out var parentNetId))
|
||||
if (!NetEntity.TryParse(args[1], out var parentNetId) || !EntityManager.TryGetEntity(parentNetId, out var parentId))
|
||||
{
|
||||
shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
|
||||
shell.WriteError(Loc.GetString("shell-invalid-entity-uid", ("uid", args[1])));
|
||||
return;
|
||||
}
|
||||
|
||||
var childId = _entManager.GetEntity(childNetId);
|
||||
var parentId = _entManager.GetEntity(parentNetId);
|
||||
var bodySystem = _entManager.System<BodySystem>();
|
||||
|
||||
|
||||
|
||||
if (Enum.TryParse<BodyPartType>(args[3], out var partType) &&
|
||||
bodySystem.TryCreatePartSlotAndAttach(parentId, args[2], childId, partType))
|
||||
_bodySystem.TryCreatePartSlotAndAttach(parentId.Value, args[2], childId.Value, partType))
|
||||
{
|
||||
shell.WriteLine($@"Added {childId} to {parentId}.");
|
||||
}
|
||||
else
|
||||
{
|
||||
shell.WriteError($@"Could not add {childId} to {parentId}.");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,2 @@
|
||||
cmd-addbodypart-desc = Adds a given entity to a containing body.
|
||||
cmd-addbodypart-help = Usage: addbodypart <entity uid> <body uid> <part slot> <part type>
|
||||
Reference in New Issue
Block a user