fix bug in adding items to an item node

1 files changed, 6 insertions, 3 deletions

diff --git a/mods/vzxv_itemstacks/container.lua b/mods/vzxv_itemstacks/container.lua
index 8899707..119789b 100644
--- a/mods/vzxv_itemstacks/container.lua
+++ b/mods/vzxv_itemstacks/container.lua
@@ -102,9 +102,13 @@ end
 
 function vzxv.add_container_items(pos, stack)
 	local contents = vzxv.get_container_stack(pos)
-	local leftover = contents:add_item(stack)
+	local free_space = contents:get_free_space()
+	local amount = math.min(free_space, stack:get_count())
+	contents:set_count(contents:get_count() + amount)
 	vzxv.set_container_stack(pos, contents)
-	return leftover
+
+	stack:set_count(stack:get_count() - amount)
+	return stack
 end
 
 function vzxv.take_container_items(pos, n)
@@ -123,7 +127,6 @@ function vzxv.player_take_container_items(pos, _, player)
 end
 
 function vzxv.player_add_container_items(pos, _, player, stack)
-	stack:set_count(math.min(stack:get_count(), stack:get_stack_max()))
 	local items = vzxv.remove_items(player, stack)
 	local leftover = vzxv.add_container_items(pos, items)
 	vzxv.give_items(player, leftover)