fix(utils): fix exports finder for enum that were transpilled with tsc (

#1340)
callstack · Sep 19, 2023 · 6390233 · 6390233
1 parent f3a4f2a
commit 6390233
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 2 deletions.
diff --git a/.changeset/orange-islands-juggle.md b/.changeset/orange-islands-juggle.md
@@ -0,0 +1,6 @@
+---
+'@linaria/testkit': patch
+'@linaria/utils': patch
+---
+
+The exports finder didn't support enums that were transpiled to esm by tsc. Fixed.
diff --git a/packages/testkit/src/collectExportsAndImports.test.ts b/packages/testkit/src/collectExportsAndImports.test.ts
@@ -22,6 +22,14 @@ function typescriptCommonJS(source: string): string {
   return result.outputText;
 }
 
+function typescriptES2022(source: string): string {
+  const result = ts.transpileModule(source, {
+    compilerOptions: { module: ts.ModuleKind.ES2022 },
+  });
+
+  return result.outputText;
+}
+
 const withoutLocal = <T extends { local: NodePath }>({
   local,
   ...obj
@@ -103,6 +111,7 @@ const compilers: [name: string, compiler: (code: string) => string][] = [
   ['swcCommonJSes5', swcCommonJS('es5')],
   ['swcCommonJSes2015', swcCommonJS('es2015')],
   ['typescriptCommonJS', typescriptCommonJS],
+  ['typescriptES2022', typescriptES2022],
 ];
 
 function runWithCompiler(

diff --git a/packages/utils/src/collectExportsAndImports.ts b/packages/utils/src/collectExportsAndImports.ts
@@ -246,14 +246,51 @@ function importFromVariableDeclarator(
   return [];
 }
 
+const findIIFE = (path: NodePath): NodePath<CallExpression> | null => {
+  if (path.isCallExpression() && path.get('callee').isFunctionExpression()) {
+    return path;
+  }
+
+  if (!path.parentPath) {
+    return null;
+  }
+
+  return findIIFE(path.parentPath);
+};
+
 function exportFromVariableDeclarator(
   path: NodePath<VariableDeclarator>
 ): Exports {
   const id = path.get('id');
   const init = path.get('init');
 
-  // If there is no init expression, we can ignore this export
-  if (!init || !init.isExpression()) return {};
+  // If there is no init and id is an identifier, we should find IIFE
+  if (!init.node && id.isIdentifier()) {
+    const binding = getScope(path).getBinding(id.node.name);
+    if (!binding) {
+      return {};
+    }
+
+    const iife = [
+      ...binding.referencePaths,
+      ...binding.constantViolations,
+      binding.path,
+    ]
+      .map(findIIFE)
+      .find(isNotNull);
+
+    if (!iife) {
+      return {};
+    }
+
+    return {
+      [id.node.name]: iife,
+    };
+  }
+
+  if (!init || !init.isExpression()) {
+    return {};
+  }
 
   if (id.isIdentifier()) {
     // It is `export const a = 1;`