add MethodDefinitionComputedOpening and MethodDefinitionComputedClosing support. fixes #470

Author: millermedeiros

lib/hooks/MethodDefinition.js

@@ -7,13 +7,24 @@ var ws = require('rocambole-whitespace');
 
 exports.format = function MethodDefinition(node) {
   br.limitAfter(node.startToken, 0);
-  // limit to one space after get/set/static
-  if (node.startToken !== node.key.startToken) {
-    ws.limitAfter(node.startToken, 1);
+
+  // key name
+  if (node.startToken.value === '[') {
+    // computed key
+    limit.around(node.startToken, 'MethodDefinitionComputedOpening');
+    limit.around(
+      tk.findNext(node.key.endToken, ']'),
+      'MethodDefinitionComputedClosing'
+    );
+  } else {
+    if (node.startToken !== node.key.startToken) {
+      // limit to one space after get/set/static
+      ws.limitAfter(node.startToken, 1);
+    }
+    ws.limitAfter(node.key.endToken, 'MethodDefinitionName');
   }
-  ws.limitAfter(node.key.endToken, 'MethodDefinitionName');
 
-  // parentheses {}
+  // curly braces {}
   var opening = node.value.body.startToken;
   limit.around(opening, 'MethodDefinitionOpeningBrace');
   limit.around(node.endToken, 'MethodDefinitionClosingBrace');

lib/preset/default-whitespace-after.js

@@ -74,6 +74,7 @@ module.exports = {
       "MemberExpressionClosing": 0,
       "MemberExpressionOpening": 0,
       "MemberExpressionPeriod": 0,
+      "MethodDefinitionComputedOpening": 0,
       "MethodDefinitionName": 0,
       "MethodName": 0,
       "ModuleSpecifierClosingBrace": 0,

lib/preset/default-whitespace-before.js

@@ -75,6 +75,7 @@ module.exports = {
       "MemberExpressionClosing": 0,
       "MemberExpressionOpening": 0,
       "MemberExpressionPeriod": 0,
+      "MethodDefinitionComputedClosing": 0,
       "ModuleSpecifierClosingBrace": 1,
       "ModuleSpecifierComma": 0,
       "ModuleSpecifierOpeningBrace": 1,

test/compare/default/class_declaration-in.js

@@ -45,3 +45,11 @@ Bar
 class Foo extends BarNamespace.Bar{}
 class Foo extends BarNamespace['Bar'].Bar["Bar"]('bar').Bar("bar", "bar"){}
 class Foo extends(BarNamespace.bar)(){}
+
+// #470
+const mySym = Symbol('some-symbol');
+class Foo {
+  [  mySym      ]() {
+    return true;
+  }
+}

test/compare/default/class_declaration-out.js

@@ -47,3 +47,11 @@ class Foo extends BarNamespace['Bar'].Bar["Bar"]('bar').Bar("bar", "bar") {
 }
 class Foo extends (BarNamespace.bar)() {
 }
+
+// #470
+const mySym = Symbol('some-symbol');
+class Foo {
+  [mySym]() {
+    return true;
+  }
+}