Support expressions in rest arg access

The current implementation assumes a numeric literal although it just checks the base type.
2016-01-06 17:07:05 -08:00
parent 88d57b58a4
commit 59ea034df2
2 changed files with 11 additions and 3 deletions
--- a/packages/babel-plugin-transform-es2015-parameters/src/rest.js
+++ b/packages/babel-plugin-transform-es2015-parameters/src/rest.js
@@ -107,9 +107,17 @@ function hasRest(node) {
 }

 function optimiseIndexGetter(path, argsId, offset) {
+  let index;
+
+  if (t.isNumericLiteral(path.parent.property)) {
+    index = t.numericLiteral(path.parent.property.value + offset);
+  } else {
+    index = t.binaryExpression("+", path.parent.property, t.numericLiteral(offset));
+  }
+
  path.parentPath.replaceWith(loadRest({
    ARGUMENTS: argsId,
-    INDEX: t.numericLiteral(path.parent.property.value + offset)
+    INDEX: index,
  }));
 }