local ADDON_NAME, GF = ...

GF.Icons = {}

local NAV_CARD_NORMAL_ATLAS = "transmog-outfit-card"
local NAV_CARD_SELECTED_ATLAS = "transmog-outfit-card-selected"
local COMMON_BUTTON_PATH = GF.COMMON_BUTTON_TEXTURE or ("Interface\\AddOns\\" .. (ADDON_NAME or "GroupFinder") .. "\\Art\\UI\\RedButton.png")
local COMMON_BUTTON_ATLAS_W = 392
local COMMON_BUTTON_ATLAS_H = 168
local COMMON_BUTTON_SQUARE = { 344 / COMMON_BUTTON_ATLAS_W, 392 / COMMON_BUTTON_ATLAS_W }
local COMMON_BUTTON_STATES = {
	normal = { 0, 48 / COMMON_BUTTON_ATLAS_H },
	highlight = { 60 / COMMON_BUTTON_ATLAS_H, 108 / COMMON_BUTTON_ATLAS_H },
	pressed = { 120 / COMMON_BUTTON_ATLAS_H, 168 / COMMON_BUTTON_ATLAS_H },
}
local NAV_EXPANDER_SIZE = 24
local NAV_EXPANDER_PLUS_W = 12
local NAV_EXPANDER_PLUS_H = 12
local NAV_EXPANDER_MINUS_W = 12
local NAV_EXPANDER_LEFT = 4
local NAV_EXPANDER_TEXT_GAP = 4
local NAV_EXPANDER_GLYPH_TEXTURE = "Interface\\QuestFrame\\QuestTracker2x"
local NAV_EXPANDER_GLYPH_ATLAS_W = 1024
local NAV_EXPANDER_GLYPH_ATLAS_H = 512
local NAV_EXPANDER_PLUS_TEXCOORD = { 985 / 1024, 1011 / 1024, 165 / 512, 191 / 512 }
local NAV_EXPANDER_PLUS_PRESSED_TEXCOORD = { 987 / 1024, 1014 / 1024, 201 / 512, 228 / 512 }
local NAV_EXPANDER_MINUS_TEXCOORD = { 993 / 1024, 1019 / 1024, 71 / 512, 81 / 512 }
local NAV_EXPANDER_MINUS_PRESSED_TEXCOORD = { 949 / 1024, 976 / 1024, 215 / 512, 225 / 512 }
local NAV_EXPANDER_HIGHLIGHT_ATLAS = "ui-questtrackerbutton-red-highlight"

local function navLevel(level)
	level = tonumber(level) or 0
	if level < 0 then
		return 0
	end
	if level > 3 then
		return 3
	end
	return level
end

local function navIndent(level)
	return 8 + level * (GF.NAV_INDENT or 12)
end

local function levelValue(value, level, fallback)
	if type(value) == "table" then
		local resolved = value[level]
		if resolved == nil then
			resolved = value[0]
		end
		if resolved == nil then
			resolved = fallback
		end
		return resolved
	end
	if value == nil then
		return fallback
	end
	return value
end

local function navButtonMetrics(rowW, level)
	rowW = math.max(tonumber(rowW) or 1, 1)
	level = navLevel(level)
	local widthPad = levelValue(GF.NAV_TRANSMOG_BUTTON_WIDTH_PAD, level, 0)
	local buttonW = math.max(rowW - widthPad, 1)
	buttonW = math.min(buttonW, rowW)
	local buttonH = level == 0 and (GF.NAV_L0_BUTTON_H or 43) or levelValue(GF.NAV_TRANSMOG_BUTTON_H, level, 43)
	local offsetX = tonumber(GF.NAV_TRANSMOG_BUTTON_OFFSET_X) or 0
	local buttonX = math.floor(((rowW - buttonW) * 0.5) + offsetX + 0.5)
	return buttonX, buttonW, buttonH
end

local function setAtlas(texture, atlas)
	if not texture or not texture.SetAtlas then
		return false
	end
	local ok = pcall(texture.SetAtlas, texture, atlas, false)
	if not ok then
		ok = pcall(texture.SetAtlas, texture, atlas)
	end
	return ok
end

