From 053b59cf006884ecede14ff2b095e8cc5eacde74 Mon Sep 17 00:00:00 2001
From: Gabriel Tofvesson <gabriel@tofvesson.se>
Date: Mon, 14 Oct 2024 12:59:37 +0200
Subject: [PATCH] Fix recursive tracking

---
 logging.lua | 26 +++++++++++---------------
 1 file changed, 11 insertions(+), 15 deletions(-)

diff --git a/logging.lua b/logging.lua
index 61a3b5a..7ae6d32 100644
--- a/logging.lua
+++ b/logging.lua
@@ -73,7 +73,7 @@ end
 
 local RecursionSentinel = {}
 function RecursionSentinel.make(table, index)
-  local obj = { table = table }
+  local obj = { table = table, index = index }
   setmetatable(obj, RecursionSentinel)
   return obj
 end
@@ -87,42 +87,38 @@ function RecursionSentinel.isKnown(knownTables, value)
 end
 
 function RecursionSentinel.getSentinel(knownTables, value)
-    local sentinel = knownTables[value]
-    if sentinel == nil then
-        sentinel = knownTables[value]
-        knownTables[#knownTables+1] = value
-        knownTables[value] = RecursionSentinel.make(value, #knownTables)
-    end
-    return sentinel
+    return knownTables[value] or RecursionSentinel.add(knownTables, value)
 end
 
 function RecursionSentinel.add(knownTables, value)
-  local sentinel = RecursionSentinel.make(value)
+  local sentinel = RecursionSentinel.make(value, #knownTables)
   knownTables[value] = sentinel
   return sentinel
 end
 
-function RecursionSentinel.pop(knownTables)
-  local value = knownTables[#knownTables]
-  knownTables[#knownTables] = nil
+function RecursionSentinel.remove(knownTables, value)
+  local sentinel = knownTables[value]
+  if sentinel then
+    sentinel.index = nil
+  end
   knownTables[value] = nil
 end
 
 local function cloneNonRecursive(value, sentinels)
   if type(value) == "table" then
+    local sentinel = RecursionSentinel.getSentinel(sentinels, value)
     if RecursionSentinel.isKnown(sentinels, value) then
-      return RecursionSentinel.getSentinel(sentinels, value)
+      return sentinel
     end
 
     local clone = {}
-    local sentinel = RecursionSentinel.add(sentinels, value)
     for i,v in ipairs(value) do
       clone[i] = cloneNonRecursive(v, sentinels)
     end
     for k,v in pairs(value) do
       clone[cloneNonRecursive(k, sentinels)] = cloneNonRecursive(v, sentinels)
     end
-    RecursionSentinel.pop(sentinels)
+    RecursionSentinel.remove(sentinels, sentinel.table)
 
     sentinel.value = clone
     return sentinel.value