ConfigFile: Fix export specifiers

storybookjs · Jan 13, 2024 · 4e2aa4c · 4e2aa4c
1 parent ee06c80
commit 4e2aa4c
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 6 deletions.
diff --git a/code/lib/csf-tools/src/ConfigFile.test.ts b/code/lib/csf-tools/src/ConfigFile.test.ts
@@ -216,6 +216,17 @@ describe('ConfigFile', () => {
           )
         ).toEqual([{ directory: '../src', titlePrefix: 'Demo' }]);
       });
+      it('export specfier', () => {
+        expect(
+          getField(
+            ['foo'],
+            dedent`
+              const foo = 'bar';
+              export { foo };
+            `
+          )
+        ).toEqual('bar');
+      });
     });
   });
 
@@ -435,6 +446,24 @@ describe('ConfigFile', () => {
         `);
       });
     });
+
+    describe('export specifiers', () => {
+      it('found object', () => {
+        expect(
+          setField(
+            ['core', 'builder'],
+            'webpack5',
+            dedent`
+              const core = { builder: 'webpack4' };
+              export { core };
+            `
+          )
+        ).toMatchInlineSnapshot(`
+          const core = { builder: 'webpack5' };
+          export { core };
+        `);
+      });
+    });
   });
 
   describe('appendToArray', () => {

diff --git a/code/lib/csf-tools/src/ConfigFile.ts b/code/lib/csf-tools/src/ConfigFile.ts
@@ -78,10 +78,12 @@ const _getPathProperties = (path: string[], node: t.Node): t.ObjectProperty[] |
   }
   return undefined;
 };
-
 // eslint-disable-next-line @typescript-eslint/naming-convention
-const _findVarInitialization = (identifier: string, program: t.Program) => {
-  let init: t.Expression | null | undefined = null;
+const _findVarDeclarator = (
+  identifier: string,
+  program: t.Program
+): t.VariableDeclarator | null | undefined => {
+  let declarator: t.VariableDeclarator | null | undefined = null;
   let declarations: t.VariableDeclarator[] | null = null;
   program.body.find((node: t.Node) => {
     if (t.isVariableDeclaration(node)) {
@@ -92,20 +94,26 @@ const _findVarInitialization = (identifier: string, program: t.Program) => {
 
     return (
       declarations &&
-      declarations.find((decl: t.Node) => {
+      declarations.find((decl: t.VariableDeclarator) => {
         if (
           t.isVariableDeclarator(decl) &&
           t.isIdentifier(decl.id) &&
           decl.id.name === identifier
         ) {
-          init = decl.init;
+          declarator = decl;
           return true; // stop looking
         }
         return false;
       })
     );
   });
-  return init;
+  return declarator;
+};
+
+// eslint-disable-next-line @typescript-eslint/naming-convention
+const _findVarInitialization = (identifier: string, program: t.Program) => {
+  const declarator = _findVarDeclarator(identifier, program);
+  return declarator?.init;
 };
 
 // eslint-disable-next-line @typescript-eslint/naming-convention
@@ -213,6 +221,16 @@ export class ConfigFile {
                 self._exportDecls[exportName] = decl;
               }
             });
+          } else if (node.specifiers) {
+            // export { X };
+            node.specifiers.forEach((spec) => {
+              if (t.isExportSpecifier(spec) && t.isIdentifier(spec.exported)) {
+                const { name: exportName } = spec.exported;
+                const decl = _findVarDeclarator(exportName, parent as t.Program) as any;
+                self._exports[exportName] = decl.init;
+                self._exportDecls[exportName] = decl;
+              }
+            });
           } else {
             logger.warn(
               getCsfParsingErrorMessage({