BundleLoader.lua

--[[
Copyright (c) [2023] [Flash_Hit a/k/a Bree_Arnold]

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
--]]

if Class then
	-- Using LoggingClass
	---@class BundleLoader:Class
	---@overload fun():BundleLoader
	---@diagnostic disable-next-line: assign-type-mismatch
	BundleLoader = Class("BundleLoader")
else
	BundleLoader = class("BundleLoader")

	if DEBUG == nil then DEBUG = true end

	function BundleLoader:debug(message, ...)
		if not DEBUG then return end
		message = string.format("[BundleLoader] DEBUG: " .. message, ...)
	end

	function BundleLoader:error(message, ...)
		message = string.format("[BundleLoader] ERROR: " .. message, ...)
		error(message)
	end

	function BundleLoader:warn(message, ...)
		message = string.format("[BundleLoader] WARNING: " .. message, ...)
	end
end

---@enum UiBundleTypes
UiBundleTypes = {
	Unknown = 0,
	Loading = 1,
	Playing = 2,
	PreEndOfRound = 3,
	EndOfRound = 4
}

function BundleLoader:__init()
	self.currentLevelConfig = {}
	self.currentGameModeConfig = {}
	self.currentLevelGameModeConfig = {}
	self.commonConfig = BundleLoader.GetCommonBundleConfig()
	self.LevelName = nil
	Hooks:Install('ResourceManager:LoadBundles', 999, self, self.OnLoadBundles)
	Hooks:Install("Terrain:Load", 999, self, self.OnTerrainLoad)
	Hooks:Install("VisualTerrain:Load", 999, self, self.OnTerrainLoad)
	Events:Subscribe('Level:RegisterEntityResources', self, self.OnLevelRegisterEntityResources)
	return self
end

function BundleLoader:UpdateConfig()
	self.currentLevelConfig = BundleLoader.GetLevelBundleConfig()
	self.currentGameModeConfig = BundleLoader.GetGameModeBundleConfig()
	self.currentLevelGameModeConfig = BundleLoader.GetLevelAndGameModeBundleConfig()

	local s_LevelName = SharedUtils:GetLevelName()
	self.LevelName = s_LevelName
	if s_LevelName and self.currentGameModeConfig.exceptionLevelList then
		for _, l_LevelName in ipairs(self.currentGameModeConfig.exceptionLevelList) do
			if s_LevelName:match(l_LevelName) then
				self.currentGameModeConfig = {}
				break
			end
		end
	end

	local s_GameMode = SharedUtils:GetCurrentGameMode()
	if s_GameMode and self.currentLevelConfig.exceptionGameModeList then
		for _, l_GameMode in ipairs(self.currentGameModeConfig.exceptionGameModeList) do
			if s_GameMode:match(l_GameMode) then
				self.currentLevelConfig = {}
				break
			end
		end
	end
end

function BundleLoader:OnEntityCreateFromBlueprint(p_HookCtx, p_Blueprint, p_Transform, p_Variation, p_ParentRepresentative)
	if self.currentLevelConfig.blueprintGuidsToBlock and self.currentLevelConfig.blueprintGuidsToBlock[tostring(p_Blueprint.instanceGuid)] then
		p_HookCtx:Return()
	end
end

function BundleLoader:GetUIBundleType(p_Bundles)
	for _, l_Bundle in ipairs(p_Bundles) do
		if l_Bundle:match("UiLoading") then
			return UiBundleTypes.Loading
		elseif l_Bundle:match("UiPlaying") then
			return UiBundleTypes.Playing
		elseif l_Bundle:match("UiPreEndOfRound") then
			return UiBundleTypes.PreEndOfRound
		elseif l_Bundle:match("UiEndOfRound") then
			return UiBundleTypes.EndOfRound
		end
	end

	return UiBundleTypes.Unknown
end

local function _ContainsBundle(p_Bundles, p_Bundle)
	p_Bundle = p_Bundle:lower()

	for _, l_Bundle in ipairs(p_Bundles) do
		if l_Bundle:lower() == p_Bundle then
			return true
		end
	end

	return false
end

function BundleLoader:AddBundles(p_Bundles, p_BundlesToAdd)
	for l_Index, l_Bundle in ipairs(p_BundlesToAdd) do
		if (self.LevelName == "Levels/XP1_004/XP1_004" and l_Bundle == "Levels/MP_001/Conquest_Large") or (self.LevelName == "Levels/XP4_Parl/XP4_Parl" and l_Bundle == "Levels/COOP_009/AB00_Parent") then
			goto continue
		end
		if _ContainsBundle(p_Bundles, l_Bundle) then
			self:debug("Ignoring bundle '%s'. It is already in the list.", l_Bundle)
		else
			self:debug("%s: %s", l_Index, l_Bundle)
			table.insert(p_Bundles, l_Bundle)
		end
		::continue::
	end
