fixup! fix: correctly transform __proto__ properties

packages/babel-plugin-transform-computed-properties/test/fixtures/spec/proto-shorthand/input.js

@@ -0,0 +1,11 @@
+var shorthand = {
+  __proto__,
+}
+
+var method = {
+  __proto__() {}
+}
+
+var methodComputed = {
+  ["__proto__"]() {}
+}

packages/babel-plugin-transform-computed-properties/test/fixtures/spec/proto-shorthand/options.json

@@ -0,0 +1,7 @@
+{
+  "plugins": [
+    "external-helpers",
+    "transform-computed-properties",
+    "transform-shorthand-properties"
+  ]
+}

packages/babel-plugin-transform-computed-properties/test/fixtures/spec/proto-shorthand/output.js

@@ -0,0 +1,3 @@
+var shorthand = babelHelpers.defineProperty({}, "__proto__", __proto__);
+var method = babelHelpers.defineProperty({}, "__proto__", function () {});
+var methodComputed = babelHelpers.defineProperty({}, "__proto__", function () {});

packages/babel-plugin-transform-shorthand-properties/src/index.js

@@ -29,12 +29,14 @@ export default declare(api => {
         }
       },
 
-      ObjectProperty({ node }) {
+      ObjectProperty(path) {
+        const { node } = path;
         if (node.shorthand) {
-          node.shorthand = false;
-          if (t.isIdentifier(node.key, { name: "__proto__" })) {
-            node.key = t.stringLiteral(node.key.name);
-            node.computed = true;
+          const computedKey = t.toComputedKey(node);
+          if (t.isStringLiteral(computedKey, { value: "__proto__" })) {
+            path.replaceWith(t.objectProperty(computedKey, node.value, true));
+          } else {
+            node.shorthand = false;
           }
         }
       },