revenge-bundle-next/lib/modules/src/metro/_internal.ts
2025-06-03 16:24:58 +07:00

64 lines
2.1 KiB
TypeScript

import { _execInitSubs, _execReqSubs } from './subscriptions/_internal'
import { isModuleExportsBad } from './utils'
import type { Metro } from '../types'
/** Initializing ID */
export let _initing = -1
/** Known uninitialized IDs */
export const _uninits = new Set<Metro.ModuleID>()
/** Known initialized IDs (not blacklisted) */
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 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
_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)
}) satisfies Metro.FactoryFn,
id,
deps,
)
}) satisfies Metro.DefineFn
}