Add autocomplete to nukearm (#14819)

Content.Server/Nuke/Commands/ToggleNukeCommand.cs

@@ -4,57 +4,73 @@ using Content.Shared.Administration;
 using JetBrains.Annotations;
 using Robust.Shared.Console;
 
-namespace Content.Server.Nuke.Commands
+namespace Content.Server.Nuke.Commands;
+
+[UsedImplicitly]
+[AdminCommand(AdminFlags.Fun)]
+public sealed class ToggleNukeCommand : LocalizedCommands
 {
-    [UsedImplicitly]
+    [Dependency] private readonly IEntityManager _entManager = default!;
-    [AdminCommand(AdminFlags.Fun)]
+
-    public sealed class ToggleNukeCommand : IConsoleCommand
+    public override string Command => "nukearm";
+
+    public override void Execute(IConsoleShell shell, string argStr, string[] args)
     {
-        public string Command => "nukearm";
+        EntityUid bombUid;
-        public string Description => "Toggle nuclear bomb timer. You can set timer directly. Uid is optional.";
+        NukeComponent? bomb = null;
-        public string Help => "nukearm <timer> <uid>";
 
-        public void Execute(IConsoleShell shell, string argStr, string[] args)
+        if (args.Length >= 2)
         {
-            EntityUid bombUid;
+            if (!EntityUid.TryParse(args[1], out bombUid))
-            NukeComponent? bomb = null;
-
-            if (args.Length >= 2)
             {
-                if (!EntityUid.TryParse(args[1], out bombUid))
+                shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
-                {
+                return;
-                    shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
-                    return;
-                }
             }
-            else
-            {
-                var entManager = IoCManager.Resolve<IEntityManager>();
-                var bombs = entManager.EntityQuery<NukeComponent>();
-
-                bomb = bombs.FirstOrDefault();
-                if (bomb == null)
-                {
-                    shell.WriteError("Can't find any entity with a NukeComponent");
-                    return;
-                }
-
-                bombUid = bomb.Owner;
-            }
-
-            var nukeSys = EntitySystem.Get<NukeSystem>();
-            if (args.Length >= 1)
-            {
-                if (!float.TryParse(args[0], out var timer))
-                {
-                    shell.WriteError("shell-argument-must-be-number");
-                    return;
-                }
-
-                nukeSys.SetRemainingTime(bombUid, timer, bomb);
-            }
-
-            nukeSys.ToggleBomb(bombUid, bomb);
         }
+        else
+        {
+            var query = _entManager.EntityQueryEnumerator<NukeComponent>();
+
+            while (query.MoveNext(out bombUid, out bomb))
+            {
+                break;
+            }
+
+            if (bomb == null)
+            {
+                shell.WriteError(Loc.GetString("cmd-nukearm-not-found"));
+                return;
+            }
+        }
+
+        var nukeSys = _entManager.System<NukeSystem>();
+
+        if (args.Length >= 1)
+        {
+            if (!float.TryParse(args[0], out var timer))
+            {
+                shell.WriteError("shell-argument-must-be-number");
+                return;
+            }
+
+            nukeSys.SetRemainingTime(bombUid, timer, bomb);
+        }
+
+        nukeSys.ToggleBomb(bombUid, bomb);
+    }
+
+    public override CompletionResult GetCompletion(IConsoleShell shell, string[] args)
+    {
+        if (args.Length == 1)
+        {
+            return CompletionResult.FromHint(Loc.GetString(Loc.GetString("cmd-nukearm-1-help")));
+        }
+
+        if (args.Length == 2)
+        {
+            return CompletionResult.FromHintOptions(CompletionHelper.Components<NukeComponent>(args[1]), Loc.GetString("cmd-nukearm-2-help"));
+        }
+
+        return CompletionResult.Empty;
     }
 }

Resources/Locale/en-US/nuke/nuke-command.ftl

@@ -0,0 +1,6 @@
+cmd-nukearm-desc = Toggle nuclear bomb timer. You can set timer directly. Uid is optional.
+cmd-nukearm-help = nukearm <timer> <uid>
+cmd-nukearm-not-found = Can't find any entity with a NukeComponent.
+
+cmd-nukearm-1-help = Time (in seconds)
+cmd-nukearm-2-help = Nuke