local T = wml.tag

function wesnoth.wml_actions.heal_unit(cfg)
	local healers = wml.get_child(cfg, "filter_second")
	if healers then
		healers = wesnoth.units.find_on_map{
			ability_type = "heals",
			T["and"](healers)
		}
	else
		healers = {}
	end

	local who = wml.get_child(cfg, "filter")
	if who then
		who = wesnoth.units.find_on_map(who)
	else
		who = wesnoth.units.find_on_map{
			x = wesnoth.current.event_context.x1,
			y = wesnoth.current.event_context.y1
		}
	end

	local heal_full = cfg.amount == "full" or cfg.amount == nil
	local moves_full = cfg.moves == "full"
	local variable = cfg.variable
	for i,u in ipairs(who) do
		local heal_amount = u.max_hitpoints - u.hitpoints
		local new_hitpoints
		if heal_full then
			new_hitpoints = u.max_hitpoints
		else
			heal_amount = tonumber(cfg.amount) or heal_amount
			new_hitpoints = math.floor(math.max(1, math.min(u.max_hitpoints, u.hitpoints + heal_amount)))
			heal_amount = new_hitpoints - u.hitpoints
		end

		if cfg.animate then
			local animator = wesnoth.units.create_animator()
			if heal_amount > 0 then
				animator:add(u, 'healed', 'hit', {value = heal_amount, text = heal_amount, color = {0, 255, 0}})
			end
			if #healers > 0 then
				animator:add(healers[1], 'healing', 'hit', {value = heal_amount})
			end
			animator:run()
		end

		u.hitpoints = new_hitpoints

		if moves_full then
			u.moves = u.max_moves
		else
			u.moves = math.min(u.max_moves, u.moves + (cfg.moves or 0))
		end

		if cfg.restore_attacks then
			u.attacks_left = u.max_attacks
		end

		if cfg.restore_statuses == true or cfg.restore_statuses == nil then
			u.status.poisoned = false
			u.status.petrified = false
			u.status.slowed = false
			u.status.unhealable = false
		end

		if variable then
			wml.variables[string.format("%s[%d]", variable, i-1)] = {id = u.id, heal_amount = heal_amount}
		end
	end
end