local function setupCardTexture(texture, atlas, blendMode)
	if not texture then
		return
	end
	setAtlas(texture, atlas)
	if texture.SetBlendMode then
		texture:SetBlendMode(blendMode or "BLEND")
	end
	texture:SetVertexColor(1, 1, 1, 1)
end

local function layoutCardTexture(texture, row, x, width, height)
	if not texture or not row then
		return
	end
	texture:ClearAllPoints()
	texture:SetPoint("LEFT", row, "LEFT", x or 0, 0)
	texture:SetSize(math.max(width or 1, 1), math.max(height or 1, 1))
end

local function setDesaturated(texture, desaturated)
	if texture and texture.SetDesaturated then
		texture:SetDesaturated(desaturated == true)
	end
end

local function navTextColor(row, disabled, selected, hover, dimmed)
	if disabled then
		return 0.55, 0.55, 0.55, 1
	end
	if (row and row._navPressed) or selected then
		return 1, 1, 1, 1
	end
	if hover then
		return 1, 0.95, 0.45, 1
	end
	if dimmed then
		return 0.48, 0.48, 0.48, 1
	end
	return 1, 0.82, 0, 1
end

local function setCommonSquareTexCoord(texture, state)
	if not texture then
		return
	end
	state = state or COMMON_BUTTON_STATES.normal
	texture:SetTexCoord(COMMON_BUTTON_SQUARE[1], COMMON_BUTTON_SQUARE[2], state[1], state[2])
end

local function setNavExpanderGlyphTexCoord(texture, coords)
	if not texture or not coords then
		return
	end
	texture:SetTexCoord(coords[1], coords[2], coords[3], coords[4])
end

local function getNavExpanderGlyphHeight(width, coords)
	local coordW = coords and ((coords[2] - coords[1]) * NAV_EXPANDER_GLYPH_ATLAS_W) or 0
	local coordH = coords and ((coords[4] - coords[3]) * NAV_EXPANDER_GLYPH_ATLAS_H) or 0
	if coordW <= 0 or coordH <= 0 then
		return width
	end
	return width * (coordH / coordW)
end

local function setupNavExpander(row)
	if not row or not row.expander then
		return
	end
	row.expander:SetTexture(COMMON_BUTTON_PATH)
	setCommonSquareTexCoord(row.expander, COMMON_BUTTON_STATES.normal)
	row.expander:SetVertexColor(1, 1, 1, 1)
	row.expander:SetShown(false)
	if row.expander.SetBlendMode then
		row.expander:SetBlendMode("BLEND")
	end
	if not row.expanderGlyph then
		row.expanderGlyph = row:CreateTexture(nil, "OVERLAY", nil, 5)
	end
	row.expanderGlyph:SetTexture(NAV_EXPANDER_GLYPH_TEXTURE)
	setNavExpanderGlyphTexCoord(row.expanderGlyph, NAV_EXPANDER_PLUS_TEXCOORD)
	if row.expanderGlyph.SetBlendMode then
		row.expanderGlyph:SetBlendMode("BLEND")
	end
	row.expanderGlyph:SetVertexColor(1, 1, 1, 1)
	row.expanderGlyph:SetShown(false)
	if not row.expanderHighlight then
		row.expanderHighlight = row:CreateTexture(nil, "OVERLAY", nil, 4)
	end
	setAtlas(row.expanderHighlight, NAV_EXPANDER_HIGHLIGHT_ATLAS)
	if row.expanderHighlight.SetBlendMode then
		row.expanderHighlight:SetBlendMode("ADD")
	end
	row.expanderHighlight:SetVertexColor(1, 1, 1, 1)
	row.expanderHighlight:SetAlpha(0.85)
	row.expanderHighlight:SetShown(false)
	if row.expanderSign then
		row.expanderSign:Hide()
	end
end

