OldThink/Content.Server/_White/_Engi/StationGoalCommand/StationGoalCommand.cs

using System.Linq;
using Content.Server.Administration;
using Content.Server.Commands;
using Content.Shared.Administration;
using Robust.Shared.Console;
using Robust.Shared.Prototypes;

namespace Content.Server._White._Engi.StationGoal
{
    [AdminCommand(AdminFlags.Fun)]
    public sealed class StationGoalCommand : IConsoleCommand
    {
        [Dependency] private readonly IEntityManager _entManager = default!;

        public string Command => "sendstationgoal";
        public string Description => Loc.GetString("send-station-goal-command-description");
        public string Help => Loc.GetString("send-station-goal-command-help-text", ("command", Command));

        public void Execute(IConsoleShell shell, string argStr, string[] args)
        {
            if (args.Length != 2)
            {
                shell.WriteError(Loc.GetString("shell-wrong-arguments-number"));
                return;
            }

            if (!NetEntity.TryParse(args[0], out var euidNet) || !_entManager.TryGetEntity(euidNet, out var euid))
            {
                shell.WriteError($"Failed to parse euid '{args[0]}'.");
                return;
            }

            var protoId = args[1];
            var prototypeManager = IoCManager.Resolve<IPrototypeManager>();
            if (!prototypeManager.TryIndex<StationGoalPrototype>(protoId, out var proto))
            {
                shell.WriteError($"No station goal found with ID {protoId}!");
                return;
            }

            var stationGoalPaper = IoCManager.Resolve<IEntityManager>().System<StationGoalPaperSystem>();
            if (!stationGoalPaper.SendStationGoal(euid, protoId))
            {
                shell.WriteError("Station goal was not sent");
                return;
            }
        }

        public CompletionResult GetCompletion(IConsoleShell shell, string[] args)
        {
            switch (args.Length)
            {
                case 1:
                    var stations = ContentCompletionHelper.StationIds(_entManager);
                    return CompletionResult.FromHintOptions(stations, "[StationId]");
                case 2:
                    var options = IoCManager.Resolve<IPrototypeManager>()
                        .EnumeratePrototypes<StationGoalPrototype>()
                        .Select(p => new CompletionOption(p.ID));

                    return CompletionResult.FromHintOptions(options, Loc.GetString("send-station-goal-command-arg-id"));
            }
            return CompletionResult.Empty;
        }
    }
}