Module format codemap: fix block formatting

lib/codemod/src/transforms/__testfixtures__/convert-to-module-format/basic.output.js

@@ -5,21 +5,19 @@ import Button from './Button';
 import { storiesOf, configure } from '@storybook/react';
 import { action } from '@storybook/addon-actions';
 
-{
+export default {
-  export default {
+  title: 'Button',
-    title: 'Button',
-  };
-
-  export const story1 = () => <Button label="Story 1" />;
-  export const story2 = () => <Button label="Story 2" onClick={action('click')} />;
-  story2.title = 'second story';
-
-  export const story3 = () => (
-      <div>
-        <Button label="The Button" onClick={action('onClick')} />
-        <br />
-      </div>
-    );
-
-  story3.title = 'complex story';
 };
+
+export const story1 = () => <Button label="Story 1" />;
+export const story2 = () => <Button label="Story 2" onClick={action('click')} />;
+story2.title = 'second story';
+
+export const story3 = () => (
+    <div>
+      <Button label="The Button" onClick={action('onClick')} />
+      <br />
+    </div>
+  );
+
+story3.title = 'complex story';

lib/codemod/src/transforms/__testfixtures__/convert-to-module-format/decorators.output.js

@@ -2,13 +2,10 @@
 import React from 'react';
 import Button from './Button';
 
-// This isn't a valid story, but it tests the `import { comp } from ...` case
+export default {
-{
+  title: 'Some.Button',
-  export default {
+  decorators: [withKnobs, storyFn => <div className="foo">{storyFn}</div>],
-    title: 'Some.Button',
-    decorators: [withKnobs, storyFn => <div className="foo">{storyFn}</div>],
-  };
-
-  export const story1 = () => <Button label="The Button" />;
-  story1.title = 'with decorator';
 };
+
+export const story1 = () => <Button label="The Button" />;
+story1.title = 'with decorator';

lib/codemod/src/transforms/__testfixtures__/convert-to-module-format/multi.output.js

@@ -2,23 +2,16 @@
 import React from 'react';
 import Button from './Button';
 
-// If we have multiple storiesOf calls, export multiple defaults. It's not valid
+export default {
-// JS but will still save the user time in converting.
+  title: 'Button1',
-
-{
-  export default {
-    title: 'Button1',
-  };
-
-  export const story1 = () => <Button label="Button1.1" />;
-  export const story2 = () => <Button label="Button1.2" />;
 };
 
-{
+export const story1 = () => <Button label="Button1.1" />;
-  export default {
+export const story2 = () => <Button label="Button1.2" />;
-    title: 'Button2',
-  };
 
-  export const story1 = () => <Button label="Button2.1" />;
+export default {
-  export const story2 = () => <Button label="Button2.2" />;
+  title: 'Button2',
-};
+};
+
+export const story1 = () => <Button label="Button2.1" />;
+export const story2 = () => <Button label="Button2.2" />;

lib/codemod/src/transforms/__testfixtures__/convert-to-module-format/parameters.output.js

@@ -4,17 +4,15 @@ import Button from './Button';
 
 import { storiesOf } from '@storybook/react';
 
-{
+export default {
-  export default {
+  title: 'Button',
-    title: 'Button',
 
-    parameters: {
+  parameters: {
-      component: Button,
+    component: Button,
-      foo: 1,
+    foo: 1,
-      bar: 2,
+    bar: 2,
-    },
+  },
-  };
-
-  export const story1 = () => <Button label="The Button" />;
-  story1.title = 'with kind parameters';
 };
+
+export const story1 = () => <Button label="The Button" />;
+story1.title = 'with kind parameters';

lib/codemod/src/transforms/__testfixtures__/convert-to-module-format/story-parameters.output.js

@@ -4,22 +4,20 @@ import Button from './Button';
 
 import { storiesOf } from '@storybook/react';
 
-{
+export default {
-  export default {
+  title: 'Button',
-    title: 'Button',
+};
+
+export const story1 = () => <Button label="The Button" />;
+story1.title = 'with story parameters';
+
+story1.parameters = {
+    header: false,
+    inline: true,
   };
 
-  export const story1 = () => <Button label="The Button" />;
+export const foo = () => <Button label="Foo" />;
-  story1.title = 'with story parameters';
 
-  story1.parameters = {
+foo.parameters = {
-      header: false,
+    bar: 1,
-      inline: true,
+  };
-    };
-
-  export const foo = () => <Button label="Foo" />;
-
-  foo.parameters = {
-      bar: 1,
-    };
-};

lib/codemod/src/transforms/convert-to-module-format.js

@@ -140,7 +140,9 @@ export default function transformer(file, api) {
       counter += 1;
     });
 
-    return j.blockStatement(statements);
+    statements.reverse();
+    statements.forEach(s => path.parent.insertAfter(s));
+    base.remove();
   }
 
   // each top-level add expression corresponds to the last "add" of the chain.
@@ -149,9 +151,8 @@ export default function transformer(file, api) {
     .find(j.CallExpression)
     .filter(add => add.node.callee.property && add.node.callee.property.name === 'add')
     .filter(add => add.node.arguments.length >= 2 && add.node.arguments[0].type === 'Literal')
-    .filter(add => add.parentPath.node.type === 'ExpressionStatement');
+    .filter(add => add.parentPath.node.type === 'ExpressionStatement')
-
+    .forEach(convertToModuleExports);
-  topLevelAddExpressions.replaceWith(convertToModuleExports);
 
   return root.toSource({ quote: 'single', trailingComma: 'true', tabWidth: 2 });
 }