ctf_respawn_immunity = {}

local IMMUNE_TIME = 5
local immune_serial = 0
local immune_players = {}

function ctf_respawn_immunity.is_immune(player)
	return immune_players[player:get_player_name()]
end

function ctf_respawn_immunity.set_immune(player)
	immune_serial = immune_serial + 1
	immune_players[player:get_player_name()] = immune_serial
	minetest.after(1, function()
		ctf_respawn_immunity.update_effects(player)
	end)

	-- Set time out
	minetest.after(IMMUNE_TIME, function(name, id)
		if immune_players[name] == id then
			immune_players[name] = nil
			ctf_respawn_immunity.update_effects(minetest.get_player_by_name(name))
		end
	end, player:get_player_name(), immune_serial)
end

function ctf_respawn_immunity.update_effects(player)
	-- TODO: transparent player when immune
	--
	-- if immune_players[player:get_player_name()] then
	-- 	player:set_texture_mod("[multiply:#1f1")
	-- else
	-- 	player:set_texture_mod(nil)
	-- end
end

local old_can_attack = ctf.can_attack
function ctf.can_attack(player, hitter, ...)
	if not player or not hitter then
		return
	end

	local pname = player:get_player_name()
	local hname = hitter:get_player_name()

	if ctf_respawn_immunity.is_immune(player) then
		hud_event.new(hname, {
			name  = "ctf_respawn_immunity:hit",
			color = 0xEE8822,
			value = pname .. " has respawn immunity!",
		})
		return false
	end

	if ctf_respawn_immunity.is_immune(hitter) then
		hud_event.new(hname, {
			name  = "ctf_respawn_immunity:end",
			color = 0xFF8C00,
			value = "Your immunity has ended!",
		})
		immune_players[hname] = nil
		ctf_respawn_immunity.update_effects(hitter)
	end

	return old_can_attack(player, hitter, ...)
end

minetest.register_on_joinplayer(ctf_respawn_immunity.set_immune)
minetest.register_on_respawnplayer(ctf_respawn_immunity.set_immune)