-- This file provides the actual flow logic that makes liquids
-- move through the pipes.

local finite_liquids = minetest.setting_getbool("liquid_finite")
local pipe_liquid_shows_loaded = 1

if mesecon then
	pipereceptor_on = {
		receptor = {
			state = mesecon.state.on,
			rules = pipeworks.mesecons_rules
		}
	}

	pipereceptor_off = {
		receptor = {
			state = mesecon.state.off,
			rules = pipeworks.mesecons_rules
		}
	}
end

-- Evaluate and balance liquid in all pipes

minetest.register_abm({
	nodenames = pipeworks.pipe_nodenames, 
	interval = 1,
	chance = 1,
	action = function(pos, node, active_object_count, active_object_count_wider)
		local coords = {
			{x = pos.x,   y = pos.y,   z = pos.z},
			{x = pos.x,   y = pos.y-1, z = pos.z},
			{x = pos.x,   y = pos.y+1, z = pos.z},
			{x = pos.x-1, y = pos.y,   z = pos.z},
			{x = pos.x+1, y = pos.y,   z = pos.z},
			{x = pos.x,   y = pos.y,   z = pos.z-1},
			{x = pos.x,   y = pos.y,   z = pos.z+1},
		}

		local num_connections = 0
		local connection_list = {}
		local total_level = 0
		
		for _,adjacentpos in ipairs(coords) do
			local adjacent_node = minetest.get_node(adjacentpos)
			if adjacent_node and string.find(dump(pipeworks.pipe_nodenames), adjacent_node.name) then

				local node_level = (minetest.get_meta(adjacentpos):get_float("liquid_level")) or 0
				if node_level < 0 then node_level = 0 end

				total_level = total_level + node_level
				num_connections = num_connections + 1
				table.insert(connection_list, adjacentpos)
			end
		end

		local average_level = total_level / num_connections

		for _, connected_pipe_pos in ipairs(connection_list) do

			local newnode
			local connected_pipe = minetest.get_node(connected_pipe_pos)
			local pipe_name = string.match(connected_pipe.name, "pipeworks:pipe_%d.*_")

			if connected_pipe and pipe_name then
				minetest.get_meta(connected_pipe_pos):set_float("liquid_level", average_level)

				if average_level > pipe_liquid_shows_loaded then
					newnode = pipe_name.."loaded"
				else
					newnode = pipe_name.."empty"
				end
			end

			if newnode and connected_pipe.name ~= newnode then 
				minetest.swap_node(connected_pipe_pos, {name = newnode, param2 = connected_pipe.param2})
			end
		end
	end
})

-- Process all pumps in the area

minetest.register_abm({
	nodenames = {"pipeworks:pump_on", "pipeworks:pump_off"},
	interval = 2,
	chance = 1,
	action = function(pos, node, active_object_count, active_object_count_wider)
		local minp =		{x = pos.x-1, y = pos.y-1, z = pos.z-1}
		local maxp =		{x = pos.x+1, y = pos.y, z = pos.z+1}
		local pos_above =	{x = pos.x, y = pos.y+1, z = pos.z}
		local node_above = minetest.get_node(pos_above)
		if not node_above then return end

		local meta = minetest.get_meta(pos_above)
		local node_level_above = meta:get_float("liquid_level")
		if node_level_above == nil then node_level_above = 0 end
		local pipe_name = string.match(node_above.name, "pipeworks:pipe_%d.*_")

		if pipe_name then
			if node.name == "pipeworks:pump_on" then
				local water_nodes = minetest.find_nodes_in_area(minp, maxp, 
									{"default:water_source", "default:water_flowing"})

				if (node_level_above < 4 ) and #water_nodes > 1 then
					meta:set_float("liquid_level", node_level_above + 4) -- add water to the pipe
				end
			else
				if node_level_above > 0 then
					meta:set_float("liquid_level", node_level_above - 0.5 ) -- leak the pipe down
				end
			end
		end
	end
})

-- Process all fountainheads in the area

minetest.register_abm({
	nodenames = {"pipeworks:fountainhead"},
	interval = 2,
	chance = 1,
	action = function(pos, node, active_object_count, active_object_count_wider)
		local pos_above = {x = pos.x, y = pos.y+1, z = pos.z}
		local node_above = minetest.get_node(pos_above) 
		if not node_above then return end

		local pos_below = {x = pos.x, y = pos.y-1, z = pos.z}
		local node_below = minetest.get_node(pos_below)
		if not node_below then return end

		local node_level_below = (minetest.get_meta(pos_below):get_float("liquid_level")) or 0

		if node_level_below > 0.9
		  and (node_above.name == "air" or node_above.name == "default:water_flowing") then
			minetest.set_node(pos_above, {name = "default:water_source"})
		elseif node_level_below < 0.75 and node_above.name == "default:water_source" then
			minetest.set_node(pos_above, {name = "air"})
		end

		if node_level_below >= 1
		  and (node_above.name == "air" or node_above.name == "default:water_source") then
			minetest.get_meta(pos_below):set_float("liquid_level", node_level_below - 1)
		end
	end
})

-- Process all spigots in the area

