AddPolymorphActionCommand.cs 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. using Content.Server.Polymorph.Components;
  2. using Content.Server.Polymorph.Systems;
  3. using Content.Shared.Administration;
  4. using Robust.Shared.Console;
  5. namespace Content.Server.Administration.Commands;
  6. [AdminCommand(AdminFlags.Fun)]
  7. public sealed class AddPolymorphActionCommand : IConsoleCommand
  8. {
  9. [Dependency] private readonly IEntityManager _entityManager = default!;
  10. public string Command => "addpolymorphaction";
  11. public string Description => Loc.GetString("add-polymorph-action-command-description");
  12. public string Help => Loc.GetString("add-polymorph-action-command-help-text");
  13. public void Execute(IConsoleShell shell, string argStr, string[] args)
  14. {
  15. if (args.Length != 2)
  16. {
  17. shell.WriteError(Loc.GetString("shell-wrong-arguments-number"));
  18. return;
  19. }
  20. if (!NetEntity.TryParse(args[0], out var entityUidNet) || !_entityManager.TryGetEntity(entityUidNet, out var entityUid))
  21. {
  22. shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
  23. return;
  24. }
  25. var polySystem = _entityManager.EntitySysManager.GetEntitySystem<PolymorphSystem>();
  26. var polymorphable = _entityManager.EnsureComponent<PolymorphableComponent>(entityUid.Value);
  27. polySystem.CreatePolymorphAction(args[1], (entityUid.Value, polymorphable));
  28. }
  29. }