fix: correct indentation of lambda argument bodies in method chains

Author: jtkiesel

packages/prettier-plugin-java/src/printers/expressions.ts

@@ -12,6 +12,7 @@ import {
   each,
   findBaseIndent,
   flatMap,
+  hasLeadingComments,
   indentInParentheses,
   isBinaryExpression,
   isNonTerminal,
@@ -276,8 +277,11 @@ export default {
       },
       "primarySuffix"
     );
+    const hasSuffixComments = children.primarySuffix.some(suffix =>
+      hasLeadingComments(suffix)
+    );
     return group(
-      canBreakForCallExpressions || willBreak(suffixes)
+      canBreakForCallExpressions || hasSuffixComments
         ? [prefix, indent(suffixes)]
         : [prefix, ...suffixes]
     );
@@ -766,7 +770,7 @@ function printTemplate<
   const parts = [begin, ...mids, end].map(image =>
     join(hardline, image.split(prefix))
   );
-  return [
+  return indent([
     parts[0],
     ...map(
       path,
@@ -780,5 +784,5 @@ function printTemplate<
       "embeddedExpression" as IterProperties<T["children"]>
     ),
     parts.at(-1)!
-  ];
+  ]);
 }

packages/prettier-plugin-java/test/unit-test/lambda/_input.java

@@ -212,6 +212,10 @@ void argumentAfterLambdaWithBlock() {
     }
 
     void huggableArguments() {
+        A.b().c(() -> {
+            return d;
+        });
+
         aaaaaaaaaaaaaaaaaaaaaaaa((bbbbbbbbbbbbbbbbbbbbbbbb, cccccccccccccccccccccccc, dddddddddddddddddddddddd) -> eeeeeeeeeeeeeeeeeeeeeeee.ffffffffffffffffffffffff());
 
         a.b(c -> d -> eeeeeeeeee.ffffffffff(gggggggggg, hhhhhhhhhh, iiiiiiiiii, jjjjjjjjjj, kkkkkkkkkk));

packages/prettier-plugin-java/test/unit-test/lambda/_output.java

@@ -228,6 +228,10 @@ void argumentAfterLambdaWithBlock() {
   }
 
   void huggableArguments() {
+    A.b().c(() -> {
+      return d;
+    });
+
     aaaaaaaaaaaaaaaaaaaaaaaa(
       (
         bbbbbbbbbbbbbbbbbbbbbbbb,

packages/prettier-plugin-java/test/unit-test/template-expression/_output.java

@@ -7,8 +7,8 @@ class TemplateExpression {
   String s = STR."You have a \{getOfferType()} waiting for you!";
 
   String msg = STR."The file \{filePath} \{
-    file.exists() ? "does" : "does not"
-  } exist";
+      file.exists() ? "does" : "does not"
+    } exist";
 
   String time = STR."The time is \{
       // The java.time.format package is very useful
@@ -55,6 +55,6 @@ class TemplateExpression {
     """;
 
   PreparedStatement ps = DB."SELECT * FROM Person p WHERE p.last_name = \{
-    name
-  }";
+      name
+    }";
 }