local mMT, DB, M, E, P, L, MEDIA = unpack(ElvUI_mMediaTag)
local DT = E:GetModule("DataTexts")

-- code is partially based on ElvUI's Durability and Itemlevel datatext

--WoW API / Variables
local _G = _G
local strupper = strupper
local UIParentLoadAddOn = UIParentLoadAddOn
local GetNumRandomDungeons = GetNumRandomDungeons
local GetLFGRandomDungeonInfo = GetLFGRandomDungeonInfo
local GetNumRFDungeons = GetNumRFDungeons
local GetLFGRoleShortageRewards = GetLFGRoleShortageRewards
local GetRFDungeonInfo = GetRFDungeonInfo
local TANK_ICON = E:TextureString(E.Media.Textures.Tank, ":14:14")
local HEALER_ICON = E:TextureString(E.Media.Textures.Healer, ":14:14")
local DPS_ICON = E:TextureString(E.Media.Textures.DPS, ":14:14")

local textString = ""

local function MakeIconString(tank, healer, damage)
	local str = ""
	if tank then str = str .. TANK_ICON end
	if healer then str = str .. HEALER_ICON end
	if damage then str = str .. DPS_ICON end

	return str
end

local function GetCallToTheArmsInfos()
	local tankReward = false
	local healerReward = false
	local dpsReward = false
	local unavailable = true
	local raid = false
	local dungeon = false

	--Dungeons
	for i = 1, GetNumRandomDungeons() do
		local id = GetLFGRandomDungeonInfo(i)
		for x = 1, LFG_ROLE_NUM_SHORTAGE_TYPES do
			local eligible, forTank, forHealer, forDamage, itemCount = GetLFGRoleShortageRewards(id, x)
			if eligible and forTank and itemCount > 0 then
				tankReward = true
				unavailable = false
				dungeon = true
			end
			if eligible and forHealer and itemCount > 0 then
				healerReward = true
				unavailable = false
				dungeon = true
			end
			if eligible and forDamage and itemCount > 0 then
				dpsReward = true
				unavailable = false
				dungeon = true
			end
		end
	end

	--LFR
	for i = 1, GetNumRFDungeons() do
		local id = GetRFDungeonInfo(i)
		for x = 1, LFG_ROLE_NUM_SHORTAGE_TYPES do
			local eligible, forTank, forHealer, forDamage, itemCount = GetLFGRoleShortageRewards(id, x)
			if eligible and forTank and itemCount > 0 then
				tankReward = true
				unavailable = false
				raid = true
			end
			if eligible and forHealer and itemCount > 0 then
				healerReward = true
				unavailable = false
				raid = true
			end
			if eligible and forDamage and itemCount > 0 then
				dpsReward = true
				unavailable = false
				raid = true
			end
		end
	end

	return not unavailable and MakeIconString(tankReward, healerReward, dpsReward), raid, dungeon
end

local function OnClick(self, button)
	if button == "LeftButton" or button == "MiddleButton" then
		_G.ToggleLFDParentFrame()
		if button == "MiddleButton" then _G.PVEFrameTab3:Click() end
	else
		if IsShiftKeyDown() then
			DB.keystones = {}
		else
			if not _G.WeeklyRewardsFrame then UIParentLoadAddOn("Blizzard_WeeklyRewards") end
			if _G.WeeklyRewardsFrame:IsVisible() then
				_G.WeeklyRewardsFrame:Hide()
			else
				_G.WeeklyRewardsFrame:Show()
			end
		end
	end
end