end

function BundleLoader:GetBundles(p_Bundles, p_Compartment)
	local s_Bundles = {}
	self:debug("Loading compartment %s", p_Compartment)

	if self.commonConfig.bundles and self.commonConfig.bundles[p_Compartment] then
		self:debug("Common Config Bundles:")
		self:AddBundles(s_Bundles, self.commonConfig.bundles[p_Compartment])
	end

	if self.currentLevelConfig.bundles and self.currentLevelConfig.bundles[p_Compartment] then
		self:debug("Current Level Config Bundles:")
		self:AddBundles(s_Bundles, self.currentLevelConfig.bundles[p_Compartment])
	end

	if self.currentLevelGameModeConfig.bundles and self.currentLevelGameModeConfig.bundles[p_Compartment] then
		self:debug("Current Level + GameMode Config Bundles:")
		self:AddBundles(s_Bundles, self.currentLevelGameModeConfig.bundles[p_Compartment])
	end

	if self.currentGameModeConfig.bundles and self.currentGameModeConfig.bundles[p_Compartment] then
		self:debug("Current GameMode Config Bundles:")
		self:AddBundles(s_Bundles, self.currentGameModeConfig.bundles[p_Compartment])
	end

	-- Handle special client compartment
	if p_Compartment == ResourceCompartment.ResourceCompartment_Frontend then
		local s_Type = self:GetUIBundleType(p_Bundles)
		if self.commonConfig.uiBundles and self.commonConfig.uiBundles[s_Type] then
			self:debug("Common Config UI Bundles:")
			self:AddBundles(s_Bundles, self.commonConfig.uiBundles[s_Type])
		end

		if self.currentLevelConfig.uiBundles and self.currentLevelConfig.uiBundles[s_Type] then
			self:debug("Current Level Config UI Bundles:")
			self:AddBundles(s_Bundles, self.currentLevelConfig.uiBundles[s_Type])
		end

		if self.currentLevelGameModeConfig.bundles and self.currentLevelGameModeConfig.uiBundles[s_Type] then
			self:debug("Current Level + GameMode Config UI Bundles:")
			self:AddBundles(s_Bundles, self.currentLevelGameModeConfig.uiBundles[s_Type])
		end

		if self.currentGameModeConfig.bundles and self.currentGameModeConfig.uiBundles[s_Type] then
			self:debug("Current GameMode Config UI Bundles:")
			self:AddBundles(s_Bundles, self.currentGameModeConfig.uiBundles[s_Type])
		end
	end

	self:debug("Game Bundles:")
	self:AddBundles(s_Bundles, p_Bundles)

	return s_Bundles
end

function BundleLoader:OnMountSuperBundles()
	if self.commonConfig.superBundles then
		for l_Index, l_SuperBundle in ipairs(self.commonConfig.superBundles) do
			if self.LevelName == "Levels/XP1_004/XP1_004" and l_SuperBundle == "Levels/MP_001/MP_001" then
				goto continue
			end
			self:debug("Mounting Common SuperBundle %s: %s.", l_Index, l_SuperBundle)
			ResourceManager:MountSuperBundle(l_SuperBundle)
			::continue::
		end
	end

	if self.currentLevelConfig.superBundles then
		for l_Index, l_SuperBundle in ipairs(self.currentLevelConfig.superBundles) do
			self:debug("Mounting Level SuperBundle %s: %s.", l_Index, l_SuperBundle)
			ResourceManager:MountSuperBundle(l_SuperBundle)
		end
	end

	if self.currentLevelGameModeConfig.superBundles then
		for l_Index, l_SuperBundle in ipairs(self.currentLevelGameModeConfig.superBundles) do
			self:debug("Mounting Level + GameMode SuperBundle %s: %s.", l_Index, l_SuperBundle)
			ResourceManager:MountSuperBundle(l_SuperBundle)
		end
	end

	if self.currentGameModeConfig.superBundles then
		for l_Index, l_SuperBundle in ipairs(self.currentGameModeConfig.superBundles) do
			self:debug("Mounting GameMode SuperBundle %s: %s.", l_Index, l_SuperBundle)
			ResourceManager:MountSuperBundle(l_SuperBundle)
		end
	end

	self:debug("Mounted all SuperBundles.")
end

