Dont even look what I did at first I'm silly

space-wizards · Sep 22, 2024 · a012744 · a012744
1 parent b430c70
commit a012744
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/Content.Shared/Storage/EntitySystems/SharedStorageSystem.cs b/Content.Shared/Storage/EntitySystems/SharedStorageSystem.cs
@@ -740,11 +740,11 @@ private void OnInsertAttempt(EntityUid uid, StorageComponent component, Containe
         }
     }
 
-    public void UpdateAppearance(Entity<StorageComponent?, AppearanceComponent?, StorageFillVisualizerComponent?> entity)
+    public void UpdateAppearance(Entity<StorageComponent?, AppearanceComponent?> entity)
     {
         // TODO STORAGE remove appearance data and just use the data on the component.
-        var (uid, storage, appearance, storageFillVis) = entity;
-        if (!Resolve(uid, ref storage, ref appearance, ref storageFillVis, false))
+        var (uid, storage, appearance) = entity;
+        if (!Resolve(uid, ref storage, ref appearance, false))
             return;
 
         // ReSharper disable once ConditionIsAlwaysTrueOrFalseAccordingToNullableAPIContract
@@ -753,15 +753,19 @@ public void UpdateAppearance(Entity<StorageComponent?, AppearanceComponent?, Sto
 
         var capacity = storage.Grid.GetArea();
         var used = GetCumulativeItemAreas((uid, storage));
-        var level = ContentHelpers.RoundToLevels(used, capacity, storageFillVis.MaxFillLevels);
 
         var isOpen = _ui.IsUiOpen(entity.Owner, StorageComponent.StorageUiKey.Key);
 
         _appearance.SetData(uid, StorageVisuals.StorageUsed, used, appearance);
         _appearance.SetData(uid, StorageVisuals.Capacity, capacity, appearance);
         _appearance.SetData(uid, StorageVisuals.Open, isOpen, appearance);
         _appearance.SetData(uid, SharedBagOpenVisuals.BagState, isOpen ? SharedBagState.Open : SharedBagState.Closed, appearance);
-        _appearance.SetData(uid, StorageFillVisuals.FillLevel, level, appearance);
+
+        if (TryComp<StorageFillVisualizerComponent>(uid, out var storageFillVisualizerComp))
+        {
+            var level = ContentHelpers.RoundToLevels(used, capacity, storageFillVisualizerComp.MaxFillLevels);
+            _appearance.SetData(uid, StorageFillVisuals.FillLevel, level, appearance);
+        }
 
         // HideClosedStackVisuals true sets the StackVisuals.Hide to the open state of the storage.
         // This is for containers that only show their contents when open. (e.g. donut boxes)