@wyw-in-js/transform 0.1.1 → 0.2.0

package/lib/utils/collectExportsAndImports.js

@@ -250,21 +250,60 @@ function getCalleeName(path) {
   }
   return undefined;
 }
-function getImportExportTypeByInteropFunction(path) {
-  const name = getCalleeName(path);
+const matchCall = (p, rules) => {
+  const name = getCalleeName(p);
   if (name === undefined) {
-    return undefined;
+    return false;
   }
-  if (name.startsWith('__exportStar')) {
+  return rules.some(([n, ...args]) => {
+    if (name !== n) return false;
+    const fnArgs = p.get('arguments');
+    if (fnArgs.length !== args.length) return false;
+    return args.every((arg, i) => {
+      if (arg === '*') return true;
+      if (typeof arg === 'function') return arg(fnArgs[i]);
+      return arg === fnArgs[i];
+    });
+  });
+};
+function getImportExportTypeByInteropFunction(path, argPath) {
+  if (matchCall(path, [['__exportStar', argPath, p => (0, _isExports.isExports)(p)]])) {
     return 're-export:*';
   }
-  if (name.startsWith('_interopRequireDefault') || name.startsWith('__importDefault')) {
+  if (matchCall(path, [['_interopRequireDefault', argPath],
+  // babel and swc <1.3.50
+  ['_interop_require_default', argPath],
+  // swc >=1.3.50
+  ['__importDefault', argPath] // ?
+  ])) {
     return 'default';
   }
-  if (name.startsWith('_interopRequireWildcard') || name.startsWith('__importStar') || name.startsWith('__toESM')) {
+  if (matchCall(path, [['_interopRequireWildcard', argPath],
+  // babel and swc <1.3.50
+  ['_interop_require_wildcard', argPath],
+  // swc >=1.3.50
+  ['__importStar', argPath],
+  // ?
+  ['__toESM', argPath],
+  // esbuild >=0.14.7
+  ['__toModule', argPath] // esbuild <0.14.7
+  ])) {
     return 'import:*';
   }
-  if (name.startsWith('__rest') || name.startsWith('__objRest') || name.startsWith('_objectDestructuringEmpty')) {
+  if (matchCall(path, [['_extends', isEmptyObject, argPath],
+  // babel and swc
+  ['__rest', argPath, isArrayExpression],
+  // tsc and esbuild <=0.11.3
+  ['__objRest', argPath, isArrayExpression],
+  // esbuild >0.11.3
+  ['_objectWithoutProperties', argPath, isArrayExpression],
+  // babel and swc <1.3.50
+  ['_object_without_properties', argPath, isArrayExpression],
+  // swc >=1.3.50
+  ['_objectDestructuringEmpty', argPath],
+  // swc <1.3.50
+  ['_object_destructuring_empty', argPath] // swc >=1.3.50
+  ])) {
     return 'import:*';
   }
   return undefined;
@@ -278,6 +317,31 @@ function isAlreadyProcessed(path) {
   }
   return false;
 }
+function isRequireCall(path) {
+  return path.isCallExpression() && (0, _isRequire.isRequire)(path.get('callee'));
+}
+function isEmptyObject(path) {
+  return path.isObjectExpression() && path.node.properties.length === 0;
+}
+function isArrayExpression(path) {
+  return path.isArrayExpression();
+}
+function isCallExpression(pathOrName) {
+  if (typeof pathOrName === 'string') {
+    return p => p.isCallExpression() && p.get('callee').isIdentifier({
+      name: pathOrName
+    });
+  }
+  return pathOrName.isCallExpression();
+}
+function isObjectExpression(path) {
+  return path.isObjectExpression();
+}
+function isIdentifier(name) {
+  return path => path.isIdentifier({
+    name
+  });
+}
 function collectFromRequire(path, state) {
   if (!(0, _isRequire.isRequire)(path)) return;
 
@@ -302,10 +366,10 @@ function collectFromRequire(path, state) {
     parentPath: container,
     key
   } = callExpression;
-  if (container.isCallExpression() && key === 0) {
+  if (container.isCallExpression()) {
     // It may be transpiled import such as
     // `var _atomic = _interopRequireDefault(require("@linaria/atomic"));`
-    const imported = getImportExportTypeByInteropFunction(container);
+    const imported = getImportExportTypeByInteropFunction(container, callExpression);
     if (!imported) {
       // It's not a transpiled import.
       // TODO: Can we guess that it's a namespace import?
@@ -562,10 +626,7 @@ function collectFromExports(path, state) {
       // It is `exports.foo = void 0`
       return;
     }
-    const {
-      name
-    } = property.node;
-    if (name === '__esModule') {
+    if (exportName === '__esModule') {
       // eslint-disable-next-line no-param-reassign
       state.isEsModule = true;
       return;
@@ -632,7 +693,10 @@ function unfoldNamespaceImport(importItem) {
   if (!(binding !== null && binding !== void 0 && binding.referenced)) {
     // Imported namespace is not referenced and probably not used,
     // but it can have side effects, so we should keep it as is
-    return [importItem];
+    return [{
+      ...importItem,
+      imported: 'side-effect'
+    }];
   }
   for (const referencePath of (_binding$referencePat = binding === null || binding === void 0 ? void 0 : binding.referencePaths) !== null && _binding$referencePat !== void 0 ? _binding$referencePat : []) {
     var _binding$referencePat;
@@ -675,7 +739,7 @@ function unfoldNamespaceImport(importItem) {
     }
     if (parentPath !== null && parentPath !== void 0 && parentPath.isCallExpression() && referencePath.listKey === 'arguments') {
       // The defined variable is used as a function argument. Let's try to figure out what is imported.
-      const importType = getImportExportTypeByInteropFunction(parentPath);
+      const importType = getImportExportTypeByInteropFunction(parentPath, referencePath);
       if (!importType) {
         // Imported value is used as an unknown function argument,
         // so we can't predict usage and import it as is.
@@ -821,8 +885,10 @@ function collectFromAssignmentExpression(path, state) {
   let exported;
   if (left.isMemberExpression() && (0, _isExports.isExports)(left.get('object'))) {
     const property = left.get('property');
-    if (property.isIdentifier()) {
+    if (!left.node.computed && property.isIdentifier()) {
       exported = property.node.name;
+    } else if (left.node.computed && property.isStringLiteral()) {
+      exported = property.node.value;
     }
   } else if ((0, _isExports.isExports)(left)) {
     // module.exports = ...
@@ -831,9 +897,23 @@ function collectFromAssignmentExpression(path, state) {
     }
   }
   if (!exported) return;
-  if (!right.isCallExpression() || !(0, _isRequire.isRequire)(right.get('callee'))) {
+  if (exported === '__esModule') {
     // eslint-disable-next-line no-param-reassign
-    state.exports[exported] = right;
+    state.isEsModule = true;
+    return;
+  }
+  if (!isRequireCall(right)) {
+    const relatedImport = state.imports.find(imp => imp.local === right);
+    if (relatedImport) {
+      state.reexports.push({
+        exported,
+        ...relatedImport
+      });
+    } else {
+      // eslint-disable-next-line no-param-reassign
+      state.exports[exported] = right;
+    }
+    path.skip();
     return;
   }
   const sourcePath = (_right$get = right.get('arguments')) === null || _right$get === void 0 ? void 0 : _right$get[0];
@@ -854,10 +934,9 @@ function collectFromAssignmentExpression(path, state) {
   });
   path.skip();
 }
-function collectFromExportStarCall(path, state) {
+function collectAllFromCall(path, require, state) {
   var _requireCall$get;
-  const [requireCall, exports] = path.get('arguments');
-  if (!(0, _isExports.isExports)(exports)) return;
+  const requireCall = typeof require === 'number' ? path.get('arguments')[require] : require;
   if (!requireCall.isCallExpression()) return;
   const callee = requireCall.get('callee');
   const sourcePath = (_requireCall$get = requireCall.get('arguments')) === null || _requireCall$get === void 0 ? void 0 : _requireCall$get[0];
@@ -887,25 +966,32 @@ function collectFromMap(map, state) {
     addExport(returnValue, exported, state);
   });
 }
-function collectFromEsbuildExportCall(path, state) {
-  const [sourceExports, map] = path.get('arguments');
-  if (!sourceExports.isIdentifier({
-    name: 'source_exports'
-  })) return;
+function collectMapFromCall(path, mapPosition, state) {
+  const map = path.get('arguments')[mapPosition];
   if (!map.isObjectExpression()) return;
   collectFromMap(map, state);
   path.skip();
 }
 function collectFromEsbuildReExportCall(path, state) {
   var _requireCall$get2;
-  const [sourceExports, requireCall, exports] = path.get('arguments');
+  const [sourceExports, someCall, exports] = path.get('arguments');
   if (!sourceExports.isIdentifier({
     name: 'source_exports'
-  })) return;
-  if (!requireCall.isCallExpression()) return;
-  if (!(0, _isExports.isExports)(exports)) return;
-  const callee = requireCall.get('callee');
-  if (!(0, _isRequire.isRequire)(callee)) return;
+  }) && !(0, _isExports.isExports)(sourceExports)) return;
+  if (!someCall.isCallExpression()) return;
+  let requireCall = someCall;
+  while (!(0, _isRequire.isRequire)(requireCall.get('callee'))) {
+    const args = requireCall.get('arguments');
+    if (args.length !== 1) {
+      return;
+    }
+    const firstArg = args[0];
+    if (!firstArg.isCallExpression()) {
+      return;
+    }
+    requireCall = firstArg;
+  }
+  if (exports !== undefined && !(0, _isExports.isExports)(exports)) return;
   const sourcePath = (_requireCall$get2 = requireCall.get('arguments')) === null || _requireCall$get2 === void 0 ? void 0 : _requireCall$get2[0];
   if (!sourcePath.isStringLiteral()) return;
   state.reexports.push({
@@ -916,41 +1002,43 @@ function collectFromEsbuildReExportCall(path, state) {
   });
   path.skip();
 }
-function collectFromSwcExportCall(path, state) {
-  const [exports, map] = path.get('arguments');
-  if (!(0, _isExports.isExports)(exports)) return;
-  if (!map.isObjectExpression()) return;
-  collectFromMap(map, state);
-  path.skip();
-}
 function collectFromCallExpression(path, state) {
   const maybeExportStart = path.get('callee');
   if (!maybeExportStart.isIdentifier()) {
     return;
   }
-  const {
-    name
-  } = maybeExportStart.node;
+  if (matchCall(path, [['__exportStar', _isExports.isExports, isCallExpression('__toModule')]])) {
+    // __exportStar(exports, __toModule(require('…')));
 
-  // TypeScript
-  if (name.startsWith('__exportStar')) {
-    collectFromExportStarCall(path, state);
+    const secondArg = path.get('arguments')[1];
+    collectAllFromCall(path, secondArg.get('arguments')[0], state);
     return;
   }
 
-  // swc
-  if (name === '_exportStar') {
-    collectFromExportStarCall(path, state);
+  // TypeScript & swc
+  if (matchCall(path, [['__exportStart', isRequireCall, _isExports.isExports], ['_exportStar', isRequireCall, _isExports.isExports], ['_export_star', isRequireCall, _isExports.isExports], ['__export', isRequireCall] // TypeScript <=3.8.3
+  ])) {
+    collectAllFromCall(path, 0, state);
+    return;
+  }
+  if (matchCall(path, [['_export', _isExports.isExports, isObjectExpression]])) {
+    collectMapFromCall(path, 1, state);
+    return;
   }
-  if (name === '_export') {
-    collectFromSwcExportCall(path, state);
+  if (matchCall(path, [['_extends', isEmptyObject, isRequireCall] // swc <=1.3.16
+  ])) {
+    collectAllFromCall(path, 1, state);
+    return;
   }
 
   // esbuild
-  if (name === '__export') {
-    collectFromEsbuildExportCall(path, state);
+  if (matchCall(path, [['__export', _isExports.isExports, isObjectExpression], ['__export', isIdentifier('source_exports'), isObjectExpression]])) {
+    collectMapFromCall(path, 1, state);
+    return;
   }
-  if (name === '__reExport') {
+  if (matchCall(path, [
+  // Different variants of re-exports in esbuild
+  ['__reExport', isIdentifier('source_exports'), isCallExpression, _isExports.isExports], ['__reExport', isIdentifier('source_exports'), isCallExpression], ['__reExport', _isExports.isExports, isCallExpression]])) {
     collectFromEsbuildReExportCall(path, state);
   }
 }