fix: gun info not applying (too early)

Author: HAHOOS

KeepInventory/Helper/InventoryHelper.cs

@@ -14,6 +14,8 @@ public static class InventoryHelper
     {
         public static Dictionary<string, string> Aliases { get; } = new() { { "HeadSlotContainer", "Head" } };
 
+        internal static Dictionary<InventorySlotReceiver, Il2CppSystem.Action> Callback = [];
+
         public static List<InventorySlotReceiver> GetAllSlots(this RigManager rigManager)
         {
             return rigManager?.GetComponentsInChildren<InventorySlotReceiver>().ToList();
@@ -64,7 +66,11 @@ public static void SpawnInSlot(this InventorySlotReceiver receiver, Barcode barc
             if (callback != null)
             {
                 var awaiter = task.GetAwaiter();
-                awaiter.OnCompleted((Il2CppSystem.Action)(() => callback?.Invoke(receiver?._slottedWeapon?.GetComponentInParent<Poolee>()?.gameObject)));
+                Il2CppSystem.Action action = (Il2CppSystem.Action)(() => callback?.Invoke(receiver?._slottedWeapon?.GetComponentInParent<Poolee>()?.gameObject));
+                if (!Utilities.Fusion.IsConnected)
+                    awaiter.OnCompleted(action);
+                else
+                    Callback.Add(receiver, action);
             }
         }
     }

KeepInventory/Patches/ReceiverPatches.cs

@@ -0,0 +1,36 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+using BoneLib;
+
+using HarmonyLib;
+
+using Il2CppSLZ.Marrow;
+
+using KeepInventory.Helper;
+
+namespace KeepInventory.Patches
+{
+    [HarmonyPatch(typeof(InventorySlotReceiver))]
+    public static class ReceiverPatches
+    {
+        [HarmonyPostfix]
+        [HarmonyPatch(nameof(InventorySlotReceiver.OnHandDrop))]
+        public static void Postfix(InventorySlotReceiver __instance)
+        {
+            if (!Core.HasFusion || !Utilities.Fusion.IsConnected)
+                return;
+
+            if (__instance.GetComponentInParent<RigManager>() != Player.RigManager)
+                return;
+
+            if (!InventoryHelper.Callback.TryGetValue(__instance, out var action))
+                return;
+
+            action?.Invoke();
+        }
+    }
+}