Uast: deprecating getLanguagePlugin and removing it's usages from the UClass (IDEA-220367, EA-141201)

GitOrigin-RevId: 71c7b24a73bc0e4103b12878e0d51a31649b8485

uast/uast-common/src/org/jetbrains/uast/UastLanguagePlugin.kt

@@ -118,6 +118,7 @@ inline fun <reified T : UElement> UastLanguagePlugin.convertOpt(element: PsiElem
   return convertElement(element, parent) as? T
 }
 
+@Deprecated("will throw exception if conversion fails", ReplaceWith("convertOpt"))
 inline fun <reified T : UElement> UastLanguagePlugin.convert(element: PsiElement, parent: UElement?): T {
   return convertElement(element, parent, T::class.java) as T
 }

uast/uast-common/src/org/jetbrains/uast/UastUtils.kt

@@ -182,6 +182,7 @@ tailrec fun UElement.getUastContext(): UastContext {
   return (uastParent ?: error("PsiElement should exist at least for UFile")).getUastContext()
 }
 
+@Deprecated("could unexpectedly throw exception", ReplaceWith("UastFacade.findPlugin"))
 tailrec fun UElement.getLanguagePlugin(): UastLanguagePlugin {
   val psi = this.sourcePsi
   if (psi != null) {

uast/uast-common/src/org/jetbrains/uast/declarations/UClass.kt

@@ -1,8 +1,11 @@
 // Copyright 2000-2019 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
 package org.jetbrains.uast
 
+import com.intellij.openapi.diagnostic.Attachment
+import com.intellij.openapi.diagnostic.Logger
 import com.intellij.psi.PsiAnonymousClass
 import com.intellij.psi.PsiClass
+import com.intellij.psi.PsiElement
 import org.jetbrains.uast.internal.acceptList
 import org.jetbrains.uast.internal.log
 import org.jetbrains.uast.visitor.UastTypedVisitor
@@ -39,17 +42,39 @@ interface UClass : UDeclaration, PsiClass {
    */
   val uastDeclarations: List<UDeclaration>
 
+  private inline fun <reified T : UElement> convertOrReport(psiElement: PsiElement, parent: UElement): T? =
+    convertOrReport(psiElement, parent, T::class.java)
+
+  private fun <T : UElement> convertOrReport(psiElement: PsiElement, parent: UElement, expectedType: Class<T>): T? {
+    fun getInfoString() = buildString {
+      appendln("context:${this@UClass.javaClass}")
+      appendln("psiElement:${psiElement.javaClass}")
+      appendln("psiElementContent:${runCatching { psiElement.text }}")
+    }
+
+    val plugin = this.sourcePsi?.let { UastFacade.findPlugin(it) } ?: UastFacade.findPlugin(psiElement)
+    if (plugin == null) {
+      LOG.error("cant get UAST plugin for $this to convert element $psiElement", Attachment("info.txt", getInfoString()))
+      return null
+    }
+    val result = expectedType.cast(plugin.convertElement(psiElement, parent, expectedType))
+    if (result == null) {
+      LOG.error("failed to convert element $psiElement in $this", Attachment("info.txt", getInfoString()))
+    }
+    return result
+  }
+
   override fun getFields(): Array<UField> =
-    javaPsi.fields.map { getLanguagePlugin().convert<UField>(it, this) }.toTypedArray()
+    javaPsi.fields.mapNotNull { convertOrReport<UField>(it, this) }.toTypedArray()
 
   override fun getInitializers(): Array<UClassInitializer> =
-    javaPsi.initializers.map { getLanguagePlugin().convert<UClassInitializer>(it, this) }.toTypedArray()
+    javaPsi.initializers.mapNotNull { convertOrReport<UClassInitializer>(it, this) }.toTypedArray()
 
   override fun getMethods(): Array<UMethod> =
-    javaPsi.methods.map { getLanguagePlugin().convert<UMethod>(it, this) }.toTypedArray()
+    javaPsi.methods.mapNotNull { convertOrReport<UMethod>(it, this) }.toTypedArray()
 
   override fun getInnerClasses(): Array<UClass> =
-    javaPsi.innerClasses.map { getLanguagePlugin().convert<UClass>(it, this) }.toTypedArray()
+    javaPsi.innerClasses.mapNotNull { convertOrReport<UClass>(it, this) }.toTypedArray()
 
   override fun asLogString(): String = log("name = $name")
 
@@ -85,6 +110,8 @@ interface UClass : UDeclaration, PsiClass {
     visitor.visitClass(this, data)
 }
 
+private val LOG = Logger.getInstance(UClass::class.java)
+
 interface UAnonymousClass : UClass, PsiAnonymousClass {
   override val psi: PsiAnonymousClass
 }