Skip to content

Sanitize some of the BPmodloader logic #886

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Kein wants to merge 12 commits into
base: main
Choose a base branch
from
Open

Sanitize some of the BPmodloader logic #886

Changes from 6 commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
9fb042b
Sanitize some of the BPmodloader logic
Kein May 22, 2025
31692f7
Catch a case where games has no Paks/ folder
Kein May 23, 2025
0123187
Fixes to local/global
Kein May 25, 2025
558dc6b
Remove unusual logic, fix the mod count to use OrderedMods
Kein May 28, 2025
096f57f
Remove useless param
Kein May 28, 2025
55f9d66
Remove useless param#2
Kein May 28, 2025
2e68e38
fix error messages, change the reload hook to sue ALT
Kein Jun 9, 2025
bcc6e7b
fix: newlines
Kein Jun 9, 2025
2a1476e
cleanup of unused annotations
Kein Jun 9, 2025
937a011
cleanup of every logger message
Kein Jun 9, 2025
1f8954b
changed: manual reload hotkey from INS to Alt+INS
Kein Jun 9, 2025
c7b52e0
BPModLoader breaking changes
Kein Jun 10, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
193 changes: 118 additions & 75 deletions assets/Mods/BPModLoaderMod/Scripts/main.lua
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,24 +1,43 @@
---########################
--- DEFINES
---########################
package.path = '.\\Mods\\ModLoaderMod\\?.lua;' .. package.path
package.path = '.\\Mods\\ModLoaderMod\\BPMods\\?.lua;' .. package.path
local UEHelpers = require("UEHelpers")

local VerboseLogging = false
local SpawnModsOnLuaInit = true -- spawn on lua unit exec? common cause of random crashes (game-specific race?)
local AssetRegistryHelpers = CreateInvalidObject() ---@cast AssetRegistryHelpers UAssetRegistryHelpers
local AssetRegistry = CreateInvalidObject() ---@cast AssetRegistry IAssetRegistry

local Mods = {}
local OrderedMods = {}
local ModOrderList = {} -- contains entries from Mods/BPModLoaderMod/load_order.txt; used for the load order of BP mods.
local DefaultModConfig = {
AssetName = "ModActor_C",
AssetNameAsFName = UEHelpers.FindOrAddFName("ModActor_C")
}


---########################
--- HELPERS
---########################

local function Log(Message, OnlyLogIfVerbose)
if not VerboseLogging and OnlyLogIfVerbose then return end
print("[BPModLoaderMod] " .. Message)
end

package.path = '.\\Mods\\ModLoaderMod\\?.lua;' .. package.path
package.path = '.\\Mods\\ModLoaderMod\\BPMods\\?.lua;' .. package.path

local Mods = {}
local OrderedMods = {}

-- Contains mod names from Mods/BPModLoaderMod/load_order.txt and is used to determine the load order of BP mods.
local ModOrderList = {}

local DefaultModConfig = {}
DefaultModConfig.AssetName = "ModActor_C"
DefaultModConfig.AssetNameAsFName = UEHelpers.FindOrAddFName("ModActor_C")
--- get table count, never rely on #t
--- @param T table
--- @return integer
local function GetModCount()
local c = 0
for _, ModInfo in ipairs(OrderedMods) do
if type(ModInfo) == "table" then c = c + 1 end
end
return c
end

-- Checks if the beginning of a string contains a certain pattern.
local function StartsWith(String, StringToCompare)
Expand Down Expand Up @@ -49,6 +68,17 @@ local function LinesFrom(file, ignoreLinesStartingWith)
return lines
end

local function LogOrderedMods()
for _, v in ipairs(OrderedMods) do
Log(string.format("%s == %s\n", v.Name, v))
if type(v) == "table" then
for k2, v2 in pairs(v) do
Log(string.format(" %s == %s\n", k2, v2))
end
end
end
end

