SheWhoReignsSupreme/revenge-bundle-next
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(lib/modules/metro): improve perf in patching module factories

Browse source
This commit is contained in:
PalmDevs 2025-07-03 21:28:16 +07:00
parent b56e6153a5
commit 75982bfb45
No known key found for this signature in database
GPG key ID: 6FAE8F8677ACA8BA
1 changed files with 53 additions and 37 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

90
lib/modules/src/metro/_internal.ts
View file

@ -10,55 +10,71 @@ export const _uninits = new Set<Metro.ModuleID>()
export const _inits = new Set<Metro.ModuleID>()
export const _paths = new Map<string, Metro.ModuleID>()
export const _metas = new Map<
Metro.ModuleID,
[deps: Metro.DependencyMap, initialized: boolean, module?: Metro.Module]
>()
export const _metas = new Map<Metro.ModuleID, Metadata>()
export type Metadata = [
deps: Metro.DependencyMap,
initialized: boolean,
module?: Metro.Module,
]
export function patchMetroDefine(metroDefine: Metro.DefineFn) {
return ((origFactory, id, deps) => {
// deps won't be undefined last time I checked
const metadata = [deps!, false] as typeof _metas extends Map<
any,
infer V
>
? V
: never
const metadata = [deps!, false] satisfies Metadata
_metas.set(id, metadata)
_uninits.add(id)
metroDefine(
((global, req, importDefault, importAll, module, exports, deps) => {
metadata[2] = module
const prevIId = _initing
_initing = id
_execReqSubs(id)
try {
origFactory(
global,
req,
importDefault,
importAll,
module,
exports,
deps,
)
metadata[1] = true
// Add the module to the initialized set only if the factory doesn't error or the exports aren't bad
// Don't use exports here, as modules can set module.exports to a different object
if (!isModuleExportsBad(module.exports)) _inits.add(id)
_execInitSubs(id, module.exports)
} catch {}
_initing = prevIId
_uninits.delete(id)
handleFactoryCall(
global,
req,
importDefault,
importAll,
module,
exports,
deps!,
id,
metadata,
origFactory,
)
}) satisfies Metro.FactoryFn,
id,
deps,
)
}) satisfies Metro.DefineFn
}
function handleFactoryCall(
global: object,
req: Metro.RequireFn,
importDefault: Metro.RequireFn,
importAll: Metro.RequireFn,
module: Metro.Module,
exports: unknown,
deps: Metro.ModuleID[],
id: Metro.ModuleID,
metadata: Metadata,
factory: Metro.FactoryFn,
) {
metadata[2] = module
const prevIId = _initing
_initing = id
_execReqSubs(id)
try {
factory(global, req, importDefault, importAll, module, exports, deps)
metadata[1] = true
// Add the module to the initialized set only if the factory doesn't error or the exports aren't bad
// Don't use exports here, as modules can set module.exports to a different object
if (!isModuleExportsBad(module.exports)) _inits.add(id)
_execInitSubs(id, module.exports)
} catch {}
_initing = prevIId
_uninits.delete(id)
}