From 505d7277e9f3c2206927c6b05c42e2ead94da9e4 Mon Sep 17 00:00:00 2001
From: Emil Stenberg <esandin@gmail.com>
Date: Wed, 10 Dec 2014 12:45:17 +0100
Subject: [PATCH] generating es6 modules if --es6 flag is used

---
 component/index.js                 | 7 +++++++
 templates/javascript/Component.jsx | 6 +++++-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/component/index.js b/component/index.js
index 5e37c8f..1d50772 100644
--- a/component/index.js
+++ b/component/index.js
@@ -3,12 +3,19 @@ var util = require('util');
 var ScriptBase = require('../script-base.js');
 
 var ComponentGenerator = module.exports = function ComponentGenerator(args, options, config) {
+
   ScriptBase.apply(this, arguments);
 };
 
 util.inherits(ComponentGenerator, ScriptBase);
 
 ComponentGenerator.prototype.createComponentFile = function createComponentFile() {
+    this.option('es6');
+
+
+  this.es6 = this.options.es6;
+    console.log('es6:', this.es6)
+
   this.generateSourceAndTest(
     'Component',
     'spec/Component',
diff --git a/templates/javascript/Component.jsx b/templates/javascript/Component.jsx
index ae0869a..fe84508 100644
--- a/templates/javascript/Component.jsx
+++ b/templates/javascript/Component.jsx
@@ -19,5 +19,9 @@ var <%= classedName %> = React.createClass({
       );
   }
 });
-
+<% if (es6) { %>
+export default <%= classedName %>;
+<% } else { %>
 module.exports = <%= classedName %>;
+<% } %>
+