local MogIt,mog = ...;
local L = mog.L;

local f = mog:CreateFilter("quality");
local colours = ITEM_QUALITY_COLORS;
-- Add Heirloom to colours - they are missing from the constant for
-- some reason. The values are from GetItemQualityColor(7) on a 3.3.5
-- client.
colours[7] = {
	hex = "|cffe6cc80",
	r = 0.90196078431373,
	g = 0.8,
	b = 0.50196078431373,
}
local selected;
local num;
local all;

f:SetHeight(41);

f.quality = f:CreateFontString(nil,"ARTWORK","GameFontHighlightSmall");
f.quality:SetPoint("TOPLEFT",f,"TOPLEFT",0,0);
f.quality:SetPoint("RIGHT",f,"RIGHT",0,0);
f.quality:SetText(QUALITY..":");
f.quality:SetJustifyH("LEFT");

f.dd = CreateFrame("Frame","MogItFiltersQualityDropdown",f,"UIDropDownMenuTemplate");
f.dd:SetPoint("TOPLEFT",f.quality,"BOTTOMLEFT",-16,-2);
UIDropDownMenu_SetWidth(f.dd,125);
UIDropDownMenu_SetButtonWidth(f.dd,140);
UIDropDownMenu_JustifyText(f.dd,"LEFT");

function f.dd.SelectAll(self)
	num = 0;
	for k,v in ipairs(L.quality) do
		selected[v] = all;
		num = num + (all and 1 or 0);
	end
	all = not all;
	UIDropDownMenu_SetText(f.dd,L["%d selected"]:format(num));
	ToggleDropDownMenu(1,nil,f.dd);
	mog:BuildList();
end

function f.dd.Tier1(self)
	if selected[self.value] and (not self.checked) then
		num = num - 1;
	elseif (not selected[self.value]) and self.checked then
		num = num + 1;
	end
	selected[self.value] = self.checked;
	UIDropDownMenu_SetText(f.dd,L["%d selected"]:format(num));
	mog:BuildList();
end

function f.dd.initialize(self)
	local info;
	info = UIDropDownMenu_CreateInfo();
	info.text =	all and L["Select All"] or L["Select None"];
	info.func = f.dd.SelectAll;
	info.notCheckable = true;
	UIDropDownMenu_AddButton(info);
	for i,v in ipairs(L.quality) do
			info = UIDropDownMenu_CreateInfo();
			info.text =	_G["ITEM_QUALITY"..v.."_DESC"];
			info.value = v;
			info.colorCode = colours[v].hex;
			info.func = f.dd.Tier1;
			info.keepShownOnClick = true;
			info.isNotRadio = true;
			info.checked = selected[v];
			UIDropDownMenu_AddButton(info);
	end
end

function f.Filter(qual)
	return (not qual) or selected[qual];
end

function f.Default()
	selected = {};
	num = 0;
	all = nil;
	for i,v in ipairs(L.quality) do
		selected[v] = true;
		num = num + 1;
	end
	UIDropDownMenu_SetText(f.dd,L["%d selected"]:format(num));
end
f.Default();
