local E, L, V, P, G = unpack(ElvUI);
local NP = E:GetModule('NamePlates')
local ipairs = ipairs
local print = print
local select = select
local strsplit = strsplit
local strtrim = strtrim
local string = string
local table = table
local _GetTime = GetTime

local loaded = false
local quantity = 0
local lastKill = {0} -- To be populated later, do not remove the initial value. The zero means inconclusive/invalid data.
local currentPullUpdateTimer = 0
local activeNameplates = {}
local LSM = LibStub('LibSharedMedia-3.0');

mppDebug = false

mppSimulationMode = false
local simulationMax = 220
local simulationCurrent = 28

local version = "0.3e"
local numVersion = 6
local defaultSettings = {
	version = version,
	dbVersion = numVersion,
	enabled = true,

	inconclusiveDataThreshold = 100, -- Mobs killed within this span of time (in milliseconds) will not be processed since we might not get the criteria update fast enough to know which mob gave what progress. Well, that's the theory anyway.
	maxTimeSinceKill = 600, -- Lag tolerance between a mob dying and the progress criteria updating, in milliseconds.

	enableTooltip = true,
	tooltipColor = "82E0FF", -- Color of the tooltip. Duh.

	enablePullEstimate = true,
	pullEstimateCombatOnly = false,

	nameplateUpdateRate = 200, -- Rate (in milliseconds) at which we update the progress we get from the current pull, as estimated by active name plates you're in combat with. Also the update rate of getting new values for nameplate text overlay if enabled.

	enableNameplateText = true,
	nameplateTextColor = "FFFFFF",
}

local warnings = {}

