feat(API): support options with the plugings API (#118)

Author: oliviertassinari

package.json

@@ -11,7 +11,7 @@
     "prebuild": "rm -rf lib/",
     "build": "babel src --out-dir lib",
     "flow": "flow",
-    "prepublish": "npm run build && pkgfiles"
+    "source": "npm run build && pkgfiles"
   },
   "repository": {
     "type": "git",

src/index.js

@@ -1,7 +1,7 @@
 // @flow weak
 /* eslint-disable global-require, import/no-dynamic-require */
 
-import { visitors } from 'babel-traverse';
+import traverse, { visitors } from 'babel-traverse';
 // import generate from 'babel-generator';
 // console.log(generate(node).code);
 import isAnnotatedForRemoval from './isAnnotatedForRemoval';
@@ -79,12 +79,29 @@ export default function ({ template, types }) {
         };
 
         if (state.opts.plugins) {
-          const pluginsVisitors = state.opts.plugins.map((pluginName) => {
+          const pluginsState = state;
+          const pluginsVisitors = state.opts.plugins.map((pluginOpts) => {
+            const pluginName =
+              typeof pluginOpts === 'string' ? pluginOpts : pluginOpts[0];
+
+            if (typeof pluginOpts !== 'string') {
+              pluginsState.opts = {
+                ...pluginsState.opts,
+                ...pluginOpts[1],
+              };
+            }
+
             const plugin = require(pluginName);
             return plugin({ template, types }).visitor;
           });
 
-          programPath.traverse(visitors.merge(pluginsVisitors));
+          traverse(
+            programPath.parent,
+            visitors.merge(pluginsVisitors),
+            programPath.scope,
+            pluginsState,
+            programPath.parentPath,
+          );
         }
 
         // On program start, do an explicit traversal up front for this plugin.

test/fixtures/flow/actual.js

@@ -5,7 +5,7 @@ class Foo1 extends Component {
   render() {}
 }
 
-type Props2 = {
+export type Props2 = {
   bar2?: string
 }
 

test/fixtures/flow/expected-wrap-es5.js

@@ -28,6 +28,9 @@ var Foo1 = function (_Component) {
 Foo1.propTypes = process.env.NODE_ENV !== "production" ? {
   bar1: require("prop-types").string
 } : {};
+var babelPluginFlowReactPropTypes_proptype_Props2 = {
+  bar2: require("prop-types").string
+};
 
 var Foo2 = function (_Component2) {
   _inherits(Foo2, _Component2);

test/fixtures/flow/options.json

@@ -1,6 +1,8 @@
 {
   "plugins": [
-    "babel-plugin-flow-react-proptypes",
+    ["babel-plugin-flow-react-proptypes", {
+      "omitRuntimeTypeExport": true
+    }],
     "babel-plugin-transform-flow-strip-types"
   ]
 }