local function applyNavExpander(row)
	if not row or not row.expander then
		return
	end
	if not row._navCanExpand then
		row.expander:Hide()
		if row.expanderGlyph then
			row.expanderGlyph:Hide()
		end
		if row.expanderHighlight then
			row.expanderHighlight:Hide()
		end
		if row.expanderSign then
			row.expanderSign:Hide()
		end
		return
	end
	local disabled = row.nodeData and row.nodeData.disabled
	local selected = row._navSelected and not disabled
	local pressed = row._navPressed and not disabled
	local hover = row._navHover and not disabled
	local dimmed = row._navDimmed and not selected and not pressed and not disabled
	local bgState = (row._navPressed and COMMON_BUTTON_STATES.pressed)
		or (hover and COMMON_BUTTON_STATES.highlight)
		or COMMON_BUTTON_STATES.normal
	setCommonSquareTexCoord(row.expander, bgState)
	local glyphTexCoord
	local glyphW
	local glyphH
	if row._navExpanded then
		glyphTexCoord = row._navPressed and NAV_EXPANDER_MINUS_PRESSED_TEXCOORD or NAV_EXPANDER_MINUS_TEXCOORD
		glyphW = NAV_EXPANDER_MINUS_W
		glyphH = getNavExpanderGlyphHeight(glyphW, glyphTexCoord)
	else
		glyphTexCoord = row._navPressed and NAV_EXPANDER_PLUS_PRESSED_TEXCOORD or NAV_EXPANDER_PLUS_TEXCOORD
		glyphW = NAV_EXPANDER_PLUS_W
		glyphH = NAV_EXPANDER_PLUS_H
	end
	setDesaturated(row.expander, dimmed and not hover)
	if dimmed and not hover then
		row.expander:SetVertexColor(0.58, 0.58, 0.58, 1)
	else
		row.expander:SetVertexColor(1, 1, 1, disabled and 0.5 or 1)
	end
	if row.expanderGlyph then
		local r, g, b, a = navTextColor(row, disabled, selected, hover, dimmed)
		row.expanderGlyph:SetTexture(NAV_EXPANDER_GLYPH_TEXTURE)
		setNavExpanderGlyphTexCoord(row.expanderGlyph, glyphTexCoord)
		row.expanderGlyph:ClearAllPoints()
		row.expanderGlyph:SetPoint("CENTER", row.expander, "CENTER", 0, 0)
		row.expanderGlyph:SetSize(glyphW, glyphH)
		setDesaturated(row.expanderGlyph, dimmed and not hover)
		row.expanderGlyph:SetVertexColor(r, g, b, a)
		row.expanderGlyph:SetShown(true)
	end
	if row.expanderHighlight then
		row.expanderHighlight:ClearAllPoints()
		row.expanderHighlight:SetPoint("CENTER", row.expander, "CENTER", 0, 0)
		row.expanderHighlight:SetSize(NAV_EXPANDER_SIZE, NAV_EXPANDER_SIZE)
		row.expanderHighlight:SetShown(row._navHover and not disabled)
	end
	if row.expanderSign then
		row.expanderSign:Hide()
	end
	row.expander:Show()
end

local function ensureNavTextureSetup(row)
	if not row then
		return
	end
	if not row._gfTransmogNavTextures then
		setupCardTexture(row.bg, NAV_CARD_NORMAL_ATLAS)
		setupCardTexture(row.hover, NAV_CARD_NORMAL_ATLAS, "ADD")
		setupCardTexture(row.sel, NAV_CARD_SELECTED_ATLAS)
		if row.cover then
			row.cover:Hide()
		end
		if row.expander then
			row.expander:Hide()
			setupNavExpander(row)
		end
		row._gfTransmogNavTextures = true
	end
	local w = row._navWidth or row:GetWidth() or 1
	local buttonX, buttonW, buttonH = navButtonMetrics(w, row._navLevel or 0)
	row._navButtonX = buttonX
	row._navButtonW = buttonW
	row._navButtonH = buttonH
	layoutCardTexture(row.bg, row, buttonX, buttonW, buttonH)
	layoutCardTexture(row.hover, row, buttonX, buttonW, buttonH)
	layoutCardTexture(row.sel, row, buttonX, buttonW, buttonH)
	if row.bg then
		row.bg:Show()
	end
	if row.cover then
		row.cover:Hide()
	end
	if row.expander then
		row.expander:Hide()
		if row.expanderGlyph then
			row.expanderGlyph:Hide()
		end
		if row.expanderHighlight then
			row.expanderHighlight:Hide()
		end
		if row.expanderSign then
			row.expanderSign:Hide()
		end
	end
end

