Skip to content

Commit

Permalink
Inject IIFE when variables shadow binding in rest param
Browse files Browse the repository at this point in the history
  • Loading branch information
nicolo-ribaudo committed Jul 7, 2022
1 parent 373f733 commit 7b2108c
Show file tree
Hide file tree
Showing 8 changed files with 106 additions and 29 deletions.
16 changes: 5 additions & 11 deletions packages/babel-plugin-transform-parameters/src/params.ts
Expand Up @@ -47,18 +47,17 @@ export default function convertFunctionParams(

const { node, scope } = path;

const state = {
needsOuterBinding: false,
scope,
};

const body = [];
const shadowedParams = new Set<string>();

for (const param of params) {
collectShadowedParamsNames(param, scope, shadowedParams);
}

const state = {
needsOuterBinding: false,
scope,
};
if (shadowedParams.size === 0) {
for (const param of params) {
if (!param.isIdentifier()) param.traverse(iifeVisitor, state);
Expand Down Expand Up @@ -156,12 +155,7 @@ export default function convertFunctionParams(
path.ensureBlock();

if (state.needsOuterBinding || shadowedParams.size > 0) {
body.push(
buildScopeIIFE(
shadowedParams,
(path.get("body") as NodePath<t.BlockStatement>).node,
),
);
body.push(buildScopeIIFE(shadowedParams, path.node.body));

path.set("body", t.blockStatement(body as t.Statement[]));

Expand Down
33 changes: 33 additions & 0 deletions packages/babel-plugin-transform-parameters/src/rest.ts
@@ -1,6 +1,12 @@
import { template, types as t } from "@babel/core";
import type { NodePath, Visitor } from "@babel/traverse";

import {
iifeVisitor,
collectShadowedParamsNames,
buildScopeIIFE,
} from "./shadow-utils";

const buildRest = template.statement(`
for (var LEN = ARGUMENTS.length,
ARRAY = new Array(ARRAY_LEN),
Expand Down Expand Up @@ -292,6 +298,33 @@ export default function convertFunctionRest(path: NodePath<t.Function>) {
const { node, scope } = path;
if (!hasRest(node)) return false;

const restPath = path.get(
`params.${node.params.length - 1}.argument`,
) as NodePath<t.Pattern | t.Identifier>;

if (!restPath.isIdentifier()) {
const shadowedParams = new Set<string>();
collectShadowedParamsNames(restPath, path.scope, shadowedParams);

let needsIIFE = shadowedParams.size > 0;
if (!needsIIFE) {
const state = {
needsOuterBinding: false,
scope,
};
restPath.traverse(iifeVisitor, state);
needsIIFE = state.needsOuterBinding;
}

if (needsIIFE) {
path.ensureBlock();
path.set(
"body",
t.blockStatement([buildScopeIIFE(shadowedParams, path.node.body)]),
);
}
}

let rest = (node.params.pop() as t.RestElement).argument as
| t.Pattern
| t.Identifier;
Expand Down

This file was deleted.

This file was deleted.

@@ -0,0 +1,9 @@
function f(...{ length: x = 0, y = x }) {
var x;
return x;
}

function g(...{ length: x = 0, y = x }) {
var x = 1;
return x;
}
@@ -0,0 +1,26 @@
function f() {
for (var _len = arguments.length, _ref = new Array(_len), _key = 0; _key < _len; _key++) {
_ref[_key] = arguments[_key];
}

var _ref$length = _ref.length,
x = _ref$length === void 0 ? 0 : _ref$length,
_ref$y = _ref.y,
y = _ref$y === void 0 ? x : _ref$y;
return x;
}

function g() {
for (var _len2 = arguments.length, _ref2 = new Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
_ref2[_key2] = arguments[_key2];
}

var _ref2$length = _ref2.length,
x = _ref2$length === void 0 ? 0 : _ref2$length,
_ref2$y = _ref2.y,
y = _ref2$y === void 0 ? x : _ref2$y;
return function (x) {
var x = 1;
return x;
}(x);
}
@@ -0,0 +1,9 @@
function f(...[x, y = x]) {
var x;
return x;
}

function g(...[x, y = x]) {
var x = 1;
return x;
}
@@ -0,0 +1,24 @@
function f() {
for (var _len = arguments.length, _ref = new Array(_len), _key = 0; _key < _len; _key++) {
_ref[_key] = arguments[_key];
}

var x = _ref[0],
_ref$ = _ref[1],
y = _ref$ === void 0 ? x : _ref$;
return x;
}

function g() {
for (var _len2 = arguments.length, _ref2 = new Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
_ref2[_key2] = arguments[_key2];
}

var x = _ref2[0],
_ref2$ = _ref2[1],
y = _ref2$ === void 0 ? x : _ref2$;
return function (x) {
var x = 1;
return x;
}(x);
}

0 comments on commit 7b2108c

Please sign in to comment.