201: PsiElementVisitor.visitElement nullability

2020-01-28 21:55:24 +03:00
parent 98c6efaed9
commit 890109beb8
1 changed files with 15 additions and 14 deletions
@@ -13,22 +13,23 @@ import org.jetbrains.kotlin.psi.KtFile

 internal fun KtFile.dumpStructureText(): String {
    val sb = StringBuilder()
-    this.accept(object : PsiElementVisitor() {
-        override fun visitElement(element: PsiElement?) {
-            if (element is PsiWhiteSpace) {
-                if (sb.lastOrNull() !in listOf(' ', '{', '}', '(', ')')) {
-                    sb.append(" ")
+    this.accept(
+        object : PsiElementVisitor() {
+            override fun visitElement(element: PsiElement) {
+                if (element is PsiWhiteSpace) {
+                    if (sb.lastOrNull() !in listOf(' ', '{', '}', '(', ')')) {
+                        sb.append(" ")
+                    }
+                    return
                }
-                return
+                if (element is LeafPsiElement) {
+                    sb.append(element.text)
+                    return
+                }
+                element.acceptChildren(this)
            }
-            if (element is LeafPsiElement) {
-                sb.append(element.text)
-                return
-            }
-            element?.acceptChildren(this)
-
-        }
-    })
+        },
+    )

    return sb.toString().trim()
 }