Add command to delete all instances of a component (#2580)
* Add command to delete all instances of a component * Fix help string * Consistent naming * Use TryGet for registrations * Remove newline Co-authored-by: Metal Gear Sloth <metalgearsloth@gmail.com>
This commit is contained in:
53
Content.Server/Administration/Commands/DeleteComponent.cs
Normal file
53
Content.Server/Administration/Commands/DeleteComponent.cs
Normal file
@@ -0,0 +1,53 @@
|
||||
#nullable enable
|
||||
using Content.Shared.Administration;
|
||||
using Robust.Server.Interfaces.Console;
|
||||
using Robust.Server.Interfaces.Player;
|
||||
using Robust.Shared.Interfaces.GameObjects;
|
||||
using Robust.Shared.IoC;
|
||||
|
||||
namespace Content.Server.Administration.Commands
|
||||
{
|
||||
[AdminCommand(AdminFlags.Admin)]
|
||||
public class DeleteComponent : IClientCommand
|
||||
{
|
||||
public string Command => "deletecomponent";
|
||||
public string Description => "Deletes all instances of the specified component.";
|
||||
public string Help => $"Usage: {Command} <name>";
|
||||
|
||||
public void Execute(IConsoleShell shell, IPlayerSession? player, string[] args)
|
||||
{
|
||||
switch (args.Length)
|
||||
{
|
||||
case 0:
|
||||
shell.SendText(player, $"Not enough arguments.\n{Help}");
|
||||
break;
|
||||
default:
|
||||
var name = string.Join(" ", args);
|
||||
var componentFactory = IoCManager.Resolve<IComponentFactory>();
|
||||
var entityManager = IoCManager.Resolve<IEntityManager>();
|
||||
|
||||
if (!componentFactory.TryGetRegistration(name, out var registration))
|
||||
{
|
||||
shell.SendText(player, $"No component exists with name {name}.");
|
||||
break;
|
||||
}
|
||||
|
||||
var componentType = registration.Type;
|
||||
var components = entityManager.ComponentManager.GetAllComponents(componentType);
|
||||
|
||||
var i = 0;
|
||||
|
||||
foreach (var component in components)
|
||||
{
|
||||
var uid = component.Owner.Uid;
|
||||
entityManager.ComponentManager.RemoveComponent(uid, component);
|
||||
i++;
|
||||
}
|
||||
|
||||
shell.SendText(player, $"Removed {i} components with name {name}.");
|
||||
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user