local _, ns = ...
local B, C, L, DB = unpack(ns)
local T = B:GetModule("Tooltip")
-----------------------
-- MountsSource
-- Author：HopeAsd
-----------------------
T.MountTable = {}

function T:IsCollected(spell)
	local index = T.MountTable[spell].index
	local isCollected = select(11, C_MountJournal.GetMountInfoByID(index))

	return isCollected
end

function T:GetOrCreateMountTable(spell)
	if issecretvalue(spell) then
		return
	end

	if not self.MountTable[spell] then
		local index = C_MountJournal.GetMountFromSpell(spell)
		if index then
			local _, mSpell = C_MountJournal.GetMountInfoByID(index)
			if spell == mSpell then
				local _, _, source = C_MountJournal.GetMountInfoExtraByID(index)
				self.MountTable[spell] = {source = source, index = index}
			end
			return self.MountTable[spell]
		end
		return nil
	end
	return self.MountTable[spell]
end

local function AddLine(self, source, isCollectedText, type, noadd)
	for i = 1, self:NumLines() do
		local line = _G[self:GetName() .. "TextLeft" .. i]
		if not line then break end
		local text = line:GetText()
		if text and B:NotSecretValue(text) and text == type then return end
	end
	if not noadd then self:AddLine(" ") end
	self:AddDoubleLine(type, isCollectedText)
	self:AddLine(source, 1, 1, 1)
	self:Show()
end

local function AddSourceByIndex(self, ...)
	if not C.db["Tooltip"]["MountsSource"] then return end

	local data = C_UnitAuras.GetAuraDataByIndex(...)
	if not data then return end

	local table = data.spellId and T:GetOrCreateMountTable(data.spellId)
	if table then
		AddLine(self, table.source, T:IsCollected(data.spellId) and COLLECTED or NOT_COLLECTED, SOURCE)
	end
end

local function AddSourceByAuraInstanceID(self, unit, auraInstanceID)
	if not C.db["Tooltip"]["MountsSource"] then return end

	local data = C_UnitAuras.GetAuraDataByAuraInstanceID(unit, auraInstanceID)
	if not data then return end

	local table = data.spellId and T:GetOrCreateMountTable(data.spellId)
	if table then
		AddLine(self, table.source, T:IsCollected(data.spellId) and COLLECTED or NOT_COLLECTED, SOURCE)
	end
end

function T:MountsSource()
	if C_AddOns.IsAddOnLoaded("MountsSource") then return end

	hooksecurefunc(GameTooltip, "SetUnitAura", AddSourceByIndex)
	hooksecurefunc(GameTooltip, "SetUnitBuffByAuraInstanceID", AddSourceByAuraInstanceID)
	hooksecurefunc(GameTooltip, "SetUnitAuraByAuraInstanceID", AddSourceByAuraInstanceID)

	B:UnregisterEvent("PLAYER_ENTERING_WORLD", T.MountsSource)
end

B:RegisterEvent("PLAYER_ENTERING_WORLD", T.MountsSource)