201: PsiElementVisitor.visitElement nullability

nj2k/tests/org/jetbrains/kotlin/nj2k/StructuralPsiCompare.kt

@@ -13,22 +13,23 @@ import org.jetbrains.kotlin.psi.KtFile
 
 internal fun KtFile.dumpStructureText(): String {
     val sb = StringBuilder()
-    this.accept(object : PsiElementVisitor() {
+    this.accept(
-        override fun visitElement(element: PsiElement?) {
+        object : PsiElementVisitor() {
-            if (element is PsiWhiteSpace) {
+            override fun visitElement(element: PsiElement) {
-                if (sb.lastOrNull() !in listOf(' ', '{', '}', '(', ')')) {
+                if (element is PsiWhiteSpace) {
-                    sb.append(" ")
+                    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()
 }