Predict StorageComponent in PseudoItem

DeltaV-Station · Sep 27, 2023 · f73dd59 · f73dd59
1 parent dd6799f
commit f73dd59
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Content.Server/Nyanotrasen/Item/PseudoItem/PseudoItemSystem.cs b/Content.Server/Nyanotrasen/Item/PseudoItem/PseudoItemSystem.cs
@@ -5,6 +5,7 @@
 using Content.Shared.DoAfter;
 using Content.Shared.IdentityManagement;
 using Content.Shared.Item.PseudoItem;
+using Content.Shared.Storage.Components;
 using Content.Server.Storage.Components;
 using Content.Server.Storage.EntitySystems;
 using Content.Server.DoAfter;
@@ -35,7 +36,7 @@ private void AddInsertVerb(EntityUid uid, PseudoItemComponent component, GetVerb
         if (component.Active)
             return;
 
-        if (!TryComp<ServerStorageComponent>(args.Target, out var targetStorage))
+        if (!TryComp<StorageComponent>(args.Target, out var targetStorage))
             return;
 
         if (component.Size > targetStorage.StorageCapacityMax - targetStorage.StorageUsed)
@@ -67,7 +68,7 @@ private void AddInsertAltVerb(EntityUid uid, PseudoItemComponent component, GetV
         if (args.Hands == null)
             return;
 
-        if (!TryComp<ServerStorageComponent>(args.Hands.ActiveHandEntity, out var targetStorage))
+        if (!TryComp<StorageComponent>(args.Hands.ActiveHandEntity, out var targetStorage))
             return;
 
         AlternativeVerb verb = new()
@@ -113,7 +114,7 @@ private void OnDoAfter(EntityUid uid, PseudoItemComponent component, DoAfterEven
         args.Handled = TryInsert(args.Args.Used.Value, uid, component);
     }
 
-    public bool TryInsert(EntityUid storageUid, EntityUid toInsert, PseudoItemComponent component, ServerStorageComponent? storage = null)
+    public bool TryInsert(EntityUid storageUid, EntityUid toInsert, PseudoItemComponent component, StorageComponent? storage = null)
     {
         if (!Resolve(storageUid, ref storage))
             return false;