MineClone2/mods/ITEMS/mcl_buckets/init.lua

-- See README.txt for licensing and other information.
local modname = minetest.get_current_modname()
local S = minetest.get_translator(modname)
local modpath = minetest.get_modpath(modname)

-- Compatibility with old bucket mod
minetest.register_alias("bucket:bucket_empty", "mcl_buckets:bucket_empty")
minetest.register_alias("bucket:bucket_water", "mcl_buckets:bucket_water")
minetest.register_alias("bucket:bucket_lava", "mcl_buckets:bucket_lava")

local mod_doc = minetest.get_modpath("doc")
local mod_mcl_core = minetest.get_modpath("mcl_core")
--local mod_mclx_core = minetest.get_modpath("mclx_core")

-- Localize some functions for faster access
local vector = vector
local math = math
local string = string

local raycast = minetest.raycast
local get_node = minetest.get_node
local set_node = minetest.set_node
local add_node = minetest.add_node
local add_item = minetest.add_item

local registered_nodes = minetest.registered_nodes
local get_item_group = minetest.get_item_group

if mod_mcl_core then
	minetest.register_craft({
		output = "mcl_buckets:bucket_empty 1",
		recipe = {
			{"mcl_core:iron_ingot", "", "mcl_core:iron_ingot"},
			{"", "mcl_core:iron_ingot", ""},
		},
	})
end

mcl_buckets = {
	liquids = {},
	buckets = {},
}

-- Sound helper functions for placing and taking liquids
local function sound_place(itemname, pos)
	local def = registered_nodes[itemname]
	if def and def.sounds and def.sounds.place then
		minetest.sound_play(def.sounds.place, {gain=1.0, pos = pos, pitch = 1 + math.random(-10, 10)*0.005}, true)
	end
end

local function sound_take(itemname, pos)
	local def = registered_nodes[itemname]
	if def and def.sounds and def.sounds.dug then
		minetest.sound_play(def.sounds.dug, {gain=1.0, pos = pos, pitch = 1 + math.random(-10, 10)*0.005}, true)
	end
end

local function place_liquid(pos, itemstring)
	local fullness = registered_nodes[itemstring].liquid_range
	sound_place(itemstring, pos)
	add_node(pos, {name=itemstring, param2=fullness})
end

local function give_bucket(new_bucket, itemstack, user)
	local inv = user:get_inventory()
	if minetest.is_creative_enabled(user:get_player_name()) then
		--TODO: is a full bucket added if inv doesn't contain one?
		return itemstack
	else
		if itemstack:get_count() == 1 then
			return new_bucket
		else
			if inv:room_for_item("main", new_bucket) then
				inv:add_item("main", new_bucket)
			else
				add_item(user:get_pos(), new_bucket)
			end
			itemstack:take_item()
			return itemstack
		end
	end
end

local pointable_sources = {}

local function bucket_raycast(user)
	--local pos = user:get_pos()
	local pos = user:get_pos()
	--local pos = vector.add(user:get_pos(), user:get_bone_position("Head_Control"))
	pos.y = pos.y + user:get_properties().eye_height
	local look_dir = user:get_look_dir()
	look_dir = vector.multiply(look_dir, 5)
	local pos2 = vector.add(pos, look_dir)

	local ray = raycast(pos, pos2, false, true)
	if ray then
		for pointed_thing in ray do
			if pointed_thing and pointable_sources[get_node(pointed_thing.above).name] then
				--minetest.chat_send_all("found!")
				return {under=pointed_thing.under,above=pointed_thing.above}
			end
		end
	end
	return nil
end

local function get_node_place(source_place, place_pos)
	local node_place
	if type(source_place) == "function" then
		node_place = source_place(place_pos)
	else
		node_place = source_place
	end
	return node_place
end

local function get_extra_check(check, pos, user)
	local result
	local take_bucket
	if check then
		result, take_bucket = check(pos, user)
		if result == nil then result = true end
		if take_bucket == nil then take_bucket = true end
	else
		result = true
		take_bucket = true
	end
	return result, take_bucket
end

local function get_bucket_drop(itemstack, user, take_bucket)
	-- Handle bucket item and inventory stuff
	if take_bucket and not minetest.is_creative_enabled(user:get_player_name()) then
		-- Add empty bucket and put it into inventory, if possible.
		-- Drop empty bucket otherwise.
		local new_bucket = ItemStack("mcl_buckets:bucket_empty")
		if itemstack:get_count() == 1 then
			return new_bucket
		else
			local inv = user:get_inventory()
			if inv:room_for_item("main", new_bucket) then
				inv:add_item("main", new_bucket)
			else
				add_item(user:get_pos(), new_bucket)
			end
			itemstack:take_item()
			return itemstack
		end
	else
		return itemstack
	end
