local cl_toolsearch_autoselect = CreateClientConVar("cl_toolsearch_autoselect", "1")
local cl_toolsearch_favoritesonly = CreateClientConVar("cl_toolsearch_favoritesonly", "0")
local cl_toolsearch_favoritestyle = CreateClientConVar("cl_toolsearch_favoritestyle", "1")

local favorites = util.JSONToTable(file.Read("tools_favorites.txt", "DATA") or "{}") or {}
local init = false

hook.Add("PostReloadToolsMenu", "ToolSearch", function()
	if not IsValid(g_SpawnMenu) then return end

	local toolPanel = g_SpawnMenu.ToolMenu.ToolPanels[1]
	local divider = toolPanel.HorizontalDivider

	local list = toolPanel.List

	if not IsValid(divider) then
		error("Something is modifying the spawnmenu and is preventing the tool search addon from working!")
		return
	end

	local panel = vgui.Create("EditablePanel", divider)
	list:SetParent(panel)
	list:Dock(FILL)

	local textEntry = panel:Add("EditablePanel")
	textEntry:Dock(TOP)
	textEntry:DockMargin(0, 0, 0, 2)
	textEntry:SetTall(20)

	local search = textEntry:Add("DTextEntry")
	search:Dock(FILL)
	search:DockMargin(0, 0, 2, 0)
	search._Paint = search.Paint
	function search:Paint(w, h)
		local ret = self:_Paint(w, h)

		if self:GetValue() == "" then
			surface.SetFont("DermaDefault")
			local txt = L.search2_hint
			local txtW, txtH = surface.GetTextSize(txt)
			surface.SetTextPos(3, h * 0.5 - txtH * 0.5 + 1)
			surface.SetTextColor(Color(0, 0, 0, 192))
			surface.DrawText(txt)
		end

		return ret
	end
	search:SetUpdateOnType(true)
	function search:OnValueChange(str, init)
		local i = 0
		for _, cat in next, list.pnlCanvas:GetChildren() do
			local hidden = 0
			for k, pnl in next, cat:GetChildren() do
				if pnl.ClassName ~= "DCategoryHeader" then
					if utf8.lower(language.GetPhrase(pnl:GetText())):match(utf8.lower(str)) and (not cl_toolsearch_favoritesonly:GetBool() or (cl_toolsearch_favoritesonly:GetBool() and favorites[pnl.Name])) then
						pnl:SetVisible(true)
						if cl_toolsearch_autoselect:GetBool() and not init then
							i = i + 1
							if i == 1 then
								pnl:SetSelected(true)
								pnl:DoClick()
							else
								pnl:SetSelected(false)
							end
						end
					else
						pnl:SetVisible(false)
						hidden = hidden + 1
					end
				end
			end
			if hidden >= #cat:GetChildren() - 1 then
				cat:SetVisible(false)
			else
				cat:SetVisible(true)
			end
			cat:InvalidateLayout()
			list.pnlCanvas:InvalidateLayout()
		end
	end

	local clear = textEntry:Add("DButton")
	clear:Dock(RIGHT)
	clear:SetWide(20)
	clear:SetText("")
	clear:SetTooltip(L.clear_field )
	function clear:DoClick()
		search:SetValue("")
	end
	local close = Material("icon16/cross.png")
	function clear:Paint(w, h)
		derma.SkinHook("Paint", "Button", self, w, h)

		surface.SetMaterial(close)
		surface.SetDrawColor(Color(255, 255, 255))
		surface.DrawTexturedRect(w * 0.5 - 16 * 0.5, h * 0.5 - 16 * 0.5, 16, 16)
	end

	local favsOnly = panel:Add("EditablePanel")
	favsOnly:Dock(TOP)
	favsOnly:DockMargin(0, 0, 0, 2)
	favsOnly:SetTall(20)

	local check = favsOnly:Add("DCheckBoxLabel")
	check:SetConVar("cl_toolsearch_favoritesonly")
	local showFavsOnly = cl_toolsearch_favoritesonly:GetBool()
	check:SetChecked(showFavsOnly)
	check:SetText(L.only_features)
	check:SetPos(0, 3)
	check:SetBright(true)

	local small_star = Material("icon16/bullet_star.png")
	local function showFavoritesOnly(showFavs)
		for _, cat in next, list.pnlCanvas:GetChildren() do
			for k, pnl in next, cat:GetChildren() do
				if pnl.ClassName ~= "DCategoryHeader" then
					if showFavs then
						if favorites[pnl.Name] then
							pnl:SetVisible(true)
							pnl.Favorite = true
						else
							pnl:SetVisible(false)
							pnl.Favorite = false
						end
					end
					pnl.Favorite = favorites[pnl.Name]
					if not pnl._Paint then
						local bgCol = CFG.skinColors.b
						pnl._Paint = pnl.Paint
						function pnl:Paint(w, h)
							local ret = self:_Paint(w, h)

							if self.Favorite then
								surface.SetMaterial(small_star)
								surface.SetDrawColor(Color(255, 255, 255))
								surface.DrawTexturedRect(w - 16, h * 0.5 - 8, 16, 16)
							end

							return ret
						end
					end
					function pnl:DoRightClick(w, h)
						self.Favorite = not self.Favorite
						favorites[self.Name] = self.Favorite
						file.Write("tools_favorites.txt", util.TableToJSON(favorites))
						surface.PlaySound("garrysmod/content_downloaded.wav")
					end
				end
				cat:InvalidateLayout()
				list.pnlCanvas:InvalidateLayout()
			end
		end
		search:OnValueChange(search:GetValue(), not init or not showFavs)
		init = true
	end

	local fix = true
	function check:OnChange()
		if fix then fix = nil return end
		showFavoritesOnly(cl_toolsearch_favoritesonly:GetBool())
	end

	divider:SetLeft(panel)

	showFavoritesOnly(showFavsOnly)
end)

hook.Add("PopulateToolMenu", "ToolSearch", function()
	spawnmenu.AddToolMenuOption("Utilities",
		"User",
		"ToolSearch",
		"Tool Search",	"",	"",
		function(pnl)
			pnl:AddControl("Header", {
				Description = "Configure the Tool Search's behavior."
			})

			pnl:AddControl("CheckBox", {
				Label = "Auto-Select",
				Command = "cl_toolsearch_autoselect",
			})
			pnl:ControlHelp("If enabled, this will select the top most tool automatically when you do a search query.")

			pnl:AddControl("Header", {
				Description = "Right-click tools to make them your favorites!"
			})
		end
	)
end)

-- RunConsoleCommand("spawnmenu_reload")