feat: add support for ES6 module type

juanjoDiaz · Jul 15, 2023 · c083a2e · c083a2e
1 parent 39d10fb
commit c083a2e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/javascript.js b/src/javascript.js
@@ -23,7 +23,7 @@ function createJSMiddlewareHandler(handlers, pathToRoot) {
     }, {});
 
   const imports = Object.keys(handlersInfo)
-    .map((handler) => `const ${handlersInfo[handler]} = require('${pathToRoot}/${handler}');`).join('\n');
+    .map((handler) => `const ${handlersInfo[handler]} = require('${pathToRoot}/${handler}.js');`).join('\n');
 
   const promiseChain = handlers.map((handler) => {
     if (handler.then && handler.catch) {

diff --git a/src/typescript.js b/src/typescript.js
@@ -23,7 +23,7 @@ function createTSMiddlewareHandler(handlers, pathToRoot) {
     }, {});
 
   const imports = Object.keys(handlersInfo)
-    .map((handler) => `import * as ${handlersInfo[handler]} from '${pathToRoot}/${handler}';`).join('\n');
+    .map((handler) => `import * as ${handlersInfo[handler]} from '${pathToRoot}/${handler}.js';`).join('\n');
 
   const promiseChain = handlers.map((handler) => {
     if (handler.then && handler.catch) {

diff --git a/test/utils/generatedFunctionTester.js b/test/utils/generatedFunctionTester.js
@@ -19,9 +19,9 @@ class GeneratedFunctionTester {
     // eslint-disable-next-line no-new-func
     return new Function('event', 'context', 'dependencies', `
       const require = (dependencyName) => {
-        const dependency = dependencies[dependencyName.replace('../', '')];
+        const dependency = dependencies[dependencyName.replace('../', '').replace(/^(.+).js$/, '$1')];
         if (!dependency) {
-          throw new Error(\`Unknow dependency (\${dep})\`);
+          throw new Error(\`Unknow dependency (\${dependencyName})\`);
         }
 
         return dependency;