Support shorthand object initializer syntax (fixes #9)

package.json

@@ -22,6 +22,7 @@
   },
   "homepage": "https://github.com/captainsafia/fony#readme",
   "dependencies": {
+    "babel-core": "^6.24.1",
     "chance": "^1.0.6",
     "commander": "^2.9.0"
   }

src/index.js

@@ -5,6 +5,7 @@ const program = require('commander');
 
 const _version_ = require('../package.json').version;
 const createData = require('./utils');
+const transform = require('./transform');
 
 program
   .version(_version_);
@@ -16,7 +17,7 @@ program
 
 var template;
 try {
-  template = JSON.parse(program.template);
+  template = eval(transform(program.template));
 } catch (error) {
   return console.log(error);
 }

src/transform.js

@@ -0,0 +1,26 @@
+// Generate AST from an es6 object initializer string
+// May want to support syntax like "word*3" to generate an array of 3 words
+// or other more complex syntaxes
+const babel = require("babel-core");
+
+// transforms ({ a: a }) or equivalently ({ a }) into
+// ({a: "a" }), note the value is stringified
+function transformPlugin(babel) {
+  const { types: t } = babel;
+
+  return {
+    visitor: {
+      Property({ node }) {
+        if (t.isIdentifier(node.value))
+          node.value = t.stringLiteral(node.value.name);
+      }
+    }
+  };
+}
+
+module.exports = code => {
+  return babel.transform(`(${code})`, {
+    plugins: [transformPlugin],
+    ast: false // don't need the ast (for now?)
+  }).code;
+};