fix: prevent argument list comments from moving to next argument

Author: jtkiesel

src/comments.ts

@@ -1,6 +1,13 @@
 import { util, type AstPath, type Doc } from "prettier";
 import { builders } from "prettier/doc";
-import { SyntaxType, type CommentNode, type SyntaxNode } from "./node-types.ts";
+import {
+  SyntaxType,
+  type ArrayAccessNode,
+  type CommentNode,
+  type FieldAccessNode,
+  type MethodInvocationNode,
+  type SyntaxNode
+} from "./node-types.ts";
 import parser from "./parser.ts";
 import printer from "./printer.ts";
 import {
@@ -188,21 +195,22 @@ function handleLabeledStatementComments(commentNode: CommentNode) {
 }
 
 function handleMemberChainComments(commentNode: CommentNode) {
-  const { enclosingNode, precedingNode, followingNode } = commentNode;
+  const { enclosingNode, followingNode, precedingNode } = commentNode;
   if (
     (enclosingNode?.type === SyntaxType.FieldAccess ||
-      (enclosingNode?.type === SyntaxType.MethodInvocation &&
-        precedingNode?.end.row !== commentNode.start.row)) &&
-    (followingNode?.type === SyntaxType.Identifier ||
-      followingNode?.type === SyntaxType.TypeArguments)
+      enclosingNode?.type === SyntaxType.MethodInvocation) &&
+    precedingNode &&
+    precedingNode === enclosingNode.objectNode &&
+    precedingNode.end.row !== commentNode.start.row
   ) {
     util.addLeadingComment(enclosingNode, commentNode);
     return true;
   } else if (
     followingNode &&
     isMember(followingNode) &&
-    precedingNode !== enclosingNode &&
-    !isPrettierIgnore(commentNode)
+    (!precedingNode ||
+      (precedingNode !== getMemberObject(followingNode) &&
+        precedingNode.end.row < commentNode.start.row))
   ) {
     util.addDanglingComment(followingNode, commentNode, undefined);
     return true;
@@ -291,6 +299,14 @@ function isMember(node: SyntaxNode) {
   );
 }
 
+function getMemberObject(
+  node: ArrayAccessNode | FieldAccessNode | MethodInvocationNode
+) {
+  return node.type === SyntaxType.ArrayAccess
+    ? node.arrayNode
+    : node.objectNode;
+}
+
 const binaryOperators = new Set([
   "<<",
   ">>",

test/unit-test/member_chain/_input.java

@@ -175,5 +175,20 @@ void argumentComment() {
         a(
             b.c() // comment
         );
+
+        a(
+            b, // comment
+            c.d
+        );
+
+        a(
+            b, // comment
+            c.d[0]
+        );
+
+        a(
+            b, // comment
+            c.d()
+        );
     }
 }

test/unit-test/member_chain/_output.java

@@ -230,5 +230,20 @@ void argumentComment() {
     a(
       b.c() // comment
     );
+
+    a(
+      b, // comment
+      c.d
+    );
+
+    a(
+      b, // comment
+      c.d[0]
+    );
+
+    a(
+      b, // comment
+      c.d()
+    );
   }
 }