--{{{ Libraries

local RL = AceLibrary("RosterLib-2.0")

--}}}

GridStatusHealth = GridStatus:NewModule("GridStatusHealth")
GridStatusHealth.menuName = "Health"

--{{{ AceDB defaults

GridStatusHealth.defaultDB = {
	debug = true,
	unit_health = {
		enable = true,
		color = { r = 1, g = 1, b = 1, a = 1 },
		priority = 30,
		range = true,
		deadAsFullHealth = true,
		useClassColors = true,
	},
	unit_healthDeficit = {
		enable = true,
		color = { r = 1, g = 1, b = 1, a = 1 },
		priority = 30,
		threshold = 80,
		range = true,
		useClassColors = true,
	},
	alert_lowHealth = {
		text = "Low HP",
		enable = true,
		color = { r = 1, g = 1, b = 1, a = 1 },
		priority = 30,
		threshold = 80,
		range = true,
	},
	alert_death = {
		text = "DEAD",
		enable = true,
		color = { r = 0, g = 0, b = 0, a = 1 },
		priority = 50,
		range = false,
	},
	alert_offline = {
		text = "Offline",
		enable = true,
		color = { r = 0, g = 0, g = 0, a = 0.6 },
		priority = 0,
		range = false,
	},
}

--}}}

--{{{ AceOptions table

--GridStatusHealth.extraOptions = {}

--}}}

local healthOptions = {
	["deadAsFullHealth"] = {
		type = "toggle",
		name = "Show dead as full health",
		desc = "Treat dead units as being full health.",
		get = function ()
			      return GridStatusHealth.db.profile.unit_health.deadAsFullHealth
		      end,
		set = function (v)
			      GridStatusHealth.db.profile.unit_health.deadAsFullHealth = v
		      end,
	},
	["useClassColors"] = {
		type = "toggle",
		name = "Use class color",
		desc = "Color health based on class.",
		get = function ()
			      return GridStatusHealth.db.profile.unit_health.useClassColors
		      end,
		set = function (v)
			      GridStatusHealth.db.profile.unit_health.useClassColors = v
		      end,
	},
}

local healthDeficitOptions = {
	["threshold"] = {
		type = "range",
		name = "Health threshold",
		desc = "Only show deficit above % damage.",
		max = 100,
		min = 0,
		step = 1,
		get = function ()
			      return GridStatusHealth.db.profile.unit_healthDeficit.threshold
		      end,
		set = function (v)
			      GridStatusHealth.db.profile.unit_healthDeficit.threshold = v
		      end,
	},
	["useClassColors"] = {
		type = "toggle",
		name = "Use class color",
		desc = "Color deficit based on class.",
		get = function ()
			      return GridStatusHealth.db.profile.unit_healthDeficit.useClassColors
		      end,
		set = function (v)
			      GridStatusHealth.db.profile.unit_healthDeficit.useClassColors = v
		      end,
	},
}

local low_healthOptions = {
	["threshold"] = {
		type = "range",
		name = "Low HP threshold",
		desc = "Set the HP % for the low HP warning.",
		max = 100,
		min = 0,
		step = 1,
		get = function ()
			      return GridStatusHealth.db.profile.alert_lowHealth.threshold
		      end,
		set = function (v)
			      GridStatusHealth.db.profile.alert_lowHealth.threshold = v
		      end,
	},
}

function GridStatusHealth:OnInitialize()
	self.super.OnInitialize(self)
	self:RegisterStatus("unit_health", "Unit health", healthOptions)
	self:RegisterStatus("unit_healthDeficit", "Health deficit", healthDeficitOptions)
	self:RegisterStatus("alert_lowHealth", "Low HP warning", low_healthOptions)
	self:RegisterStatus("alert_death", "Death warning")
	self:RegisterStatus("alert_offline", "Offline warning")
end

function GridStatusHealth:OnEnable()
	self:RegisterEvent("Grid_UnitCreated")
	self:RegisterBucketEvent("UNIT_HEALTH", 0.2)
end

function GridStatusHealth:UNIT_HEALTH(units)
	local unitid
	local settings = self.db.profile.unit_health

	for unitid in pairs(units) do
		self:UpdateUnit(unitid)
	end
end