-- Loads mod order data from load_order.txt and pushes it into ModOrderList.
local function LoadModOrder()
local file = 'Mods/BPModLoaderMod/load_order.txt'
Expand Down Expand Up @@ -110,20 +140,42 @@ local function SetupModOrder()
end
end

local function CacheAssetRegistry()
---@cast AssetRegistryHelpers UObject
if (AssetRegistryHelpers:IsValid() and AssetRegistry:IsValid()) then return end

AssetRegistryHelpers = StaticFindObject("/Script/AssetRegistry.Default__AssetRegistryHelpers") --[[@as UObject]]
if (AssetRegistryHelpers ~= nil and AssetRegistryHelpers:IsValid()) then
---@cast AssetRegistryHelpers UAssetRegistryHelpers
AssetRegistry = AssetRegistryHelpers:GetAssetRegistry() --[[@as IAssetRegistry]]
end
if not AssetRegistry:IsValid() then
print("Failed to fetch AssetRegistry via ARHelpers, falling back to SFO search\n")
AssetRegistry = StaticFindObject("/Script/AssetRegistry.Default__AssetRegistryImpl") --[[@as IAssetRegistry]]
end
if not AssetRegistry:IsValid() then error("Unable to continue - failed to validate UE game provides instance of AssetRegistry!\n") end
end

---########################
--- MAIN LOGIC
---########################
local function LoadModConfigs()
-- Load configurations for mods.
local Dirs = IterateGameDirectories();
if not Dirs then
if not Dirs or not Dirs.Game.Content.Paks then
error("[BPModLoader] UE4SS does not support loading mods for this game.")
end
local LogicModsDir = Dirs.Game.Content.Paks.LogicMods
if not Dirs then error("[BPModLoader] IterateGameDirectories failed, cannot load BP mod configurations.") end
if not LogicModsDir then
CreateLogicModsDirectory();
Dirs = IterateGameDirectories();
Dirs = IterateGameDirectories(); ---@cast Dirs -nil
LogicModsDir = Dirs.Game.Content.Paks.LogicMods
if not LogicModsDir then error("[BPModLoader] Unable to find or create Content/Paks/LogicMods directory. Try creating manually.") end
if not LogicModsDir then
error("[BPModLoader] Unable to find or create Content/Paks/LogicMods directory. Try creating manually.")
end
return
end

for ModDirectoryName, ModDirectory in pairs(LogicModsDir) do
Log(string.format("Mod: %s\n", ModDirectoryName))
for _, ModFile in pairs(ModDirectory.__files) do
Expand Down Expand Up @@ -163,24 +215,9 @@ local function LoadModConfigs()
end

LoadModOrder()

SetupModOrder()
end

LoadModConfigs()

for _, v in ipairs(OrderedMods) do
Log(string.format("%s == %s\n", v.Name, v))
if type(v) == "table" then
for k2, v2 in pairs(v) do
Log(string.format(" %s == %s\n", k2, v2))
end
end
end

local AssetRegistryHelpers = nil
local AssetRegistry = nil

local function LoadMod(ModName, ModInfo, World)
if ModInfo.Priority ~= nil then
Log(string.format("Loading mod [Priority: #%i]: %s\n", ModInfo.Priority, ModName))
Expand All @@ -204,7 +241,12 @@ local function LoadMod(ModName, ModInfo, World)
["AssetName"] = UEHelpers.FindOrAddFName(ModInfo.AssetName),
}
end
---@cast AssetRegistryHelpers UObject
if (not AssetRegistryHelpers:IsValid()) then
error("Unable to continue - AssetRegistryHelpers is invalid")
end

---@cast AssetRegistryHelpers UAssetRegistryHelpers
local ModClass = AssetRegistryHelpers:GetAsset(AssetData)
if not ModClass:IsValid() then
local ObjectPath = AssetData.ObjectPath and AssetData.ObjectPath:ToString() or ""
Expand Down Expand Up @@ -235,23 +277,6 @@ local function LoadMod(ModName, ModInfo, World)
end
end

