tbd-station-14/Content.Server/Nuke/Commands/SendNukeCodesCommand.cs

using System.Linq;
using Content.Server.Administration;
using Content.Server.Station.Components;
using Content.Server.Station.Systems;
using Content.Shared.Administration;
using JetBrains.Annotations;
using Robust.Shared.Console;

namespace Content.Server.Nuke.Commands
{
    [UsedImplicitly]
    [AdminCommand(AdminFlags.Fun)]
    public sealed class SendNukeCodesCommand : IConsoleCommand
    {
        public string Command => "nukecodes";
        public string Description => "Send nuke codes to a station's communication consoles";
        public string Help => "nukecodes [station EntityUid]";

        [Dependency] private readonly IEntityManager _entityManager = default!;

        public SendNukeCodesCommand()
        {
            IoCManager.InjectDependencies(this);
        }

        public void Execute(IConsoleShell shell, string argStr, string[] args)
        {
            if (args.Length != 1)
            {
                shell.WriteError(Loc.GetString("shell-need-exactly-one-argument"));
                return;
            }

            if (!EntityUid.TryParse(args[0], out var uid))
            {
                shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
                return;
            }

            _entityManager.System<NukeCodePaperSystem>().SendNukeCodes(uid);
        }

        public CompletionResult GetCompletion(IConsoleShell shell, string[] args)
        {
            if (args.Length != 1)
            {
                return CompletionResult.Empty;
            }

            var stations = _entityManager
                .EntityQuery<StationDataComponent>()
                .Select(stationData =>
                {
                    var meta = _entityManager.GetComponent<MetaDataComponent>(stationData.Owner);

                    return new CompletionOption(stationData.Owner.ToString(), meta.EntityName);
                });

            return CompletionResult.FromHintOptions(stations, null);
        }
    }
}