--[[
Copyright (c) 2008 Chris Bannister,
All rights reserved.
 
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions
are met:
1. Redistributions of source code must retain the above copyright
   notice, this list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright
   notice, this list of conditions and the following disclaimer in the
   documentation and/or other materials provided with the distribution.
3. The name of the author may not be used to endorse or promote products
   derived from this software without specific prior written permission.
 
THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
]]
 
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
 
	local n, s = UnitName(unit)
	self.name = string.utf8sub(n, 1, 3)
 
	if UnitIsDeadOrGhost(unit) or not UnitIsConnected(unit) then
		self.Health.bg:SetVertexColor(0.3, 0.3, 0.3)
	else
		self.Health.bg:SetVertexColor(GetClassColor(unit))
	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)
	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.bg:SetVertexColor(0.3, 0.3, 0.3)
	else
		bar.bg:SetVertexColor(GetClassColor(unit))
	end
 
	-- Hopefully this fixes everything ...
	local incHeal = self.incHeal
	if incHeal > 0 then
		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)
	end
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:SetMovable(true)
	--self:RegisterForDrag("LleftButton")
 
	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:SetStatusBarColor(0, 0, 0)
	hp:SetAlpha(0.9)
 
	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 = hp:CreateTexture(nil, "ARTWORK")
	border:SetPoint("LEFT", self, "LEFT", -4, 0)
	border:SetPoint("RIGHT", self, "RIGHT", 4, 0)
	border:SetPoint("TOP", self, "TOP", 0, 4)
	border:SetPoint("BOTTOM", self, "BOTTOM", 0, -4)
	border:SetTexture([[Interface\AddOns\oUF_Grid\media\Normal.tga]])
	border:SetAlpha(1)
	border:Hide()
	border:SetVertexColor(1, 1, 1)
 
	self.border = border
 
	local icon = hp:CreateTexture(nil, "OVERLAY")
	icon:SetPoint("CENTER")
	icon:SetAlpha(1)
	icon:SetHeight(20)
	icon:SetWidth(20)
	icon:SetTexCoord(0.07, 0.93, 0.07, 0.93)
	icon:Hide()
 
	icon.ShowText = function(s)
		self.Name:Hide()
		s:Show()
	end
 
	icon.HideText = function(s)
		self.Name:Show()
		s:Hide()
	end
 
	self.Icon = icon
 
	self.Range = true
	self.inRangeAlpha = 1
	self.outsideRangeAlpha = 0.4
 
	self.incHeal = 0
	self.healMod = 0
 
	return self
end
 
local style = setmetatable({
	["initial-height"] = height,
	["initial-width"] = width,
}, {
	__call = frame,
})
 
oUF:RegisterStyle("Kanne-Grid", style)
oUF:SetActiveStyle("Kanne-Grid")
 
local raid = {}
for i = 1, 8 do
	local r = oUF:Spawn("header", "oUF_Raid" .. i)
	r:SetPoint("TOP", UIParent, "TOP", 0, -500)
	if i == 1 then
		r:SetPoint("LEFT", UIParent, "LEFT", 10, 0)
		r:SetAttribute("showParty", true)
		r:SetAttribute("showPlayer", true)
		r:SetAttribute("showSolo", true)
	else
		r:SetPoint("LEFT", raid[i - 1], "RIGHT", 6, 0)
	end
 
	r:SetManyAttributes(
		"showRaid", true,
		"groupFilter", i,
		"yOffset", - 10
	)
 
	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")
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: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")