mppDefaultValues = {{1,40,"Test NPC"},{2,35,"Test NPC"},{104277,4,"Legion Hound"},{102253,4,"Understone Demolisher"},{98954,4,"Felsworn Myrmidon"},{97185,10,"The Grimewalker"},{113537,4,"Unknown"},{102094,4,"Risen Swordsman"},{104246,4,"Duskwatch Guard"},{104278,3,"Felbound Enforcer"},{105617,4,"Eredar Chaosbringer"},{105952,6,"Withered Manawraith"},{97043,4,"Seacursed Slaver"},{102430,1,"Tarspitter Slug"},{98366,4,"Ghostly Retainer"},{95832,2,"Valarjar Shieldmaiden"},{104295,1,"Blazing Imp"},{97171,10,"Hatecoil Arcanist"},{102287,4,"Unknown"},{98733,4,"Withered Fiend"},{105921,4,"Nightborne Spellsword"},{96247,1,"Vileshard Crawler"},{98813,4,"Bloodscent Felhound"},{95769,4,"Mindshattered Screecher"},{113699,8,"Forgotten Spirit"},{91785,2,"Wandering Shellback"},{100216,4,"Hatecoil Wrangler"},{97172,1,"Saltsea Droplet"},{98973,1,"Skeletal Warrior"},{105651,4,"Dreadborne Seer"},{92168,8,"Target Dummy Test2"},{105699,3,"Mana Saber"},{105715,4,"Watchful Inquisitor"},{98368,4,"Ghostly Protector"},{95834,2,"Valarjar Mystic"},{91786,1,"Gritslime Snail"},{98177,12,"Glayvianna Soulrender"},{97173,4,"Restless Tides"},{96934,2,"Valarjar Trapper"},{105636,4,"Understone Drudge"},{95771,4,"Dreadsoul Ruiner"},{96584,4,"Immoliant Fury"},{97365,4,"Seacursed Mistmender"},{91006,4,"Rockback Gnasher"},{96664,2,"Valarjar Runecarver"},{105876,1,"Enchanted Broodling"},{95947,4,"Mak'rana Hardshell"},{95772,4,"Frenzied Nightclaw"},{99358,4,"Rotheart Dryad"},{101414,2,"Saltscale Skulker"},{98370,4,"Ghostly Councilor"},{111901,3,"Unknown"},{105845,1,"Glowing Spiderling"},{98243,4,"Soul-Torn Champion"},{98275,4,"Risen Archer"},{99359,3,"Rotheart Keeper"},{106786,1,"Bitterbrine Slave"},{91008,4,"Rockbound Pelter"},{104300,4,"Shadow Mistress"},{98706,6,"Commander Shemdah'sohn"},{98770,4,"Wrathguard Felblade"},{105703,1,"Mana Wyrm"},{100364,4,"Spirit of Vengeance"},{99360,9,"Vilethorn Blossom"},{97097,4,"Helarjar Champion"},{91790,4,"Mak'rana Siltwalker"},{98691,4,"Risen Scout"},{105720,4,"Understone Drudge"},{104270,8,"Guardian Construct"},{101991,4,"Nightmare Dweller"},{98963,1,"Blazing Imp"},{98756,4,"Arcane Anomaly"},{98533,10,"Foul Mother"},{105705,4,"Bound Energy"},{99649,12,"Dreadlord Mendacius"},{98900,4,"Wyrmtongue Trickster"},{98677,1,"Rook Spiderling"},{92350,4,"Understone Drudge"},{102566,12,"Grimhorn the Enslaver"},{102104,4,"Enslaved Shieldmaiden"},{102375,3,"Runecarver Slave"},{96574,1,"Stormforged Sentinel"},{102232,4,"Rockbound Trapper"},{91793,1,"Seaspray Crab"},{100527,3,"Dreadfire Imp"},{98280,4,"Risen Arcanist"},{95842,2,"Valarjar Thundercaller"},{91794,1,"Saltscale Lurker"},{104251,4,"Duskwatch Sentry"},{91332,4,"Stoneclaw Hunter"},{102351,1,"Mana Wyrm"},{102584,4,"Malignant Defiler"},{96480,1,"Viletongue Belcher"},{98759,4,"Vicious Manafang"},{100441,1,"Unknown"},{98926,4,"Shadow Hunter"},{95779,10,"Festerhide Grizzly"},{99365,4,"Taintheart Stalker"},{96608,2,"Ebonclaw Worg"},{96640,2,"Valarjar Marksman"},{90998,4,"Blightshard Shaper"},{106059,4,"Warp Shade"},{98425,4,"Unstable Amalgamation"},{102788,4,"Felspite Dominator"},{97182,6,"Night Watch Mariner"},{101839,4,"Risen Companion"},{100529,1,"Hatespawn Slime"},{99307,12,"Skjal"},{98521,10,"Lord Etheldrin Ravencrest"},{98792,4,"Wyrmtongue Scavenger"},{102404,4,"Stoneclaw Grubmaster"},{100526,1,"Tormented Bloodseeker"},{106787,1,"Bitterbrine Slave"},{99366,4,"Taintheart Summoner"},{97087,2,"Valarjar Champion"},{101438,4,"Vileshard Chunk"},{97119,1,"Shroud Hound"},{90997,4,"Mightstone Breaker"},{91796,10,"Skrog Wavecrasher"},{97677,1,"Barbed Spiderling"},{113998,4,"Mightstone Breaker"},{95920,2,"Animated Storm"},{91781,4,"Hatecoil Warrior"},{95861,4,"Hatecoil Oracle"},{99188,4,"Waterlogged Soul Guard"},{91792,10,"Stormwake Hydra"},{95939,10,"Skrog Tidestomper"},{91783,4,"Hatecoil Stormweaver"},{106785,1,"Bitterbrine Slave"},{111563,4,"Duskwatch Guard"},{98681,6,"Rook Spinner"},{97197,2,"Valarjar Purifier"},{91000,8,"Vileshard Hulk"},{100451,3,"Target Dummy Test"},{91782,10,"Hatecoil Crusher"},{97678,8,"Aranasi Broodmother"},{96587,4,"Felsworn Infester"},{97200,4,"Seacursed Soulkeeper"},{100531,8,"Bloodtainted Fury"},{101549,1,"Arcane Minion"},{99033,4,"Helarjar Mistcaller"},{105629,1,"Wyrmtongue Scavenger"},{98810,6,"Wrathguard Bladelord"},{95766,4,"Crazed Razorbeak"},{98919,4,"Seacursed Swiftblade"},{105682,8,"Felguard Destroyer"},{101679,4,"Dreadsoul Poisoner"},{105915,4,"Nightborne Reclaimer"},{91001,4,"Tarspitter Lurker"},{98732,1,"Plagued Rat"},} -- Yeah ok look I know this is far from optimal, but I'm tired and this was a super quick way of just getting it in there.

local dbFixes = {}
dbFixes[4] = {{1, 40, "Test NPC"}, {99307, 12, "Skjal"}}
dbFixes[5] = {{2, 35, "Test NPC"}, {95947, 4, "Mak'rana Hardshell"}}

-- New to lua, don't judge too hard pls.

function mppGetLastKill()
	return lastKill
end

--
-- GENERAL ADDON UTILITY
-- And by "utility" I mostly mean creating a bunch of shit that should really be built-in.

