local Invoker = require("Invoker")

local LastHit = {}

local optionAwareness = Menu.AddOption({"Utility"}, "Last Hit Awareness", "On/Off")
local optionEnable = Menu.AddOption({"Utility"}, "Last Hit Enable", "On/Off")
local key = Menu.AddKeyOption({"Utility"}, "Last Hit Key", Enum.ButtonCode.KEY_SPACE)

local target1, target2

function LastHit.OnUpdate()
	if not Menu.IsEnabled(optionEnable) then return end
	if not Menu.IsKeyDown(key) then return end

	local myHero = Heroes.GetLocal()
	if not myHero then return end

	local range = NPC.GetAttackRange(myHero)
	local target

	if target2 and Entity.IsAlive(target2) and not Entity.IsDormant(target2) and NPC.IsEntityInRange(myHero, target2, range) then target = target2 end
	if target1 and Entity.IsAlive(target1) and not Entity.IsDormant(target1) and NPC.IsEntityInRange(myHero, target1, range) then target = target1 end

	if NPC.GetUnitName(myHero) == "npc_dota_hero_invoker" then Invoker.PressKey(myHero, "EEE") end

	if target then Player.AttackTarget(Players.GetLocal(), myHero, target) end
end

function LastHit.OnDraw()
	if not Menu.IsEnabled(optionAwareness) then return end

	local myHero = Heroes.GetLocal()
	if not myHero then return end
	if not NPC.IsVisible(myHero) then return end

	local radius = 1000
	local creeps = NPC.GetUnitsInRadius(myHero, radius, Enum.TeamType.TEAM_ENEMY)
	for i, npc in ipairs(creeps) do
		local oneHitDamage = LastHit.GetOneHitDamageVersus(myHero, npc)
		if NPC.IsCreep(npc) and Entity.IsAlive(npc) and not Entity.IsDormant(npc) then

			local x, y, visible = Renderer.WorldToScreen(Entity.GetAbsOrigin(npc))
			local size = 10

			if Entity.GetHealth(npc) <= oneHitDamage then
				Renderer.SetDrawColor(255, 0, 0, 150)
				Renderer.DrawFilledRect(x-size, y-size, 2*size, 2*size)
				target1 = npc
			elseif Entity.GetHealth(npc) <= 1.5 * oneHitDamage then
				Renderer.SetDrawColor(255, 255, 0, 150)
				Renderer.DrawFilledRect(x-size, y-size, 2*size, 2*size)
				target2 = npc
			end
		end
	end
end

function LastHit.GetOneHitDamageVersus(myHero, npc)
	if not myHero or not npc then return 0 end

	local damage = NPC.GetTrueDamage(myHero) * NPC.GetArmorDamageMultiplier(npc)

	if NPC.GetUnitName(myHero) == "npc_dota_hero_invoker" and Invoker.GetInstances(myHero) ~= "EEE" then
		local E = NPC.GetAbility(myHero, "invoker_exort")
		local extra_damage = 12 * Ability.GetLevel(E)
		damage = damage + extra_damage
	end

	return damage
end

return LastHit