WoWInterface/Interface/AddOns/OPie/Rings/Common.lua

local AB, _, T = assert(OPie.ActionBook:compatible(2,14), "Requires a compatible version of ActionBook"), ...
local MODERN = select(4,GetBuildInfo()) >= 8e4
local CF_WRATH = not MODERN and select(4,GetBuildInfo()) >= 3e4
local ORI, EV, L, PC = OPie.UI, T.Evie, T.L, T.OPieCore

if MODERN or CF_WRATH then -- OPieTracker
	local function generateColor(c, n)
		local hue, v, s = (15+(c-1)*360/n) % 360, 1, 0.85
		local h, f = math.floor(hue/60) % 6, (hue/60) % 1
		local p, q, t = v - v*s, v - v*f*s, v - v*s + v*f*s

		if h == 0 then return v, t, p
		elseif h == 1 then return q, v, p
		elseif h == 2 then return p, v, t
		elseif h == 3 then return p, q, v
		elseif h == 4 then return t, p, v
		elseif h == 5 then return v, p, q
		end
	end
	
	local function GetTrackingInfo(...)
		return (MODERN and C_Minimap or _G).GetTrackingInfo(...)
	end
	local function SetTracking(...)
		return (MODERN and C_Minimap or _G).SetTracking(...)
	end
	local function GetNumTrackingTypes(...)
		return (MODERN and C_Minimap or _G).GetNumTrackingTypes(...)
	end
	
	local collectionData = {}
	local function setTracking(id)
		SetTracking(id, not select(3, GetTrackingInfo(id)))
	end
	local function hint(k)
		local name, tex, on = GetTrackingInfo(k)
		return not not name, on and 1 or 0, tex, name, 0,0,0
	end
	local trackerActions = setmetatable({}, {__index=function(t, k)
		t[k] = AB:CreateActionSlot(hint, k, "func", setTracking, k)
		return t[k]
	end})
	local function preClick(selfId, _, updatedId)
		if selfId ~= updatedId then return end
		local n = GetNumTrackingTypes()
		if n ~= #collectionData then
			for i=1,n do
				local token = "OPieBundleTracker" .. i
				collectionData[i], collectionData[token] = token, trackerActions[i]
				ORI:SetDisplayOptions(token, nil, nil, generateColor(i,n))
			end
			for i=n+1,#collectionData do
				collectionData[i] = nil
			end
			AB:UpdateActionSlot(selfId, collectionData)
		end
	end
	local col = AB:CreateActionSlot(nil,nil, "collection", collectionData)
	OPie:SetRing("OPieTracking", col, {name=L"Minimap Tracking", hotkey="ALT-F"})
	AB:AddObserver("internal.collection.preopen", preClick, col)
	function EV.PLAYER_ENTERING_WORLD()
		return "remove", preClick(col, nil, col)
	end
