Merge pull request #135 from blitz-js/fix-134

fix: export list for getServerSideProps
blitz-js · Nov 22, 2022 · 25dcdaf · 25dcdaf
2 parents 81d6fd4 + f92dc32
commit 25dcdaf
Show file tree

Hide file tree

Showing 3 changed files with 67 additions and 1 deletion.
diff --git a/src/index.ts b/src/index.ts
@@ -3,7 +3,7 @@ import { addNamed as addNamedImport } from '@babel/helper-module-imports';
 import {
   arrayExpression,
   callExpression,
-  exportDefaultSpecifier,
+  exportNamedDeclaration,
   ClassDeclaration,
   classExpression,
   ExportNamedDeclaration,
@@ -192,6 +192,52 @@ function transformImportExportDefault(paths: NodePath<any>[]) {
   }
 }
 
+/**
+ * transforms `export { getServerSideProps }` import & export line
+ */
+function transformDatafetcherNamedImportExports(
+  paths: NodePath<any>[],
+  dataFetcherName: 'getServerSideProps' | 'getStaticProps'
+) {
+  for (const path of paths) {
+    if (isExportNamedDeclaration(path)) {
+      for (const specifier of path.node.specifiers) {
+        if (specifier.local.name === dataFetcherName) {
+          paths.forEach((path) =>
+            path.traverse({
+              ImportSpecifier(specifier) {
+                if (specifier.node.imported.name === dataFetcherName) {
+                  specifier.node.local.name = `_${dataFetcherName}`;
+                }
+              },
+            })
+          );
+
+          path.insertAfter(
+            exportNamedDeclaration(
+              variableDeclaration('const', [
+                variableDeclarator(
+                  identifier(dataFetcherName),
+                  identifier(`_${dataFetcherName}`)
+                ),
+              ])
+            ) as any
+          );
+
+          path.node.specifiers.splice(
+            path.node.specifiers.indexOf(specifier),
+            1
+          );
+
+          if (path.node.specifiers.length === 0) {
+            path.remove();
+          }
+        }
+      }
+    }
+  }
+}
+
 function shouldBeSkipped(filePath: string) {
   if (!filePath.includes('pages' + nodePath.sep)) {
     return true;
@@ -219,6 +265,8 @@ function superJsonWithNext(): PluginObj {
         }
 
         transformImportExportDefault(path.get('body'));
+        transformDatafetcherNamedImportExports(path.get('body'), 'getServerSideProps');
+        transformDatafetcherNamedImportExports(path.get('body'), 'getStaticProps');
 
         const body = path.get('body');
 

diff --git a/test/pages/export gssP and page from other (repro #134)/code.js b/test/pages/export gssP and page from other (repro #134)/code.js
@@ -0,0 +1,8 @@
+import {
+  getServerSideProps,
+  PreviewPage,
+} from '../../../common-components/contentful-elements/pages/preview';
+
+export { getServerSideProps };
+
+export default PreviewPage;
diff --git a/test/pages/export gssP and page from other (repro #134)/output.js b/test/pages/export gssP and page from other (repro #134)/output.js
@@ -0,0 +1,10 @@
+import { withSuperJSONPage as _withSuperJSONPage } from 'babel-plugin-superjson-next/tools';
+import { withSuperJSONProps as _withSuperJSONProps } from 'babel-plugin-superjson-next/tools';
+import {
+  getServerSideProps as _getServerSideProps,
+  PreviewPage,
+} from '../../../common-components/contentful-elements/pages/preview';
+export const getServerSideProps = _withSuperJSONProps(_getServerSideProps, [
+  'smth',
+]);
+export default _withSuperJSONPage(PreviewPage);