function GridStatusHealth:Grid_UnitCreated(name)
	local unitid = RL:GetUnitIDFromName(name)
	if unitid then
		self:UpdateUnit(unitid, true)
		self:UpdateUnit(unitid)
	end

end

function GridStatusHealth:UpdateUnit(unitid, ignoreRange)
	local cur, max = UnitHealth(unitid), UnitHealthMax(unitid)
	local name = UnitName(unitid)
	local settings = self.db.profile.unit_health
	local deficitSettings = self.db.profile.unit_healthDeficit
	local healthText
	local priority = settings.priority
	
	if not name then return end

	self:StatusDeath(unitid, UnitIsDeadOrGhost(unitid))
	self:StatusOffline(unitid, not UnitIsConnected(unitid))

	if settings.deadAsFullHealth and UnitIsDeadOrGhost(unitid) then
		cur = max
	end

	if cur < max then
		healthText = self:FormatHealthText(cur,max)
	else
		priority = 1
	end

	if self:IsLowHealth(name, cur, max) and
		not UnitIsDeadOrGhost(unitid) then
		self:StatusLowHealth(unitid, true)
	else
		self:StatusLowHealth(unitid, false)
	end

	if (cur / max * 100) <= deficitSettings.threshold then
		GridStatus:SendStatusGained(name, "unit_healthDeficit",
					    deficitSettings.priority,
					    (deficitSettings.range and 40),
					    (deficitSettings.useClassColors and self:UnitClassColor(name) or
					     deficitSettings.color),
					    healthText,
					    cur, max,
					    nil)
	else
		GridStatus:SendStatusLost(name, "unit_healthDeficit")
	end

	GridStatus:SendStatusGained(name, "unit_health",
				    priority,
				    (not ignoreRange and settings.range and 40),
				    (settings.useClassColors and self:UnitClassColor(name) or
				     settings.color),
				    healthText,
				    cur, max,
				    nil)
end

function GridStatusHealth:FormatHealthText(cur, max)
	local healthText
	local deficit = max - cur
	if deficit > 999 then
		healthText = string.format("-%.1fk", deficit/1000.0)
	else
		healthText = string.format("-%d", deficit)
	end

	return healthText
end

function GridStatusHealth:UnitClassColor(name)
	local u = RL:GetUnitObjectFromName(name)
	return (u and RAID_CLASS_COLORS[u.class])
end

function GridStatusHealth:IsLowHealth(name, cur, max)
	return (cur / max * 100) <= self.db.profile.alert_lowHealth.threshold
end

function GridStatusHealth:StatusLowHealth(unitid, gained)
	local name = UnitName(unitid)
	local settings = self.db.profile.alert_lowHealth

	-- return if this option isnt enabled
	if not settings.enable then return end

	if gained then
		GridStatus:SendStatusGained(name, "alert_lowHealth",
					    settings.priority,
					    (settings.range and 40),
					    settings.color,
					    settings.text,
					    nil,
					    nil,
					    nil)
	else
		GridStatus:SendStatusLost(name, "alert_lowHealth")
	end
end

function GridStatusHealth:StatusDeath(unitid, gained)
	local name = UnitName(unitid)
	local settings = self.db.profile.alert_death
	
	if not name then return end

	-- return if this option isnt enabled
	if not settings.enable then return end

	if gained then
		-- trigger death event for other modules as wow isnt firing a death event
		self:TriggerEvent("Grid_UnitDeath", name)
		GridStatus:SendStatusGained(name, "alert_death",
					    settings.priority,
					    (settings.range and 40),
					    settings.color,
					    settings.text,
					    (self.db.profile.unit_health.deadAsFullHealth and 100 or 0),
					    100,
					    nil)
	else
		GridStatus:SendStatusLost(name, "alert_death")
	end
end

function GridStatusHealth:StatusOffline(unitid, gained)
	local name = UnitName(unitid)
	local settings = self.db.profile.alert_offline
	
	if not name then return end

	if gained then
		-- trigger offline event for other modules
		self:TriggerEvent("Grid_UnitOffline", name)
		GridStatus:SendStatusGained(name, "alert_offline",
					    settings.priority,
					    (settings.range and 40),
					    settings.color,
					    settings.text,
					    nil,
					    nil,
					    nil)
	else
		GridStatus:SendStatusLost(name, "alert_offline")
	end
end