Skip to content

Commit

Permalink
Correctly transform spreads to use proper concat method
Browse files Browse the repository at this point in the history
  • Loading branch information
danez committed Dec 3, 2018
1 parent 282129e commit 077b54d
Show file tree
Hide file tree
Showing 7 changed files with 44 additions and 5 deletions.
Expand Up @@ -21,7 +21,7 @@ var _ref4 = [a[1], a[0]];
a[0] = _ref4[0];
a[1] = _ref4[1];

var _ref5 = babelHelpers.toConsumableArray(foo).concat([bar]),
var _ref5 = [].concat(babelHelpers.toConsumableArray(foo), [bar]),
a = _ref5[0],
b = _ref5[1];

Expand Down
20 changes: 18 additions & 2 deletions packages/babel-plugin-transform-spread/src/index.js
Expand Up @@ -57,13 +57,29 @@ export default declare((api, options) => {
if (!hasSpread(elements)) return;

const nodes = build(elements, scope);
const first = nodes.shift();
let first = nodes[0];

if (nodes.length === 0 && first !== elements[0].argument) {
// If there is only one element in the ArrayExpression and
// the element was transformed (Array.prototype.slice.call or toConsumableArray)
// we know that the transformed code already takes care of cloning the array.
// So we can simply return that element.
if (nodes.length === 1 && first !== elements[0].argument) {
path.replaceWith(first);
return;
}

// If the first element is a ArrayExpression we can directly call
// concat on it.
// `[..].concat(..)`
// If not then we have to use `[].concat(arr)` and not `arr.concat`
// because `arr` could be extended/modified (e.g. Immutable) and we do not know exactly
// what concat would produce.
if (!t.isArrayExpression(first)) {
first = t.arrayExpression([]);
} else {
nodes.shift();
}

path.replaceWith(
t.callExpression(
t.memberExpression(first, t.identifier("concat")),
Expand Down
@@ -0,0 +1,9 @@
const arr = [];

arr.concat = () => {
throw new Error('Should not be called');
};

expect(() => {
const x = [...arr];
}).not.toThrow();
@@ -0,0 +1,7 @@
const arr = [];

arr.concat = () => {
throw new Error('Should not be called');
};

const x = [...arr];
@@ -0,0 +1,7 @@
const arr = [];

arr.concat = () => {
throw new Error('Should not be called');
};

const x = [].concat(arr);
@@ -1 +1 @@
var lyrics = babelHelpers.toConsumableArray(parts).concat(["head", "and", "toes"]);
var lyrics = [].concat(babelHelpers.toConsumableArray(parts), ["head", "and", "toes"]);
Expand Up @@ -3,5 +3,5 @@ function foo() {
bar[_key] = arguments[_key];
}

return bar.concat();
return [].concat(bar);
}

0 comments on commit 077b54d

Please sign in to comment.