local S = minetest.get_translator("mcl_death_messages")

mcl_death_messages = {}

-- Death messages
local msgs = {
	["arrow"] = {
		S("%s was fatally hit by an arrow."),
		S("%s has been killed with an arrow."),
	},
	["arrow_name"] = {
		S("%s was shot by an arrow from %s."),
	},
	["fire"] = {
		S("%s has been cooked crisp."),
		S("%s felt the burn."),
		S("%s died in the flames."),
		S("%s died in a fire."),
	},
	["lava"] = {
		S("%s melted in lava."),
		S("%s took a bath in a hot lava tub."),
		S("%s died in lava."),
		S("%s could not survive in lava."),
	},
	["drown"] = {
		S("%s forgot to breathe."),
		S("%s drowned."),
		S("%s ran out of oxygen."),
	},
	["murder"] = {
		S("%s was killed by %s."),
	},
	["mob_kill"] = {
		S("%s was killed by a mob."),
	},
	["blaze_fireball"] = {
		S("%s was burned to death by a blaze's fireball."),
		S("%s was killed by a fireball from a blaze."),
	},
	["fire_charge"] = {
		S("%s was burned by a fire charge."),
	},
	["ghast_fireball"] = {
		S("A ghast scared %s to death."),
		S("%s has been fireballed by a ghast."),
	},
	["fall"] = {
		S("%s fell from a high cliff."),
		S("%s took fatal fall damage."),
		S("%s fell victim to gravity."),
	},
	["other"] = {
		S("%s died."),
	}
}

local mobkills = {
	["mobs_mc:zombie"] = S("%s was killed by a zombie."),
	["mobs_mc:baby_zombie"] = S("%s was killed by a baby zombie."),
	["mobs_mc:blaze"] = S("%s was killed by a blaze."),
	["mobs_mc:slime"] = S("%s was killed by a slime."),
	["mobs_mc:witch"] = S("%s was killed by a witch."),
	["mobs_mc:magma_cube_tiny"] = S("%s was killed by a magma cube."),
	["mobs_mc:magma_cube_small"] = S("%s was killed by a magma cube."),
	["mobs_mc:magma_cube_big"] = S("%s was killed by a magma cube."),
	["mobs_mc:wolf"] = S("%s was killed by a wolf."),
	["mobs_mc:cat"] = S("%s was killed by a cat."),
	["mobs_mc:ocelot"] = S("%s was killed by an ocelot."),
	["mobs_mc:ender_dragon"] = S("%s was killed by an ender dragon."),
	["mobs_mc:wither"] = S("%s was killed by a wither."),
	["mobs_mc:enderman"] = S("%s was killed by an enderman."),
	["mobs_mc:endermite"] = S("%s was killed by an endermite."),
	["mobs_mc:ghast"] = S("%s was killed by a ghast."),
	["mobs_mc:guardian_elder"] = S("%s was killed by an elder guardian."),
	["mobs_mc:guardian"] = S("%s was killed by a guardian."),
	["mobs_mc:iron_golem"] = S("%s was killed by an iron golem."),
	["mobs_mc:polar_bear"] = S("%s was killed by a polar_bear."),
	["mobs_mc:killer_bunny"] = S("%s was killed by a killer bunny."),
	["mobs_mc:shulker"] = S("%s was killed by a shulker."),
	["mobs_mc:silverfish"] = S("%s was killed by a silverfish."),
	["mobs_mc:skeleton"] = S("%s was killed by a skeleton."),
	["mobs_mc:stray"] = S("%s was killed by a stray."),
	["mobs_mc:slime_tiny"] = S("%s was killed by a slime."),
	["mobs_mc:slime_small"] = S("%s was killed by a slime."),
	["mobs_mc:slime_big"] = S("%s was killed by a slime."),
	["mobs_mc:spider"] = S("%s was killed by a spider."),
	["mobs_mc:cave_spider"] = S("%s was killed by a cave spider."),
	["mobs_mc:vex"] = S("%s was killed by a vex."),
	["mobs_mc:evoker"] = S("%s was killed by an evoker."),
	["mobs_mc:illusioner"] = S("%s was killed by an illusioner."),
	["mobs_mc:vindicator"] = S("%s was killed by a vindicator."),
	["mobs_mc:villager_zombie"] = S("%s was killed by a zombie villager."),
	["mobs_mc:husk"] = S("%s was killed by a husk."),
	["mobs_mc:baby_husk"] = S("%s was killed by a baby husk."),
	["mobs_mc:pigman"] = S("%s was killed by a zombie pigman."),
	["mobs_mc:baby_pigman"] = S("%s was killed by a baby zombie pigman."),
}

