FIR/UAST: commonize binary expression with type

2021-06-01 14:42:02 -07:00
parent b2e644b485
commit 9a539aac7d
16 changed files with 425 additions and 65 deletions
@@ -0,0 +1,37 @@
+/*
+ * Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.uast.kotlin
+
+import org.jetbrains.kotlin.lexer.KtTokens
+import org.jetbrains.kotlin.psi.KtBinaryExpressionWithTypeRHS
+import org.jetbrains.uast.*
+
+class KotlinUBinaryExpressionWithType(
+    override val sourcePsi: KtBinaryExpressionWithTypeRHS,
+    givenParent: UElement?
+) : KotlinAbstractUExpression(givenParent), UBinaryExpressionWithType, KotlinUElementWithType, KotlinEvaluatableUElement {
+    override val operand by lz {
+        baseResolveProviderService.baseKotlinConverter.convertOrEmpty(sourcePsi.left, this)
+    }
+
+    override val type by lz {
+        sourcePsi.right?.let {
+            baseResolveProviderService.resolveToType(it, this)
+        } ?: UastErrorType
+    }
+
+    override val typeReference by lz {
+        sourcePsi.right?.let {
+            KotlinUTypeReferenceExpression(it, this) { type }
+        }
+    }
+
+    override val operationKind = when (sourcePsi.operationReference.getReferencedNameElementType()) {
+        KtTokens.AS_KEYWORD -> UastBinaryExpressionWithTypeKind.TYPE_CAST
+        KtTokens.AS_SAFE -> KotlinBinaryExpressionWithTypeKinds.SAFE_TYPE_CAST
+        else -> UastBinaryExpressionWithTypeKind.UNKNOWN
+    }
+}