FIR/UAST: commonize dot qualified expression

2021-06-07 17:10:13 -07:00
parent 8f70ef41f0
commit 08e486d1a7
91 changed files with 600 additions and 172 deletions
@@ -0,0 +1,36 @@
+/*
+ * 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 com.intellij.psi.PsiElement
+import com.intellij.psi.PsiNamedElement
+import org.jetbrains.kotlin.psi.KtDotQualifiedExpression
+import org.jetbrains.uast.UCallExpression
+import org.jetbrains.uast.UElement
+import org.jetbrains.uast.UQualifiedReferenceExpression
+import org.jetbrains.uast.UastQualifiedExpressionAccessType
+import org.jetbrains.uast.kotlin.internal.DelegatedMultiResolve
+
+class KotlinUQualifiedReferenceExpression(
+    override val sourcePsi: KtDotQualifiedExpression,
+    givenParent: UElement?
+) : KotlinAbstractUExpression(givenParent), UQualifiedReferenceExpression, DelegatedMultiResolve,
+    KotlinUElementWithType, KotlinEvaluatableUElement {
+    override val receiver by lz { baseResolveProviderService.baseKotlinConverter.convertOrEmpty(sourcePsi.receiverExpression, this) }
+    override val selector by lz { baseResolveProviderService.baseKotlinConverter.convertOrEmpty(sourcePsi.selectorExpression, this) }
+    override val accessType = UastQualifiedExpressionAccessType.SIMPLE
+
+    override fun resolve(): PsiElement? = sourcePsi.selectorExpression?.let { baseResolveProviderService.resolveToDeclaration(it) }
+
+    override val resolvedName: String?
+        get() = (resolve() as? PsiNamedElement)?.name
+
+    override val referenceNameElement: UElement?
+        get() = when (val selector = selector) {
+            is UCallExpression -> selector.methodIdentifier
+            else -> super.referenceNameElement
+        }
+}