local regen_interval = tonumber(minetest.settings:get("regen_interval"))
if regen_interval <= 0 then
	regen_interval = 6
end

local regen_amount = tonumber(minetest.settings:get("regen_amount"))
if regen_amount <= 0 then
	regen_amount = 1
end

local function regen_update()
	local get = ctf_classes.get
	local players = minetest.get_connected_players()
	local medic_by_team = { red = {}, blue = {} }
	local tnames = {}
	local found_medic = false

	-- First get medic positions and team names
	for i=1, #players do
		local player = players[i]
		local pname = player:get_player_name()
		local class = get(player)
		local tname = ctf.player(pname).team
		tnames[pname] = tname
		if class.name == "medic" then
			if tname then
				medic_by_team[tname][#medic_by_team[tname] + 1] = player:get_pos()
				found_medic = true
			end
		end
	end

	if not found_medic then
		return
	end

	-- Next, update hp

	local function sqdist(a, b)
		local x = a.x - b.x
		local y = a.y - b.y
		local z = a.z - b.z
		return x*x + y*y + z*z
	end

	for i=1, #players do
		local player = players[i]
		local pname = player:get_player_name()
		local tname = tnames[pname]
		local hp = player:get_hp()
		local max_hp = player:get_properties().hp_max
		if tname and hp ~= max_hp and hp > 0 then
			local pos = player:get_pos()
			local medics = medic_by_team[tname]
			for j=1, #medics do
				if sqdist(pos, medics[j]) < 100 then
					hp = hp + regen_amount
					player:set_hp(hp)
					break
				end
			end
		end
	end
end

local update = regen_interval / 2
minetest.register_globalstep(function(delta)
	update = update + delta
	if update < regen_interval then
		return
	end
	update = update - regen_interval

	regen_update()
end)