minetest.register_abm({
	nodenames = {"pipeworks:spigot","pipeworks:spigot_pouring"},
	interval = 2,
	chance = 1,
	action = function(pos, node, active_object_count, active_object_count_wider)
		local pos_below = {x = pos.x, y = pos.y-1, z = pos.z}
		local below_node = minetest.get_node(pos_below)
		if not below_node then return end

		if below_node.name == "air" or below_node.name == "default:water_flowing"
		  or below_node.name == "default:water_source" then 
			local fdir = node.param2
			local fdir_to_pos = {
				{x = pos.x,   y = pos.y, z = pos.z+1},
				{x = pos.x+1, y = pos.y, z = pos.z  },
				{x = pos.x,   y = pos.y, z = pos.z-1},
				{x = pos.x-1, y = pos.y, z = pos.z  }
			}

			local pos_adjacent = fdir_to_pos[fdir+1]
			local adjacent_node = minetest.get_node(pos_adjacent)
			if not adjacent_node then return end
			local adjacent_node_level = (minetest.get_meta(pos_adjacent):get_float("liquid_level")) or 0
			local pipe_name = string.match(adjacent_node.name, "pipeworks:pipe_%d.*_")

			if pipe_name and adjacent_node_level > 0.9
			  and (below_node.name == "air" or below_node.name == "default:water_flowing") then
				minetest.set_node(pos, {name = "pipeworks:spigot_pouring", param2 = fdir})
				minetest.set_node(pos_below, {name = "default:water_source"})
			end

			if (pipe_name and adjacent_node_level < 0.75)
			  or (node.name ~= "pipeworks:spigot" and not pipe_name) then
				minetest.set_node(pos,{name = "pipeworks:spigot", param2 = fdir})
				if below_node.name == "default:water_source" then
					minetest.set_node(pos_below, {name = "air"})
				end
			end

			if adjacent_node_level >= 1
			  and (below_node.name == "air" or below_node.name == "default:water_source") then
				minetest.get_meta(pos_adjacent):set_float("liquid_level", adjacent_node_level - 1)
			end
		end
	end
})

pipeworks.device_nodenames = {}

table.insert(pipeworks.device_nodenames,"pipeworks:valve_on_empty")
table.insert(pipeworks.device_nodenames,"pipeworks:valve_off_empty")
table.insert(pipeworks.device_nodenames,"pipeworks:valve_on_loaded")
table.insert(pipeworks.device_nodenames,"pipeworks:flow_sensor_empty")
table.insert(pipeworks.device_nodenames,"pipeworks:flow_sensor_loaded")

minetest.register_abm({
	nodenames = pipeworks.device_nodenames,
	interval = 2,
	chance = 1,
	action = function(pos, node, active_object_count, active_object_count_wider)
		local fdir = node.param2
		local fdir_mod4    = fdir % 4
		local fdir_mod4_p2 = (fdir+2) % 4

		local fdir_to_pos = {
			{x = pos.x+1, y = pos.y, z = pos.z  },
			{x = pos.x,   y = pos.y, z = pos.z-1},
			{x = pos.x-1, y = pos.y, z = pos.z  },
			{x = pos.x,   y = pos.y, z = pos.z+1},
		}

		local pos_adjacent1 = fdir_to_pos[fdir_mod4    + 1]
		local pos_adjacent2 = fdir_to_pos[fdir_mod4_p2 + 1]

		local adjacent_node1 = minetest.get_node(pos_adjacent1)
		local adjacent_node2 = minetest.get_node(pos_adjacent2)

		if not adjacent_node1 or not adjacent_node2 then return end

		local my_level = (minetest.get_meta(pos):get_float("liquid_level")) or 0
		local adjacent_node_level1 = (minetest.get_meta(pos_adjacent1):get_float("liquid_level")) or 0
		local adjacent_node_level2 = (minetest.get_meta(pos_adjacent2):get_float("liquid_level")) or 0

		if not string.match(node.name, "pipeworks:valve_off") then

			local num_connections = 1
			local set1
			local set2
			local total_level = my_level

			if string.find(dump(pipeworks.pipe_nodenames), adjacent_node1.name) or
			  (string.find(dump(pipeworks.device_nodenames), adjacent_node1.name) and
			  (adjacent_node1.param2 == fdir_mod4 or adjacent_node1.param2 == fdir_mod4_p2)) then
				num_connections = num_connections + 1
				total_level = total_level + adjacent_node_level1
				set1 = true
			end

			if string.find(dump(pipeworks.pipe_nodenames), adjacent_node2.name) or
			  (string.find(dump(pipeworks.device_nodenames), adjacent_node2.name) and
			  (adjacent_node2.param2 == fdir_mod4 or adjacent_node2.param2 == fdir_mod4_p2)) then
				num_connections = num_connections + 1
				total_level = total_level + adjacent_node_level2
				set2 = true
			end

			local average_level = total_level / num_connections

			minetest.get_meta(pos):set_float("liquid_level", average_level)

			if set1 then
				minetest.get_meta(pos_adjacent1):set_float("liquid_level", average_level)
			end

			if set2 then
				minetest.get_meta(pos_adjacent2):set_float("liquid_level", average_level)
			end

			if node.name == "pipeworks:flow_sensor_empty" or
			   node.name == "pipeworks:flow_sensor_loaded" then
				local sensor = string.match(node.name, "pipeworks:flow_sensor_")
				local newnode = nil

				if my_level > 1 then
					newnode = sensor.."loaded"
				else
					newnode = sensor.."empty"
				end

				if newnode ~= node.name then
					minetest.swap_node(pos, {name = newnode, param2 = node.param2})
					if mesecon then
						if newnode == "pipeworks:flow_sensor_empty" then
							mesecon.receptor_off(pos, rules) 
						else
							mesecon.receptor_on(pos, rules) 
						end
					end
				end
			end
		end
	end
})

-- table.insert(pipeworks.device_nodenames,"pipeworks:entry_panel")