inherit temporary computed property closure name from AssignmentExpressions and VariableDeclarators - #56

lib/6to5/templates/function-return-obj-this.js

@@ -1,3 +0,0 @@
-(function (KEY) {
-  return KEY;
-}).call(this, OBJECT)

lib/6to5/transform.js

@@ -11,7 +11,7 @@ function transform(code, opts) {
   code = (code || "") + "";
 
   var file = new File(opts);
-  return util.parse(code);
+  return file.parse(code);
 }
 
 transform.test = function (task, assert) {

lib/6to5/transformers/computed-property-names.js

@@ -5,7 +5,6 @@ var _        = require("lodash");
 
 exports.ObjectExpression = function (node, parent, file) {
   var hasComputed = false;
-  var hasThis     = false;
 
   var computed = [];
 
@@ -13,7 +12,6 @@ exports.ObjectExpression = function (node, parent, file) {
     if (prop.computed) {
       hasComputed = true;
       computed.unshift(prop);
-      hasThis = hasThis || traverse.hasType(prop, "ThisExpression");
       return false;
     } else {
       return true;
@@ -22,22 +20,32 @@ exports.ObjectExpression = function (node, parent, file) {
 
   if (!hasComputed) return;
 
-  var templateName = "function-return-obj";
-  if (hasThis) templateName += "-this";
+  var objIdNode;
 
-  var objId = b.identifier(file.generateUid("ref"));
+  if (parent.type === "AssignmentExpression") {
+    objIdNode = parent.left;
+  } else if (parent.type === "VariableDeclarator") {
+    objIdNode = parent.id;
+  }
 
-  var container = util.template(templateName, {
+  var objId = "ref";
+
+  if (objIdNode && objIdNode.type === "Identifier") {
+    objId = objIdNode.name;
+  }
+
+  objId = b.identifier(file.generateUid(objId));
+
+  var container = util.template("function-return-obj", {
     KEY: objId,
     OBJECT: node
   });
 
-  var containerBody;
-  if (templateName === "function-return-obj") {
-    containerBody = container.callee.body.body;
-  } else {
-    containerBody = container.callee.object.body.body;
-  }
+  var containerCallee = container.callee;
+  var containerBody = containerCallee.body.body;
+
+  containerCallee._aliasFunction = true;
+  containerCallee._aliasFunctionStopNonArrowFunctions = true;
 
   _.each(computed, function (prop) {
     containerBody.unshift(util.template("obj-key-set", {