import-in-the-middle 1.6.0 → 1.7.0

package/hook.js

@@ -2,25 +2,41 @@
 //
 // This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2021 Datadog, Inc.
 
+const fs = require('fs')
+const { fileURLToPath } = require('url')
 const specifiers = new Map()
 const isWin = process.platform === "win32"
+const warn = require('./lib/helpers')
 
 // FIXME: Typescript extensions are added temporarily until we find a better
 // way of supporting arbitrary extensions
 const EXTENSION_RE = /\.(js|mjs|cjs|ts|mts|cts)$/
+const EXTENSION_MJS_RE = /\.mjs$/
+const EXTENSION_JS_RE = /\.js$/
 const NODE_VERSION = process.versions.node.split('.')
 const NODE_MAJOR = Number(NODE_VERSION[0])
 const NODE_MINOR = Number(NODE_VERSION[1])
+const FILE_NAME = 'hook.js'
 
 let entrypoint
-
 let getExports
+let getEsmExports
+let getPkgJsonTypeModule
+
 if (NODE_MAJOR >= 20 || (NODE_MAJOR == 18 && NODE_MINOR >= 19)) {
   getExports = require('./lib/get-exports.js')
 } else {
   getExports = (url) => import(url).then(Object.keys)
 }
 
+if (NODE_MAJOR >= 16) {
+  getEsmExports = require('./lib/get-esm-exports.js')
+  getPkgJsonTypeModule = require('./lib/get-pkg-json-type-module.js')
+} else {
+  getEsmExports = undefined
+  getPkgJsonTypeModule = undefined
+}
+
 function hasIitm (url) {
   try {
     return new URL(url).searchParams.has('iitm')
@@ -108,12 +124,23 @@ function createHook (meta) {
     ) {
       return url
     }
-
+    
+    // on Node's 16.0.0-16.12.0, url.format is undefined for the cyclical dependency test files ./test/fixtures/a.mjs & ./test/fixtures/b.mjs 
+    // so explicitly set format to 'module' for files with a .mjs extension or cjs files that have type 'module in their package.json
+    // so that they can go through the ast parsing patch for Node >= 16
+    if (NODE_MAJOR === 16 && NODE_MINOR < 13) {
+      if (
+        (url.format === undefined && EXTENSION_MJS_RE.test(url.url)) || 
+        (EXTENSION_JS_RE.test(url.url) && getPkgJsonTypeModule(fileURLToPath(url.url)))
+      ) {
+        url.format = 'module'
+      }
+    }
 
     specifiers.set(url.url, specifier)
 
     return {
-      url: addIitm(url.url),
+      url: url.format !== 'module' || NODE_MAJOR < 16 ? addIitm(url.url) : url.url,
       shortCircuit: true,
       format: url.format
     }
@@ -122,9 +149,12 @@ 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 {
+      
+      return { 
         source: `
 import { register } from '${iitmURL}'
 import * as namespace from ${JSON.stringify(url)}
@@ -139,6 +169,41 @@ set.${n} = (v) => {
 `).join('\n')}
 register(${JSON.stringify(realUrl)}, namespace, set, ${JSON.stringify(specifiers.get(realUrl))})
 `
+      } 
+    } else if (NODE_MAJOR >= 16 && context.format === 'module') {
+      let fileContents
+      const realPath = fileURLToPath(url)
+      try {
+        fileContents = fs.readFileSync(realPath, 'utf8')
+      } catch (parseError) {
+        warn(`Had trouble reading file: ${fileContents}, got error: ${parseError}`, FILE_NAME)
+        return parentGetSource(url, context, parentGetSource)
+      }      
+      try {
+        const outPut = getEsmExports(fileContents, true, url)
+        fileContents = outPut.code
+        exportAlias = outPut.exportAlias
+      } catch (parseError) {
+        warn(`Tried AST parsing ${realPath}, got error: ${parseError}`, FILE_NAME)
+        return parentGetSource(url, context, parentGetSource)
+      }
+      const src = `${fileContents}
+import { register as DATADOG_REGISTER_FUNC } from '${iitmURL}'
+{
+  const set = {}
+  const namespace = {}
+  ${Object.entries(exportAlias).map(([key, value]) => `
+  set.${key} = (v) => {
+    ${value} = v
+    return true
+  }
+  namespace.${key} = ${value}
+`).join('\n')}
+DATADOG_REGISTER_FUNC(${JSON.stringify(url)}, namespace, set, ${JSON.stringify(specifiers.get(url))})
+}
+`
+      return { 
+        source: src
       }
     }
 
@@ -147,7 +212,7 @@ register(${JSON.stringify(realUrl)}, namespace, set, ${JSON.stringify(specifiers
 
   // For Node.js 16.12.0 and higher.
   async function load (url, context, parentLoad) {
-    if (hasIitm(url)) {
+    if (hasIitm(url) || context.format === 'module') {
       const { source } = await getSource(url, context, parentLoad)
       return {
         source,
@@ -167,7 +232,7 @@ register(${JSON.stringify(realUrl)}, namespace, set, ${JSON.stringify(specifiers
       resolve,
       getSource,
       getFormat (url, context, parentGetFormat) {
-        if (hasIitm(url)) {
+        if (hasIitm(url) || context.format === 'module') {
           return {
             format: 'module'
           }

package/lib/get-esm-exports.js

@@ -1,7 +1,11 @@
 'use strict'
 
+const recast = require('recast')
 const { Parser } = require('acorn')
-const { importAssertions } = require('acorn-import-assertions');
+const { importAssertions } = require('acorn-import-assertions')
+const warn = require('./helpers')
+
+const TS_EXTENSION_RE = /\.(ts|mts|cts)$/
 
 const acornOpts = {
   ecmaVersion: 'latest',
@@ -9,87 +13,182 @@ const acornOpts = {
 }
 
 const parser = Parser.extend(importAssertions)
+const FILE_NAME = 'get-esm-exports'
+
+function getEsmExports(moduleStr, generate=false, url=undefined) {
+  const exportSpecifierNames = new Set()
+  const exportAlias = {}
+  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)
+      }
+    }})
+  }
+
+  const iitmRenamedExport = 'iitmRenamedExport';
+
+  // Loop through the top-level declarations of the AST
+  for (const statement of ast.program.body) {
+    if (statement.type === 'ExportNamedDeclaration') {
+      const node = statement;
+
+      if (node.declaration) {
+        parseDeclaration(node.declaration, exportAlias);
+      } else {
+        parseSpecifiers(node.specifiers, exportAlias, exportSpecifierNames);
+      }
+    } else if (statement.type === 'ExportDefaultDeclaration') {
+      const node = statement;
 
-function warn (txt) {
-  process.emitWarning(txt, 'get-esm-exports')
+      if (['ObjectExpression', 'ArrayExpression', 'Literal'].includes(node.declaration.type) && generate) {
+        const variableDeclaration = {
+          type: 'VariableDeclaration',
+          kind: 'let',
+          declarations: [
+            {
+              type: 'VariableDeclarator',
+              id: {
+                type: 'Identifier',
+                name: iitmRenamedExport,
+              },
+              init: node.declaration,
+            },
+          ],
+        };
+
+        // Replace the current ExportDefaultDeclaration with the new VariableDeclaration
+        ast.program.body.splice(ast.program.body.indexOf(statement), 1, variableDeclaration);
+
+        const newExportDefaultDeclaration = {
+          type: 'ExportDefaultDeclaration',
+          declaration: {
+            type: 'Identifier',
+            name: iitmRenamedExport,
+          },
+        };
+
+        // Insert the new ExportDefaultDeclaration after the VariableDeclaration
+        ast.program.body.splice(ast.program.body.indexOf(variableDeclaration) + 1, 0, newExportDefaultDeclaration);
+      } else if (['FunctionDeclaration', 'Identifier', 'ClassDeclaration'].includes(node.declaration.type) && generate) {
+        node.declaration.id = { type: 'Identifier', name: iitmRenamedExport };
+      }
+      exportAlias['default'] = iitmRenamedExport;
+    } else if (statement.type === 'ExportAllDeclaration') {
+      const node = statement;
+      const exportedName = node.exported ? node.exported.name : '*';
+
+      exportAlias[exportedName] = exportedName;
+    } else if (statement.type === 'ExportSpecifier') {
+      const node = statement;
+      exportSpecifierNames.add(node.local.name);
+
+      if (node.exported.name) {
+        exportAlias[node.exported.name] = node.local.name;
+      } else if (node.exported.value) {
+        exportAlias[node.exported.value] = node.local.name;
+      } else {
+        warn('unrecognized specifier export: ' + node.exported, FILE_NAME);
+      }
+    }
+  }
+
+  if (exportSpecifierNames.size !== 0 && generate) {
+    convertExportSpecifierToLet(exportSpecifierNames, ast);
+  }
+
+  if (generate) {
+    return {
+      exportAlias: exportAlias,
+      code: recast.print(ast).code,
+    };
+  }
+
+  return Object.keys(exportAlias);
 }
 
-function getEsmExports (moduleStr) {
-  const exportedNames = new Set()
-  const tree = parser.parse(moduleStr, acornOpts)
-  for (const node of tree.body) {
-    if (!node.type.startsWith('Export')) continue
-    switch (node.type) {
-      case 'ExportNamedDeclaration':
-        if (node.declaration) {
-          parseDeclaration(node, exportedNames)
-        } else {
-          parseSpecifiers(node, exportedNames)
+function convertExportSpecifierToLet(exportSpecifierNames, ast) {
+  for (const statement of ast.program.body) {
+    if (statement.type === 'VariableDeclaration') {
+      const declaration = statement;
+  
+      if (declaration.kind === 'const') {
+        for (const declarator of declaration.declarations) {
+          const variableName = declarator.id.name;
+          if (exportSpecifierNames.has(variableName)) {
+            declaration.kind = 'let';
+          }
         }
-        break
-      case 'ExportDefaultDeclaration':
-        exportedNames.add('default')
-        break
-      case 'ExportAllDeclaration':
-        if (node.exported) {
-          exportedNames.add(node.exported.name)
-        } else {
-          exportedNames.add('*')
-        }
-        break
-      default:
-        warn('unrecognized export type: ' + node.type)
+      }
     }
   }
-  return Array.from(exportedNames)
 }
 
-function parseDeclaration (node, exportedNames) {
-  switch (node.declaration.type) {
+function parseDeclaration(declaration, exportAlias) {
+  switch (declaration.type) {
     case 'FunctionDeclaration':
-      exportedNames.add(node.declaration.id.name)
+    case 'ClassDeclaration':
+      exportAlias[declaration.id.name] = declaration.id.name
       break
     case 'VariableDeclaration':
-      for (const varDecl of node.declaration.declarations) {
-        parseVariableDeclaration(varDecl, exportedNames)
+      for (const varDecl of declaration.declarations) {
+        if (declaration.kind === 'const') {
+          declaration.kind = 'let'
+        }
+        parseVariableDeclaration(varDecl, exportAlias)
       }
       break
-    case 'ClassDeclaration':
-      exportedNames.add(node.declaration.id.name)
-      break
     default:
-      warn('unknown declaration type: ' + node.delcaration.type)
+      warn('unknown declaration type: ' + declaration.type, FILE_NAME)
   }
 }
 
-function parseVariableDeclaration (node, exportedNames) {
-  switch (node.id.type) {
+function parseVariableDeclaration(varDecl, exportAlias) {
+  switch (varDecl.id.type) {
     case 'Identifier':
-      exportedNames.add(node.id.name)
+      exportAlias[varDecl.id.name] = varDecl.id.name
       break
     case 'ObjectPattern':
-      for (const prop of node.id.properties) {
-        exportedNames.add(prop.value.name)
+      for (const prop of varDecl.id.properties) {
+        exportAlias[prop.value.name] = prop.value.name
       }
       break
     case 'ArrayPattern':
-      for (const elem of node.id.elements) {
-        exportedNames.add(elem.name)
+      for (const elem of varDecl.id.elements) {
+        if (elem) {
+          exportAlias[elem.name] = elem.name
+        }
       }
       break
     default:
-      warn('unknown variable declaration type: ' + node.id.type)
+      warn('unknown variable declaration type: ' + varDecl.id.type, FILE_NAME)
   }
 }
 
-function parseSpecifiers (node, exportedNames) {
-  for (const specifier of node.specifiers) {
-    if (specifier.exported.type === 'Identifier') {
-      exportedNames.add(specifier.exported.name)
+function parseSpecifiers(specifiers, exportAlias, exportSpecifierNames) {
+  for (const specifier of specifiers) {
+    if (specifier.type === 'ExportSpecifier') {
+      exportSpecifierNames.add(specifier.local.name)
+      if (specifier.exported && specifier.exported.name) {
+        exportAlias[specifier.exported.name] = specifier.local.name
+      } else if (specifier.exported && specifier.exported.value) {
+        exportAlias[specifier.exported.value] = specifier.local.name
+      } else {
+        warn('unrecognized specifier export: ' + specifier, FILE_NAME)
+      }
+    }
+    else if (specifier.exported.type === 'Identifier') {
+      exportAlias[specifier.exported.name] = specifier.exported.name
     } else if (specifier.exported.type === 'Literal') {
-      exportedNames.add(specifier.exported.value)
-    } else {
-      warn('unrecognized specifier type: ' + specifier.exported.type)
+      exportAlias[specifier.exported.value] = specifier.exported.value
+    } 
+    else {
+      warn('unrecognized specifier type: ' + specifier.exported.type, FILE_NAME)
     }
   }
 }

package/lib/get-exports.js

@@ -30,7 +30,7 @@ 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)
@@ -38,7 +38,7 @@ async function getExports (url, context, parentLoad) {
 
   // 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/lib/get-pkg-json-type-module.js

@@ -0,0 +1,46 @@
+const fs = require('fs')
+const path = require('path')
+const warn = require('./helpers')
+
+const FILE_NAME = 'get-pkg-json-type-module'
+
+function findNearestPackageJson(filePath) {
+  let currentDir = path.dirname(filePath)
+  
+  while (currentDir !== '/') {
+    const packageJsonPath = path.join(currentDir, 'package.json')
+    try {
+      const packageJsonContent = fs.readFileSync(packageJsonPath, 'utf8')
+      return packageJsonContent;
+    } catch (error) {
+      // File does not exist, continue searching in the parent directory
+    }
+
+    currentDir = path.dirname(currentDir)
+  }
+
+  return null
+}
+
+function isModuleType(filePath) {
+  const packageJsonContent = findNearestPackageJson(filePath)
+
+  if (!packageJsonContent) {
+    return false
+  }
+
+  try {
+    const packageJson = JSON.parse(packageJsonContent)
+
+    if (packageJson.type === 'module') {
+      return true
+    } else {
+      return false
+    }
+  } catch (error) {
+    warn('Error reading or parsing package.json: ' + error.message, FILE_NAME)
+    return false
+  }
+}
+
+module.exports = isModuleType

package/lib/helpers.js

@@ -0,0 +1,5 @@
+function warn (txt, fileName) {
+  process.emitWarning(txt, fileName)
+}
+
+module.exports = warn

package/package.json

@@ -1,10 +1,10 @@
 {
   "name": "import-in-the-middle",
-  "version": "1.6.0",
+  "version": "1.7.0",
   "description": "Intercept imports in Node.js",
   "main": "index.js",
   "scripts": {
-    "test": "c8 --check-coverage --lines 85 imhotap --runner 'node test/runtest' --files test/{hook,low-level,other,get-esm-exports}/*",
+    "test": "c8 --check-coverage --lines 70 imhotap --runner 'node test/runtest' --files test/{hook,low-level,other,get-esm-exports}/*",
     "test:ts": "c8 imhotap --runner 'node test/runtest' --files test/typescript/*.test.mts",
     "coverage": "c8 --reporter html imhotap --runner 'node test/runtest' --files test/{hook,low-level,other,get-esm-exports}/* && echo '\nNow open coverage/index.html\n'"
   },
@@ -34,9 +34,11 @@
     "typescript": "^4.7.4"
   },
   "dependencies": {
-    "acorn": "^8.8.2",
+    "@babel/parser": "^7.23.5",
+    "acorn": "^8.11.2",
     "acorn-import-assertions": "^1.9.0",
     "cjs-module-lexer": "^1.2.2",
-    "module-details-from-path": "^1.0.3"
+    "module-details-from-path": "^1.0.3",
+    "recast": "^0.23.4"
   }
 }

package/test/fixtures/a.mjs

@@ -0,0 +1,7 @@
+import { testB } from './b.mjs';
+
+export function testA() {
+  console.log("testA");
+}
+
+testB();

package/test/fixtures/b.mjs

@@ -0,0 +1,6 @@
+import { testA } from './a.mjs';
+
+export function testB() {
+  console.log("testB");
+  testA();
+}

package/test/fixtures/env.mjs

@@ -0,0 +1,8 @@
+let env = {FOO: 'baz'};
+
+function setEnv(newEnv) {
+  console.log('setting env, env.FOO is', newEnv.FOO);
+  env = newEnv;
+}
+
+export { setEnv, env };

package/test/fixtures/export-types/declarations.mjs

@@ -0,0 +1,19 @@
+const o = { name5: 1, name6: 1 };
+const array = [1, 1]
+
+// Exporting declarations
+export let name1 = 1, name2 = 1/*, … */; // also var
+export const name3 = 1, name4 = 1/*, … */; // also var, let
+export function functionName() { return 1 }
+export class ClassName { getFoo() { return 1 } }
+export function* generatorFunctionName() { return 1 }
+export const { name5, name6: bar } = o;
+export const [ name7, name8 ] = array;
+export async function asyncFunctionName() { return 1 }
+export async function* asyncGeneratorFunctionName() { yield 1 }
+export const arrowFunction = () => {
+  return 1;
+}
+export const asyncArrowFunction = async () => {
+  return 1;
+}

package/test/fixtures/export-types/default-class-anon.mjs

@@ -0,0 +1 @@
+export default class { getFoo() { return 1 } }

package/test/fixtures/export-types/default-class.mjs

@@ -0,0 +1 @@
+export default class ClassName { getFoo() { return 1 } }

package/test/fixtures/export-types/default-expression-array.mjs

@@ -0,0 +1 @@
+export default [1]

package/test/fixtures/export-types/default-expression-num.mjs

@@ -0,0 +1 @@
+export default 1

package/test/fixtures/export-types/default-expression-string.mjs

@@ -0,0 +1 @@
+export default 'dog'

package/test/fixtures/export-types/default-function-anon.mjs

@@ -0,0 +1 @@
+export default function () { return 1 }

package/test/fixtures/export-types/default-function.mjs

@@ -0,0 +1 @@
+export default function functionName() { return 1 }

package/test/fixtures/export-types/default-generator-anon.mjs

@@ -0,0 +1 @@
+export default function* () { return 1 }

package/test/fixtures/export-types/default-generator.mjs

@@ -0,0 +1 @@
+export default function* generatorFunctionName() { return 1 }

package/test/fixtures/export-types/list.mjs

@@ -0,0 +1,12 @@
+// Export list
+const name1 = 1
+const name2 = 1
+const name5 = 1
+const variable1 = 1
+const variable2 = 1
+const variable3 = 1
+const name6 = 1
+export { name1, name2 };
+export { variable1 as name3, variable2 as name4, /* …, */ name5 };
+export { variable3 as "name" };
+export { name6 as default /*, … */ };

package/test/fixtures/reassign-let.mjs

@@ -0,0 +1,5 @@
+import { setEnv } from "./env.mjs"
+import {useEnv} from "./use-env.mjs"
+
+setEnv({ FOO: 'bar'})
+useEnv()

package/test/fixtures/test-undefined-format/a.js

@@ -0,0 +1,8 @@
+import { testB } from './b.js';
+
+export function testA() {
+  console.log("testA");
+}
+
+testB();
+

package/test/fixtures/test-undefined-format/b.js

@@ -0,0 +1,7 @@
+import { testA } from './a.js';
+
+export function testB() {
+  console.log("testB");
+  testA();
+}
+

package/test/fixtures/test-undefined-format/package.json

@@ -0,0 +1,9 @@
+{
+  "name": "test-v16.12-undefined-format",
+  "private": true,
+  "type": "module",
+  "scripts": {
+    "test-undefined-format": "node --loader ../../../hook.mjs ./a.js"
+  },
+  "dependencies": {}
+}

package/test/fixtures/use-env.mjs

@@ -0,0 +1,5 @@
+import {env} from "./env.mjs"
+
+export function useEnv() {
+  console.log('using env from another module, env.FOO is', env.FOO)
+}

package/test/get-esm-exports/v18.19-get-esm-exports.js

@@ -1,3 +1,3 @@
 // v18.19.0 backported ESM hook execution to a separate thread,
 // thus being equivalent to >=v20.
-require('./v20-get-esm-exports')
+require('./v16-get-esm-exports')

package/test/hook/default-exports.mjs

@@ -0,0 +1,71 @@
+// Unless explicitly stated otherwise all files in this repository are licensed under the Apache 2.0 License.
+//
+// This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2021 Datadog, Inc.
+
+import Hook from '../../index.js'
+import a from '../fixtures/export-types/default-expression-array.mjs'
+import n from '../fixtures/export-types/default-expression-num.mjs'
+import s from '../fixtures/export-types/default-expression-string.mjs'
+import fn from '../fixtures/export-types/default-function.mjs'
+import cn from '../fixtures/export-types/default-class.mjs'
+import gfn from '../fixtures/export-types/default-generator.mjs'
+import afn from '../fixtures/export-types/default-function-anon.mjs'
+import acn from '../fixtures/export-types/default-class-anon.mjs'
+import agfn from '../fixtures/export-types/default-generator-anon.mjs'
+import { strictEqual } from 'assert'
+
+Hook((exports, name) => {
+  if (name.match(/default-expression-array\.m?js/)) {
+    exports.default[0] += 1
+  } else if (name.match(/default-expression-num\.m?js/)) {
+    exports.default += 1
+  }
+  else if (name.match(/default-expression-string\.m?js/)) {
+    exports.default += 'dawg'
+  }
+  else if (name.match(/default-function\.m?js/)) {
+    const orig = exports.default
+    exports.default = function () {
+      return orig() + 1
+    }
+  }
+  else if (name.match(/default-class\.m?js/)) {
+    exports.default.prototype.getFoo = function () {
+      return 2
+    }
+  }
+  else if (name.match(/default-generator\.m?js/)) {
+    const orig2 = exports.default
+    exports.default = function* () {
+      return orig2().next().value + 1
+    }
+  }
+  else if (name.match(/default-function-anon\.m?js/)) {
+    const orig = exports.default
+    exports.default = function () {
+      return orig() + 1
+    }
+  }
+  else if (name.match(/default-class-anon\.m?js/)) {
+    exports.default.prototype.getFoo = function () {
+      return 2
+    }
+  }
+  else if (name.match(/default-generator-anon\.m?js/)) {
+    const orig2 = exports.default
+    exports.default = function* () {
+      return orig2().next().value + 1
+    }
+  }
+})
+
+strictEqual(a[0], 2)
+strictEqual(fn(), 2)
+strictEqual(new cn().getFoo(), 2)
+strictEqual(gfn().next().value, 2)
+strictEqual(afn(), 2)
+strictEqual(new acn().getFoo(), 2)
+strictEqual(agfn().next().value, 2)
+// the below tests won't work because literal default exports are static
+// strictEqual(n, 2)
+// strictEqual(s, 'dogdawg')

package/test/hook/v16-declaration-exports.mjs

@@ -0,0 +1,84 @@
+// Unless explicitly stated otherwise all files in this repository are licensed under the Apache 2.0 License.
+//
+// This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2021 Datadog, Inc.
+
+import Hook from '../../index.js'
+import {
+  name1 as n1,
+  name2 as n2, 
+  name3 as n3, 
+  name4 as n4, 
+  functionName as fn, 
+  ClassName as cn, 
+  generatorFunctionName as gfn, 
+  name5 as n5, 
+  bar as n6, 
+  name7 as n7, 
+  name8 as n8,
+  asyncFunctionName as afn,
+  asyncGeneratorFunctionName as agfn,
+  arrowFunction as arfn,
+  asyncArrowFunction as aarfn,
+  } from '../fixtures/export-types/declarations.mjs'
+import { strictEqual } from 'assert'
+
+Hook((exports, name) => {
+  if (name.match(/declarations\.m?js/)) {
+    exports.name1 += 1
+    exports.name2 += 1
+    exports.name3 += 1
+    exports.name4 += 1
+    const orig = exports.functionName
+    exports.functionName = function () {
+      return orig() + 1
+    }
+    exports.ClassName.prototype.getFoo = function () {
+      return 2
+    }
+    const orig2 = exports.generatorFunctionName
+    exports.generatorFunctionName = function* () {
+      return orig2().next().value + 1
+    }
+    exports.name5 += 1
+    exports.bar += 1
+    exports.name7 += 1
+    exports.name8 += 1
+    const asyncOrig = exports.asyncFunctionName
+    exports.asyncFunctionName = async function () {
+      return await asyncOrig() + 1
+    }
+    const asyncOrig2 = exports.asyncGeneratorFunctionName
+    exports.asyncGeneratorFunctionName = async function* () {
+      for await (const value of asyncOrig2()) {
+        yield value + 1;
+      }
+    }
+    const arrowOrig = exports.arrowFunction
+    exports.arrowFunction = () => {
+      return arrowOrig() + 1
+    }
+    const asyncArrowOrig = exports.asyncArrowFunction
+    exports.asyncArrowFunction = async () => {
+      return await asyncArrowOrig() + 1
+    }
+  }
+})
+
+strictEqual(n1, 2)
+strictEqual(n2, 2)
+strictEqual(n3, 2)
+strictEqual(n4, 2)
+strictEqual(fn(), 2)
+strictEqual(new cn().getFoo(), 2)
+strictEqual(gfn().next().value, 2)
+strictEqual(n5, 2)
+strictEqual(n6, 2)
+strictEqual(n7, 2)
+strictEqual(n8, 2)
+strictEqual(await afn(), 2)
+for await (const value of agfn()) {
+  strictEqual(value, 2)
+}
+strictEqual(arfn(), 2)
+strictEqual(await aarfn(), 2)
+

package/test/hook/v16-list-exports.mjs

@@ -0,0 +1,34 @@
+// Unless explicitly stated otherwise all files in this repository are licensed under the Apache 2.0 License.
+//
+// This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2021 Datadog, Inc.
+
+import Hook from '../../index.js'
+import n, {
+  name1 as n1,
+  name2 as n2, 
+  name3 as n3, 
+  name4 as n4, 
+  name5 as n5, 
+  "name" as n6,
+  } from '../fixtures/export-types/list.mjs'
+import { strictEqual } from 'assert'
+
+Hook((exports, name) => {
+  if (name.match(/list\.m?js/)) {
+    exports.name1 += 1
+    exports.name2 += 1
+    exports.name3 += 1
+    exports.name4 += 1
+    exports.name5 += 1
+    exports.name += 1
+    exports.default += 1
+  }
+})
+
+strictEqual(n1, 2)
+strictEqual(n2, 2)
+strictEqual(n3, 2)
+strictEqual(n4, 2)
+strictEqual(n5, 2)
+strictEqual(n6, 2)
+strictEqual(n, 2)

package/test/other/v16-cyclical-dependency.mjs

@@ -0,0 +1,29 @@
+// Unless explicitly stated otherwise all files in this repository are licensed under the Apache 2.0 License.
+//
+// This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2021 Datadog, Inc.
+
+import { spawn } from 'child_process'
+import { strictEqual } from 'assert'
+
+const nodeProcess = spawn('node', [
+  '--loader', 
+  './hook.mjs', 
+  './test/fixtures/a.mjs'
+])
+
+const expectedOutput = 'testB\ntestA'
+let stdout = ''
+let stderr = ''
+
+nodeProcess.stdout.on('data', (data) => {
+  stdout += data.toString()
+});
+
+nodeProcess.stderr.on('data', (data) => {
+  stderr += data.toString()
+});
+
+nodeProcess.on('close', (code) => {
+  strictEqual(stderr, '', 'There should be no errors on stderr')
+  strictEqual(stdout.trim(), expectedOutput, 'The stdout should match the expected output')
+});

package/test/other/v16-reassign-let.mjs

@@ -0,0 +1,29 @@
+// Unless explicitly stated otherwise all files in this repository are licensed under the Apache 2.0 License.
+//
+// This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2021 Datadog, Inc.
+
+import { spawn } from 'child_process'
+import { strictEqual } from 'assert'
+
+const nodeProcess = spawn('node', [
+  '--loader', 
+  './hook.mjs', 
+  './test/fixtures/reassign-let.mjs'
+])
+
+const expectedOutput = 'setting env, env.FOO is bar\nusing env from another module, env.FOO is bar'
+let stdout = ''
+let stderr = ''
+
+nodeProcess.stdout.on('data', (data) => {
+  stdout += data.toString()
+})
+
+nodeProcess.stderr.on('data', (data) => {
+  stderr += data.toString()
+})
+
+nodeProcess.on('close', (code) => {
+  strictEqual(stderr, '', 'There should be no errors on stderr')
+  strictEqual(stdout.trim(), expectedOutput, 'The stdout should match the expected output')
+})

package/test/other/v16-test-undefined-format.mjs

@@ -0,0 +1,26 @@
+// Unless explicitly stated otherwise all files in this repository are licensed under the Apache 2.0 License.
+//
+// This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2021 Datadog, Inc.
+
+import { spawn } from 'child_process'
+import { strictEqual } from 'assert'
+
+const nodeProcess = spawn('node', ['--loader', '../../../hook.mjs', './a.js'], { 
+  cwd: './test/fixtures/test-undefined-format/'})
+
+const expectedOutput = 'testB\ntestA'
+let stdout = ''
+let stderr = ''
+
+nodeProcess.stdout.on('data', (data) => {
+  stdout += data.toString()
+});
+
+nodeProcess.stderr.on('data', (data) => {
+  stderr += data.toString()
+});
+
+nodeProcess.on('close', (code) => {
+  strictEqual(stderr, '', 'There should be no errors on stderr')
+  strictEqual(stdout.trim(), expectedOutput, 'The stdout should match the expected output')
+});