end

local function on_place_bucket(itemstack, user, pointed_thing, def)
	-- Must be pointing to node
	if pointed_thing.type ~= "node" then
		return
	end
	-- Call on_rightclick if the pointed node defines it
	local new_stack = mcl_util.call_on_rightclick(itemstack, user, pointed_thing)
	if new_stack then
		return new_stack
	end

	local undernode = get_node(pointed_thing.under)
	local abovenode = get_node(pointed_thing.above)

	if registered_nodes[undernode.name].buildable_to or get_item_group(undernode.name, "cauldron") == 1 then
		local result, take_bucket = get_extra_check(def.extra_check, pointed_thing.under, user)
		if result then
			local node_place = get_node_place(def.source_place, pointed_thing.under)
			local pns = user:get_player_name()

			-- Check protection
			if minetest.is_protected(pointed_thing.under, pns) then
				minetest.record_protection_violation(pointed_thing.under, pns)
				return itemstack
			end

			-- Place liquid
			place_liquid(pointed_thing.under, node_place)

			-- Update doc mod
			if mod_doc and doc.entry_exists("nodes", node_place) then
				doc.mark_entry_as_revealed(user:get_player_name(), "nodes", node_place)
			end
		end
		return get_bucket_drop(itemstack, user, take_bucket)
	elseif registered_nodes[abovenode.name].buildable_to or get_item_group(abovenode.name, "cauldron") == 1 then
		local result, take_bucket = get_extra_check(def.extra_check, pointed_thing.above, user)
		if result then
			local node_place = get_node_place(def.source_place, pointed_thing.above)
			local pns = user:get_player_name()

			-- Check protection
			if minetest.is_protected(pointed_thing.above, pns) then
				minetest.record_protection_violation(pointed_thing.above, pns)
				return itemstack
			end

			-- Place liquid
			place_liquid(pointed_thing.above, node_place)

			-- Update doc mod
			if mod_doc and doc.entry_exists("nodes", node_place) then
				doc.mark_entry_as_revealed(user:get_player_name(), "nodes", node_place)
			end
		end
		return get_bucket_drop(itemstack, user, take_bucket)
	else
		return itemstack
	end
end


local function on_place_bucket_empty(itemstack, user, pointed_thing)
	-- Must be pointing to node
	if pointed_thing.type ~= "node" then
		return itemstack
	end

	-- Call on_rightclick if the pointed node defines it
	local new_stack = mcl_util.call_on_rightclick(itemstack, user, pointed_thing)
	if new_stack then
		return new_stack
	end

	local node = minetest.get_node(pointed_thing.under)
	local nn = node.name

	local new_bucket
	local liquid_node = bucket_raycast(user)
	if liquid_node then
		if minetest.is_protected(liquid_node.above, user:get_player_name()) then
			minetest.record_protection_violation(liquid_node.above, user:get_player_name())
		end
		local liquid_name = get_node(liquid_node.above).name
		if liquid_name then
			local liquid_def = mcl_buckets.liquids[liquid_name]
			if liquid_def then
				--minetest.chat_send_all("test")
				-- Fill bucket, but not in Creative Mode
				-- FIXME: remove this line
				--if not minetest.is_creative_enabled(user:get_player_name()) then
				if not false then
					new_bucket = ItemStack({name = liquid_def.bucketname})
					if liquid_def.on_take then
						liquid_def.on_take(user)
					end
				end
				add_node(liquid_node.above, {name="air"})
				sound_take(nn, liquid_node.above)

				if mod_doc and doc.entry_exists("nodes", liquid_name) then
					doc.mark_entry_as_revealed(user:get_player_name(), "nodes", liquid_name)
				end
				if new_bucket then
					return give_bucket(new_bucket, itemstack, user)
				end
			else
				minetest.log("error", string.format("[mcl_buckets] Node [%s] has invalid group [_mcl_bucket_pointable]!", liquid_name))
			end
		end
		return itemstack
	else
		-- FIXME: replace this ugly code by cauldrons API
		if nn == "mcl_cauldrons:cauldron_3" then
			-- Take water out of full cauldron
			set_node(pointed_thing.under, {name="mcl_cauldrons:cauldron"})
			if not minetest.is_creative_enabled(user:get_player_name()) then
				new_bucket = ItemStack("mcl_buckets:bucket_water")
			end
			sound_take("mcl_core:water_source", pointed_thing.under)
		elseif nn == "mcl_cauldrons:cauldron_3r" then
			-- Take river water out of full cauldron
			set_node(pointed_thing.under, {name="mcl_cauldrons:cauldron"})
			if not minetest.is_creative_enabled(user:get_player_name()) then
				new_bucket = ItemStack("mcl_buckets:bucket_river_water")
			end
			sound_take("mclx_core:river_water_source", pointed_thing.under)
		end
		if new_bucket then
			return give_bucket(new_bucket, itemstack, user)
		end
	end
	return itemstack
