1 files changed, 52 insertions, 0 deletions
diff --git a/node_modules/babel-plugin-transform-inline-consecutive-adds/lib/array-collapser.js b/node_modules/babel-plugin-transform-inline-consecutive-adds/lib/array-collapser.js
new file mode 100644
index 00000000..540c92aa
--- /dev/null
+++ b/node_modules/babel-plugin-transform-inline-consecutive-adds/lib/array-collapser.js
@@ -0,0 +1,52 @@
+"use strict";
+
+const Collapser = require("./collapser");
+
+class ArrayCollapser extends Collapser {
+  isInitTypeValid(init) {
+    return init.isArrayExpression();
+  }
+
+  isExpressionTypeValid(expr) {
+    return expr.isCallExpression();
+  }
+
+  getExpressionChecker(objName, checkReference) {
+    return expr => {
+      // checks expr is of form:
+      // foo.push(rval1, ...nrvals)
+      const callee = expr.get("callee");
+
+      if (!callee.isMemberExpression()) {
+        return false;
+      }
+
+      const obj = callee.get("object"),
+            prop = callee.get("property");
+
+      if (!obj.isIdentifier() || obj.node.name !== objName || !prop.isIdentifier() || prop.node.name !== "push") {
+        return false;
+      }
+
+      const args = expr.get("arguments");
+
+      if (args.some(checkReference)) {
+        return false;
+      }
+
+      return true;
+    };
+  }
+
+  extractAssignment(expr) {
+    return expr.node.arguments;
+  }
+
+  addSuccessfully(t, args, init) {
+    args.map(a => init.elements.push(a));
+    return true;
+  }
+
+}
+
+module.exports = ArrayCollapser;
+\ No newline at end of file