feat: expose jiti.evalModule

.eslintignore

@@ -3,3 +3,4 @@ test/fixtures/error-*
 renovate.json
 package.json
 tsconfig.json
+coverage

src/jiti.ts

@@ -50,9 +50,21 @@ const defaults: JITIOptions = {
 };
 
 type Require = typeof require;
+type Module = typeof module;
+
+type ModuleCache = Record<string, Module>;
+
+export type EvalModuleOptions = Partial<{
+  id: string;
+  filename: string;
+  ext: string;
+  cache: ModuleCache;
+}>;
+
 export interface JITI extends Require {
   transform: (opts: TransformOptions) => string;
   register: () => () => void;
+  evalModule: (source: string, options?: EvalModuleOptions) => unknown;
 }
 
 const JS_EXT_RE = /\.(c|m)?j(sx?)$/;
@@ -61,8 +73,8 @@ const TS_EXT_RE = /\.(c|m)?t(sx?)$/;
 export default function createJITI(
   _filename: string,
   opts: JITIOptions = {},
-  parentModule?: typeof module,
-  parentCache?: Record<string, typeof module>
+  parentModule?: Module,
+  parentCache?: ModuleCache
 ): JITI {
   opts = { ...defaults, ...opts };
 
@@ -263,7 +275,7 @@ export default function createJITI(
   }
 
   function jiti(id: string) {
-    const requiredModules = parentCache || {};
+    const cache = parentCache || {};
 
     // Check for node: and file: protocol
     if (id.startsWith("node:")) {
@@ -302,15 +314,30 @@ export default function createJITI(
     }
 
     // Check for CJS cache
-    if (requiredModules[filename]) {
-      return _interopDefault(requiredModules[filename]?.exports);
+    if (cache[filename]) {
+      return _interopDefault(cache[filename]?.exports);
     }
     if (opts.requireCache && nativeRequire.cache[filename]) {
       return _interopDefault(nativeRequire.cache[filename]?.exports);
     }
 
     // Read source
-    let source = readFileSync(filename, "utf8");
+    const source = readFileSync(filename, "utf8");
+
+    // Evaluate module
+    return evalModule(source, { id, filename, ext, cache });
+  }
+
+  function evalModule(source: string, evalOptions: EvalModuleOptions = {}) {
+    // Resolve options
+    const id =
+      evalOptions.id ||
+      (evalOptions.filename
+        ? basename(evalOptions.filename)
+        : `_jitiEval.${evalOptions.ext || ".js"}`);
+    const filename = evalOptions.filename || _resolve(id);
+    const ext = evalOptions.ext || extname(filename);
+    const cache = (opts.cache || parentCache || {}) as ModuleCache;
 
     // Transpile
     const isTypescript = ext === ".ts" || ext === ".mts" || ext === ".cts";
@@ -359,7 +386,7 @@ export default function createJITI(
       }
     }
 
-    mod.require = createJITI(filename, opts, mod, requiredModules);
+    mod.require = createJITI(filename, opts, mod, cache);
 
     // @ts-ignore
     mod.path = dirname(filename);
@@ -368,7 +395,7 @@ export default function createJITI(
     mod.paths = Module._nodeModulePaths(mod.path);
 
     // Set CJS cache before eval
-    requiredModules[filename] = mod;
+    cache[filename] = mod;
     if (opts.requireCache) {
       nativeRequire.cache[filename] = mod;
     }
@@ -440,6 +467,7 @@ export default function createJITI(
   jiti.main = nativeRequire.main;
   jiti.transform = transform;
   jiti.register = register;
+  jiti.evalModule = evalModule;
 
   return jiti;
 }

test/__snapshots__/fixtures.test.ts.snap

@@ -39,6 +39,7 @@ TypeError: The \\"listener\\" argument must be of type function. Received undefi
     at _addListener (events)
     at process.addListener (events)
     at <cwd>/index.ts
+    at evalModule (<root>/dist/jiti)
     at jiti (<root>/dist/jiti)
     at Object.<anonymous> (<root>/bin/jiti)
     at Module._compile (internal/modules/cjs/loader)
@@ -65,6 +66,7 @@ exports[`fixtures > esm > stdout 1`] = `
     'at getStack (<cwd>/test)',
     'at test (<cwd>/test)',
     'at <cwd>/index.js',
+    'at evalModule (<root>/dist/jiti)',
     'at jiti (<root>/dist/jiti)',
     'at Object.<anonymous> (<root>/bin/jiti)',
     'at Module._compile (internal/modules/cjs/loader)',