local function debugLog(s)
	if mppDebug then
		print(s)
	end
end

local function split(str)
	a = {}
	for s in string.gmatch(str, "%S+") do
		table.insert(a, s)
	end
	return a
end

local function round(number, decimals)
    return (("%%.%df"):format(decimals)):format(number)
end

local function tlen(t)
	local length = 0
	for _ in pairs(t) do
		length = length + 1
	end
	return length
end

local function GetTime()
	return _GetTime() * 1000
end

local function hasWarned(message)
	for _,warning in pairs(warnings) do
		if warning == message then
			return true
		end
	end
	return false
end

local function warning(message)
	if not hasWarned(message) then
		print(message)
		table.insert(warnings, message)
		return true
	end
	return false
end

local function getSetting(setting)
	if (not setting or MythicPlusProgressDB["settings"][setting] == nil) then
		warning("MPP attempted to get missing setting: " .. setting or "nil")
		return
	end
	return MythicPlusProgressDB["settings"][setting]
end

function mppGetSetting(setting)
	return getSetting(setting)
end

local function setSetting(setting, value)
	if (not setting or MythicPlusProgressDB["settings"][setting] == nil) then
		warning("MPP attempted to set missing setting: " .. setting or "nil")
		return
	end
	MythicPlusProgressDB["settings"][setting] = value
	return value
end

local function toggleSetting(setting)
	return setSetting(setting, not getSetting(setting))
end

--
-- WOW GENERAL WRAPPERS/EZUTILITIES
--

local function getNPCID(guid)
	local targetType, _,_,_,_, npcID = strsplit("-", guid)
	if targetType == "Creature" or targetType == "Vehicle" and npcID then
		return tonumber(npcID)
	end
end

-- TODO: Figure out how to filter out bosses.
local function isValidTarget(targetToken)
	if UnitCanAttack("player", targetToken) and not UnitIsDead(targetToken) then
		return true
	end
end

local function getSteps()
	return select(3, C_Scenario.GetStepInfo())
end

local function isDungeonFinished()
	if mppSimulationMode then return false end
	return (getSteps() and getSteps() < 1)
end

-- Will also return true in challenge modes if those are ever re-implemented as M+ is basically recycled Challenge Mode.
local function isMythicPlus()
	if mppSimulationMode then return true end
	local difficulty = select(3, GetInstanceInfo()) or -1
	if difficulty == 8 and not isDungeonFinished() then
		return true
	else
		return false
	end
end

-- DEPRECATED
local function isTeeming()
	warning("Deprecated function isTeeming called!")
	return
	--[[local affixes = select(2, C_ChallengeMode.GetActiveKeystoneInfo())
	if affixes then
		for k,v in pairs(affixes) do
			if v == 5 then
				return true
			end
		return false
		end
	end--]]
end

local function getProgressInfo()
	if isMythicPlus() then
		local numSteps = select(3, C_Scenario.GetStepInfo()) -- GetStepInfo tells us how many steps there are.
		if numSteps and numSteps > 0 then
			local info = {C_Scenario.GetCriteriaInfo(numSteps)} -- It should be the last step.
			if info[13] == true then -- if isWeightedProgress
				return info
			end
		end
	end
end

function mppGetProgressInfo()
	return getProgressInfo()
end

local function getMaxQuantity()
	if mppSimulationMode then return simulationMax end
	local progInfo = getProgressInfo()
	if progInfo then
		return getProgressInfo()[5]
	end
end

local function getCurrentQuantity()
	if mppSimulationMode then return simulationCurrent end
	return strtrim(getProgressInfo()[8], "%")
end

local function getEnemyForcesProgress()
--	debugLog("getEnemyForcesProgress called.")
	-- Returns exact float value of current enemies killed progress (1-100).
	local quantity, maxQuantity = getCurrentQuantity(), getMaxQuantity()
	local progress = quantity / maxQuantity
	return progress * 100
end

--
-- DB READ/WRITES
--

local function getValue(npcID)
	debugLog("getValue called. Args: " .. npcID)
	local npcData = MythicPlusProgressDB["npcData"][npcID]
	if npcData then
		local hiValue, hiOccurrence = nil, -1
		for value, occurrence in pairs(npcData["values"]) do
			if occurrence > hiOccurrence then
				hiValue, hiOccurrence = value, occurrence
			end
		end
		if hiValue ~= nil then
			return hiValue
		end
	end
end