local function CacheAssetRegistry()
if AssetRegistryHelpers and AssetRegistry then return end

AssetRegistryHelpers = StaticFindObject("/Script/AssetRegistry.Default__AssetRegistryHelpers")
if not AssetRegistryHelpers:IsValid() then Log("AssetRegistryHelpers is not valid\n") end

if AssetRegistryHelpers then
AssetRegistry = AssetRegistryHelpers:GetAssetRegistry()
if AssetRegistry:IsValid() then return end
end

AssetRegistry = StaticFindObject("/Script/AssetRegistry.Default__AssetRegistryImpl")
if AssetRegistry:IsValid() then return end

error("AssetRegistry is not valid\n")
end

local function LoadMods(World)
if not World or not World:IsValid() then
Log("[Warning] Invalid UWorld object was passed to LoadMods.\n")
Expand All @@ -266,36 +291,54 @@ local function LoadMods(World)
end
end

RegisterKeyBind(Key.INS, function()
ExecuteInGameThread(function()
LoadMods(UEHelpers.GetWorld())

---########################
--- ENTRY POINT
---########################

LoadModConfigs()
LogOrderedMods()

--- Only execute/add any hooks if we have at least one mod entry
--- It does not matter we can do it manually on a hotkey, PAKs arent mounted
--- automatially at runtime on LUA reload and we dont need potentially unstable and moot hooks
if (GetModCount() > 0) then

RegisterKeyBind(Key.INS, function()
ExecuteInGameThread(function()
LoadMods(UEHelpers.GetWorld())
end)
end)
end)

RegisterBeginPlayPostHook(function(ContextParam)
local Context = ContextParam:get()
for _, ModConfig in ipairs(OrderedMods) do
if Context:GetClass():GetFName() ~= ModConfig.AssetNameAsFName then return end
local AssetPathWithClassPrefix = string.format("BlueprintGeneratedClass %s.%s", ModConfig.AssetPath, ModConfig.AssetName)
if AssetPathWithClassPrefix == Context:GetClass():GetFullName() then
local PostBeginPlay = Context.PostBeginPlay
if PostBeginPlay:IsValid() then
Log(string.format("Executing 'PostBeginPlay' for mod '%s'\n", Context:GetFullName()))
PostBeginPlay()
else
Log(string.format("PostBeginPlay not valid for mod %s\n", Context:GetFullName()), true)

RegisterBeginPlayPostHook(function(ContextParam)
local Context = ContextParam:get()
for _, ModConfig in ipairs(OrderedMods) do
if Context:GetClass():GetFName() ~= ModConfig.AssetNameAsFName then return end
local AssetPathWithClassPrefix = string.format("BlueprintGeneratedClass %s.%s", ModConfig.AssetPath, ModConfig.AssetName)
if AssetPathWithClassPrefix == Context:GetClass():GetFullName() then
local PostBeginPlay = Context.PostBeginPlay
if PostBeginPlay:IsValid() then
Log(string.format("Executing 'PostBeginPlay' for mod '%s'\n", Context:GetFullName()))
PostBeginPlay()
else
Log(string.format("PostBeginPlay not valid for mod %s\n", Context:GetFullName()), true)
end
end
end
end
end)
end)

RegisterLoadMapPostHook(function(Engine, World)
LoadMods(World:get())
end)
RegisterLoadMapPostHook(function(Engine, World)
LoadMods(World:get())
end)

ExecuteInGameThread(function()
local ExistingActor = FindFirstOf("Actor")
if ExistingActor:IsValid() then
LoadMods(ExistingActor:GetWorld())
if (SpawnModsOnLuaInit) then
ExecuteInGameThread(function()
local ExistingActor = FindFirstOf("Actor")
if ExistingActor:IsValid() then
LoadMods(ExistingActor:GetWorld())
end
end)
end
end)
else
Log(string.format("No PAK mod entries found, skipping hooking the events for this run\n"), false)
end