Add strange pills, RandomFillSolutionComponent (#15067)

This commit is contained in:
MisterMecky
2023-05-01 22:43:31 +08:00
committed by GitHub
parent 860ed0ac2f
commit 57cb533e9c
4 changed files with 168 additions and 0 deletions

View File

@@ -0,0 +1,37 @@
using Content.Server.Chemistry.Components.SolutionManager;
using Content.Shared.Chemistry.Reagent;
using Content.Shared.Random;
using Content.Shared.Random.Helpers;
using Robust.Shared.Prototypes;
using Robust.Shared.Random;
namespace Content.Server.Chemistry.EntitySystems;
public sealed class SolutionRandomFillSystem : EntitySystem
{
[Dependency] private readonly SolutionContainerSystem _solutionsSystem = default!;
[Dependency] private readonly IPrototypeManager _proto = default!;
[Dependency] private readonly IRobustRandom _random = default!;
public override void Initialize()
{
base.Initialize();
SubscribeLocalEvent<RandomFillSolutionComponent, MapInitEvent>(OnRandomSolutionFillMapInit);
}
public void OnRandomSolutionFillMapInit(EntityUid uid, RandomFillSolutionComponent component, MapInitEvent args)
{
var target = _solutionsSystem.EnsureSolution(uid, component.Solution);
var reagent = _proto.Index<WeightedRandomPrototype>(component.WeightedRandomId).Pick(_random);
if (!_proto.TryIndex<ReagentPrototype>(reagent, out ReagentPrototype? reagentProto))
{
Logger.Error(
$"Tried to add invalid reagent Id {reagent} using SolutionRandomFill.");
return;
}
target.AddReagent(reagent, component.Quantity);
}
}