local function deleteEntry(npcID)
	local exists = (MythicPlusProgressDB["npcData"][npcID] ~= nil)
	MythicPlusProgressDB["npcData"][npcID] = nil
	return exists
end

local function updateValue(npcID, value, npcName, forceQuantity)
	--debugLog("updateValue called. Args: " .. npcID or "nil" .. ", " .. value or "nil" .. ", " .. npcName or "nil" .. ", " .. forceQuantity or "nil")
	if value <= 0 then
		debugLog("Discarding update for " .. toString(npcName) .. "(" .. npcID .. ") due to value being " .. tovalue)
		return
	end
	local npcData = MythicPlusProgressDB["npcData"][npcID]
	if not npcData then
		MythicPlusProgressDB["npcData"][npcID] = {values={}, name=npcName or "Unknown"}
		return updateValue(npcID, value, npcName, forceQuantity)
	end
	local values = npcData["values"]
	if values[value] == nil then
		values[value] = (forceQuantity or 1)
	else
		values[value] = values[value] + (forceQuantity or 1)
	end
	for val, occurrence in pairs(values) do
		if val ~= value then
			values[val] = occurrence * 0.75 -- Newer values will quickly overtake old ones
		end
	end
end

-- Temp testing global access
function mppUpdateValue(npcID, value, npcName, forceQuantity)
	return updateValue(npcID, value, npcName, forceQuantity)
end

-- Temp testing global access
function mppGetValue(npcID, value)
	return getValue(npcID)
end

local function exportData()
	local a = string.format("Export ver %s (%s) - %i mobs: {", getSetting("version"), getSetting("dbVersion"), tlen(MythicPlusProgressDB["npcData"]))
	for npcID,t in pairs(MythicPlusProgressDB["npcData"]) do
	   local value = getValue(npcID)
	   local name = t["name"]
	   a = a .. "{".. npcID..","..value..",\""..name.."\"},"
	end
	a = a .. "}"
	local f = CreateFrame('EditBox', "MPPExportBox", UIParent, "InputBoxTemplate")
	f:SetSize(100, 50)
	f:SetPoint("CENTER", 0, 150)
	f:SetScript("OnEnterPressed", f.Hide)
	f:SetText(a)
end

--
-- Light DB wrap
--

-- Returns a nil-100 number representing the percentual progress that npcID is expected to give you.
local function getEstimatedProgress(npcID)
	debugLog("getEstimatedProgress called. Args: " .. npcID)
	local npcValue, maxQuantity = getValue(npcID), getMaxQuantity()
	if npcValue and maxQuantity then
		return (npcValue / maxQuantity) * 100
	end
end

--
-- TRIGGERS/HOOKS
--

-- Called when our enemy forces criteria increases, no matter how small the increase (but >0).
local function onProgressUpdated(deltaProgress)
	debugLog("onProgressUpdated called. Args: " .. deltaProgress)
	if currentQuantity == getMaxQuantity() then
		return -- Disregard data that caps us as we don't know if we got the full value.
	end
	local timestamp, npcID, npcName, isDataUseful = unpack(lastKill) -- See what the last mob we killed was
	if timestamp and npcID and deltaProgress and isDataUseful then -- Assert that we have some useful data to work with
		local timeSinceKill = GetTime() - timestamp
		debugLog("timeSinceKill: " .. timestamp .. " Current Time: " .. GetTime() .. "Timestamp of kill: " .. timeSinceKill)
		if timeSinceKill <= getSetting("maxTimeSinceKill") then
			debugLog(string.format("Gained %f%s. Last mob killed was %s (%i) %fs ago", deltaProgress, "%", npcName, npcID, timeSinceKill/1000))
			updateValue(npcID, deltaProgress, npcName) -- Looks like we have ourselves a valid entry. Set this in our database/list/whatever.
		else
			debugLog(string.format("Gained %f%s. Last mob killed was %s (%i) %fs ago (PAST CUTOFF!)", deltaProgress, "%", npcName, npcID, timeSinceKill))
		end
	end
end

-- Called directly by our hook
local function onCriteriaUpdate()
	debugLog("onCriteriaUpdate called")
	if not currentQuantity then
		currentQuantity = 0
	end
	if not isMythicPlus() or not loaded or not getSetting("enabled") then return end
	newQuantity = getCurrentQuantity()
	deltaQuantity = newQuantity - currentQuantity
	if deltaQuantity > 0 then
		currentQuantity = newQuantity
		onProgressUpdated(deltaQuantity)
	end

end

