place wrapped export default at end of file (Closes #129)

this ensures that any static properties are correctly transferred onto the HOC

src/index.ts

@@ -3,6 +3,7 @@ import { addNamed as addNamedImport } from '@babel/helper-module-imports';
 import {
   arrayExpression,
   callExpression,
+  exportDefaultSpecifier,
   ClassDeclaration,
   classExpression,
   ExportNamedDeclaration,
@@ -20,9 +21,9 @@ import {
   variableDeclarator,
   importDeclaration,
   importDefaultSpecifier,
-  importSpecifier,
   identifier,
   exportDefaultDeclaration,
+  Program,
 } from '@babel/types';
 import * as nodePath from 'path';
 
@@ -129,8 +130,11 @@ function wrapExportDefaultDeclaration(path: NodePath<any>) {
     isClassDeclaration(node.declaration)
   ) {
     if (node.declaration.id) {
-      path.insertBefore(node.declaration);
-      node.declaration = wrapInHOC(node.declaration.id);
+      path.replaceInline(node.declaration);
+      (path.parentPath as NodePath<Program>).pushContainer(
+        'body',
+        exportDefaultDeclaration(wrapInHOC(node.declaration.id)) as any
+      );
     } else {
       if (isFunctionDeclaration(node.declaration)) {
         node.declaration = wrapInHOC(

test/pages/static property/code.js

@@ -0,0 +1,17 @@
+import Text from '../components/Text';
+
+export const getServerSideProps = () => {
+  return {
+    props: {
+      text: 'hi',
+    },
+  };
+};
+
+export default function Page() {
+  return 'foo';
+}
+
+Page.getLayout = function getLayout(page) {
+  return 'bar';
+};

test/pages/static property/output.js

@@ -0,0 +1,20 @@
+import { withSuperJSONPage as _withSuperJSONPage } from 'babel-plugin-superjson-next/tools';
+import { withSuperJSONProps as _withSuperJSONProps } from 'babel-plugin-superjson-next/tools';
+import Text from '../components/Text';
+export const getServerSideProps = _withSuperJSONProps(() => {
+  return {
+    props: {
+      text: 'hi',
+    },
+  };
+}, ['smth']);
+
+function Page() {
+  return 'foo';
+}
+
+Page.getLayout = function getLayout(page) {
+  return 'bar';
+};
+
+export default _withSuperJSONPage(Page);