cerxes/babel
3
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

shim in function arity when proxying named functions - fixes #896

Browse Source
This commit is contained in:
Sebastian McKenzie 2015-02-26 08:21:37 +11:00
parent d7ec337567
commit 1f79445ee5
2 changed files with 22 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/babel/transformation/helpers/get-function-arity.js Normal file
View File

@ -0,0 +1,9 @@
var t = require("../../types");
module.exports = function (node) {
var lastNonDefault = 0;
for (var i = 0; i < node.params.length; i++) {
if (!t.isAssignmentPattern(node.params[i])) lastNonDefault = i + 1;
}
return lastNonDefault;
};

16
src/babel/transformation/helpers/name-method.js
View File

@ -1,5 +1,6 @@
var util = require("../../util");
var t = require("../../types");
var getFunctionArity = require("./get-function-arity");
var util = require("../../util");
var t = require("../../types");
var visitor = {
enter(node, parent, scope, state) {
@ -21,12 +22,21 @@ var wrap = function (state, method, id, scope) {
if (state.selfReference) {
var templateName = "property-method-assignment-wrapper";
if (method.generator) templateName += "-generator";
return util.template(templateName, {
var template = util.template(templateName, {
FUNCTION: method,
FUNCTION_ID: id,
FUNCTION_KEY: scope.generateUidIdentifier(id.name),
WRAPPER_KEY: scope.generateUidIdentifier(id.name + "Wrapper")
});
// shim in dummy params to retain function arity, if you try to read the
// source then you'll get the original since it's proxied so it's all good
var params = template.callee.body.body[0].declarations[0].init.params;
for (var i = 0, len = getFunctionArity(method); i < len; i++) {
params.push(scope.generateUidIdentifier("x"));
}
return template;
} else {
method.id = id;
return method;