Splits.lua

function WarpDeplete:UpdateSplits()
	self:PrintDebug("Updating splits")
	local splits = self:GetSplitsForCurrentInstance()

	if not splits then
		self:PrintDebug("Could not get splits for current instance")
		return
	end

	local best = splits.best
	if not best then
		best = {}
		splits.best = best
	end

	local current = splits.current
	if not current then
		current = {}
		splits.current = current
	end

	local currentDiff = splits.currentDiff
	if not currentDiff then
		currentDiff = {}
		splits.currentDiff = currentDiff
	end

	for i, boss in ipairs(self.state.objectives) do
		if boss.time and boss.time ~= current[i] then
			self:PrintDebug("Setting current time for " .. tostring(i) .. ": " .. tostring(boss.time))
			current[i] = boss.time

			if best[i] then
				currentDiff[i] = boss.time - best[i]
				self:PrintDebug("Setting diff for " .. tostring(i) .. " to " .. tostring(currentDiff[i]))
			end
		elseif not boss.time then
			current[i] = nil
			currentDiff[i] = nil
		end
	end

	if self.state.forcesCompleted and self.state.forcesCompletionTime ~= current.forces then
		self:PrintDebug("Setting current time for forces")
		current.forces = self.state.forcesCompletionTime

		if best.forces then
			currentDiff.forces = self.state.forcesCompletionTime - best.forces
			self:PrintDebug("Setting diff for forces to " .. tostring(currentDiff.forces))
		end
	elseif not self.state.forcesCompleted then
		current.forces = nil
		currentDiff.forces = nil
	end

	if self.state.challengeCompleted and self.state.completionTimeMs ~= current.challenge then
		self:PrintDebug("Setting current time for challenge")
		current.challenge = self.state.completionTimeMs

		if best.challenge then
			currentDiff.challenge = self.state.completionTimeMs - best.challenge
			self:PrintDebug("Setting diff for challenge to " .. tostring(currentDiff.challenge))
		end
	elseif not self.state.challengeCompleted then
		current.challenge = nil
		currentDiff.challenge = nil
	end

	self:PrintDebug("Splits updated")
end

---@param objective integer|"forces"|"challenge"
function WarpDeplete:GetCurrentDiff(objective)
	if self.state.demoModeActive then
		if type(objective) == "number" then
			return -60 + (objective * 30)
		end

		if objective == "forces" then
			return -40
		end

		if objective == "challenge" then
			return 100 * 1000
		end

		return 0
	end

	local splits = self:GetSplitsForCurrentInstance()
	if not splits then
		return nil
	end

	local currentDiff = splits.currentDiff
	if not currentDiff then
		return nil
	end

	return currentDiff[objective]
end

---@param objective integer|"forces"|"challenge"
function WarpDeplete:GetBestSplit(objective)
	if self.state.demoModeActive then
		if type(objective) == "number" then
			return 60 * 3 * objective
		end

		if objective == "forces" then
			return 60 * 30
		end

		if objective == "challenge" then
			return 60 * 36 * 1000
		end

		return 0
	end

	local splits = self:GetSplitsForCurrentInstance()
	if not splits then
		return nil
	end

	local best = splits.best
	if not best then
		return nil
	end

	return best[objective]
end

function WarpDeplete:GetSplitsForCurrentInstance()
	if not self.state.mapId or not self.state.level then
		return nil
	end

	return self:GetSplits(self.state.mapId, self.state.level)
end

function WarpDeplete:GetSplits(mapId, keystoneLevel)
	local mapSplits = self.db.global.splits[mapId]
	if not mapSplits then
		mapSplits = {}
		self.db.global.splits[mapId] = mapSplits
	end

	local keystoneSplits = mapSplits[keystoneLevel]
	if not keystoneSplits then
		keystoneSplits = {}
		mapSplits[keystoneLevel] = keystoneSplits
	end

	return keystoneSplits
end

function WarpDeplete:ResetCurrentSplits()
	local splits = self:GetSplitsForCurrentInstance()
	if not splits then
		return
	end

	splits.current = {}
	splits.currentDiff = {}
end

function WarpDeplete:UpdateBestSplits()
	local splits = self:GetSplitsForCurrentInstance()
	if not splits or not splits.current then
		return
	end

	if not splits.best then
		splits.best = {}
	end

	for k, v in pairs(splits.current) do
		self:PrintDebug("Updating best split for objective " .. tostring(k))
		if not splits.best[k] then
			self:PrintDebug("No best time found, setting " .. tostring(v))
			splits.best[k] = v
		elseif splits.best[k] > v then
			self:PrintDebug("Better time found, setting " .. tostring(v))
			splits.best[k] = v
		end
	end
end