Bypass multiple exports (#41)

* Add test case

* Bypass multiple exports

Author: orionmiz

src/lib.rs

@@ -68,6 +68,8 @@ struct NextSuperJsonTransformer {
 
     has_init_props: bool,
     use_init_props: bool,
+
+    has_multiple_props: bool,
 }
 
 #[derive(Debug, Default, Clone, Deserialize)]
@@ -90,6 +92,8 @@ pub fn transform(config: Config) -> impl VisitMut {
 
         has_init_props: false,
         use_init_props: false,
+
+        has_multiple_props: false,
     }
 }
 
@@ -106,7 +110,7 @@ impl VisitMut for NextSuperJsonTransformer {
         self.find_ssg_prop(items);
 
         if self.props.export.orig.is_none() {
-            if !self.use_init_props {
+            if !self.use_init_props || self.has_multiple_props {
                 return;
             }
 
@@ -603,11 +607,13 @@ impl NextSuperJsonTransformer {
     pub fn find_ssg_prop(&mut self, items: &mut Vec<ModuleItem>) {
         let mut ssg_prop_ident = None;
 
-        self.props.export.orig = items.iter_mut().position(|item| {
+        let mut first = None;
+
+        items.iter_mut().enumerate().any(|(pos, item)| {
             // check initial props
             item.visit_mut_children_with(self);
 
-            match item {
+            let found = match item {
                 // check has ssg props
                 ModuleItem::ModuleDecl(ModuleDecl::ExportDecl(ExportDecl { decl, .. })) => {
                     match decl {
@@ -657,9 +663,25 @@ impl NextSuperJsonTransformer {
                     self.props.export.spec.is_some()
                 }
                 _ => false,
+            };
+
+            if found {
+                if first.is_some() || self.use_init_props {
+                    self.has_multiple_props = true;
+                    return true;
+                }
+                first = Some(pos);
             }
+
+            false
         });
 
+        if self.has_multiple_props {
+            return;
+        }
+
+        self.props.export.orig = first;
+
         if ssg_prop_ident.is_some() && !self.props.skip {
             let mut n = items.len();
 

tests/fixture/skip/multiple-exports/code.js

@@ -0,0 +1,19 @@
+export default function Page() {
+  return <div>Page</div>;
+}
+
+export const getStaticProps = () => {
+  return {
+    props: {},
+  };
+}
+
+export const getServerSideProps = () => {
+  return {
+    props: {},
+  };
+}
+
+Page.getInitialProps = () => {
+  return {};
+}

tests/fixture/skip/multiple-exports/output.js

@@ -0,0 +1,19 @@
+export default function Page() {
+  return <div>Page</div>;
+}
+
+export const getStaticProps = () => {
+  return {
+    props: {},
+  };
+}
+
+export const getServerSideProps = () => {
+  return {
+    props: {},
+  };
+}
+
+Page.getInitialProps = () => {
+  return {};
+}