local _G = getfenv(0)
local oUF = _G.oufgrid or _G.oUF
 
if not oUF then
	return error("oUF_Grid requires oUF")
end
 
local UnitName = UnitName
local UnitClass = UnitClass
local select = select
local unpack = unpack
local UnitDebuff = UnitDebuff
local UnitInRaid = UnitInRaid
 
local width, height = 32, 32
 
local supernova = [[Interface\AddOns\oUF_Grid\media\nokiafc22.ttf]]
local texture = [[Interface\AddOns\oUF_Grid\media\gradient32x32.tga]]
local hightlight = [[Interface\AddOns\oUF_Grid\media\mouseoverHighlight.tga]]
 
local colors = {
	class ={
		-- I accept patches you know
		["DEATHKNIGHT"] = { 0.77, 0.12, 0.23 },
		["DRUID"] = { 1.0 , 0.49, 0.04 },
		["HUNTER"] = { 0.67, 0.83, 0.45 },
		["MAGE"] = { 0.41, 0.8 , 0.94 },
		["PALADIN"] = { 0.96, 0.55, 0.73 },
		["PRIEST"] = { 1.0 , 1.0 , 1.0 },
		["ROGUE"] = { 1.0 , 0.96, 0.41 },
		["SHAMAN"] = { 0,0.86,0.73 },
		["WARLOCK"] = { 0.58, 0.51, 0.7 },
		["WARRIOR"] = { 0.78, 0.61, 0.43 },
	},
}
setmetatable(colors.class, {
	__index = function(self, key)
		return { 0.78, 0.61, 0.43 }
	end
})
 
local GetClassColor = function(unit)
	return unpack(colors.class[select(2, UnitClass(unit))])
end
 
local ColorGradient = function(perc, r1, g1, b1, r2, g2, b2, r3, g3, b3)
	if perc >= 1 then
		return { r3, g3, b3 }
	elseif perc <= 0 then
		return { r1, g1, b1 }
	end
 
	local segment, relperc = math.modf(perc*(2))
	local offset = (segment*3)+1
 
	if(offset == 1) then
		return { r1 + (r2-r1)*relperc, g1 + (g2-g1)*relperc, b1 + (b2-b1)*relperc }
	end
 
	return { r2 + (r3-r2)*relperc, g2 + (g3-g2)*relperc, b2 + (b3-b2)*relperc }
end
 
local Name_Update = function(self, event, unit)
	if self.unit ~= unit then return end
 
	--self:Reset()
 
	local n, s = UnitName(unit)
	self.name = string.utf8sub(n, 1, 3)
 
	if UnitIsDeadOrGhost(unit) or not UnitIsConnected(unit) then
		self.Health:SetStatusBarColor(0.3, 0.3, 0.3)
	else
		local r, g, b = GetClassColor(unit)
		self.Health:SetStatusBarColor(r, g, b)
		self.Health.bg:SetVertexColor(r * 0.3, g * 0.3, b * 0.3)
	end
end
 
local round = function(x, y)
	return math.floor((x * 10 ^ y)+ 0.5) / 10 ^ y
end
 
local Health_Update = function(self, event, unit, bar, current, max)
	if not self.unit then self.unit = unit end
	if unit ~= self.unit then return end
 
	local def = max - current
	bar:SetValue(current)
 
	local per = round(current/max, 100)
	local col = ColorGradient(per, 1, 0, 0, 1, 1, 0, 1, 1, 1)
	self.Name:SetTextColor(unpack(col))
 
	if per > 0.9 or UnitIsDeadOrGhost(unit) then
		self.Name:SetText(self.name)
	else
		self.Name:SetFormattedText("-%0.1f", math.floor(def/100)/10)
	end
 
	if UnitIsDeadOrGhost(unit) or not UnitIsConnected(unit) then
		bar:SetStatusBarColor(0.3, 0.3, 0.3)
	else
		local r, g, b = GetClassColor(unit)
		bar:SetStatusBarColor(r, g, b)
		bar.bg:SetVertexColor(r * 0.3, g * 0.3, b * 0.3)
	end
 
	-- Hopefully this fixes everything ...
	local incHeal = self.incHeal
	if incHeal > 0 then
		if max == current then
			self.heal:Hide()
			return
		end
		local size = height * per
		local incSize = ((self.healMod or 1 * incHeal) / max) * height
 
		if incSize + size >= height then
			incSize = height - size
		end
 
		self.heal:SetHeight(incSize)
		self.heal:SetPoint("BOTTOM", self, "BOTTOM", 0, size)
		self.heal:Show()
	else
		self.heal:Hide()
	end
