add WIP build plugins

2015-08-24 15:34:31 -04:00
parent 56c2903811
commit 8408244d93
3 changed files with 160 additions and 0 deletions
--- a/scripts/build-plugins/inline-node-builders.js
+++ b/scripts/build-plugins/inline-node-builders.js
@@ -0,0 +1,52 @@
+module.exports = function (babel) {
+  var t = babel.types;
+
+  return new babel.Plugin("inline-node-builders", {
+    visitor: {
+      CallExpression: function (node, parent, scope) {
+        var callee = this.get("callee");
+        if (!callee.isMemberExpression()) return;
+
+        var obj = callee.get("object");
+        if (!obj.referencesImport("babel-types", "*")) return;
+
+        var prop = callee.get("property");
+        if (!prop.isIdentifier()) return;
+
+        var type = prop.node.name;
+        var builder = t.BUILDER_KEYS[type] || t.BUILDER_KEYS[type[0].toUpperCase() + type.slice(1)];
+        if (!builder) return;
+
+        var props = [];
+
+        var i = 0;
+
+        for (var key in builder) {
+          var def = builder[key];
+          var arg = node.arguments[i];
+
+          if (arg) {
+            if (t.isLiteral(arg)) {
+              props.push(t.property("init", t.identifier(key), arg));
+            } else {
+              var uid = scope.generateDeclaredUidIdentifier("temp");
+              props.push(t.property("init", t.identifier(key), t.sequenceExpression([
+                t.assignmentExpression("=", uid, arg),
+                t.conditionalExpression(
+                  t.logicalExpression("===", uid, t.identifier("undefined")),
+                  t.valueToNode(def),
+                  uid
+                )
+              ])));
+            }
+          } else {
+            props.push(t.property("init", t.identifier(key), t.valueToNode(def)));
+          }
+        }
+
+
+        return t.objectExpression(props);
+      }
+    }
+  });
+};