-- Select death message
local dmsg = function(mtype, ...)
	local r = math.random(1, #msgs[mtype])
	return string.format(msgs[mtype][r], ...)
end

-- Select death message for death by mob
local mmsg = function(mtype, ...)
	if mobkills[mtype] then
		return string.format(mobkills[mtype], ...)
	else
		return dmsg("mob_kill", ...)
	end
end

local last_damages = { }

minetest.register_on_dieplayer(function(player, reason)
	-- Death message
	local message = minetest.settings:get_bool("mcl_showDeathMessages")
	if message == nil then
		message = true
	end
	if message then
		local name = player:get_player_name()
		if not name then
			return
		end
		local msg
		if reason.type == "node_damage" then
			local pos = player:get_pos()
			-- Check multiple nodes because players occupy multiple nodes
			-- (we add one additional node because the check may fail if the player was
			-- just barely touching the node with the head)
			local posses = { pos, {x=pos.x,y=pos.y+1,z=pos.z}, {x=pos.x,y=pos.y+2,z=pos.z}}
			for p=1, #posses do
				local node = minetest.registered_nodes[minetest.get_node(posses[p]).name]
				-- Lava
				if minetest.get_item_group(node.name, "lava") ~= 0 then
					msg = dmsg("lava", name)
					break
				-- Fire
				elseif minetest.get_item_group(node.name, "fire") ~= 0 then
					msg = dmsg("fire", name)
					break
				end
			end
		elseif reason.type == "drown" then
			msg = dmsg("drown", name)
		elseif reason.type == "punch" then
		-- Punches
			local hitter = reason.object
			local hittername, hittertype, hittersubtype, shooter
			-- Unknown hitter
			if hitter == nil then
				msg = dmsg("murder_any")
			-- Player
			elseif hitter:is_player() then
				hittername = hitter:get_player_name()
				if hittername ~= nil then
					msg = dmsg("murder", name, hittername)
				else
					msg = dmsg("murder_any", name)
				end
			-- Mob (according to Common Mob Interface)
			elseif hitter:get_luaentity()._cmi_is_mob then
				if hitter:get_luaentity().nametag and hitter:get_luaentity().nametag ~= "" then
					hittername = hitter:get_luaentity().nametag
				end
				hittersubtype = hitter:get_luaentity().name
				if hittername then
					msg = dmsg("murder", name, hittername)
				elseif hittersubtype ~= nil and hittersubtype ~= "" then
					msg = mmsg(hittersubtype, name)
				else
					msg = dmsg("murder_any", name)
				end
			-- Arrow
			elseif hitter:get_luaentity().name == "mcl_bows:arrow_entity" or hitter:get_luaentity().name == "mobs_mc:arrow_entity" then
				local shooter
				if hitter:get_luaentity()._shooter then
					shooter = hitter:get_luaentity()._shooter
				end
				if shooter == nil then
					msg = dmsg("arrow", name)
				elseif shooter:is_player() then
					msg = dmsg("arrow_name", name, shooter:get_player_name())
				elseif shooter:get_luaentity()._cmi_is_mob then
					if shooter:get_luaentity().nametag ~= "" then
						msg = dmsg("arrow_name", name, shooter:get_player_name())
					else
						msg = dmsg("arrow", name)
					end
				else
					msg = dmsg("arrow", name)
				end
			-- Blaze fireball
			elseif hitter:get_luaentity().name == "mobs_mc:blaze_fireball" then
				if hitter:get_luaentity()._shot_from_dispenser then
					msg = dmsg("fire_charge", name)
				else
					msg = dmsg("blaze_fireball", name)
				end
			-- Ghast fireball
			elseif hitter:get_luaentity().name == "mobs_monster:fireball" then
				msg = dmsg("ghast_fireball", name)
			end
		-- Falling
		elseif reason.type == "fall" then
			msg = dmsg("fall", name)
		-- Other
		elseif reason.type == "set_hp" then
			if last_damages[name] and last_damages[name].custom then
				msg = last_damages[name].message
			end
		end
		if not msg then
			msg = dmsg("other", name)
		end
		minetest.chat_send_all(msg)
	end
end)

local start_damage_reset_countdown = function (player)
	minetest.after(1, function(playername)
		-- FIXME: Fix race condition with many damages in quick succession
		if last_damages[playername] and last_damages[playername].custom then
			last_damages[playername] = nil
		end
	end, player:get_player_name())
end

-- To be called BEFORE damaging a player via set_hp. The next time the player dies due to a set_hp,
-- the message will be shown. This must happen within one second, otherwise it won't work.
function mcl_death_messages.player_damage(player, message)
	last_damages[player:get_player_name()] = { custom = true, message = message }
	start_damage_reset_countdown(player)
end