From 701240bc3a17a73d73f3a96cd81e559d662ac42c Mon Sep 17 00:00:00 2001
From: Thomas Rudin <thomas@rudin.li>
Date: Sun, 09 Dec 2018 16:01:10 +0100
Subject: [PATCH] Quarry: Check for valid facedir to fix server freeze #465 (#466)

---
 technic/machines/register/common.lua |  124 +++++++++++++++++++++++++++++-----------
 1 files changed, 89 insertions(+), 35 deletions(-)

diff --git a/technic/machines/register/common.lua b/technic/machines/register/common.lua
index b269b06..dfa2948 100644
--- a/technic/machines/register/common.lua
+++ b/technic/machines/register/common.lua
@@ -1,39 +1,62 @@
 
 local S = technic.getter
 
+-- handles the machine upgrades every tick
 function technic.handle_machine_upgrades(meta)
 	-- Get the names of the upgrades
 	local inv = meta:get_inventory()
-	local upg_item1
-	local upg_item2
+
 	local srcstack = inv:get_stack("upgrade1", 1)
-	if srcstack then
-		upg_item1 = srcstack:to_table()
-	end
+	local upg_item1 = srcstack and srcstack:get_name()
+
 	srcstack = inv:get_stack("upgrade2", 1)
-	if srcstack then
-		upg_item2 = srcstack:to_table()
-	end
+	local upg_item2 = srcstack and srcstack:get_name()
 
 	-- Save some power by installing battery upgrades.
 	-- Tube loading speed can be upgraded using control logic units.
 	local EU_upgrade = 0
 	local tube_upgrade = 0
-	if upg_item1 then
-		if     upg_item1.name == "technic:battery" then
-			EU_upgrade = EU_upgrade + 1
-		elseif upg_item1.name == "technic:control_logic_unit" then
-			tube_upgrade = tube_upgrade + 1
-		end
+
+	if upg_item1 == "technic:control_logic_unit" then
+		tube_upgrade = tube_upgrade + 1
+	elseif upg_item1 == "technic:battery" then
+		EU_upgrade = EU_upgrade + 1
 	end
-	if upg_item2 then
-		if     upg_item2.name == "technic:battery" then
-			EU_upgrade = EU_upgrade + 1
-		elseif upg_item2.name == "technic:control_logic_unit" then
-			tube_upgrade = tube_upgrade + 1
-		end
+
+	if upg_item2 == "technic:control_logic_unit" then
+		tube_upgrade = tube_upgrade + 1
+	elseif  upg_item2 == "technic:battery" then
+		EU_upgrade = EU_upgrade + 1
 	end
+
 	return EU_upgrade, tube_upgrade
+end
+
+-- handles the machine upgrades when set or removed
+local function on_machine_upgrade(meta, stack)
+	local stack_name = stack:get_name()
+	if stack_name == "default:chest" then
+		meta:set_int("public", 1)
+		return 1
+	elseif stack_name ~= "technic:control_logic_unit"
+	   and stack_name ~= "technic:battery" then
+		return 0
+	end
+	return 1
+end
+
+-- something is about to be removed
+local function on_machine_downgrade(meta, stack, list)
+	if stack:get_name() == "default:chest" then
+		local inv = meta:get_inventory()
+		local upg1, upg2 = inv:get_stack("upgrade1", 1), inv:get_stack("upgrade2", 1)
+
+		-- only set 0 if theres not a nother chest in the other list too
+		if (not upg1 or not upg2 or upg1:get_name() ~= upg2:get_name()) then
+			meta:set_int("public", 0)
+		end
+	end
+	return 1
 end
 
 
@@ -115,46 +138,77 @@
 	meta:set_int("tube_time", tube_time)
 end
 
-
 function technic.machine_can_dig(pos, player)
 	local meta = minetest.get_meta(pos)
 	local inv = meta:get_inventory()
-	if not inv:is_empty("src") or not inv:is_empty("dst") or
-	   not inv:is_empty("upgrade1") or not inv:is_empty("upgrade2") then
+	if not inv:is_empty("src") or not inv:is_empty("dst") then
 		if player then
 			minetest.chat_send_player(player:get_player_name(),
 				S("Machine cannot be removed because it is not empty"))
 		end
 		return false
-	else
-		return true
+	end
+
+	return true
+end
+
+function technic.machine_after_dig_node(pos, oldnode, oldmetadata, player)
+	if oldmetadata.inventory then
+		if oldmetadata.inventory.upgrade1 and oldmetadata.inventory.upgrade1[1] then
+			local stack = ItemStack(oldmetadata.inventory.upgrade1[1])
+			if not stack:is_empty() then
+				minetest.add_item(pos, stack)
+			end
+		end
+		if oldmetadata.inventory.upgrade2 and oldmetadata.inventory.upgrade2[1] then
+			local stack = ItemStack(oldmetadata.inventory.upgrade2[1])
+			if not stack:is_empty() then
+				minetest.add_item(pos, stack)
+			end
+		end
+	end
+
+	if minetest.registered_nodes[oldnode.name].tube then
+		pipeworks.after_dig(pos, oldnode, oldmetadata, player)
 	end
 end
 
-local function inv_change(pos, player, count, from_list, to_list)
-	if minetest.is_protected(pos, player:get_player_name()) then
-		minetest.chat_send_player(player:get_player_name(),
-			S("Inventory move disallowed due to protection"))
+local function inv_change(pos, player, count, from_list, to_list, stack)
+	local playername = player:get_player_name()
+	local meta = minetest.get_meta(pos);
+	local public = (meta:get_int("public") == 1)
+	local to_upgrade = to_list == "upgrade1" or to_list == "upgrade2"
+	local from_upgrade = from_list == "upgrade1" or from_list == "upgrade2"
+
+	if (not public or to_upgrade or from_upgrade) and minetest.is_protected(pos, playername) then
+		minetest.chat_send_player(playername, S("Inventory move disallowed due to protection"))
 		return 0
 	end
-	if to_list == "upgrade1" or to_list == "upgrade2" then
+	if to_upgrade then
 		-- only place a single item into it, if it's empty
-		local empty = minetest.get_meta(pos):get_inventory():is_empty(to_list)
-		return empty and 1 or 0
+		local empty = meta:get_inventory():is_empty(to_list)
+		if empty then
+			return on_machine_upgrade(meta, stack)
+		end
+		return 0
+	elseif from_upgrade then
+		-- only called on take (not move)
+		on_machine_downgrade(meta, stack, from_list)
 	end
 	return count
 end
 
 function technic.machine_inventory_put(pos, listname, index, stack, player)
-	return inv_change(pos, player, stack:get_count(), nil, listname)
+	return inv_change(pos, player, stack:get_count(), nil, listname, stack)
 end
 
 function technic.machine_inventory_take(pos, listname, index, stack, player)
-	return inv_change(pos, player, stack:get_count(), listname, nil)
+	return inv_change(pos, player, stack:get_count(), listname, nil, stack)
 end
 
 function technic.machine_inventory_move(pos, from_list, from_index,
 		to_list, to_index, count, player)
-	return inv_change(pos, player, count, from_list, to_list)
+	local stack = minetest.get_meta(pos):get_inventory():get_stack(from_list, from_index)
+	return inv_change(pos, player, count, from_list, to_list, stack)
 end
 

--
Gitblit v1.8.0