moreblocks/stairsplus/api/station.lua

-- for creating the circular saw and similar nodes

local api = stairsplus.api

local default_stack_max = tonumber(minetest.settings:get("default_stack_max")) or 99

local station = {}

function station.update_infotext(pos)
	local node = minetest.get_node(pos)
	local def = minetest.registered_nodes[node.name]
	return def.update_infotext(pos)
end

function station.build_formspec(pos)
	local node = minetest.get_node(pos)
	local def = minetest.registered_nodes[node.name]
	return def.build_formspec(pos)
end

function station.get_cost(shaped_node)
	if shaped_node == "" then
		return 0
	end

	local shape = api.get_shape_of_shaped_node(shaped_node)
	if shape == "node" then
		return 8
	end

	local shape_def = api.registered_shapes[shape]
	return shape_def and shape_def.eighths
end

function station.get_current_node(pos)
	local meta = minetest.get_meta(pos)
	local inv = meta:get_inventory()

	local input_stack = inv:get_stack("input", 1)
	if not input_stack:is_empty() then
		return input_stack:get_name()
	end

	local micro_stack = inv:get_stack("micro", 1)
	if not micro_stack:is_empty() then
		return api.get_node_of_shaped_node(micro_stack:get_name())
	end

	local recycle_stack = inv:get_stack("recycle", 1)
	if not recycle_stack:is_empty() then
		return api.get_node_of_shaped_node(recycle_stack:get_name())
	end
end

function station.can_dig(pos)
	local meta = minetest.get_meta(pos)
	local inv = meta:get_inventory()
	return inv:is_empty("input") and inv:is_empty("micro")
end

function station.on_receive_fields(pos, formname, fields, sender)
	local meta = minetest.get_meta(pos)
	local max = tonumber(fields.max_offered)
	if max and max > 0 then
		meta:set_int("max_offered", max)
		-- Update to show the correct number of items:
		station.update_inventory(pos)
		station.update_infotext(pos)
	end
end

function station.update_inventory(pos, taken_stack)
	local meta = minetest.get_meta(pos)
	local inv = meta:get_inventory()

	local node = station.get_current_node(pos)

	if not node then
		for i = 1, inv:get_size("output") do
			inv:set_stack("output", i, ItemStack())
		end
		return
	end

	local input_stack = inv:get_stack("input", 1)
	local micro_stack = inv:get_stack("micro", 1)
	local recycle_stack = inv:get_stack("recycle", 1)

	local current_value = 8 * input_stack:get_count() + micro_stack:get_count()
	local new_value = current_value + station.get_cost(recycle_stack:get_name()) * recycle_stack:get_count()
	if taken_stack then
		new_value = new_value - station.get_cost(taken_stack:get_name()) * taken_stack:get_count()
	end
	local new_micros = new_value % 8
	local new_blocks = math.floor(new_value / 8)

	inv:set_stack("input", 1, ItemStack({name = node, count = new_blocks}))
	inv:set_stack("micro", 1, ItemStack({name = api.get_micronode(node), count = new_micros}))
	inv:set_stack("recycle", 1, ItemStack())

	if new_value == 0 then
		for i = 1, inv:get_size("output") do
			inv:set_stack("output", i, "")
		end
		return
	end

	local valid_shapes = api.shapes_by_node[node]
	local shape_groups = minetest.registered_nodes[minetest.get_node(pos).name]._stairsplus_shape_groups

	local max_offered = meta:get_int("max_offered")

	local i = 1
	for _, group in ipairs(shape_groups) do
		for _, shape in ipairs(api.shapes_by_group[group]) do
			if valid_shapes[shape] then
				local shape_def = api.registered_shapes[shape]
				local shaped_node = api.format_name(node, shape)
				local stack_max = math.min(max_offered, ItemStack(shaped_node):get_stack_max())
				local count = math.min(stack_max, math.floor(new_value / shape_def.eighths))
				local stack
				if count > 0 then
					stack = ItemStack({name = shaped_node, count = count})
				else
					stack = ""
				end
				inv:set_stack("output", i, stack)
				i = i + 1
			end
		end
	end

	for j = i, inv:get_size("output") do
		inv:set_stack("output", j, "")
	end
end

function station.on_metadata_inventory_put(pos, listname, index, stack, player)
	station.update_inventory(pos)
	station.update_infotext(pos)
end

function station.on_metadata_inventory_take(pos, listname, index, stack, player)
	if listname == "output" then
		station.update_inventory(pos, stack)
	else
		station.update_inventory(pos)
	end

	station.update_infotext(pos)
end

-- Moving the inventory of the station around is not allowed because it
-- is a fictional inventory. Moving inventory around would be rather
-- impractical and make things more difficult to calculate:
function station.allow_metadata_inventory_move()
	return 0
end

function station.allow_metadata_inventory_put(pos, listname, index, stack, player)
	if listname ~= "recycle" then
		return 0
	end

	local shaped_node = stack:get_name()
	local node = api.get_node_of_shaped_node(shaped_node)
	local shape = api.get_shape_of_shaped_node(shaped_node)

	if not (node and shape) then
		return 0
	end

	local meta = minetest.get_meta(pos)
	local inv = meta:get_inventory()
	local current_node = station.get_current_node(pos)

	if current_node and node ~= current_node then
		return 0
	end

	local count = stack:get_count()
	local cost = station.get_cost(shaped_node)

	local input_stack = inv:get_stack("input", 1)
	local micro_stack = inv:get_stack("micro", 1)

	local current_value = 8 * input_stack:get_count() + micro_stack:get_count()
	local max_value = 8 * ItemStack(node):get_stack_max() + 7

	local available_value = max_value - current_value
	local available_count = math.floor(available_value / cost)

	return math.min(count, available_count)
end

function station.on_construct(pos)
	local meta = minetest.get_meta(pos)

	meta:set_string("formspec", station.build_formspec(pos))
	meta:set_string("max_offered", default_stack_max) -- How many items of this kind are offered by default?

	local inv = meta:get_inventory()

	inv:set_size("input", 1)    -- Input slot for full blocks of material x.
	inv:set_size("micro", 1)    -- Storage for 1-7 surplus microblocks.
	inv:set_size("recycle", 1)  -- Surplus partial blocks can be placed here.
	inv:set_size("output", 6 * 9) -- 6x9 versions of stair-parts of material x.

	station.update_infotext(pos)
end

function station.after_place_node(pos, placer)
	local meta = minetest.get_meta(pos)
	if minetest.is_player(placer) then
		meta:set_string("owner", placer:get_player_name())
	end
end

function api.register_station(name, shape_groups, def)
	def.after_place_node = def.after_place_node or station.after_place_node
	def.on_construct = def.on_construct or station.on_construct
	def.can_dig = def.can_dig or station.can_dig
	def.on_receive_fields = def.on_receive_fields or station.on_receive_fields
	def.allow_metadata_inventory_move = def.allow_metadata_inventory_move or station.allow_metadata_inventory_move
	def.allow_metadata_inventory_put = def.allow_metadata_inventory_put or station.allow_metadata_inventory_put
	def.allow_metadata_inventory_take = def.allow_metadata_inventory_take or station.allow_metadata_inventory_take
	def.on_metadata_inventory_put = def.on_metadata_inventory_put or station.on_metadata_inventory_put
	def.on_metadata_inventory_take = def.on_metadata_inventory_take or station.on_metadata_inventory_take
	def._stairsplus_shape_groups = shape_groups

	minetest.register_node(name, def)
end

api.station = station