local _, env = ...;
local HANDLER, LDD = CPAPI.CreateEventHandler({'Frame'}, {'PLAYER_REGEN_DISABLED'}), LibStub('LibUIDropDownMenu-4.0');

local function CacheAvailableBinding(bindings, prune, binding, category, key, ...)
	if key then
		if prune and IsBindingForGamePad(key) then
			return -- handle overlap if pruning enabled
		end
	else
		if binding then
			category = category or BINDING_HEADER_OTHER;
			bindings[category] = bindings[category] or {};
			bindings[category][#bindings[category] + 1] = binding;
		end
		return
	end
	return CacheAvailableBinding(bindings, prune, binding, category, ...)
end

local function GetRealBindingName(binding)
	return _G[('BINDING_NAME_%s'):format(binding) or binding]
		or (select(3, env.db.Bindings:GetDescriptionForBinding(binding)))
		or GetBindingName(binding)
end

local function OnBindingClick(self)
	if SetBinding(HANDLER:GetBindingString(), self.value) then
		CPAPI.Log('Binding %s was successfully changed to |cFFFFFFFF%s|r.', 
			ConsolePort:GetFormattedButtonCombination(HANDLER:GetBinding()),
			GetRealBindingName(self.value)
		);
	end
	SaveBindings(GetCurrentBindingSet())
	HANDLER:Close()
end

local function OnButtonEnter(self)
	if self.arg1 and self.arg1 ~= CPAPI.ExtraActionButtonID then 
		GameTooltip:SetOwner(self, 'ANCHOR_BOTTOMRIGHT')
		GameTooltip:SetAction(self.arg1)
	end
end

local function OnButtonLeave(self)
	if GameTooltip:IsOwned(self) then
		GameTooltip:Hide()
	end
end

local function OnBindingCancel(self)
	HANDLER:Close()
end

local function DivideTable(tbl, num)
	if not next(tbl) then return end
	return {unpack(tbl, 1, num)}, DivideTable({unpack(tbl, num+1)}, num)
end

local function ShowBindingDropdown(frame, level, menuList)
	local info = LDD:UIDropDownMenu_CreateInfo()
	local allBindings = {};
	local pruningEnabled = not env.db('bindingOverlapEnable')

	for i=1, GetNumBindings() do
		CacheAvailableBinding(allBindings, pruningEnabled, GetBinding(i))
	end

	local customHeader = (' |T%s:0|t %s '):format(CPAPI.GetAsset('Textures\\Logo\\CP_Tiny.blp'), SPECIAL)
	for i, data in env.db:For('Bindings/Special') do
		if not data.readonly then
			CacheAvailableBinding(allBindings, pruningEnabled, data.binding, customHeader, GetBindingKey(data.binding))
		end
	end

	local bindings = {}
	for category, set in pairs(allBindings) do
		local title = _G[category] or category
		local subsets = {DivideTable(set, 26)}
		if ( #subsets == 1 ) then
			bindings[title] = subsets[1];
		else
			for i, subset in ipairs(subsets) do
				bindings[('%s (%d)'):format(title, i)] = subset;
			end
		end
	end

	info.notCheckable = 1;
	if (level == 1) then
		-- show which binding is being modified
		info.func = nop;
		info.text = ('%s: %s'):format(
			NORMAL_FONT_COLOR:WrapTextInColorCode(KEY_BINDING),
			ConsolePort:GetFormattedButtonCombination(HANDLER:GetBinding()) or ''
		);

		LDD:UIDropDownMenu_AddButton(info)

		-- add cancel button
		info.func = OnBindingCancel;
		info.text = CANCEL;
		LDD:UIDropDownMenu_AddButton(info)

		-- separate, prepare for binding menus
		LDD:UIDropDownMenu_AddSeparator(level)
		info.func = nil;

		-- binding menus
		for category, set in env.db.table.spairs(bindings) do
			info.text = category;
			info.hasArrow = true;
			info.menuList = category;
			LDD:UIDropDownMenu_AddButton(info)
		end
	else
		local set = bindings[menuList];
		if set then
			local lastIndexWasSeparator = true;
			for i, binding in ipairs(set) do
				if binding:match('^HEADER_BLANK') then
					if ( not lastIndexWasSeparator and i ~= #set ) then
						LDD:UIDropDownMenu_AddSeparator(level)
						lastIndexWasSeparator = true;
					end
				elseif binding:match('^HEADER') then
					-- do something?
				else
					local action = env.db('Actionbar/Binding/'..binding)
					local icon = action and GetActionTexture(action)
					local desc, _, name = env.db.Bindings:GetDescriptionForBinding(binding, true)
					lastIndexWasSeparator = false;
					info.text = GetRealBindingName(binding);
					info.value = binding;
					info.owner = frame;
					info.func = OnBindingClick;
					info.icon = icon;
					info.arg1 = action;
					info.tooltipTitle = name;
					info.tooltipText = desc;
					info.tooltipOnButton = not not (desc and name);
					info.funcOnEnter = OnButtonEnter;
					info.funcOnLeave = OnButtonLeave;
					LDD:UIDropDownMenu_AddButton(info, level)
				end
			end
		end
	end
end

function HANDLER:Close()
	LDD:CloseDropDownMenus()
	self:Hide()
	self:ClearAllPoints()
end

function HANDLER:GetBindingString()
	return (self.mod .. self.btn);
end

function HANDLER:GetBinding()
	return self.btn, self.mod;
end

function HANDLER:SetFrame(owner)
	self:Show()
	self:SetParent(owner)
	self:SetAllPoints(owner)

	local state, mod = owner:GetAttribute('state');
	if owner.isMainButton then
		mod = state;
	elseif state:match('ALT') then
		mod = 'ALT-'..owner.mod;
	else
		mod = owner.mod;
	end

	self.btn = owner.plainID;
	self.mod = mod;

	LDD:UIDropDownMenu_Initialize(self, ShowBindingDropdown)
	LDD:ToggleDropDownMenu(nil, nil, self, 'cursor')
end

HANDLER.PLAYER_REGEN_DISABLED = HANDLER.Close;

function env:OpenBindingDropdown(frame)
	if InCombatLockdown() then return end

	HANDLER:SetFrame(frame)
	if not HANDLER.initialized then
		ConsolePort:AddInterfaceCursorFrame('L_DropDownList1')
		ConsolePort:AddInterfaceCursorFrame('L_DropDownList2')
		HANDLER.initialized = true;
		if L_DropDownList1.Backdrop then
			L_DropDownList1.Backdrop:SetBackdrop(CPAPI.Backdrops.Frame)
			L_DropDownList1.Backdrop:SetBackdropColor(0, 0, 0, 0.75)
		end
		if L_DropDownList2.Backdrop then
			L_DropDownList2.Backdrop:SetBackdrop(CPAPI.Backdrops.Frame)
			L_DropDownList2.Backdrop:SetBackdropColor(0, 0, 0, 0.75)
		end
	end
end