FIR/UAST: commonize named/vararg expressions

2021-06-18 00:16:09 -07:00
parent dcf99e9fd1
commit 33ff90a6da
3 changed files with 111 additions and 108 deletions
@@ -0,0 +1,50 @@
+/*
+ * 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 org.jetbrains.kotlin.psi.ValueArgument
+import org.jetbrains.uast.*
+
+class KotlinUNamedExpression private constructor(
+    override val name: String?,
+    override val sourcePsi: PsiElement?,
+    givenParent: UElement?,
+    expressionProducer: (UElement) -> UExpression
+) : KotlinAbstractUElement(givenParent), UNamedExpression {
+
+    override val expression: UExpression by lz { expressionProducer(this) }
+
+    override val uAnnotations: List<UAnnotation> = emptyList()
+
+    override val psi: PsiElement? = null
+
+    override val javaPsi: PsiElement? = null
+
+    companion object {
+        fun create(
+            name: String?,
+            valueArgument: ValueArgument,
+            uastParent: UElement?
+        ): UNamedExpression {
+            val expression = valueArgument.getArgumentExpression()
+            return KotlinUNamedExpression(name, valueArgument.asElement(), uastParent) { expressionParent ->
+                expression?.let { expressionParent.getLanguagePlugin().convertOpt(it, expressionParent) }
+                    ?: UastEmptyExpression(expressionParent)
+            }
+        }
+
+        fun create(
+            name: String?,
+            valueArguments: List<ValueArgument>,
+            uastParent: UElement?
+        ): UNamedExpression {
+            return KotlinUNamedExpression(name, null, uastParent) { expressionParent ->
+                KotlinUVarargExpression(valueArguments, expressionParent)
+            }
+        }
+    }
+}
@@ -0,0 +1,61 @@
+/*
+ * 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.PsiType
+import org.jetbrains.kotlin.psi.ValueArgument
+import org.jetbrains.uast.*
+import org.jetbrains.uast.kotlin.internal.DelegatedMultiResolve
+
+class KotlinUVarargExpression(
+    private val valueArgs: List<ValueArgument>,
+    uastParent: UElement?
+) : KotlinAbstractUExpression(uastParent), UCallExpression, DelegatedMultiResolve {
+    override val kind: UastCallKind = UastCallKind.NESTED_ARRAY_INITIALIZER
+
+    override val valueArguments: List<UExpression> by lz {
+        valueArgs.map {
+            it.getArgumentExpression()?.let { argumentExpression ->
+                getLanguagePlugin().convertOpt(argumentExpression, this)
+            } ?: UastEmptyExpression(this)
+        }
+    }
+
+    override fun getArgumentForParameter(i: Int): UExpression? = valueArguments.getOrNull(i)
+
+    override val valueArgumentCount: Int
+        get() = valueArgs.size
+
+    override val psi: PsiElement?
+        get() = null
+
+    override val methodIdentifier: UIdentifier?
+        get() = null
+
+    override val classReference: UReferenceExpression?
+        get() = null
+
+    override val methodName: String?
+        get() = null
+
+    override val typeArgumentCount: Int
+        get() = 0
+
+    override val typeArguments: List<PsiType>
+        get() = emptyList()
+
+    override val returnType: PsiType?
+        get() = null
+
+    override fun resolve() = null
+
+    override val receiver: UExpression?
+        get() = null
+
+    override val receiverType: PsiType?
+        get() = null
+}