Added flow support to getstorybook upgrade

lib/cli/README.md

@@ -40,4 +40,14 @@ If you don't want to use `yarn` always you can use the `--use-npm` option like t
 getstorybook --use-npm
 ```
 
+* * *
+
+## [Flow](https://flow.org/) support
+
+It also supports flow files. By default, [jscodeshift](https://github.com/facebook/jscodeshift), the tool used to transform the source files, uses babel to read the files. To be able to transform any flow annotated file, you need to use the flow parser.
+
+```sh
+getstorybook --parser flow
+```
+
 For more information visit: [storybook.js.org](https://storybook.js.org)

lib/cli/bin/generate.js

@@ -19,6 +19,7 @@ program
   .version(pkg.version)
   .option('-f --force', 'Forcely add storybook')
   .option('-N --use-npm', 'Use npm to install deps')
+  .option('-p --parser <babel | babylon | flow>', 'jscodeshift parser')
   .parse(process.argv);
 
 const welcomeMessage = 'getstorybook - the simplest way to add a storybook to your project.';
@@ -78,7 +79,7 @@ switch (projectType) {
 
   case types.UPDATE_PACKAGE_ORGANIZATIONS:
     // eslint-disable-next-line
-    require('../generators/UPDATE_PACKAGE_ORGANIZATIONS')
+    require('../generators/UPDATE_PACKAGE_ORGANIZATIONS')(program.parser)
       .then(() => null) // commmandLog doesn't like to see output
       .then(commandLog('Upgrading your project to the new storybook packages.'))
       .then(end);

lib/cli/generators/UPDATE_PACKAGE_ORGANIZATIONS/index.js

@@ -32,7 +32,7 @@ function updatePackageJson() {
   });
 }
 
-function updateSourceCode() {
+function updateSourceCode(parser) {
   const jscodeshiftPath = path.dirname(require.resolve('jscodeshift'));
   const jscodeshiftCommand = path.join(jscodeshiftPath, 'bin', 'jscodeshift.sh');
 
@@ -43,8 +43,9 @@ function updateSourceCode() {
   );
 
   const args = ['-t', codemodPath, '--silent', '--ignore-pattern', '"node_modules|dist"', '.'];
+  if (parser) args.push('--parser', parser);
 
   return spawn(jscodeshiftCommand, args, { stdio: 'inherit' });
 }
 
-module.exports = updatePackageJson().then(updateSourceCode);
+module.exports = parser => updatePackageJson().then(() => updateSourceCode(parser));