make changes

hook.js

@@ -134,8 +134,11 @@ function createHook (meta) {
   const iitmURL = new URL('lib/register.js', meta.url).toString()
   async function getSource (url, context, parentGetSource) {
     if (hasIitm(url)) {
+
       const realUrl = deleteIitm(url)
+
       const exportNames = await getExports(realUrl, context, parentGetSource)
+
       return { 
         source: `
 import { register } from '${iitmURL}'
@@ -162,7 +165,7 @@ register(${JSON.stringify(realUrl)}, namespace, set, ${JSON.stringify(specifiers
         return parentGetSource(url, context, parentGetSource)
       }      
       try {
-        const outPut = astParse(fileContents)
+        const outPut = astParse(fileContents, url)
         fileContents = outPut.code
         exportAlias = outPut.exportAlias
       } catch (parseError) {

lib/ast-parse.js

@@ -2,8 +2,8 @@
 
 const getEsmExports = require('./get-esm-exports.js')
 
-function astParse(fileContents) {
-  return getEsmExports(fileContents, true)
+function astParse(fileContents, url) {
+  return getEsmExports(fileContents, true, url)
 }
 
 module.exports = astParse

lib/get-esm-exports.js

@@ -4,6 +4,8 @@ const recast = require('recast');
 const { Parser } = require('acorn')
 const { importAssertions } = require('acorn-import-assertions');
 
+const TS_EXTENSION_RE = /\.(ts|mts|cts)$/
+
 const acornOpts = {
   ecmaVersion: 'latest',
   sourceType: 'module'
@@ -15,14 +17,21 @@ function warn (txt) {
   process.emitWarning(txt, 'get-esm-exports')
 }
 
-function getEsmExports(moduleStr, generate=false) {
+function getEsmExports(moduleStr, generate=false, url=undefined) {
   const exportSpecifierNames = new Set();
   const exportAlias = {}
-  const ast = recast.parse(moduleStr, {parser: {
-    parse(source) {
-      return parser.parse(source, acornOpts);
-    }
-  }})
+  let ast
+
+  // if it's a typescript file, we need to parse it with recasts typescript parser
+  if (url && TS_EXTENSION_RE.test(url)) {
+    ast = recast.parse(moduleStr, {parser: require("recast/parsers/typescript")})
+  } else {
+    ast = recast.parse(moduleStr, {parser: {
+      parse(source) {
+        return parser.parse(source, acornOpts);
+      }
+    }})
+  }
 
   recast.visit(ast, {
     visitExportNamedDeclaration(path) {
@@ -41,7 +50,7 @@ function getEsmExports(moduleStr, generate=false) {
       const node = path.node;
       const iitmRenamedExport = 'iitmRenamedExport';
 
-      if (node.declaration.type === 'ObjectExpression' || node.declaration.type === 'ArrayExpression' || node.declaration.type === 'Literal') {
+      if ((node.declaration.type === 'ObjectExpression' || node.declaration.type === 'ArrayExpression' || node.declaration.type === 'Literal') && generate) {
         const variableDeclaration = {
           type: 'VariableDeclaration',
           kind: 'let',
@@ -97,7 +106,7 @@ function getEsmExports(moduleStr, generate=false) {
     },
   });
 
-  if (exportSpecifierNames.size !== 0) {
+  if (exportSpecifierNames.size !== 0 && generate) {
     convertExportSpecifierToLet(exportSpecifierNames, ast)
   }
 

lib/get-exports.js

@@ -30,15 +30,15 @@ async function getExports (url, context, parentLoad) {
   }
 
   if (format === 'module') {
-    return getEsmExports(source)
+    return getEsmExports(source, false, url)
   }
   if (format === 'commonjs') {
     return addDefault(getCjsExports(source).exports)
   }
 
   // At this point our `format` is either undefined or not known by us. Fall
   // back to parsing as ESM/CJS.
-  const esmExports = getEsmExports(source)
+  const esmExports = getEsmExports(source, false, url)
   if (!esmExports.length) {
     // TODO(bengl) it's might be possible to get here if somehow the format
     // isn't set at first and yet we have an ESM module with no exports.

package.json

@@ -34,6 +34,7 @@
     "typescript": "^4.7.4"
   },
   "dependencies": {
+    "@babel/parser": "^7.23.5",
     "acorn": "^8.11.2",
     "acorn-import-assertions": "^1.9.0",
     "cjs-module-lexer": "^1.2.2",