support generators in nameMethod helper

This commit is contained in:
Sebastian McKenzie 2015-02-08 20:20:49 +11:00
parent 9e3c67a8a2
commit e06c8cd106

View File

@ -36,15 +36,18 @@ exports.property = function (node, file, scope) {
scope.traverse(node, visitor, state); scope.traverse(node, visitor, state);
var method = node.value;
if (state.selfReference) { if (state.selfReference) {
// todo: support generators var templateName = "property-method-assignment-wrapper";
node.value = util.template("property-method-assignment-wrapper", { if (method.generator) templateName += "-generator";
FUNCTION: node.value, node.value = util.template(templateName, {
FUNCTION: method,
FUNCTION_ID: key, FUNCTION_ID: key,
FUNCTION_KEY: scope.generateUidIdentifier(id), FUNCTION_KEY: scope.generateUidIdentifier(id),
WRAPPER_KEY: scope.generateUidIdentifier(id + "Wrapper") WRAPPER_KEY: scope.generateUidIdentifier(id + "Wrapper")
}); });
} else { } else {
node.value.id = key; method.id = key;
} }
}; };