From ef0bb33def0a88d243e9ca7b2492410690daeffa Mon Sep 17 00:00:00 2001
From: EvilFactory <evilfactory@users.noreply.github.com>
Date: Sun, 30 Oct 2022 19:21:15 -0300
Subject: [PATCH] Register Cargo and Upgrade types and remove old debug print

---
 .../Lua/DefaultRegister/RegisterShared.lua          |  6 ++++++
 Barotrauma/BarotraumaShared/Lua/LuaSetup.lua        | 13 +++++++++++--
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/Barotrauma/BarotraumaShared/Lua/DefaultRegister/RegisterShared.lua b/Barotrauma/BarotraumaShared/Lua/DefaultRegister/RegisterShared.lua
index 7589ead94..7f3c87660 100644
--- a/Barotrauma/BarotraumaShared/Lua/DefaultRegister/RegisterShared.lua
+++ b/Barotrauma/BarotraumaShared/Lua/DefaultRegister/RegisterShared.lua
@@ -81,6 +81,12 @@ RegisterBarotrauma("CharacterParams+SoundParams")
 
 RegisterBarotrauma("Item")
 RegisterBarotrauma("DeconstructItem")
+RegisterBarotrauma("PurchasedItem")
+RegisterBarotrauma("PurchasedItemSwap")
+RegisterBarotrauma("PurchasedUpgrade")
+RegisterBarotrauma("SoldItem")
+RegisterBarotrauma("UpgradeManager")
+RegisterBarotrauma("CargoManager")
 RegisterBarotrauma("FabricationRecipe")
 RegisterBarotrauma("PreferredContainer")
 RegisterBarotrauma("SwappableItem")
diff --git a/Barotrauma/BarotraumaShared/Lua/LuaSetup.lua b/Barotrauma/BarotraumaShared/Lua/LuaSetup.lua
index 7c2aae69a..89c9216e0 100644
--- a/Barotrauma/BarotraumaShared/Lua/LuaSetup.lua
+++ b/Barotrauma/BarotraumaShared/Lua/LuaSetup.lua
@@ -42,14 +42,23 @@ if not CSActive then
     LuaUserData.RegisterType = function (typeName)
         local descriptor = Descriptors[typeName]
 
-        print(typeName)
-
         if descriptor == nil then
             error("Type '" .. typeName .. "' can't be registered", 2)
         else
             return descriptor
         end
     end
+
+    local originalCreateStatic = LuaUserData.CreateStatic
+    LuaUserData.CreateStatic = function (typeName, addCallMethod)
+        local descriptor = Descriptors[typeName]
+
+        if descriptor == nil then
+            error("Unable to create static reference to type " .. typeName, 2)
+        end
+
+        return originalCreateStatic(typeName, addCallMethod)
+    end
 end
 
 LuaSetup = nil