Fix metamorphic glass name mispredict (#16550)

2023-06-30 18:57:28 -04:00
parent 69e1c7e7a5
commit 53e9bc8236
4 changed files with 131 additions and 8 deletions
--- a/Content.Server/Chemistry/EntitySystems/TransformableContainerSystem.cs
+++ b/Content.Server/Chemistry/EntitySystems/TransformableContainerSystem.cs
@@ -0,0 +1,84 @@
+using Content.Server.Chemistry.Components;
+using Content.Shared.Chemistry.Reagent;
+using Robust.Shared.GameObjects;
+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)
+        {
+            return;
+        }
+
+        //Only reagents with spritePath property can change appearance of transformable containers!
+        if (!string.IsNullOrWhiteSpace(reagentId)
+            && _prototypeManager.TryIndex(reagentId, 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);
+        }
+    }
+}