local function OnEnter(self)
	DT.tooltip:ClearLines()
	local instanceInfos = mMT:GetDungeonInfo()

	if instanceInfos then
		DT.tooltip:AddLine(L["Dungeon Infos:"], mMT:GetRGB("title"))
		DT.tooltip:AddDoubleLine(L["Name:"], instanceInfos.name, mMT:GetRGB("text", "text"))

		if instanceInfos.difficultyShort or instanceInfos.difficultyName then
			local difficulty = (instanceInfos.difficultyShort or instanceInfos.difficultyName) .. ((instanceInfos.isChallengeMode and instanceInfos.level) or "")
			DT.tooltip:AddDoubleLine(L["Difficulty:"], instanceInfos.difficultyColor:WrapTextInColorCode(difficulty), mMT:GetRGB())
		end

		DT.tooltip:AddDoubleLine(L["Guild Party:"], instanceInfos.isGuildParty and L["Yes"] or L["No"], mMT:GetRGB("text", instanceInfos.isGuildParty and "green" or "red"))
		DT.tooltip:AddLine(" ")
	end

	local playerDifficultyInfos = mMT:GetPlayerDifficulty()
	if playerDifficultyInfos then
		DT.tooltip:AddLine(L["Difficulty Infos:"], mMT:GetRGB("title"))
		DT.tooltip:AddDoubleLine(L["Dungeon:"], playerDifficultyInfos.dungeon.color:WrapTextInColorCode(playerDifficultyInfos.dungeon.name), mMT:GetRGB())
		DT.tooltip:AddDoubleLine(L["Raid:"], playerDifficultyInfos.raid.color:WrapTextInColorCode(playerDifficultyInfos.raid.name), mMT:GetRGB())
		if E.Retail then DT.tooltip:AddDoubleLine(L["Legacy Raid:"], playerDifficultyInfos.legacy.color:WrapTextInColorCode(playerDifficultyInfos.legacy.name), mMT:GetRGB()) end
		DT.tooltip:AddLine(" ")
	end

	local ctaInfo, raid, dungeon = GetCallToTheArmsInfos()
	if ctaInfo then
		DT.tooltip:AddLine(L["Call to Arms:"], mMT:GetRGB("title"))
		DT.tooltip:AddDoubleLine((dungeon and L["Dungeon"] or "") .. ((dungeon and raid) and "/" or "") .. (raid and L["Raid"] or "") .. ":", ctaInfo, mMT:GetRGB("text", "text"))
		DT.tooltip:AddLine(" ")
	end

	if E.Retail and E:XPIsLevelMax() then
		local myScore = mMT:GetMyMythicPlusScore()
		local myKeystone = mMT:GetMyKeystone()

		if myKeystone and myScore then
			DT.tooltip:AddLine(L["My Info"], mMT:GetRGB("title"))
			DT.tooltip:AddDoubleLine(DUNGEON_SCORE, myScore, mMT:GetRGB())
			DT.tooltip:AddDoubleLine(L["Keystone"], myKeystone, mMT:GetRGB())
			DT.tooltip:AddLine(" ")
		end

		if DB.keystones and next(DB.keystones) then
			DT.tooltip:AddLine(L["Keystones on your Account"], mMT:GetRGB("title"))
			for _, characters in pairs(DB.keystones) do
				DT.tooltip:AddDoubleLine(characters.name, characters.key)
			end
			DT.tooltip:AddLine(" ")
		end

		local weeklyAffixes = mMT:GetWeeklyAffixes()
		if weeklyAffixes then
			DT.tooltip:AddLine(L["This week's Affix"], mMT:GetRGB("title"))
			DT.tooltip:AddLine(weeklyAffixes, mMT:GetRGB())
			DT.tooltip:AddLine(" ")
		end

		local vaultInfoRaid, vaultInfoDungeons, vaultInfoWorld = mMT:GetVaultInfo()
		if vaultInfoRaid and vaultInfoDungeons and vaultInfoWorld then
			DT.tooltip:AddLine(GREAT_VAULT_REWARDS, mMT:GetRGB("title"))
			DT.tooltip:AddDoubleLine(RAID, vaultInfoRaid, mMT:GetRGB("text", "text"))
			DT.tooltip:AddDoubleLine(DUNGEONS, vaultInfoDungeons, mMT:GetRGB("text", "text"))
			DT.tooltip:AddDoubleLine(WORLD, vaultInfoWorld, mMT:GetRGB("text", "text"))
			DT.tooltip:AddLine(" ")
		end
	end

	DT.tooltip:AddLine(MEDIA.leftClick .. " " .. L["left click to open LFD Frame"], mMT:GetRGB("tip"))
	DT.tooltip:AddLine(MEDIA.rightClick .. " " .. L["right click to open Great Vault"], mMT:GetRGB("tip"))
	DT.tooltip:AddLine(MEDIA.rightClick .. " " .. L["SHIFT + right click to clear all saved keystones."], mMT:GetRGB("tip"))
	DT.tooltip:Show()
end

local function OnEvent(self, event)
	if event == "ELVUI_FORCE_UPDATE" then
		if mMT:GetWeeklyResetTime() then DB.keystones = {} end
	end

	local text = L["Dungeon"]
	if E.db.mMediaTag.datatexts.dungeon.icon ~= "none" then text = E:TextureString(MEDIA.icons.lfg[E.db.mMediaTag.datatexts.dungeon.icon], ":14:14") .. " " .. text end

	if E.db.mMediaTag.datatexts.dungeon.dungeon_name then
		local instanceInfos = mMT:GetDungeonInfo()
		if instanceInfos and (instanceInfos.difficultyShort or instanceInfos.difficultyName) then
			local challengeModeInfo = instanceInfos.isChallengeMode and instanceInfos.level
			local delveInfo = instanceInfos.isDelve and "-" .. instanceInfos.level
			local difficulty = (instanceInfos.difficultyShort or instanceInfos.difficultyName) .. ((challengeModeInfo or delveInfo) or "")
			local guild = instanceInfos.isGuildParty and MEDIA.color.GUILD
			text = strupper(instanceInfos.shortName or instanceInfos.name)
			text = guild and guild:WrapTextInColorCode(text) or text
			text = instanceInfos.difficultyColor:WrapTextInColorCode(difficulty) .. " - " .. text
		end
	end

	self.text:SetFormattedText(textString, text)
end

local function OnLeave(self)
	DT.tooltip:Hide()
end

local function ValueColorUpdate(self, hex)
	local textHex = E.db.mMediaTag.datatexts.text.override_text and "|c" .. MEDIA.color.override_text.hex or hex
	textString = strjoin("", textHex, "%s|r")

	OnEvent(self)
end

local events = { "CHALLENGE_MODE_START", "UPDATE_INSTANCE_INFO", "SCENARIO_UPDATE", "PLAYER_DIFFICULTY_CHANGED", "LFG_UPDATE_RANDOM_INFO" }

DT:RegisterDatatext("mMT - Dungeon", mMT.Name, events, OnEvent, nil, OnClick, OnEnter, OnLeave, L["Dungeon"], nil, ValueColorUpdate)
