Add handleAssignmentPatternComments helper function.

yamafaktory · yamafaktory · commit fc998c92508d · 2017-02-14T23:10:50.000+01:00
diff --git a/src/comments.js b/src/comments.js
@@ -175,6 +175,8 @@ function attach(comments, ast, text) {
     } else {
       if (handleIfStatementComments(enclosingNode, followingNode, comment)) {
         // We're good
+      } else if (handleAssignmentPatternComments(enclosingNode, followingNode, comment)) {
+        // We're good
       } else if (precedingNode && followingNode) {
         // Otherwise, text exists both before and after the comment on
         // the same line. If there is both a preceding and following
@@ -381,6 +383,20 @@ function handleConditionalExpressionComments(enclosingNode, followingNode, comme
   return false;
 }
 
+function handleAssignmentPatternComments(enclosingNode, followingNode, comment) {
+  if (
+    enclosingNode &&
+      (enclosingNode.type === "Property" || enclosingNode.type === "ObjectProperty") &&
+      enclosingNode.value && enclosingNode.value.type === "AssignmentPattern"
+  ) {
+    addTrailingComment(enclosingNode.value.left, comment);
+
+    return true;
+  }
+
+  return false;
+}
+
 function printComment(commentPath) {
   const comment = commentPath.getValue();
   comment.printed = true;
