Support export all declaration

phated · May 4, 2019 · 79a814f · 79a814f
1 parent 840ca88
commit 79a814f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/src/__tests__/fixtures/component_32.js b/src/__tests__/fixtures/component_32.js
@@ -6,3 +6,4 @@ export function SuperDuperCustomButton({color, ...otherProps}) {
 }
 
 SuperDuperCustomButton.propTypes = sharedProps;
+export * from './component_31';
diff --git a/src/__tests__/fixtures/component_33.js b/src/__tests__/fixtures/component_33.js
@@ -1,4 +1,4 @@
-import * as C31 from './component_31';
+import * as C31 from './component_32';
 import PropTypes from 'prop-types';
 
 export function SuperDuperCustomButton({color, ...otherProps}) {

diff --git a/src/utils/resolveImportedValue.js b/src/utils/resolveImportedValue.js
@@ -17,8 +17,6 @@ import fs from 'fs';
 const { namedTypes: t, NodePath } = types;
 
 export default function resolveImportedValue(path: NodePath, name: string) {
-  t.ImportDeclaration.assert(path.node);
-
   // Bail if no filename was provided for the current source file.
   // Also never traverse into react itself.
   const source = path.node.source.value;
@@ -101,7 +99,14 @@ function findExportedValue(ast, name) {
 
       return false;
     },
-    // TODO: visitExportAllDeclaration
+    visitExportAllDeclaration(path: NodePath) {
+      const resolvedPath = resolveImportedValue(path, name);
+      if (resolvedPath) {
+        resultPath = resolvedPath;
+      }
+
+      return false;
+    }
   });
 
   return resultPath;
Original file line number	Diff line number	Diff line change
Expand Up		@@ -6,3 +6,4 @@ export function SuperDuperCustomButton({color, ...otherProps}) {
		}

		SuperDuperCustomButton.propTypes = sharedProps;
		export * from './component_31';