-- Called directly by our hook
local function onCombatLogEvent(args)
	--local _,combatType,_,_,_,_,_, destGUID, destName = unpack(args)
	--if combatType == "UNIT_DIED" then
	local timestamp, combatType, something, srcGUID, srcName, srcFlags, something2, destGUID, destName, destFlags = unpack(args)
	if combatType == "PARTY_KILL" then
		if not isMythicPlus() then return end
		if mppDebug then
			--foreach(args, print)
		end
		local npcID = getNPCID(destGUID)
		if npcID then
			local isDataUseful = true
			local timeSinceLastKill = GetTime() - lastKill[1]
			if timeSinceLastKill <= getSetting("inconclusiveDataThreshold") then
				debugLog("Data not useful: " .. timeSinceLastKill .. " - " .. lastKill[1] .. " - " .. GetTime())
				isDataUseful = false
			end
			lastKill = {GetTime(), npcID, destName, isDataUseful} -- timestamp is not at all accurate, we use GetTime() instead.
			if mppDebug then
				foreach(lastKill, print)
			end
		end
	end
end

local function verifySettings(forceWipe)
	for setting, value in pairs(defaultSettings) do
		if MythicPlusProgressDB["settings"][setting] == nil or forceWipe then
			MythicPlusProgressDB["settings"][setting] = value
		end
	end
	-- At last, update version string

	setSetting("version", version)
end

local function upgradeDB()
	local oldVer = MythicPlusProgressDB["settings"]["dbVersion"]
	for ver, fixes in pairs(dbFixes) do
		if ver > oldVer then
			for _,fixTable in pairs(fixes) do
				local npcID, value, name = unpack(fixTable)
				deleteEntry(npcID)
				updateValue(npcID, value, name)
			end
		end
	end
	setSetting("dbVersion", numVersion)
end


local function verifyDB(forceWipe)
	if not MythicPlusProgressDB or not MythicPlusProgressDB["settings"] or not MythicPlusProgressDB["npcData"] or forceWipe then
		print("Running first time MPP setup. This should only happen once. Enjoy! ;)")
		MythicPlusProgressDB = {}
		MythicPlusProgressDB["settings"] = {}
		MythicPlusProgressDB["settings"]["dbVersion"] = defaultSettings["dbVersion"]
		MythicPlusProgressDB["npcData"] = {}
	else
		MythicPlusProgressDB["settings"]["dbVersion"] = MythicPlusProgressDB["settings"]["dbVersion"] or 0
	end
	verifySettings()
	if mppDefaultValues ~= nil then
		for k,v in pairs(mppDefaultValues) do
			local npcID, value, name = unpack(v)
			if getValue(npcID) == nil then
				updateValue(npcID, value, name, 1)
			end
		end
	end

	-- DB Fixes per version
	upgradeDB()
end

-- Called directly by our hook
local function onAddonLoad(addonName)
	if addonName == "MythicPlusProgress" then
		verifyDB()
		if isMythicPlus() then
			quantity = getEnemyForcesProgress()
			debugLog("MPP Loaded in progress: " .. quantity .. "in.")
		else
			quantity = 0
			debugLog("MPP loaded not in progress.")
		end
		loaded = true
	end
end

---
--- TOOLTIPS
---

local function addLineToTooltip(str)
    GameTooltip:AddDoubleLine(str)
    GameTooltip:Show()
end

local function shouldAddTooltip(unit)
	if loaded and getSetting("enabled") and getSetting("enableTooltip") and isMythicPlus() and isValidTarget(unit) then
		return true
	end
	return false
end

local function getTooltipMessage(npcID)
	local tempMessage = "|cFF"..getSetting("tooltipColor").."M+Progress: "
	local estProg = getEstimatedProgress(npcID)
	if not estProg then
		return tempMessage .. "No record."
	end
	mobsLeft = (100 - getEnemyForcesProgress()) / estProg
	tempMessage = string.format("%s%.2f%s (%i left)", tempMessage, estProg, "%", math.ceil(mobsLeft))
	return tempMessage
end

local function onNPCTooltip(self)
	local unit = select(2, self:GetUnit())
	if unit then
		local guid = UnitGUID(unit)
		npcID = getNPCID(guid)
		if npcID and shouldAddTooltip(unit) then
			local tooltipMessage = getTooltipMessage(npcID)
			if tooltipMessage then
				addLineToTooltip(tooltipMessage)
			end
		end
	end
end

---
--- SHITTY CURRENT PULL FRAME
---

