3d3c9e3348
* fixe * Save work. * Rune-aware parser. * oogh * pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests pass tests * Publicizes a lot of common generic commands, so custom toolshed envs can include them. * i think i might implode * Tests. * a * b * awa --------- Co-authored-by: moonheart08 <moonheart08@users.noreply.github.com>
85 lines
2.7 KiB
C#
85 lines
2.7 KiB
C#
using System.Linq;
|
|
using Content.Server.Chemistry.EntitySystems;
|
|
using Content.Shared.Administration;
|
|
using Content.Shared.Chemistry.Components;
|
|
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.TryRemoveReagent(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}";
|
|
}
|
|
}
|