civ14/Content.Server/Atmos/Commands/SetMapAtmosCommand.cs

using Content.Server.Administration;
using Content.Server.Atmos.Components;
using Content.Server.Atmos.EntitySystems;
using Content.Shared.Administration;
using Content.Shared.Atmos;
using Robust.Shared.Console;
using Robust.Shared.Map;

namespace Content.Server.Atmos.Commands;

[AdminCommand(AdminFlags.Admin)]
public sealed class AddMapAtmosCommand : LocalizedCommands
{
    [Dependency] private readonly IEntityManager _entities = default!;
    [Dependency] private readonly IMapManager _map = default!;

    private const string _cmd = "cmd-set-map-atmos";
    public override string Command => "setmapatmos";
    public override string Description => Loc.GetString($"{_cmd}-desc");
    public override string Help => Loc.GetString($"{_cmd}-help");

    public override void Execute(IConsoleShell shell, string argStr, string[] args)
    {
        if (args.Length < 2)
        {
            shell.WriteLine(Help);
            return;
        }

        int.TryParse(args[0], out var id);
        var map = _map.GetMapEntityId(new MapId(id));
        if (!map.IsValid())
        {
            shell.WriteError(Loc.GetString("cmd-parse-failure-mapid",  ("arg", args[0])));
            return;
        }

        if (!bool.TryParse(args[1], out var space))
        {
            shell.WriteError(Loc.GetString("cmd-parse-failure-bool", ("arg", args[1])));
            return;
        }

        if (space || args.Length < 4)
        {
            _entities.RemoveComponent<MapAtmosphereComponent>(map);
            shell.WriteLine(Loc.GetString($"{_cmd}-removed", ("map", id)));
            return;
        }

        if (!float.TryParse(args[2], out var temp))
        {
            shell.WriteError(Loc.GetString("cmd-parse-failure-float", ("arg", args[2])));
            return;
        }

        var mix = new GasMixture(Atmospherics.CellVolume) {Temperature = Math.Max(temp, Atmospherics.TCMB)};
        for (var i = 0; i < Atmospherics.TotalNumberOfGases; i++)
        {
            if (args.Length == 3 + i)
                break;

            if (!float.TryParse(args[3+i], out var moles))
            {
                shell.WriteError(Loc.GetString("cmd-parse-failure-float", ("arg", args[3+i])));
                return;
            }

            mix.AdjustMoles(i, moles);
        }

        var atmos = _entities.EntitySysManager.GetEntitySystem<AtmosphereSystem>();
        atmos.SetMapAtmosphere(map, space, mix);
        shell.WriteLine(Loc.GetString($"{_cmd}-updated", ("map", id)));
    }

    public override CompletionResult GetCompletion(IConsoleShell shell, string[] args)
    {
        if (args.Length == 1)
            return CompletionResult.FromHintOptions(CompletionHelper.MapIds(_entities), Loc.GetString($"{_cmd}-hint-map"));

        if (args.Length == 2)
            return CompletionResult.FromHintOptions(new[]{ "false", "true"}, Loc.GetString($"{_cmd}-hint-space"));

        if (!bool.TryParse(args[1], out var space) || space)
            return CompletionResult.Empty;

        if (args.Length == 3)
            return CompletionResult.FromHint(Loc.GetString($"{_cmd}-hint-temp"));

        var gas = (Gas) args.Length - 4;
        return CompletionResult.FromHint(Loc.GetString($"{_cmd}-hint-gas" , ("gas", gas.ToString())));
    }
}