Update upgrades on inventory move (#99)

mt-mods · Feb 14, 2022 · 631879b · 631879b
1 parent 9e0f6ea
commit 631879b
Showing 1 changed file with 23 additions and 8 deletions.
diff --git a/engines/default.lua b/engines/default.lua
@@ -204,20 +204,29 @@ minetest.register_node("jumpdrive:engine", {
 	end,
 
 	-- inventory protection
+	allow_metadata_inventory_move = function(pos, _, _, _, _, count, player)
+		if (not player)
+			or (not player:is_player())
+			or minetest.is_protected(pos, player:get_player_name())
+		then return 0 end
+
+		return count
+	end,
+
 	allow_metadata_inventory_take = function(pos, listname, index, stack, player)
-		if player and player:is_player() and minetest.is_protected(pos, player:get_player_name()) then
-			-- protected
-			return 0
-		end
+		if (not player)
+			or (not player:is_player())
+			or minetest.is_protected(pos, player:get_player_name())
+		then return 0 end
 
 		return stack:get_count()
 	end,
 
 	allow_metadata_inventory_put = function(pos, listname, index, stack, player)
-		if player and player:is_player() and minetest.is_protected(pos, player:get_player_name()) then
-			-- protected
-			return 0
-		end
+		if (not player)
+			or (not player:is_player())
+			or minetest.is_protected(pos, player:get_player_name())
+		then return 0 end
 
 		return stack:get_count()
 	end,
@@ -235,6 +244,12 @@ minetest.register_node("jumpdrive:engine", {
 		end
 	end,
 
+	on_metadata_inventory_move = function(pos, from_list, _, to_list)
+		if from_list == "upgrade" or to_list == "upgrade" then
+			jumpdrive.upgrade.calculate(pos)
+		end
+	end,
+
 	on_punch = function(pos, node, puncher)
 
 		if minetest.is_protected(pos, puncher:get_player_name()) then