function GF.Icons.ApplyNavButtonState(row)
	if not row or not row.bg then
		return
	end
	ensureNavTextureSetup(row)
	local disabled = row.nodeData and row.nodeData.disabled
	local selected = row._navSelected and not disabled
	local pressed = row._navPressed and not disabled
	local hover = row._navHover and not disabled
	local dimmed = row._navDimmed and not selected and not disabled
	if row.bg then
		setDesaturated(row.bg, false)
		row.bg:SetVertexColor(1, 1, 1, 1)
		row.bg:SetAlpha((disabled and 0.35) or 1)
		row.bg:Show()
	end
	if row.hover then
		setDesaturated(row.hover, false)
		row.hover:SetVertexColor(1, 1, 1, 1)
		row.hover:SetAlpha(hover and not pressed and 0.8 or 0)
		row.hover:SetShown(hover and not pressed)
	end
	if row.sel then
		setDesaturated(row.sel, false)
		row.sel:SetVertexColor(1, 1, 1, 1)
		row.sel:SetAlpha((selected and 1) or (pressed and 0.82) or 0)
		row.sel:SetShown(selected or pressed)
	end
	if row.cover then
		row.cover:SetAlpha(0)
		row.cover:Hide()
	end
	applyNavExpander(row)
	if row.label then
		row.label:SetTextColor(navTextColor(row, disabled, selected, hover, dimmed))
	end
end

function GF.Icons.ApplyNavButton(row, level, label, rowW, rowH, opts)
	if not row then
		return
	end
	opts = opts or {}
	level = navLevel(level)
	row._navLevel = level
	row._navWidth = rowW
	row._navHeight = rowH
	row._navSelected = opts.selected == true
	row._navCanExpand = opts.canExpand == true
	row._navExpanded = opts.expanded == true
	row._navDimmed = opts.dimmed == true

	if row.label then
		row.label:SetText(label or "")
		row.label:ClearAllPoints()
		local buttonX, buttonW, buttonH = navButtonMetrics(rowW, level)
		local textLeft = levelValue(GF.NAV_TRANSMOG_BUTTON_TEXT_LEFT, level, 12)
		local textRight = levelValue(GF.NAV_TRANSMOG_BUTTON_TEXT_RIGHT, level, 8)
		local hasExpander = row._navCanExpand
		if hasExpander then
			if row.expander then
				row.expander:ClearAllPoints()
				row.expander:SetPoint("CENTER", row, "LEFT", buttonX + NAV_EXPANDER_LEFT + (NAV_EXPANDER_SIZE * 0.5), 0)
				row.expander:SetSize(NAV_EXPANDER_SIZE, NAV_EXPANDER_SIZE)
			end
			textLeft = math.max(textLeft, NAV_EXPANDER_LEFT + NAV_EXPANDER_SIZE + NAV_EXPANDER_TEXT_GAP)
		end
		local textInset = math.max(textLeft, textRight)
		local textW = math.max(buttonW - (textInset * 2), 1)
		row.label:SetSize(textW, buttonH)
		row.label:SetPoint("CENTER", row, "LEFT", buttonX + textInset + (textW * 0.5), 0)
		row.label:SetJustifyH("CENTER")
		row.label:SetJustifyV("MIDDLE")
		row.label:SetWordWrap(false)
		row.label:SetShown(true)
	end

	GF.Icons.ApplyNavButtonState(row)
end

function GF.Icons.ApplyNavL0Banner(row, _navKey, label, rowW, rowH)
	GF.Icons.ApplyNavButton(row, 0, label, rowW, rowH, {
		selected = row and row._navSelected,
	})
end

function GF.Icons.ApplySelectionHighlight(tex, selected)
	if not tex then
		return
	end
	local ok = setAtlas(tex, GF.NAV_FLYOUT_SELECTED_ATLAS or "groupfinder-highlightbar-yellow")
	if ok then
		tex:SetBlendMode("ADD")
		tex:SetAlpha(1)
	else
		tex:SetColorTexture(1, 0.74, 0.18, selected and 0.45 or 0)
	end
	tex:SetShown(selected == true)
end

function GF.Icons.EnsureNavHighlightLayout(tex, row, width, height, offsetX, offsetY)
	if not tex or not row then
		return
	end
	tex:ClearAllPoints()
	tex:SetPoint("LEFT", row, "LEFT", offsetX or 0, offsetY or 0)
	tex:SetSize(math.max(width or 1, 1), math.max(height or 1, 1))
end
