RemoveEntityStorageCommand.cs 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. using Content.Server.Storage.Components;
  2. using Content.Server.Storage.EntitySystems;
  3. using Content.Shared.Administration;
  4. using Robust.Shared.Console;
  5. namespace Content.Server.Administration.Commands
  6. {
  7. [AdminCommand(AdminFlags.Admin)]
  8. public sealed class RemoveEntityStorageCommand : IConsoleCommand
  9. {
  10. [Dependency] private readonly IEntityManager _entManager = default!;
  11. public string Command => "rmstorage";
  12. public string Description => "Removes a given entity from it's containing storage, if any.";
  13. public string Help => "Usage: rmstorage <uid>";
  14. public void Execute(IConsoleShell shell, string argStr, string[] args)
  15. {
  16. if (args.Length != 1)
  17. {
  18. shell.WriteError(Loc.GetString("shell-wrong-arguments-number"));
  19. return;
  20. }
  21. if (!NetEntity.TryParse(args[0], out var entityNet) || !_entManager.TryGetEntity(entityNet, out var entityUid))
  22. {
  23. shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
  24. return;
  25. }
  26. if (!_entManager.EntitySysManager.TryGetEntitySystem<EntityStorageSystem>(out var entstorage))
  27. return;
  28. if (!_entManager.TryGetComponent<TransformComponent>(entityUid, out var transform))
  29. return;
  30. var parent = transform.ParentUid;
  31. if (_entManager.TryGetComponent<EntityStorageComponent>(parent, out var storage))
  32. {
  33. entstorage.Remove(entityUid.Value, parent, storage);
  34. }
  35. else
  36. {
  37. shell.WriteError("Could not remove from storage.");
  38. }
  39. }
  40. }
  41. }