Fix NPE in NoConsecutiveBlankLinesRule.

pinterest · Mar 8, 2021 · 44418ef · 44418ef
1 parent eecd43d
commit 44418ef
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/...dard/src/main/kotlin/com/pinterest/ktlint/ruleset/standard/NoConsecutiveBlankLinesRule.kt b/...dard/src/main/kotlin/com/pinterest/ktlint/ruleset/standard/NoConsecutiveBlankLinesRule.kt
@@ -9,14 +9,16 @@ import org.jetbrains.kotlin.com.intellij.lang.ASTNode
 import org.jetbrains.kotlin.com.intellij.psi.PsiWhiteSpace
 import org.jetbrains.kotlin.com.intellij.psi.impl.source.tree.LeafPsiElement
 
-class NoConsecutiveBlankLinesRule : Rule("no-consecutive-blank-lines") {
+public class NoConsecutiveBlankLinesRule : Rule("no-consecutive-blank-lines") {
 
     override fun visit(
         node: ASTNode,
         autoCorrect: Boolean,
         emit: (offset: Int, errorMessage: String, canBeAutoCorrected: Boolean) -> Unit
     ) {
-        if (node is PsiWhiteSpace) {
+        if (node is PsiWhiteSpace &&
+            node.prevSibling != null
+        ) {
             val text = node.getText()
             val lfcount = text.count { it == '\n' }
             if (lfcount < 2) {

diff --git a/.../src/test/kotlin/com/pinterest/ktlint/ruleset/standard/NoConsecutiveBlankLinesRuleTest.kt b/.../src/test/kotlin/com/pinterest/ktlint/ruleset/standard/NoConsecutiveBlankLinesRuleTest.kt
@@ -207,4 +207,20 @@ class NoConsecutiveBlankLinesRuleTest {
             )
         ).isEmpty()
     }
+
+    @Test
+    fun `should not raise NPE on linting Kotlin script file`() {
+        assertThat(
+            NoConsecutiveBlankLinesRule().lint(
+                """
+                import java.net.URI
+
+                plugins {
+                    `java-library`
+                }
+                """.trimIndent(),
+                script = true
+            )
+        ).isEmpty()
+    }
 }