From aab36cd28f2e5d7f1de42bc86d686abf7f97857a Mon Sep 17 00:00:00 2001
From: Chr1Z93 <puls.christian@gmail.com>
Date: Thu, 5 Oct 2023 10:27:03 +0200
Subject: [PATCH] return drawn tokens

---
 src/accessories/CleanUpHelper.ttslua | 1 +
 src/chaosbag/ChaosBagApi.ttslua      | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/src/accessories/CleanUpHelper.ttslua b/src/accessories/CleanUpHelper.ttslua
index 3d25e52c..4a3548c0 100644
--- a/src/accessories/CleanUpHelper.ttslua
+++ b/src/accessories/CleanUpHelper.ttslua
@@ -190,6 +190,7 @@ function cleanUp(_, color)
   removeLines()
   discardHands()
   tokenSpawnTrackerApi.resetAll()
+  chaosBagApi.returnChaosTokens()
   chaosBagApi.releaseAllSealedTokens(color)
 
   printToAll("Tidying main play area...", "White")
diff --git a/src/chaosbag/ChaosBagApi.ttslua b/src/chaosbag/ChaosBagApi.ttslua
index e0252266..c83dd081 100644
--- a/src/chaosbag/ChaosBagApi.ttslua
+++ b/src/chaosbag/ChaosBagApi.ttslua
@@ -28,6 +28,11 @@ do
     return Global.call("releaseAllSealedTokens", playerColor)
   end
 
+  -- returns all drawn tokens to the chaos bag
+  ChaosBagApi.returnChaosTokens = function(playerColor)
+    return Global.call("returnChaosTokens", playerColor)
+  end
+
   -- removes the specified chaos token from the chaos bag
   ---@param id String ID of the chaos token
   ChaosBagApi.removeChaosToken = function(id)