From f9d61a8b5e0fddda2132113325e7edcb75af6f27 Mon Sep 17 00:00:00 2001
From: Gabriel Tofvesson <contact@w1zzrd.dev>
Date: Wed, 25 Sep 2024 10:11:27 +0000
Subject: [PATCH] Refresh cache entries on change

---
 items.lua | 44 ++++++++++++++++++++++++++++++++++----------
 1 file changed, 34 insertions(+), 10 deletions(-)

diff --git a/items.lua b/items.lua
index e07d1c8..11f0a55 100644
--- a/items.lua
+++ b/items.lua
@@ -17,6 +17,18 @@ local function countNamed(tbl)
   return count
 end
 
+local function cacheFromItemDetail(detail)
+  return {
+    name = detail.name,
+    enchantments = detail.enchantments,
+    count = detail.count,
+    maxCount = detail.maxCount,
+    displayName = detail.displayName,
+    damage = detail.damage,
+    maxDamage = detail.maxDamage
+  }
+end
+
 local function scanChest(chest)
   local entry = {}
   for slot=1,chest.size() do
@@ -24,15 +36,7 @@ local function scanChest(chest)
     if detail ~= nil then
       table.insert(
         entry,
-        {
-          name = detail.name,
-          enchantments = detail.enchantments,
-          count = detail.count,
-          maxCount = detail.maxCount,
-          displayName = detail.displayName,
-          damage = detail.damage,
-          maxDamage = detail.maxDamage
-        }
+        cacheFromItemDetail(detail)
       )
     else
       table.insert(
@@ -94,6 +98,16 @@ function Cache.makeCache(cacheLoader)
   return Cache:from(state)
 end
 
+function Cache:reloadSlot(chestName, slot)
+  local chest = self[chestName]
+  if chest == nil then
+    return false
+  end
+
+  chest[slot] = cacheFromItemDetail(store[chestName].getItemDetail(slot))
+  return true
+end
+
 function Cache:from(o)
   setmetatable(o, self)
   self.__index = self
@@ -231,6 +245,11 @@ function Cache:insertStack(nodeSlot)
     if not result then
       print("Slot insert error for chest \""..slot.chest.."\"! Make sure it's still attached to the controller")
     else
+      if change == slot.count then
+        self:reloadSlot(slot.chest, slot.slot)
+      else
+        slot.count = slot.count - change
+      end
       count = count + change
     end
   end
@@ -246,7 +265,7 @@ function Cache:extractStack(detail)
   end)
 
   local count = 0
-  for _,slot in ipairs(foundSlots) do
+  for index,slot in ipairs(foundSlots) do
     if count == detail.count then
       break
     end
@@ -256,6 +275,11 @@ function Cache:extractStack(detail)
     if not result then
       print("Slot extract error for chest \""..slot.chest.."\"! Make sure it's still attached to the controller")
     else
+      if change == slot.count then
+        self:reloadSlot(slot.chest, slot.slot)
+      else
+        slot.count = slot.count - change
+      end
       count = count + change
     end
   end