From 545ceeee339c2cc7cca1940f63c494751f4c7f7b Mon Sep 17 00:00:00 2001
From: Nicolay Mitropolsky <Nicolay.Mitropolsky@jetbrains.com>
Date: Mon, 12 Aug 2019 14:50:48 +0300
Subject: [PATCH] Uast: deprecating `getLanguagePlugin` and removing it's
 usages from the `UClass` (IDEA-220367, EA-141201)

GitOrigin-RevId: 71c7b24a73bc0e4103b12878e0d51a31649b8485
---
 .../org/jetbrains/uast/UastLanguagePlugin.kt  |  1 +
 .../src/org/jetbrains/uast/UastUtils.kt       |  1 +
 .../org/jetbrains/uast/declarations/UClass.kt | 35 ++++++++++++++++---
 3 files changed, 33 insertions(+), 4 deletions(-)

diff --git a/uast/uast-common/src/org/jetbrains/uast/UastLanguagePlugin.kt b/uast/uast-common/src/org/jetbrains/uast/UastLanguagePlugin.kt
index 5ba2e4b943e0..80d25b33021e 100644
--- a/uast/uast-common/src/org/jetbrains/uast/UastLanguagePlugin.kt
+++ b/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
 }
diff --git a/uast/uast-common/src/org/jetbrains/uast/UastUtils.kt b/uast/uast-common/src/org/jetbrains/uast/UastUtils.kt
index 2c2eecd78ed4..8f2ba45afacd 100644
--- a/uast/uast-common/src/org/jetbrains/uast/UastUtils.kt
+++ b/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) {
diff --git a/uast/uast-common/src/org/jetbrains/uast/declarations/UClass.kt b/uast/uast-common/src/org/jetbrains/uast/declarations/UClass.kt
index cef00af76714..316cf335379f 100644
--- a/uast/uast-common/src/org/jetbrains/uast/declarations/UClass.kt
+++ b/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
 }