fix(register): remove file extension tests in compile (#825)

packages/integrate-module/package.json

@@ -13,6 +13,7 @@
     "@swc-node/register": "workspace:*",
     "@types/react": "^18.3.3",
     "@types/react-dom": "^18.3.0",
+    "esmock": "^2.6.6",
     "ipaddr.js": "^2.2.0",
     "react": "^18.3.1",
     "react-dom": "^18.3.1",

packages/integrate-module/src/index.ts

@@ -4,6 +4,7 @@ import test from 'node:test'
 
 import { RepositoryState } from '@napi-rs/simple-git'
 import { bar as subBar } from '@subdirectory/bar.mjs'
+import esmock from 'esmock'
 import { supportedExtensions } from 'file-type'
 import { renderToString } from 'react-dom/server'
 import { simpleGit } from 'simple-git'
@@ -79,3 +80,13 @@ await test('resolve json file', () => {
 await test('resolve ipaddr.js', () => {
   assert.ok(ipaddr.isValid('192.168.1.1'))
 })
+
+await test('esmock should work', async () => {
+  const main = await esmock('./mocked.ts', {
+    path: {
+      basename: () => 'hello',
+    },
+  })
+
+  assert.strictEqual(main.pathbasenamewrap(), 'hello')
+})

packages/integrate-module/src/mocked.ts

@@ -0,0 +1,5 @@
+import path from 'path'
+
+export default {
+  pathbasenamewrap: (n: string) => path.basename(n),
+}

packages/register/esm.mts

@@ -17,7 +17,7 @@ import ts from 'typescript'
 // @ts-expect-error
 import { readDefaultTsConfig } from '../lib/read-default-tsconfig.js'
 // @ts-expect-error
-import { compile } from '../lib/register.js'
+import { compile, DEFAULT_EXTENSIONS } from '../lib/register.js'
 
 const debug = debugFactory('@swc-node')
 

packages/register/register.ts

@@ -5,7 +5,7 @@ import * as ts from 'typescript'
 
 import { readDefaultTsConfig, tsCompilerOptionsToSwcConfig } from './read-default-tsconfig'
 
-const DEFAULT_EXTENSIONS = [
+const DEFAULT_EXTENSIONS = new Set([
   ts.Extension.Js,
   ts.Extension.Ts,
   ts.Extension.Jsx,
@@ -16,17 +16,7 @@ const DEFAULT_EXTENSIONS = [
   ts.Extension.Cts,
   '.es6',
   '.es',
-]
-
-export const AVAILABLE_TS_EXTENSION_PATTERN = new RegExp(
-  `((?<!\\.d)(${[ts.Extension.Ts, ts.Extension.Tsx, ts.Extension.Mts, ts.Extension.Cts].map((ext) => ext.replace(/^\./, '\\.')).join('|')}))$`,
-  'i',
-)
-
-export const AVAILABLE_EXTENSION_PATTERN = new RegExp(
-  `((?<!\\.d)(${DEFAULT_EXTENSIONS.map((ext) => ext.replace(/^\./, '\\.')).join('|')}))$`,
-  'i',
-)
+])
 
 const injectInlineSourceMap = ({
   filename,
@@ -89,12 +79,8 @@ export function compile(
   },
   async = false,
 ) {
-  if (
-    typeof sourcecode === 'undefined' ||
-    (filename.includes('node_modules') && !AVAILABLE_TS_EXTENSION_PATTERN.test(filename)) ||
-    !AVAILABLE_EXTENSION_PATTERN.test(filename)
-  ) {
-    return sourcecode
+  if (sourcecode == null) {
+    return
   }
   if (options && typeof options.fallbackToTs === 'function' && options.fallbackToTs(filename)) {
     delete options.fallbackToTs
@@ -134,7 +120,7 @@ export function register(options: Partial<ts.CompilerOptions> = {}, hookOpts = {
   options.module = ts.ModuleKind.CommonJS
   installSourceMapSupport()
   return addHook((code, filename) => compile(code, filename, options), {
-    exts: DEFAULT_EXTENSIONS,
+    exts: Array.from(DEFAULT_EXTENSIONS),
     ...hookOpts,
   })
 }