85 lines
3.2 KiB
C#
85 lines
3.2 KiB
C#
using Content.Server.Chemistry.Components;
|
|
using Content.Shared.Chemistry.EntitySystems;
|
|
using Content.Shared.Chemistry.Reagent;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Server.Chemistry.EntitySystems;
|
|
|
|
public sealed class TransformableContainerSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
|
[Dependency] private readonly SolutionContainerSystem _solutionsSystem = default!;
|
|
[Dependency] private readonly MetaDataSystem _metadataSystem = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<TransformableContainerComponent, MapInitEvent>(OnMapInit);
|
|
SubscribeLocalEvent<TransformableContainerComponent, SolutionChangedEvent>(OnSolutionChange);
|
|
}
|
|
|
|
private void OnMapInit(EntityUid uid, TransformableContainerComponent component, MapInitEvent args)
|
|
{
|
|
var meta = MetaData(uid);
|
|
if (string.IsNullOrEmpty(component.InitialName))
|
|
{
|
|
component.InitialName = meta.EntityName;
|
|
}
|
|
if (string.IsNullOrEmpty(component.InitialDescription))
|
|
{
|
|
component.InitialDescription = meta.EntityDescription;
|
|
}
|
|
}
|
|
|
|
private void OnSolutionChange(EntityUid owner, TransformableContainerComponent component,
|
|
SolutionChangedEvent args)
|
|
{
|
|
if (!_solutionsSystem.TryGetFitsInDispenser(owner, out var solution))
|
|
return;
|
|
//Transform container into initial state when emptied
|
|
if (component.CurrentReagent != null && solution.Contents.Count == 0)
|
|
{
|
|
CancelTransformation(owner, component);
|
|
}
|
|
|
|
//the biggest reagent in the solution decides the appearance
|
|
var reagentId = solution.GetPrimaryReagentId();
|
|
|
|
//If biggest reagent didn't changed - don't change anything at all
|
|
if (component.CurrentReagent != null && component.CurrentReagent.ID == reagentId?.Prototype)
|
|
{
|
|
return;
|
|
}
|
|
|
|
//Only reagents with spritePath property can change appearance of transformable containers!
|
|
if (!string.IsNullOrWhiteSpace(reagentId?.Prototype)
|
|
&& _prototypeManager.TryIndex(reagentId.Value.Prototype, out ReagentPrototype? proto))
|
|
{
|
|
var metadata = MetaData(owner);
|
|
var val = Loc.GetString("transformable-container-component-glass", ("name", proto.LocalizedName));
|
|
_metadataSystem.SetEntityName(owner, val, metadata);
|
|
_metadataSystem.SetEntityDescription(owner, proto.LocalizedDescription, metadata);
|
|
component.CurrentReagent = proto;
|
|
component.Transformed = true;
|
|
}
|
|
}
|
|
|
|
private void CancelTransformation(EntityUid owner, TransformableContainerComponent component)
|
|
{
|
|
component.CurrentReagent = null;
|
|
component.Transformed = false;
|
|
|
|
var metadata = MetaData(owner);
|
|
|
|
if (!string.IsNullOrEmpty(component.InitialName))
|
|
{
|
|
_metadataSystem.SetEntityName(owner, component.InitialName, metadata);
|
|
}
|
|
if (!string.IsNullOrEmpty(component.InitialDescription))
|
|
{
|
|
_metadataSystem.SetEntityDescription(owner, component.InitialDescription, metadata);
|
|
}
|
|
}
|
|
}
|