diff --git a/main.js b/main.js index 6dc06e0339..fc67157a3a 100644 --- a/main.js +++ b/main.js @@ -1,10 +1,16 @@ 'use strict'; -var visitors = require('./vendor/fbtransform/visitors').transformVisitors; +var visitors = require('./vendor/fbtransform/visitors'); var transform = require('jstransform').transform; module.exports = { - transform: function(code) { - return transform(visitors.react, code).code; + transform: function(code, options) { + var visitorList; + if (options && options.harmony) { + visitorList = visitors.getAllVisitors(); + } else { + visitorList = visitors.transformVisitors.react; + } + return transform(visitorList, code).code; } };