tbd-station-14/Content.Server/Administration/Toolshed/SolutionCommand.cs

using System.Linq;
using Content.Shared.Administration;
using Content.Shared.Chemistry.Components;
using Content.Shared.Chemistry.EntitySystems;
using Content.Shared.Chemistry.Reagent;
using Content.Shared.FixedPoint;
using Robust.Shared.Toolshed;
using Robust.Shared.Toolshed.Syntax;
using Robust.Shared.Toolshed.TypeParsers;

namespace Content.Server.Administration.Toolshed;

[ToolshedCommand, AdminCommand(AdminFlags.Debug)]
public sealed class SolutionCommand : ToolshedCommand
{
    private SolutionContainerSystem? _solutionContainer;

    [CommandImplementation("get")]
    public SolutionRef? Get(
            [CommandInvocationContext] IInvocationContext ctx,
            [PipedArgument] EntityUid input,
            [CommandArgument] ValueRef<string> name
        )
    {
        _solutionContainer ??= GetSys<SolutionContainerSystem>();

        _solutionContainer.TryGetSolution(input, name.Evaluate(ctx)!, out var solution);

        if (solution is not null)
            return new SolutionRef(input, solution);

        return null;
    }

    [CommandImplementation("get")]
    public IEnumerable<SolutionRef> Get(
        [CommandInvocationContext] IInvocationContext ctx,
        [PipedArgument] IEnumerable<EntityUid> input,
        [CommandArgument] ValueRef<string> name
    )
    {
        return input.Select(x => Get(ctx, x, name)).Where(x => x is not null).Cast<SolutionRef>();
    }

    [CommandImplementation("adjreagent")]
    public SolutionRef AdjReagent(
            [CommandInvocationContext] IInvocationContext ctx,
            [PipedArgument] SolutionRef input,
            [CommandArgument] Prototype<ReagentPrototype> name,
            [CommandArgument] ValueRef<FixedPoint2> amountRef
        )
    {
        _solutionContainer ??= GetSys<SolutionContainerSystem>();

        var amount = amountRef.Evaluate(ctx);
        if (amount > 0)
        {
            _solutionContainer.TryAddReagent(input.Owner, input.Solution, name.Value.ID, amount, out _);
        }
        else if (amount < 0)
        {
            _solutionContainer.RemoveReagent(input.Owner, input.Solution, name.Value.ID, -amount);
        }

        return input;
    }

    [CommandImplementation("adjreagent")]
    public IEnumerable<SolutionRef> AdjReagent(
            [CommandInvocationContext] IInvocationContext ctx,
            [PipedArgument] IEnumerable<SolutionRef> input,
            [CommandArgument] Prototype<ReagentPrototype> name,
            [CommandArgument] ValueRef<FixedPoint2> amountRef
        )
        => input.Select(x => AdjReagent(ctx, x, name, amountRef));
}

public readonly record struct SolutionRef(EntityUid Owner, Solution Solution)
{
    public override string ToString()
    {
        return $"{Owner} {Solution}";
    }
}