end
 
local reset = function(self)
	self.border:Hide()
	self.Icon:Hide()
	self.Dispell = false
end
 
local OnEnter = function(self)
	UnitFrame_OnEnter(self)
	self.Highlight:Show()
end
 
local OnLeave = function(self)
	UnitFrame_OnLeave(self)
	self.Highlight:Hide()
end
 
local frame = function(settings, self, unit)
	self.menu = menu
 
	self:EnableMouse(true)
 
	self:SetScript("OnEnter", OnEnter)
	self:SetScript("OnLeave", OnLeave)
 
	self:RegisterForClicks("anyup")
 
	self:SetBackdrop({
		bgFile = "Interface\\Tooltips\\UI-Tooltip-Background", tile = true, tileSize = 16,
		insets = { left = -2, right = -2, top = -2, bottom = -2 },
	})
 
	self:SetBackdropColor(0, 0, 0, 1)
 
	self:SetAttribute("*type2", "menu")
 
	local hp = CreateFrame("StatusBar", nil, self)
	hp:SetAllPoints(self)
	hp:SetStatusBarTexture(texture)
	hp:SetOrientation("VERTICAL")
	-- hp:SetFrameLevel(5)
	--hp:SetAlpha(0.)
 
	local hpbg = hp:CreateTexture(nil, "BACKGROUND")
	hpbg:SetAllPoints(hp)
	hpbg:SetTexture(texture)
	hpbg:SetAlpha(1)
 
	local heal = hp:CreateTexture(nil, "OVERLAY")
	heal:SetHeight(height)
	heal:SetWidth(width)
	heal:SetPoint("BOTTOM")
	heal:SetTexture(texture)
	heal:SetVertexColor(0, 1, 0)
	heal:Hide()
 
	self.heal = heal
 
	hp.bg = hpbg
	self.Health = hp
	self.OverrideUpdateHealth = Health_Update
 
	local hl = hp:CreateTexture(nil, "OVERLAY")
	hl:SetAllPoints(self)
	hl:SetTexture("Interface\\QuestFrame\\UI-QuestTitleHighlight")
	hl:SetBlendMode("ADD")
	hl:Hide()
 
	self.Highlight = hl
 
	local name = hp:CreateFontString(nil, "OVERLAY")
	name:SetAlpha(1)
	name:SetPoint("CENTER")
	name:SetJustifyH("CENTER")
	name:SetFont(supernova, 10, "THINOUTLINE")
	name:SetShadowColor(0,0,0,1)
	name:SetShadowOffset(1, -1)
	name:SetTextColor(1, 1, 1, 1)
 
	self.Name = name
	self.UNIT_NAME_UPDATE = Name_Update
 
	local border = self:CreateTexture(nil, "ARTWORK")
	border:SetPoint("LEFT", self, "LEFT", - 5, 0)
	border:SetPoint("RIGHT", self, "RIGHT", 5, 0)
	border:SetPoint("TOP", self, "TOP", 0, 5)
	border:SetPoint("BOTTOM", self, "BOTTOM", 0, - 5)
	border:SetTexture([[Interface\AddOns\oUF_Grid\media\Normal.tga]])
	border:SetAlpha(1)
	border:SetVertexColor(1, 1, 1)
	border:Hide()
 
	self.border = border
 
	local icon = hp:CreateTexture(nil, "OVERLAY")
	icon:SetPoint("CENTER")
	icon:SetAlpha(1)
	icon:SetHeight(16)
	icon:SetWidth(16)
	icon:SetTexCoord(0.07, 0.93, 0.07, 0.93)
	icon:Hide()
 
	local show, hide = icon.Show, icon.Hide
	icon.Show = function(self)
		show(self)
		name:Hide()
	end
 
	icon.Hide = function(self)
		hide(self)
		name:Show()
	end
 
	self.Icon = icon
 
	local cd = CreateFrame("Cooldown", nil, self)
	cd:SetAllPoints(icon)
	cd:SetFrameLevel(7)
	cd.noCooldownCount = true
 
	self.cd = cd
 
	self.Range = true
	self.inRangeAlpha = 1
	self.outsideRangeAlpha = 0.4
 
	self.incHeal = 0
	self.healMod = 0
 
	self.Reset = reset
 
	return self
