86 lines
3.3 KiB
C#
86 lines
3.3 KiB
C#
using Content.Server.Chemistry.Components;
|
|
using Content.Server.Chemistry.Containers.EntitySystems;
|
|
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, SolutionContainerChangedEvent>(OnSolutionChange);
|
|
}
|
|
|
|
private void OnMapInit(Entity<TransformableContainerComponent> entity, ref MapInitEvent args)
|
|
{
|
|
var meta = MetaData(entity.Owner);
|
|
if (string.IsNullOrEmpty(entity.Comp.InitialName))
|
|
{
|
|
entity.Comp.InitialName = meta.EntityName;
|
|
}
|
|
if (string.IsNullOrEmpty(entity.Comp.InitialDescription))
|
|
{
|
|
entity.Comp.InitialDescription = meta.EntityDescription;
|
|
}
|
|
}
|
|
|
|
private void OnSolutionChange(Entity<TransformableContainerComponent> entity, ref SolutionContainerChangedEvent args)
|
|
{
|
|
if (!_solutionsSystem.TryGetFitsInDispenser(entity.Owner, out _, out var solution))
|
|
return;
|
|
|
|
//Transform container into initial state when emptied
|
|
if (entity.Comp.CurrentReagent != null && solution.Contents.Count == 0)
|
|
{
|
|
CancelTransformation(entity);
|
|
}
|
|
|
|
//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 (entity.Comp.CurrentReagent != null && entity.Comp.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(entity.Owner);
|
|
var val = Loc.GetString("transformable-container-component-glass", ("name", proto.LocalizedName));
|
|
_metadataSystem.SetEntityName(entity.Owner, val, metadata);
|
|
_metadataSystem.SetEntityDescription(entity.Owner, proto.LocalizedDescription, metadata);
|
|
entity.Comp.CurrentReagent = proto;
|
|
entity.Comp.Transformed = true;
|
|
}
|
|
}
|
|
|
|
private void CancelTransformation(Entity<TransformableContainerComponent> entity)
|
|
{
|
|
entity.Comp.CurrentReagent = null;
|
|
entity.Comp.Transformed = false;
|
|
|
|
var metadata = MetaData(entity);
|
|
|
|
if (!string.IsNullOrEmpty(entity.Comp.InitialName))
|
|
{
|
|
_metadataSystem.SetEntityName(entity.Owner, entity.Comp.InitialName, metadata);
|
|
}
|
|
if (!string.IsNullOrEmpty(entity.Comp.InitialDescription))
|
|
{
|
|
_metadataSystem.SetEntityDescription(entity.Owner, entity.Comp.InitialDescription, metadata);
|
|
}
|
|
}
|
|
}
|