feat: options for nativeModules and transformModules

README.md

@@ -111,6 +111,22 @@ Return the `.default` export of a module at the top-level.
 
 Custom alias map used to resolve ids.
 
+### `nativeModules`
+
+- Type: Array
+- Default: ['typescript`]
+- Environment Variable: `JITI_NATIVE_MODULES`
+
+List of modules (within `node_modules`) to always use native require for them.
+
+### `transformModules`
+
+- Type: Array
+- Default: []
+- Environment Variable: `JITI_TRANSFORM_MODULES`
+
+List of modules (within `node_modules`) to transform them regardless of syntax.
+
 ## Development
 
 - Clone Repo

package.json

@@ -44,9 +44,11 @@
     "acorn": "^8.8.0",
     "babel-plugin-dynamic-import-node": "^2.3.3",
     "babel-plugin-parameter-decorator": "^1.0.16",
+    "config": "^3.3.7",
     "create-require": "^1.1.1",
     "cross-env": "^7.0.3",
     "destr": "^1.1.1",
+    "escape-string-regexp": "^5.0.0",
     "eslint": "^8.23.0",
     "esm": "^3.2.25",
     "estree-walker": "^3.0.1",

src/jiti.ts

@@ -7,6 +7,7 @@ import { fileURLToPath, pathToFileURL } from 'url'
 import { dirname, join, basename, extname } from 'pathe'
 import { sync as mkdirpSync } from 'mkdirp'
 import destr from 'destr'
+import escapeStringRegexp from 'escape-string-regexp'
 import createRequire from 'create-require'
 import { lt } from 'semver'
 import { normalizeAliases, resolveAlias } from 'pathe/utils'
@@ -22,6 +23,8 @@ const _EnvESMResolve = destr(process.env.JITI_ESM_RESOLVE)
 const _EnvRequireCache = destr(process.env.JITI_REQUIRE_CACHE)
 const _EnvSourceMaps = destr(process.env.JITI_SOURCE_MAPS)
 const _EnvAlias = destr(process.env.JITI_ALIAS)
+const _EnvTransform = destr(process.env.JITI_TRANSFORM_MODULES)
+const _EnvNative = destr(process.env.JITI_NATIVE_MODULES)
 
 const isWindows = platform() === 'win32'
 
@@ -35,7 +38,9 @@ const defaults: JITIOptions = {
   cacheVersion: '7',
   legacy: lt(process.version || '0.0.0', '14.0.0'),
   extensions: ['.js', '.mjs', '.cjs', '.ts'],
-  alias: _EnvAlias
+  alias: _EnvAlias,
+  nativeModules: _EnvNative || [],
+  transformModules: _EnvTransform || []
 }
 
 type Require = typeof require
@@ -60,6 +65,12 @@ export default function createJITI (_filename: string, opts: JITIOptions = {}, p
     ? normalizeAliases(opts.alias || {})
     : null
 
+  // List of modules to force transform or native
+  const nativeModules = ['typescript', 'jiti'].concat(opts.nativeModules || [])
+  const transformModules = ([] as string[]).concat(opts.transformModules || [])
+  const isNativeRe = new RegExp(`node_modules/(${nativeModules.map(m => escapeStringRegexp(m)).join('|')})/`)
+  const isTransformRe = new RegExp(`node_modules/(${transformModules.map(m => escapeStringRegexp(m)).join('|')})/`)
+
   function debug (...args: string[]) {
     if (opts.debug) {
       // eslint-disable-next-line no-console
@@ -227,6 +238,12 @@ export default function createJITI (_filename: string, opts: JITIOptions = {}, p
       return nativeRequire(id)
     }
 
+    // Force native modules
+    if (isNativeRe.test(filename)) {
+      debug('[native]', filename)
+      return nativeRequire(id)
+    }
+
     // Check for CJS cache
     if (opts.requireCache && nativeRequire.cache[filename]) {
       return _interopDefault(nativeRequire.cache[filename]?.exports)
@@ -243,12 +260,10 @@ export default function createJITI (_filename: string, opts: JITIOptions = {}, p
     const needsTranspile = !isCommonJS && (
       isTypescript ||
       isNativeModule ||
+      isTransformRe.test(filename) ||
       hasESMSyntax(source) ||
-      (opts.legacy && detectLegacySyntax(source)) ||
-      // https://github.com/unjs/jiti/issues/56
-      filename.includes('node_modules/config/')
-    ) &&
-    !filename.includes('node_modules/typescript/')
+      (opts.legacy && detectLegacySyntax(source))
+    )
 
     const start = performance.now()
     if (needsTranspile) {

src/types.ts

@@ -28,4 +28,6 @@ export type JITIOptions = {
   extensions?: string[]
   transformOptions?: Omit<TransformOptions, 'source'>
   alias?: Record<string, string>
+  nativeModules?: string[]
+  transformModules?: string[]
 }

test/__snapshots__/fixtures.test.ts.snap

@@ -64,7 +64,10 @@ exports[`fixtures > esm > stdout 1`] = `
 }"
 `;
 
-exports[`fixtures > exotic > stdout 1`] = `"Typescript: true"`;
+exports[`fixtures > exotic > stdout 1`] = `
+"Typescript: true
+Config: true"
+`;
 
 exports[`fixtures > hashbang > stdout 1`] = `"1"`;
 

test/fixtures/exotic/index.ts

@@ -1,3 +1,7 @@
+
 import * as ts from 'typescript'
 
 console.log('Typescript:', 'mapEntries' in ts)
+
+// https://github.com/unjs/jiti/issues/56
+console.log('Config:', 'getFilesOrder' in require('config/parser'))