currentPullFrame = CreateFrame("frame", "currentPullFrame12", UIParent)
mppFrame = currentPullFrame
currentPullFrame:SetPoint("CENTER", UIParent, 400, 300)
currentPullFrame:EnableMouse(true)
currentPullFrame:SetMovable(true)
currentPullFrame:RegisterForDrag("LeftButton")
currentPullFrame:SetScript("OnDragStart", currentPullFrame.StartMoving)
currentPullFrame:SetScript("OnDragStop", currentPullFrame.StopMovingOrSizing)
currentPullFrame:SetWidth(50)
currentPullFrame:SetHeight(50)
--currentPullFrame:SetAllPoints()
--currentPullFrame:SetClampRectInsets(200, 400, 200, 500)
currentPullText = currentPullFrame:CreateFontString("currentPullString", "BACKGROUND", "GameFontHighlightLarge")
currentPullText:SetPoint("CENTER");
currentPullText:SetText("MPP String Uninitialized.")

---
--- NAMEPLATES
---

local function isTargetPulled(target)
	-- debugLog("isTargetPulled with args: " ..target)
	local threat = UnitThreatSituation("player", target) or -1 -- Is nil if we're not on their aggro table, so make it -1 instead.
	if isValidTarget(target) and (threat >= 0 or UnitPlayerControlled(target.."target")) then
		return true
	end
	return false
end

local function getPulledUnits()
	local tempList = {}
	for _, nameplate in pairs(C_NamePlate.GetNamePlates()) do
		if nameplate.UnitFrame.unitExists then
			if isTargetPulled(nameplate.UnitFrame.displayedUnit) then
				table.insert(tempList, UnitGUID(nameplate.UnitFrame.displayedUnit))
			end
		end
	end
	return tempList
end

local function getPulledProgress(pulledUnits)
	local estProg = 0
	for _, guid in pairs(pulledUnits) do
		npcID = getNPCID(guid)
		if npcID then
			estProg = estProg + (getEstimatedProgress(npcID) or 0)
		end
	end
	return estProg
end

local function shouldShowCurrentPullEstimate()
	if getSetting("enabled") and getSetting("enablePullEstimate") and isMythicPlus() and not isDungeonFinished() then
		if getSetting("pullEstimateCombatOnly") and not UnitAffectingCombat("player") then
			return false
		end
		return true
	end
	return false
end

local function setCurrentPullEstimateLabel(s)
	currentPullString:SetText(s)
	currentPullFrame:SetWidth(currentPullString:GetStringWidth())
	currentPullFrame:SetHeight(currentPullString:GetStringHeight())
	--print(currentPullFrame:GetCenter())
end

local function updateCurrentPullEstimate()
	if not shouldShowCurrentPullEstimate() then
		currentPullFrame:Hide()
		return
	else
		currentPullFrame:Show()
	end
	local pulledUnits = getPulledUnits()
	local estProg = getPulledProgress(pulledUnits)
	-- debugLog(tlen(pulledUnits) .. "/" .. tlen(activeNameplates).." active nameplates for an estimated " .. estProg .. "%")
	local curProg = getEnemyForcesProgress()
	local totProg = estProg + curProg
	if estProg == 0 then
		tempMessage = "No recorded mobs pulled or nameplates inactive."
	else
		tempMessage = string.format("Current pull: %.2f%s + %.2f%s = %.2f%s", curProg, "%", estProg, "%", (math.floor(totProg*100)/100), "%")
	end
	setCurrentPullEstimateLabel(tempMessage)
end

local function createNameplateText(token)
	local npcID = getNPCID(UnitGUID(token))
	if npcID then
		if activeNameplates[token] then
			activeNameplates[token]:Hide() -- This should never happen...
		end
		local nameplate;
		if not E.private.nameplates.enable then
			nameplate = C_NamePlate.GetNamePlateForUnit(token)
		else
			nameplate = nameplate or NP:GetNamePlateForUnit(token);
		end
		if nameplate then
			if not E.private.nameplates.enable then
				activeNameplates[token] = nameplate:CreateFontString(token.."mppProgress", nameplate.UnitFrame.healthBar, "GameFontHighlightSmall")
			else
				activeNameplates[token] = nameplate:CreateFontString(token.."mppProgress", nameplate.UnitFrame.HealthBar, "GameFontHighlightSmall")
			end
			activeNameplates[token]:SetText("+?%")
		end
	end
end

