Allow calling valueOf() on nullable UAST expressions

diff --git a/uast/uast-common/src/org/jetbrains/uast/evaluation/UEvaluationContext.kt b/uast/uast-common/src/org/jetbrains/uast/evaluation/UEvaluationContext.kt
index daad664fcd0f030c671a5251551e78a85f097f4c..9967076d6efe768edcc42a8e94ec0a1346f2db16 100644 (file)
--- a/uast/uast-common/src/org/jetbrains/uast/evaluation/UEvaluationContext.kt
+++ b/uast/uast-common/src/org/jetbrains/uast/evaluation/UEvaluationContext.kt
@@ -38,14 +38,15 @@ fun UFile.analyzeAll(context: UastContext = getUastContext(), extensions: List<U
         MapBasedEvaluationContext(context, extensions).analyzeAll(this)
 
 @JvmOverloads
-fun UExpression.uValueOf(extensions: List<UEvaluatorExtension> = emptyList()): UValue? {
+fun UExpression?.uValueOf(extensions: List<UEvaluatorExtension> = emptyList()): UValue? {
+    if (this == null) return null
     val declaration = getContainingDeclaration() ?: return null
     val context = declaration.getEvaluationContextWithCaching(extensions)
     context.analyze(declaration)
     return context.valueOf(this)
 }
 
-fun UExpression.uValueOf(vararg extensions: UEvaluatorExtension): UValue? = uValueOf(extensions.asList())
+fun UExpression?.uValueOf(vararg extensions: UEvaluatorExtension): UValue? = uValueOf(extensions.asList())
 
 fun UDeclaration.getEvaluationContextWithCaching(extensions: List<UEvaluatorExtension> = emptyList()): UEvaluationContext {
     return containingFile?.let { file ->