tbd-station-14/Content.Server/Administration/Commands/SetSolutionThermalEnergy.cs

using Content.Shared.Administration;
using Content.Shared.Chemistry.Components.SolutionManager;
using Content.Shared.Chemistry.EntitySystems;
using Robust.Shared.Console;

namespace Content.Server.Administration.Commands
{
    [AdminCommand(AdminFlags.Fun)]
    public sealed class SetSolutionThermalEnergy : IConsoleCommand
    {
        [Dependency] private readonly IEntityManager _entManager = default!;

        public string Command => "setsolutionthermalenergy";
        public string Description => "Set the thermal energy of some solution.";
        public string Help => $"Usage: {Command} <target> <solution> <new thermal energy>";

        public void Execute(IConsoleShell shell, string argStr, string[] args)
        {
            if (args.Length < 3)
            {
                shell.WriteLine($"Not enough arguments.\n{Help}");
                return;
            }

            if (!NetEntity.TryParse(args[0], out var uidNet) || !_entManager.TryGetEntity(uidNet, out var uid))
            {
                shell.WriteLine($"Invalid entity id.");
                return;
            }

            if (!_entManager.TryGetComponent(uid, out SolutionContainerManagerComponent? man))
            {
                shell.WriteLine($"Entity does not have any solutions.");
                return;
            }

            if (!man.Solutions.ContainsKey(args[1]))
            {
                var validSolutions = string.Join(", ", man.Solutions.Keys);
                shell.WriteLine($"Entity does not have a \"{args[1]}\" solution. Valid solutions are:\n{validSolutions}");
                return;
            }
            var solution = man.Solutions[args[1]];

            if (!float.TryParse(args[2], out var quantity))
            {
                shell.WriteLine($"Failed to parse new thermal energy.");
                return;
            }

            if (solution.GetHeatCapacity(null) <= 0.0f)
            {
                if(quantity != 0.0f)
                {
                    shell.WriteLine($"Cannot set the thermal energy of a solution with 0 heat capacity to a non-zero number.");
                    return;
                }
            }
            else if(quantity <= 0.0f)
            {
                shell.WriteLine($"Cannot set the thermal energy of a solution with heat capacity to a non-positive number.");
                return;
            }

            _entManager.System<SolutionContainerSystem>().SetThermalEnergy(uid.Value, solution, quantity);
        }
    }
}