36 lines
1.5 KiB
C#
36 lines
1.5 KiB
C#
using System.Collections.Generic;
|
|
using Content.Server.Chemistry.EntitySystems;
|
|
using Content.Shared.Chemistry.Components;
|
|
using Content.Shared.Chemistry.Reagent;
|
|
using Content.Shared.FixedPoint;
|
|
using JetBrains.Annotations;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.Serialization.Manager.Attributes;
|
|
using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype.Set;
|
|
|
|
namespace Content.Server.Chemistry.ReagentEntityReactions
|
|
{
|
|
[UsedImplicitly]
|
|
public class AddToSolutionReaction : ReagentEntityReaction
|
|
{
|
|
[DataField("solution")]
|
|
private string _solution = "reagents";
|
|
|
|
[DataField("reagents", true, customTypeSerializer: typeof(PrototypeIdHashSetSerializer<ReagentPrototype>))]
|
|
// ReSharper disable once CollectionNeverUpdated.Local
|
|
private readonly HashSet<string> _reagents = new();
|
|
|
|
protected override void React(IEntity entity, ReagentPrototype reagent, FixedPoint2 volume, Solution? source)
|
|
{
|
|
// TODO see if this is correct
|
|
if (!EntitySystem.Get<SolutionContainerSystem>()
|
|
.TryGetSolution(entity.Uid, _solution, out var solutionContainer)
|
|
|| (_reagents.Count > 0 && !_reagents.Contains(reagent.ID))) return;
|
|
|
|
if (EntitySystem.Get<SolutionContainerSystem>()
|
|
.TryAddReagent(entity.Uid, solutionContainer, reagent.ID, volume, out var accepted))
|
|
source?.RemoveReagent(reagent.ID, accepted);
|
|
}
|
|
}
|
|
}
|