local function removeNameplateText(token)
	if activeNameplates[token] ~= nil then
		activeNameplates[token]:SetText("")
		activeNameplates[token]:Hide()
		activeNameplates[token] = nil
	end
end

local function updateNameplateValue(token)
	local npcID = getNPCID(UnitGUID(token))
	if npcID then
		local estProg = getEstimatedProgress(npcID)
		if estProg and estProg > 0 then
			local tempMessage = "|cFF"..getSetting("nameplateTextColor").."+"
			tempMessage = string.format("%s%.2f%s", tempMessage, estProg, "%")
			activeNameplates[token]:SetText(tempMessage)
			activeNameplates[token]:Show()
			return true
		end
	end
	if activeNameplates[token] then -- If mob dies, a new nameplate is created but not shown, and this ui widget will then not exist.
		activeNameplates[token]:SetText("")
		activeNameplates[token]:Hide()
	end
	return false
end

local function updateNameplateValues()
	for token,_ in pairs(activeNameplates) do
		updateNameplateValue(token)
	end
end

local function updateNameplatePosition(token)
	local nameplate;
	if not E.private.nameplates.enable then
		nameplate = C_NamePlate.GetNamePlateForUnit(token)

		if nameplate.UnitFrame.unitExists and activeNameplates[token] ~= nil then
			activeNameplates[token]:SetPoint("LEFT", nameplate.UnitFrame.name, "LEFT", nameplate.UnitFrame.name:GetWidth(), 0)
		else
			removeNameplateText(token)
			debugLog("Token " .. token or "nil" .. "does not seem to exist. Why are we trying to update it?")
		end
	else
		nameplate = nameplate or NP:GetNamePlateForUnit(token);

		if activeNameplates[token] ~= nil then
			activeNameplates[token]:SetPoint("LEFT", nameplate.UnitFrame.HealthBar, "LEFT", nameplate.UnitFrame.HealthBar:GetWidth(), 0)
		else
			removeNameplateText(token)
			debugLog("Token " .. token or "nil" .. "does not seem to exist. Why are we trying to update it?")
		end

	end
end

local function shouldShowNameplateTexts()
	if getSetting("enabled") and getSetting("enableNameplateText") and isMythicPlus() and not isDungeonFinished() then
		return true
	end
	return false
end

local function onAddNameplate(token)
	if shouldShowNameplateTexts() then
		createNameplateText(token)
		updateNameplateValue(token)
		updateNameplatePosition(token)
	end
end

local function onRemoveNameplate(token)
	removeNameplateText(token)
	activeNameplates[token] = nil -- This line has been made superflous tbh.
end

local function removeNameplates()
	for token,_ in pairs(activeNameplates) do
		removeNameplateText(token)
	end
end


local function updateNameplates()
	if shouldShowNameplateTexts() then
		for token,_ in pairs(activeNameplates) do
			updateNameplatePosition(token)
		end
	else
		removeNameplates()
	end
end

---
--- SET UP HOOKS
---

local frame = CreateFrame("FRAME")
frame:RegisterEvent("ADDON_LOADED")
frame:RegisterEvent("SCENARIO_CRITERIA_UPDATE")
frame:RegisterEvent("COMBAT_LOG_EVENT_UNFILTERED")


--frame:RegisterEvent("NAME_PLATE_CREATED")
frame:RegisterEvent("NAME_PLATE_UNIT_ADDED")
frame:RegisterEvent("NAME_PLATE_UNIT_REMOVED")

function frame:OnEvent(event, ...)
	args={...}
	if event == "ADDON_LOADED" then
		onAddonLoad(args[1])
	elseif event == "SCENARIO_CRITERIA_UPDATE" then
		onCriteriaUpdate()
	elseif event == "COMBAT_LOG_EVENT_UNFILTERED" then
		onCombatLogEvent(args)
	elseif event == "NAME_PLATE_CREATED" then
		--onCreateNameplate(...)
	elseif event == "NAME_PLATE_UNIT_ADDED" then
		onAddNameplate(...)
	elseif event == "NAME_PLATE_UNIT_REMOVED" then
		onRemoveNameplate(...)
	else
		if warning("MythicPlusProgress unhandled event: " .. event) then
			for k,v in ipairs(args) do
				print(k,v)
			end
		end
	end
end

function frame:OnUpdate(elapsed)
	currentPullUpdateTimer = currentPullUpdateTimer + elapsed * 1000 -- Not using milliseconds in 2016? WutFace
	if currentPullUpdateTimer >= getSetting("nameplateUpdateRate") then
		currentPullUpdateTimer = 0
		updateCurrentPullEstimate()
		updateNameplateValues()
	end
	updateNameplates()
