Emit a parsing error for calls with multiple trailing lambdas (#457)

Summary: Pull Request resolved: #457 Reviewed By: strulovich Differential Revision: D56153379 Pulled By: hick209 fbshipit-source-id: c4b264e5251c15e8855e7fe9447458c49efe74d6
facebook · Apr 16, 2024 · fad9d0d · fad9d0d
1 parent 6a639f0
commit fad9d0d
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 8 deletions.
diff --git a/core/src/main/java/com/facebook/ktfmt/format/KotlinInputAstVisitor.kt b/core/src/main/java/com/facebook/ktfmt/format/KotlinInputAstVisitor.kt
@@ -771,13 +771,17 @@ class KotlinInputAstVisitor(
           }
         }
       }
-      if (lambdaArguments.isNotEmpty()) {
-        builder.space()
-        visitArgumentInternal(
-            lambdaArguments.single(),
-            wrapInBlock = false,
-            brokeBeforeBrace = brokeBeforeBrace,
-        )
+      when (lambdaArguments.size) {
+        0 -> {}
+        1 -> {
+          builder.space()
+          visitArgumentInternal(
+              lambdaArguments.single(),
+              wrapInBlock = false,
+              brokeBeforeBrace = brokeBeforeBrace,
+          )
+        }
+        else -> throw ParseError("Maximum one trailing lambda is allowed", lambdaArguments[1])
       }
     }
   }

diff --git a/core/src/main/java/com/facebook/ktfmt/format/ParseError.kt b/core/src/main/java/com/facebook/ktfmt/format/ParseError.kt
@@ -17,7 +17,24 @@
 package com.facebook.ktfmt.format
 
 import org.jetbrains.kotlin.com.intellij.openapi.util.text.LineColumn
+import org.jetbrains.kotlin.com.intellij.psi.PsiElement
 
 class ParseError(val errorDescription: String, val lineColumn: LineColumn) :
     IllegalArgumentException(
-        "${lineColumn.line + 1}:${lineColumn.column + 1}: error: $errorDescription")
+        "${lineColumn.line + 1}:${lineColumn.column + 1}: error: $errorDescription") {
+
+  constructor(
+      errorDescription: String,
+      element: PsiElement,
+  ) : this(errorDescription, positionOf(element))
+
+  companion object {
+    private fun positionOf(element: PsiElement): LineColumn {
+      val doc = element.containingFile.viewProvider.document!!
+      val offset = element.textOffset
+      val lineZero = doc.getLineNumber(offset)
+      val colZero = offset - doc.getLineStartOffset(lineZero)
+      return LineColumn.of(lineZero, colZero)
+    }
+  }
+}
diff --git a/core/src/test/java/com/facebook/ktfmt/cli/MainTest.kt b/core/src/test/java/com/facebook/ktfmt/cli/MainTest.kt
@@ -159,6 +159,18 @@ class MainTest {
     assertThat(err.toString(testCharset)).contains("foo.kt:1:14: error: ")
   }
 
+  @Test
+  fun `Parsing error for multiple trailing lambdas`() {
+    val fooBar = root.resolve("foo.kt")
+    fooBar.writeText("val x = foo(bar { } { zap = 2 })")
+    val returnValue =
+        Main(emptyInput, PrintStream(out), PrintStream(err), arrayOf(fooBar.toString())).run()
+
+    assertThat(returnValue).isEqualTo(1)
+    assertThat(err.toString(testCharset))
+        .contains("foo.kt:1:21: error: Maximum one trailing lambda is allowed")
+  }
+
   @Test
   fun `all files in args are processed, even if one of them has an error`() {
     val file1 = root.resolve("file1.kt")