99 lines
2.4 KiB
JavaScript
99 lines
2.4 KiB
JavaScript
"use strict";
|
|
|
|
var util = require("../../../util");
|
|
var t = require("../../../types");
|
|
|
|
exports.check = function (node) {
|
|
return t.isFunction(node) && hasDefaults(node);
|
|
};
|
|
|
|
var hasDefaults = function (node) {
|
|
for (var i = 0; i < node.params.length; i++) {
|
|
if (!t.isIdentifier(node.params[i])) return true;
|
|
}
|
|
return false;
|
|
};
|
|
|
|
var iifeVisitor = {
|
|
enter: function (node, parent, scope, state) {
|
|
if (t.isReferencedIdentifier(node, parent) && state.scope.hasOwnReference(node.name)) {
|
|
state.iife = true;
|
|
this.stop();
|
|
}
|
|
}
|
|
};
|
|
|
|
exports.Function = function (node, parent, scope, file) {
|
|
if (!hasDefaults(node)) return;
|
|
|
|
t.ensureBlock(node);
|
|
|
|
var body = [];
|
|
|
|
var argsIdentifier = t.identifier("arguments");
|
|
argsIdentifier._ignoreAliasFunctions = true;
|
|
|
|
var lastNonDefaultParam = 0;
|
|
|
|
var state = { iife: false, scope: scope };
|
|
|
|
var pushDefNode = function (left, right, i) {
|
|
var defNode = util.template("default-parameter", {
|
|
VARIABLE_NAME: left,
|
|
DEFAULT_VALUE: right,
|
|
ARGUMENT_KEY: t.literal(i),
|
|
ARGUMENTS: argsIdentifier
|
|
}, true);
|
|
file.checkNode(defNode);
|
|
defNode._blockHoist = node.params.length - i;
|
|
body.push(defNode);
|
|
};
|
|
|
|
for (var i = 0; i < node.params.length; i++) {
|
|
var param = node.params[i];
|
|
|
|
if (!t.isAssignmentPattern(param)) {
|
|
lastNonDefaultParam = i + 1;
|
|
|
|
if (!t.isIdentifier(param)) {
|
|
scope.traverse(param, iifeVisitor, state);
|
|
}
|
|
|
|
if (file.transformers["es6.blockScopingTDZ"].canRun()) {
|
|
pushDefNode(param, t.identifier("undefined"), i);
|
|
}
|
|
|
|
continue;
|
|
}
|
|
|
|
var left = param.left;
|
|
var right = param.right;
|
|
|
|
node.params[i] = scope.generateUidIdentifier("x");
|
|
|
|
if (!state.iife) {
|
|
if (t.isIdentifier(right) && scope.hasOwnReference(right.name)) {
|
|
state.iife = true;
|
|
} else {
|
|
scope.traverse(right, iifeVisitor, state);
|
|
}
|
|
}
|
|
|
|
pushDefNode(left, right, i);
|
|
}
|
|
|
|
// we need to cut off all trailing default parameters
|
|
node.params = node.params.slice(0, lastNonDefaultParam);
|
|
|
|
if (state.iife) {
|
|
var container = t.functionExpression(null, [], node.body, node.generator);
|
|
container._aliasFunction = true;
|
|
|
|
body.push(t.returnStatement(t.callExpression(container, [])));
|
|
|
|
node.body = t.blockStatement(body);
|
|
} else {
|
|
node.body.body = body.concat(node.body.body);
|
|
}
|
|
};
|