end

frame:SetScript("OnEvent", frame.OnEvent)
frame:SetScript("OnUpdate", frame.OnUpdate)
GameTooltip:HookScript("OnTooltipSetUnit", onNPCTooltip)

SLASH_MYTHICPLUSPROGRESS1,SLASH_MYTHICPLUSPROGRESS2,SLASH_MYTHICPLUSPROGRESS3 = "/mpp","/mypp","/mythicplusprogress"

-- TODO: Redo this entire thing that I threw together in 2 min
function SlashCmdList.MYTHICPLUSPROGRESS(args, editbox)
	args = split(args)
	args[1] = string.lower(args[1] or "")

	if args[1] == "toggle" then
		print("MythicPlusProgress toggled ".. (toggleSetting("enabled") and "ON" or "OFF"))

	elseif args[1] == "reset" then
		currentPullFrame:SetPoint("CENTER", UIParent, 50, 50)
		print("MythicPlusProgress' position reset.")

	elseif args[1] == "sim" then
		mppSimulationMode = not mppSimulationMode
		print("MPP simulation mode toggled ".. (mppSimulationMode and "ON" or "OFF"))

	elseif args[1] == "debug" then
		mppDebug = not mppDebug
		print("MPP debug toggled ".. (mppDebug and "ON" or "OFF"))

	elseif args[1] == "updatevalue" then
		updateValue(getNPCID(UnitGUID("target")), tonumber(args[2]), UnitName("target") .. ("(Manual)"))

	elseif args[1] == "getvalue" then
		print(getValue(tonumber(args[2])) or "No Data")

	elseif args[1] == "getvalues" then
		local npcData = MythicPlusProgressDB["npcData"][tonumber(args[2])]
		if npcData then
			print(tostring(npcData["name"]))
			foreach(npcData["values"], print)
		else
			print("No data.")
		end

	elseif args[1] == "getsetting" then
		print(getSetting(args[2]))

	elseif args[1] == "simmax" then
		simulationMax = tonumber(args[2])

	elseif args[1] == "simcur" then
		simulationCurrent = tonumber(args[2])

	elseif args[1] == "wipeall" then
		verifyDB(true)
		print("RIP Database.")

	elseif args[1] == "wipesettings" then
		verifySettings(true)

	elseif args[1] == "tooltip" then
		setSetting("enableTooltip", not getSetting("enableTooltip"))
		print("Toggled tooltips ".. (getSetting("enableTooltip") and "on" or "off"))

	elseif args[1] == "currentpull" then
		setSetting("enablePullEstimate", not getSetting("enablePullEstimate"))
		print("Toggled \"current pull\" display ".. (getSetting("enablePullEstimate") and "on" or "off"))

	elseif args[1] == "dbinfo" then
		print("Mobs recorded: " .. tlen(MythicPlusProgressDB["npcData"]))

	elseif args[1] == "combatonly" then
		setSetting("pullEstimateCombatOnly", not getSetting("pullEstimateCombatOnly"))
		print("Toggled combat only \"current pull\" display ".. (getSetting("pullEstimateCombatOnly") and "on" or "off"))

	elseif args[1] == "version" then
		print("MythicPlusProgress Version: " .. getSetting("version") .. " - " .. "DB Version: " .. getSetting("dbVersion"))

	elseif args[1] == "nameplates" then
		print("Nameplate text overlay toggled ".. (toggleSetting("enableNameplateText") and "ON" or "OFF"))

	elseif args[1] == "export" then
		exportData()
		print("Data export opened, copy it to clipboard then hit enter to close.")

	else
		print("MythicPlusProgress commands:")
		print("/mpp toggle (Toggles ALL addon visibility on/off, it will still record npc data while in Mythic+)")
		print("/mpp reset (reset position of 'current pull' text, it likes to run away)")
		print("/mpp version (displays current version of addon)")
		print("/mpp tooltip (toggles the tooltip functionality on/off)")
		print("/mpp currentpull (toggles the \"current pull\" functionality on/off)")
		print("/mpp dbinfo (Shows how many unique mobs you've recorded)")
		print("/mpp combatonly (Toggles only showing the current pull estimate while you or one of your party members are in combat)")
		print("/mpp nameplates (Toggles nameplate text overlay on/off)")
		print("/mpp wipesettings (Resets all settings to default (but not mob data!))")
	end
end