end
 
local style = setmetatable({
	["initial-height"] = height,
	["initial-width"] = width,
}, {
	__call = frame,
})
 
oUF:RegisterStyle("Kanne-Grid", style)
oUF:SetActiveStyle("Kanne-Grid")
 
local f = CreateFrame("Frame", "oUF_Grid", UIParent)
f:SetHeight(20)
f:SetWidth(20)
f:SetPoint("CENTER")
f:SetMovable(true)
f:SetUserPlaced(true)
 
local raid = {}
for i = 1, 8 do
	local r = oUF:Spawn("header", "oUF_Raid" .. i)
 
	r:SetParent(f)
 
	r:SetPoint("TOPLEFT", f, "TOPLEFT", 0, 0)
 
	if i == 1 then
		-- As Haste would say;
		-- ZA WARUDO !!!
		r:SetPoint("TOPLEFT", f, "TOPLEFT", 20, 0)
		r:SetAttribute("showParty", true)
		r:SetAttribute("showPlayer", true)
		r:SetAttribute("showSolo", true)
	else
		r:SetPoint("TOPLEFT", raid[i - 1], "TOPRIGHT", 9, 0)
	end
 
	r:SetMovable(true)
 
	r:SetManyAttributes(
		"showRaid", true,
		"groupFilter", i,
		"yOffset", - 9
	)
 
	r:Show()
	raid[i] = r
end
 
local SubGroups = function()
	local t = {}
	for i = 1, 8 do t[i] = 0 end
	for i = 1, GetNumRaidMembers() do
		local s = select(3, GetRaidRosterInfo(i))
		t[s] = t[s] + 1
	end
	return t
end
 
-- BG
local bg = CreateFrame("Frame", nil, f)
bg:SetBackdrop({
	bgFile = "Interface\\ChatFrame\\ChatFrameBackground", tile = true, tileSize = 16,
	edgeFile = "Interface\\Tooltips\\UI-Tooltip-Border", edgeSize = 10,
	insets = {left = 2, right = 2, top = 2, bottom = 2}
})
bg:SetBackdropColor(0, 0, 0, 0.6)
bg:SetBackdropBorderColor(0.4, 0.4, 0.4, 1)
bg:SetFrameLevel(0)
bg:SetMovable(true)
bg:EnableMouse(true)
bg:SetClampedToScreen(true)
 
bg:SetScript("OnMouseUp", function(self, button)
	f:StopMovingOrSizing()
end)
 
bg:SetScript("OnMouseDown", function(self, button)
	if button == "LeftButton" and IsModifiedClick("ALT") then
		f:ClearAllPoints()
		f:StartMoving()
	end
end)
 
bg:SetScript("OnEvent", function(self, event, ...)
	return self[event](self, event, ...)
end)
 
function bg:RAID_ROSTER_UPDATE()
	if not UnitInRaid("player") then
		return self:PARTY_MEMBERS_CHANGED()
	else
		self:Show()
	end
 
	local roster = SubGroups()
 
	local h, last, first = 1
	for k, v in ipairs(roster) do
		if v > 0 then
			if not first then
				first = k
			end
			last = k
		end
		if v > roster[h] then
			h = k
		end
	end
 
	self:ClearAllPoints()
	self:SetPoint("TOP", _G["oUF_Raid1"], "TOP", 0, 8)
	self:SetPoint("LEFT", _G["oUF_Raid" .. first], "LEFT", -8 , 0)
	self:SetPoint("RIGHT", _G["oUF_Raid" .. last], "RIGHT", 8, 0)
	self:SetPoint("BOTTOM", _G["oUF_Raid" .. h], "BOTTOM", 0, -8)
end
 
function bg:PARTY_MEMBERS_CHANGED()
	if UnitInRaid("player") then return end
 
	self:ClearAllPoints()
	self:SetPoint("BOTTOMRIGHT", _G["oUF_Raid1"], "BOTTOMRIGHT", 8, - 8)
	self:SetPoint("TOPLEFT", _G["oUF_Raid1"], "TOPLEFT", - 8, 8)
	self:Show()
end
 
bg.PLAYER_LOGIN = bg.RAID_ROSTER_UPDATE
bg:RegisterEvent("PLAYER_LOGIN")
bg:RegisterEvent("RAID_ROSTER_UPDATE")
bg:RegisterEvent("PARTY_MEMBERS_CHANGED")