end

controls.register_on_press(function(player, key)
	if key ~= "RMB" then
		return
	end

	local wielded_item = player:get_wielded_item()
	local itemname = wielded_item:get_name()
	local def = mcl_buckets.buckets[itemname]

	if itemname == "mcl_buckets:bucket_empty" then
		local pointed_thing = mcl_util.get_pointed_thing(player, true)

		if not pointed_thing then
			return
		end
		wielded_item = on_place_bucket_empty(wielded_item, player, pointed_thing)
	elseif def then
		local pointed_thing = mcl_util.get_pointed_thing(player, false)

		if not pointed_thing then
			return
		end
		wielded_item = on_place_bucket(wielded_item, player, pointed_thing, def)
	end
	
	player:set_wielded_item(wielded_item)
end)

function mcl_buckets.register_liquid(def)
	for _,source in ipairs(def.source_take) do
		mcl_buckets.liquids[source] = {
			source_place = def.source_place,
			source_take = source,
			on_take = def.on_take,
			bucketname = def.bucketname,
		}
		pointable_sources[source] = true
		if type(def.source_place) == "string" then
			mcl_buckets.liquids[def.source_place] = mcl_buckets.liquids[source]
		end
	end

	mcl_buckets.buckets[def.bucketname] = def

	if def.bucketname == nil or def.bucketname == "" then
		error(string.format("[mcl_bucket] Invalid itemname then registering [%s]!", def.name))
	end

	minetest.register_craftitem(def.bucketname, {
		description = def.name,
		_doc_items_longdesc = def.longdesc,
		_doc_items_usagehelp = def.usagehelp,
		_tt_help = def.tt_help,
		inventory_image = def.inventory_image,
		stack_max = 1,
		groups = def.groups,
		_on_dispense = function(stack, pos, droppos, dropnode, dropdir)
			local buildable = registered_nodes[dropnode.name].buildable_to or dropnode.name == "mcl_portals:portal"
			if not buildable then return stack end
			local result, take_bucket = get_extra_check(def.extra_check, droppos, nil)
			if result then -- Fail placement of liquid if result is false
				place_liquid(droppos, get_node_place(def.source_place, droppos))
			end
			if take_bucket then
				stack:set_name("mcl_buckets:bucket_empty")
			end
			return stack
		end,
	})
end

minetest.register_craftitem("mcl_buckets:bucket_empty", {
	description = S("Empty Bucket"),
	_doc_items_longdesc = S("A bucket can be used to collect and release liquids."),
	_doc_items_usagehelp = S("Punch a liquid source to collect it. You can then use the filled bucket to place the liquid somewhere else."),
	_tt_help = S("Collects liquids"),
	--liquids_pointable = true,
	inventory_image = "bucket.png",
	stack_max = 16,
	_on_dispense = function(stack, pos, droppos, dropnode, dropdir)
		-- Fill empty bucket with liquid or drop bucket if no liquid
		local collect_liquid = false

		local liquiddef = mcl_buckets.liquids[dropnode.name]
		local new_bucket
		if liquiddef and liquiddef.bucketname and (dropnode.name  == liquiddef.source_take) then
			-- Fill bucket
			new_bucket = ItemStack({name = liquiddef.bucketname})
			sound_take(dropnode.name, droppos)
			collect_liquid = true
		end
		if collect_liquid then
			set_node(droppos, {name="air"})

			-- Fill bucket with liquid
			stack = new_bucket
		else
			-- No liquid found: Drop empty bucket
			add_item(droppos, stack)
			stack:take_item()
		end
		return stack
	end,
})

dofile(modpath.."/register.lua")