drop upgrade list items if dug (but protect other inventories)

Tim

2015-02-06 48e02b84526733eaa0f99ccc01de6d76ad1ba7fa

 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


@@ -105,8 +128,7 @@
   if minetest.get_item_group(node1.name, "tubedevice") > 0 then
      output_tube_connected = true
   end
   tube_time = meta:get_int("tube_time")
   tube_time = tube_time + tube_upgrade
   local tube_time = meta:get_int("tube_time") + tube_upgrade
   if tube_time >= 2 then
      tube_time = 0
      if output_tube_connected then
@@ -116,41 +138,62 @@
   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
      if not inv:is_empty("upgrade1") then
         minetest.item_drop(inv:get_stack("upgrade1", 1), "", pos)
      end
      if not inv:is_empty("upgrade2") then
         minetest.item_drop(inv:get_stack("upgrade2", 1), "", pos)
      end
      return true
   end
end

local function inv_change(pos, player, count)
   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_upgrade then
      -- only place a single item into it, if it's empty
      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())
   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())
   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)
   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