function BundleLoader:AddRegistries(p_Registries)
	for l_Compartment, l_Names in pairs(p_Registries) do
		for _, l_Name in ipairs(l_Names) do
			if (self.LevelName == "Levels/XP1_004/XP1_004" and l_Name == "Levels/MP_001/Conquest_Large") or (self.LevelName == "Levels/XP4_Parl/XP4_Parl" and l_Name == "Levels/COOP_009/AB00_Parent") then
				goto continue
			end
			self:debug("Adding RegistryContainer from '%s' to compartment %s.", l_Name, l_Compartment)
			local s_SubWorldData = ResourceManager:SearchForDataContainer(l_Name)

			if not s_SubWorldData then
				self:error("Failed to find the SubWorldData for '%s'.", l_Name)
				return
			end

			s_SubWorldData = SubWorldData(s_SubWorldData)

			if not s_SubWorldData.registryContainer then
				self:error("Failed to find the RegistryContainer to add for '%s'.", l_Name)
				return
			end

			ResourceManager:AddRegistry(s_SubWorldData.registryContainer, l_Compartment)
			::continue::
		end
	end
end

---VEXT Shared Level:RegisterEntityResources Event
---@param p_LevelData DataContainer|LevelData @needs to be upcasted to LevelData
function BundleLoader:OnLevelRegisterEntityResources(p_LevelData)
	if self.commonConfig.registries then
		self:AddRegistries(self.commonConfig.registries)
	end

	if self.currentLevelConfig.registries then
		self:AddRegistries(self.currentLevelConfig.registries)
	end

	if self.currentLevelGameModeConfig.registries then
		self:AddRegistries(self.currentLevelGameModeConfig.registries)
	end

	if self.currentGameModeConfig.registries then
		self:AddRegistries(self.currentGameModeConfig.registries)
	end
end

---VEXT Shared ResourceManager:LoadBundles Hook
---@param p_HookCtx HookContext
---@param p_Bundles string[]
---@param p_Compartment ResourceCompartment|integer
function BundleLoader:OnLoadBundles(p_HookCtx, p_Bundles, p_Compartment)
	if p_Compartment == ResourceCompartment.ResourceCompartment_Game then
		self:UpdateConfig()
		self:OnMountSuperBundles()
	end

	p_HookCtx:Pass(self:GetBundles(p_Bundles, p_Compartment), p_Compartment)
end

---VEXT Shared VisualTerrain:Load Hook
---VEXT Shared Terrain:Load Hook
---@param p_HookCtx HookContext
---@param p_TerrainName string
function BundleLoader:OnTerrainLoad(p_HookCtx, p_TerrainName)
	if self.currentLevelGameModeConfig.terrainAssetName then
		if not string.find(p_TerrainName:lower(), self.currentLevelGameModeConfig.terrainAssetName:lower()) then
			p_HookCtx:Return()
		end

		return
	end

	if self.currentLevelConfig.terrainAssetName then
		if not string.find(p_TerrainName:lower(), self.currentLevelConfig.terrainAssetName:lower()) then
			p_HookCtx:Return()
		end
		return
	end

	self:warn("No terrain asset name specified. This means every terrain will be loaded.")
	self:warn("Loading terrain '%s'", p_TerrainName)
end

-- NOTE: THIS BELOW EXPECTS A SPECIFIC STRUCTURE

-- Include modifications that should get loaded every time.
function BundleLoader.GetCommonBundleConfig()
	local s_Success, s_BundleConfig = pcall(require, "__shared/BundleLoader/BundleConfig/Common")
	return s_Success and s_BundleConfig or {}
end

-- Include level specific modifications. Only get loaded when the level does.
function BundleLoader.GetLevelBundleConfig()
	local s_LevelName = SharedUtils:GetLevelName():gsub(".*/", "")
	local s_Success, s_BundleConfig = pcall(require, string.format("__shared/BundleLoader/BundleConfig/Levels/%s", s_LevelName))
	return s_Success and s_BundleConfig or {}
end

-- Include gamemode specific modifications. Only get loaded when the gamemode does.
function BundleLoader.GetGameModeBundleConfig()
	local s_Success, s_BundleConfig = pcall(require, string.format("__shared/BundleLoader/BundleConfig/GameModes/%s", SharedUtils:GetCurrentGameMode()))
	return s_Success and s_BundleConfig or {}
end

-- Include level & gamemode specific modifications. Only get loaded when the level & gamemode does.
function BundleLoader.GetLevelAndGameModeBundleConfig()
	local s_LevelName = SharedUtils:GetLevelName():gsub(".*/", "")
	local s_Success, s_BundleConfig = pcall(require, string.format("__shared/BundleLoader/BundleConfig/Levels/%s/%s", s_LevelName, SharedUtils:GetCurrentGameMode()))
	return s_Success and s_BundleConfig or {}
end

BundleLoader = BundleLoader()

return BundleLoader