end
do -- OPieAutoQuest
	local exclude, questItems, IsQuestItem = PC:RegisterPVar("AutoQuestExclude", {}), {}
	if MODERN then
		questItems[30148] = "72986 72985"
		local include = {
			[33634]=true, [35797]=true, [37888]=true, [37860]=true, [37859]=true, [37815]=true, [46847]=true, [47030]=true, [39213]=true, [42986]=true, [49278]=true,
			[86425]={31332, 31333, 31334, 31335, 31336, 31337}, [87214]={31752, 34774}, [90006]=true, [86536]=true, [86534]=true, [97268]=true, [111821]={34774, 31752},
			[180008]={60609}, [180009]={60609}, [180170]={60649},
			[174464]=true, [168035]=true,
		}
		function IsQuestItem(iid, bag, slot)
			if exclude[iid] then return false end
			local iqi = C_Container.GetContainerItemQuestInfo(bag, slot)
			local isQuest, startQuestId, isQuestActive = iqi.isQuestItem, iqi.questID, iqi.isActive
			isQuest = iid and ((isQuest and GetItemSpell(iid)) or (include[iid] == true) or (startQuestId and not isQuestActive and not C_QuestLog.IsQuestFlaggedCompleted(startQuestId)))
			if not isQuest and type(include[iid]) == "table" then
				isQuest = true
				for _, qid in pairs(include[iid]) do
					if C_QuestLog.IsQuestFlaggedCompleted(qid) then
						isQuest = false
					end
				end
			end
			return isQuest, startQuestId and not isQuestActive
		end
	else
		local include = PC:RegisterPVar("AutoQuestWhitelist", {}) do
			local hexclude, hinclude = {}, {}
			for i in ("12460 12451 12450 12455 12457 12458 12459"):gmatch("%d+") do
				hexclude[i+0] = true
			end
			for i in (CF_WRATH and "33634 35797 37888 37860 37859 37815 46847 47030 39213 42986 49278" or ""):gmatch("%d+") do
				hinclude[i+0] = true
			end
			setmetatable(exclude, {__index=hexclude})
			setmetatable(include, CF_WRATH and {__index=hinclude} or nil)
		end
		local QUEST_ITEM = Enum.ItemClass.Questitem
		function IsQuestItem(iid, bag, slot, skipTypeCheck)
			local isQuest, startsQuest = false, false
			if include[iid] then
				isQuest = true
			elseif not (iid and GetItemSpell(iid) and not exclude[iid]) then
			elseif select(12, GetItemInfo(iid)) == QUEST_ITEM then
				isQuest = true
			elseif skipTypeCheck then
				include[iid], isQuest = true, true
			end
			if CF_WRATH and bag and slot then
				local _isQuestItem, startQuestId, isQuestActive = GetContainerItemQuestInfo(bag, slot)
				startsQuest = startQuestId and not isQuestActive and not C_QuestLog.IsQuestFlaggedCompleted(startQuestId) or false
				isQuest = isQuest or startsQuest
			end
			return isQuest, startsQuest
		end

		local lastQuestAcceptTime = GetTime()-20
		function EV.QUEST_ACCEPTED()
			lastQuestAcceptTime = GetTime()
		end
		function EV:CHAT_MSG_LOOT(text)
			local iid = text:match("|Hitem:(%d+).-|h")
			if iid and (GetTime()-lastQuestAcceptTime) < 1 then
				IsQuestItem(iid+0, nil, nil, true)
			end
		end
	end
	local GetQuestLogTitle = GetQuestLogTitle or function(i)
		local q = C_QuestLog.GetInfo(i)
		if q then
			local qid = q.questID
			return nil, nil, nil, q.isHeader, q.isCollapsed, C_QuestLog.IsComplete(qid), nil, qid
		end
	end

	local colId, current, changed
	local collection, inring, ctok = MODERN and {"EB", EB=AB:GetActionSlot("extrabutton", 1)} or {}, {}, 0
	local function scanQuests(i)
		for i=i or 1, (MODERN and C_QuestLog.GetNumQuestLogEntries or GetNumQuestLogEntries)() do
			local _, _, _, isHeader, isCollapsed, isComplete, _, qid = GetQuestLogTitle(i)
			if isHeader and isCollapsed then
				ExpandQuestHeader(i)
				return scanQuests(i+1), CollapseQuestHeader(i)
			elseif questItems[qid] and not isComplete then
				for id in questItems[qid]:gmatch("%d+") do
					local iid = id+0
					local act = AB:GetActionSlot("item", iid)
					if act and not exclude[iid] then
						local tok = "OpieBundleQuest" .. id
						if not inring[tok] then
							collection[#collection+1], collection[tok], changed = tok, act, true
						end
						inring[tok] = current
						break
					end
				end
			elseif MODERN then
				local link, _, _, showWhenComplete = GetQuestLogSpecialItemInfo(i)
				if link and (showWhenComplete or not isComplete) then
					local iid = tonumber(link:match("item:(%d+)"))
					if not exclude[iid] then
						local tok = "OPieBundleQuest" .. iid
						if not inring[tok] then
							collection[#collection+1], collection[tok], changed = tok, AB:GetActionSlot("item", iid), true
						end
						inring[tok] = current
					end
				end
			end
		end
	end
	local function syncRing(_, _, upId)
		if upId ~= colId then return end
		changed, current = false, (ctok + 1) % 2
		
		local ns = MODERN and C_Container.GetContainerNumSlots or GetContainerNumSlots
		local giid = MODERN and C_Container.GetContainerItemID or GetContainerItemID
		for bag=0,4 do
			for slot=1, ns(bag) do
				local iid = giid(bag, slot)
				local isQuest, startsQuestMark = IsQuestItem(iid, bag, slot)
				if isQuest then
					local tok = "OPieBundleQuest" .. iid
					if not inring[tok] then
						collection[#collection+1], collection[tok], changed = tok, AB:GetActionSlot("item", iid), true
					end
					ORI:SetQuestHint(tok, startsQuestMark)
					inring[tok] = current
				end
			end
		end
		for i=0,INVSLOT_LAST_EQUIPPED do
			local tok = "OPieBundleQuest" .. (GetInventoryItemID("player", i) or 0)
			if inring[tok] then
				inring[tok] = current
			end
		end
		scanQuests()

		local freePos, oldCount = MODERN and 2 or 1, #collection
		for i=freePos, oldCount do
			local v = collection[i]
			collection[freePos], freePos, collection[v], inring[v] = collection[i], freePos + (inring[v] == current and 1 or 0), (inring[v] == current and collection[v] or nil), inring[v] == current and current or nil
		end
		for i=oldCount,freePos,-1 do collection[i] = nil end
		ctok = current

		if changed or freePos <= oldCount then
			AB:UpdateActionSlot(colId, collection)
		end
	end
	colId = AB:CreateActionSlot(nil,nil, "collection",collection)
	OPie:SetRing("OPieAutoQuest", colId, {name=L"Quest Items", hotkey="ALT-Q"})
	AB:AddObserver("internal.collection.preopen", syncRing)
	function EV.PLAYER_REGEN_DISABLED()
		syncRing(nil, nil, colId)
	end

	local function excludeItemID(iid)
		if iid > 0 then
			exclude[iid] = true
		else
			exclude[-iid] = nil
			if exclude[-iid] then
				exclude[-iid] = false
			end
		end
	end
	T.AddSlashSuffix(function(msg)
		local args = msg:match("^%s*%S+%s*(.*)$")
		if args:match("^[%d%s%-]+$") then
			for iid in args:gmatch("[%-]?%d+") do
				excludeItemID(tonumber(iid))
			end
		else
			local flag, _, link
			flag, args = args:match("^(%-?)(.*)$")
			_, link = GetItemInfo(args:match("|H(item:%d+)") or args)
			local iid = link and link:match("item:(%d+)")
			if iid then
				excludeItemID(tonumber(iid) * (flag == "-" and -1 or 1))
			end
		end
	end, "exclude-quest-item")
end