1 files changed, 50 insertions, 1 deletions

@@ -1,7 +1,12 @@
-using BepInEx;
+using AnodyneSharp;
+using AnodyneSharp.Entities;
+using AnodyneSharp.Entities.Gadget;
+using AnodyneSharp.Entities.Gadget.Treasures;
+using BepInEx;
 using BepInEx.NET.Common;
 using HarmonyLib;
 using HarmonyLib.Tools;
+using System;
 using System.Reflection;
 
 namespace AnodyneArchipelago
@@ -26,6 +31,15 @@ namespace AnodyneArchipelago
         }
     }
 
+    [HarmonyPatch(typeof(AnodyneGame), "Update")]
+    class GameUpdatePatch
+    {
+        static void Postfix()
+        {
+            ArchipelagoManager.Update();
+        }
+    }
+
     [HarmonyPatch(typeof(AnodyneSharp.States.PlayState), nameof(AnodyneSharp.States.PlayState.Create))]
     class PlayStateCreatePatch
     {
@@ -36,4 +50,39 @@ namespace AnodyneArchipelago
             ArchipelagoManager.Connect("localhost:38281", "Anodyne", "");
         }
     }
+
+    [HarmonyPatch(typeof(TreasureChest), nameof(TreasureChest.PlayerInteraction))]
+    class ChestInteractPatch
+    {
+        static void Prefix(TreasureChest __instance)
+        {
+            Type chestType = typeof(TreasureChest);
+            FieldInfo presetField = chestType.GetField("_preset", BindingFlags.NonPublic | BindingFlags.Instance);
+            EntityPreset preset = presetField.GetValue(__instance) as EntityPreset;
+            Plugin.Instance.Log.LogInfo($"Touched chest: {preset.EntityID.ToString()}");
+        }
+    }
+
+    [HarmonyPatch(typeof(TreasureChest), "SetTreasure")]
+    class ChestSetTreasurePatch
+    {
+        static bool Prefix(TreasureChest __instance)
+        {
+            Type chestType = typeof(TreasureChest);
+            FieldInfo presetField = chestType.GetField("_preset", BindingFlags.NonPublic | BindingFlags.Instance);
+            EntityPreset preset = presetField.GetValue(__instance) as EntityPreset;
+
+            if (Locations.LocationsByGuid.ContainsKey(preset.EntityID))
+            {
+                BaseTreasure treasure = new ArchipelagoTreasure(Locations.LocationsByGuid[preset.EntityID], __instance.Position);
+
+                FieldInfo treasureField = chestType.GetField("_treasure", BindingFlags.NonPublic | BindingFlags.Instance);
+                treasureField.SetValue(__instance, treasure);
+
+                return false;
+            }
+
+            return true;
+        }
+    }
 }