jlengrand/kotlin
1
0
Fork 0
mirror of https://github.com/jlengrand/kotlin.git synced 2026-03-10 08:31:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

FIR/UAST: commonize double colon expressions

Browse Source
This commit is contained in:
Jinseong Jeon
2021-06-08 22:58:43 -07:00
committed by TeamCityServer
parent 77e8aed995
commit 2999d0bd4b
20 changed files with 244 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
plugins/uast-kotlin-base/src/org/jetbrains/uast/kotlin/BaseKotlinUastResolveProviderService.kt
View File

@@ -7,6 +7,7 @@ package org.jetbrains.uast.kotlin
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiType
import org.jetbrains.kotlin.psi.KtDoubleColonExpression
import org.jetbrains.kotlin.psi.KtExpression
import org.jetbrains.kotlin.psi.KtTypeReference
import org.jetbrains.uast.UElement
@@ -25,6 +26,8 @@ interface BaseKotlinUastResolveProviderService {
fun resolveToType(ktTypeReference: KtTypeReference, source: UElement): PsiType?
fun getDoubleColonReceiverType(ktDoubleColonExpression: KtDoubleColonExpression, source: UElement): PsiType?
fun getExpressionType(uExpression: UExpression): PsiType?
fun evaluate(uExpression: UExpression): Any?

21
plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUCallableReferenceExpression.kt → plugins/uast-kotlin-base/src/org/jetbrains/uast/kotlin/expressions/KotlinUCallableReferenceExpression.kt
View File

@@ -1,17 +1,6 @@
/*
* Copyright 2010-2016 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* 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
@@ -20,7 +9,6 @@ import com.intellij.psi.PsiElement
import com.intellij.psi.PsiNamedElement
import com.intellij.psi.ResolveResult
import org.jetbrains.kotlin.psi.KtCallableReferenceExpression
import org.jetbrains.kotlin.resolve.BindingContext.DOUBLE_COLON_LHS
import org.jetbrains.uast.*
import org.jetbrains.uast.kotlin.internal.getResolveResultVariants
@@ -32,12 +20,11 @@ class KotlinUCallableReferenceExpression(
get() {
if (qualifierType != null) return null
val receiverExpression = sourcePsi.receiverExpression ?: return null
return KotlinConverter.convertExpression(receiverExpression, this, DEFAULT_EXPRESSION_TYPES_LIST)
return baseResolveProviderService.baseKotlinConverter.convertExpression(receiverExpression, this, DEFAULT_EXPRESSION_TYPES_LIST)
}
override val qualifierType by lz {
val ktType = sourcePsi.analyze()[DOUBLE_COLON_LHS, sourcePsi.receiverExpression]?.type ?: return@lz null
ktType.toPsiType(this, sourcePsi, boxed = true)
baseResolveProviderService.getDoubleColonReceiverType(sourcePsi, this)
}
override val callableName: String

28
plugins/uast-kotlin-base/src/org/jetbrains/uast/kotlin/expressions/KotlinUClassLiteralExpression.kt Normal file
View File

@@ -0,0 +1,28 @@
/*
* 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.psi.KtClassLiteralExpression
import org.jetbrains.uast.DEFAULT_EXPRESSION_TYPES_LIST
import org.jetbrains.uast.UClassLiteralExpression
import org.jetbrains.uast.UElement
import org.jetbrains.uast.UExpression
class KotlinUClassLiteralExpression(
override val sourcePsi: KtClassLiteralExpression,
givenParent: UElement?
) : KotlinAbstractUExpression(givenParent), UClassLiteralExpression, KotlinUElementWithType {
override val type by lz {
baseResolveProviderService.getDoubleColonReceiverType(sourcePsi, this)
}
override val expression: UExpression?
get() {
if (type != null) return null
val receiverExpression = sourcePsi.receiverExpression ?: return null
return baseResolveProviderService.baseKotlinConverter.convertExpression(receiverExpression, this, DEFAULT_EXPRESSION_TYPES_LIST)
}
}

2
plugins/uast-kotlin-fir/src/org/jetbrains/uast/kotlin/FirKotlinConverter.kt
View File

@@ -309,6 +309,8 @@ internal object FirKotlinConverter : BaseKotlinConverter {
is KtArrayAccessExpression -> expr<UArrayAccessExpression>(build(::FirKotlinUArrayAccessExpression))
is KtCallableReferenceExpression -> expr<UCallableReferenceExpression>(build(::KotlinUCallableReferenceExpression))
is KtClassLiteralExpression -> expr<UClassLiteralExpression>(build(::KotlinUClassLiteralExpression))
is KtDotQualifiedExpression -> expr<UQualifiedReferenceExpression>(build(::KotlinUQualifiedReferenceExpression))
is KtSimpleNameExpression -> expr<USimpleNameReferenceExpression>(build(::FirKotlinUSimpleReferenceExpression))

7
plugins/uast-kotlin-fir/src/org/jetbrains/uast/kotlin/FirKotlinUastResolveProviderService.kt
View File

@@ -10,6 +10,7 @@ import com.intellij.psi.PsiType
import org.jetbrains.kotlin.idea.frontend.api.analyseForUast
import org.jetbrains.kotlin.idea.references.mainReference
import org.jetbrains.kotlin.load.kotlin.TypeMappingMode
import org.jetbrains.kotlin.psi.KtDoubleColonExpression
import org.jetbrains.kotlin.psi.KtExpression
import org.jetbrains.kotlin.psi.KtReferenceExpression
import org.jetbrains.kotlin.psi.KtTypeReference
@@ -42,6 +43,12 @@ interface FirKotlinUastResolveProviderService : BaseKotlinUastResolveProviderSer
}
}
override fun getDoubleColonReceiverType(ktDoubleColonExpression: KtDoubleColonExpression, source: UElement): PsiType? {
analyseForUast(ktDoubleColonExpression) {
return ktDoubleColonExpression.getReceiverPsiType(TypeMappingMode.DEFAULT_UAST)
}
}
override fun getExpressionType(uExpression: UExpression): PsiType? {
val ktExpression = uExpression.sourcePsi as? KtExpression ?: return null
analyseForUast(ktExpression) {

2
plugins/uast-kotlin-fir/testData/legacyRenderLog/MethodReference.log.fir.txt vendored
View File

@@ -1,7 +1,7 @@
UFile (package = )
UClass (name = MethodReferenceKt)
UField (name = x)
[!] UnknownKotlinExpression (CALLABLE_REFERENCE_EXPRESSION)
UCallableReferenceExpression (name = bar)
UMethod (name = getX)
UClass (name = Foo)
UMethod (name = Foo)

2
plugins/uast-kotlin-fir/testData/legacyRenderLog/MethodReference.render.fir.txt vendored
View File

@@ -1,5 +1,5 @@
public final class MethodReferenceKt {
private static final var x: kotlin.reflect.KFunction<? extends kotlin.Unit> = [!] UnknownKotlinExpression (CALLABLE_REFERENCE_EXPRESSION)
private static final var x: kotlin.reflect.KFunction<? extends kotlin.Unit> = Foo::bar
public static final fun getX() : kotlin.reflect.KFunction<? extends kotlin.Unit> = UastEmptyExpression
}

4
plugins/uast-kotlin-fir/testData/legacyTypes/MethodReference.types.fir.txt vendored
View File

@@ -1,7 +1,7 @@
UFile (package = ) [public final class MethodReferenceKt {...]
UClass (name = MethodReferenceKt) [public final class MethodReferenceKt {...}]
UField (name = x) [private static final var x: kotlin.reflect.KFunction<? extends kotlin.Unit> = [!] UnknownKotlinExpression (CALLABLE_REFERENCE_EXPRESSION)]
[!] UnknownKotlinExpression (CALLABLE_REFERENCE_EXPRESSION) [[!] UnknownKotlinExpression (CALLABLE_REFERENCE_EXPRESSION)]
UField (name = x) [private static final var x: kotlin.reflect.KFunction<? extends kotlin.Unit> = Foo::bar]
UCallableReferenceExpression (name = bar) [Foo::bar] : PsiType:KFunction<? extends Unit>
UMethod (name = getX) [public static final fun getX() : kotlin.reflect.KFunction<? extends kotlin.Unit> = UastEmptyExpression]
UClass (name = Foo) [public final class Foo {...}]
UMethod (name = Foo) [public fun Foo() = UastEmptyExpression]

4
plugins/uast-kotlin-fir/testData/legacyValues/MethodReference.values.fir.txt vendored
View File

@@ -1,7 +1,7 @@
UFile (package = ) [public final class MethodReferenceKt {...]
UClass (name = MethodReferenceKt) [public final class MethodReferenceKt {...}]
UField (name = x) [private static final var x: kotlin.reflect.KFunction<? extends kotlin.Unit> = [!] UnknownKotlinExpression (CALLABLE_REFERENCE_EXPRESSION)]
[!] UnknownKotlinExpression (CALLABLE_REFERENCE_EXPRESSION) [[!] UnknownKotlinExpression (CALLABLE_REFERENCE_EXPRESSION)] = Undetermined
UField (name = x) [private static final var x: kotlin.reflect.KFunction<? extends kotlin.Unit> = Foo::bar]
UCallableReferenceExpression (name = bar) [Foo::bar] = external Foo::bar()
UMethod (name = getX) [public static final fun getX() : kotlin.reflect.KFunction<? extends kotlin.Unit> = UastEmptyExpression]
UClass (name = Foo) [public final class Foo {...}]
UMethod (name = Foo) [public fun Foo() = UastEmptyExpression]

15
plugins/uast-kotlin-fir/testData/type/classLiteral.kt vendored Normal file
View File

@@ -0,0 +1,15 @@
class Foo(val s: String) {
override fun equals(other: Any?): Boolean {
if (other == null) return false
if (other::class != this::class) return false
return s == (other as Foo).s
}
}
internal val stringClass = String::class
fun box(): String {
val x: CharSequence = ""
val xClass = x::class
return if (xClass == stringClass) "OK" else "$xClass"
}

53
plugins/uast-kotlin-fir/testData/type/classLiteral.types.fe10.txt vendored Normal file
View File

@@ -0,0 +1,53 @@
UFile (package = ) [public final class ClassLiteralKt {...]
UClass (name = ClassLiteralKt) [public final class ClassLiteralKt {...}]
UField (name = stringClass) [@org.jetbrains.annotations.NotNull private static final var stringClass: kotlin.reflect.KClass<java.lang.String> = java.lang.String]
UAnnotation (fqName = org.jetbrains.annotations.NotNull) [@org.jetbrains.annotations.NotNull]
UClassLiteralExpression [java.lang.String] : PsiType:KClass<String>
UMethod (name = getStringClass) [public static final fun getStringClass() : kotlin.reflect.KClass<java.lang.String> = UastEmptyExpression]
UMethod (name = box) [public static final fun box() : java.lang.String {...}]
UBlockExpression [{...}] : PsiType:Void
UDeclarationsExpression [var x: java.lang.CharSequence = ""]
ULocalVariable (name = x) [var x: java.lang.CharSequence = ""]
ULiteralExpression (value = "") [""] : PsiType:String
UDeclarationsExpression [var xClass: kotlin.reflect.KClass<? extends java.lang.CharSequence> = java.lang.CharSequence]
ULocalVariable (name = xClass) [var xClass: kotlin.reflect.KClass<? extends java.lang.CharSequence> = java.lang.CharSequence]
UClassLiteralExpression [java.lang.CharSequence] : PsiType:KClass<? extends CharSequence>
UReturnExpression [return if (xClass == stringClass) "OK" else xClass] : PsiType:Void
UIfExpression [if (xClass == stringClass) "OK" else xClass] : PsiType:String
UBinaryExpression (operator = ==) [xClass == stringClass] : PsiType:boolean
USimpleNameReferenceExpression (identifier = xClass) [xClass] : PsiType:KClass<? extends CharSequence>
USimpleNameReferenceExpression (identifier = stringClass) [stringClass] : PsiType:KClass<String>
ULiteralExpression (value = "OK") ["OK"] : PsiType:String
USimpleNameReferenceExpression (identifier = xClass) [xClass] : PsiType:KClass<? extends CharSequence>
UClass (name = Foo) [public final class Foo {...}]
UField (name = s) [@org.jetbrains.annotations.NotNull private final var s: java.lang.String]
UAnnotation (fqName = org.jetbrains.annotations.NotNull) [@org.jetbrains.annotations.NotNull]
UMethod (name = equals) [public fun equals(@org.jetbrains.annotations.Nullable other: java.lang.Object) : boolean {...}]
UParameter (name = other) [@org.jetbrains.annotations.Nullable var other: java.lang.Object]
UAnnotation (fqName = org.jetbrains.annotations.Nullable) [@org.jetbrains.annotations.Nullable]
UBlockExpression [{...}] : PsiType:Void
UIfExpression [if (other == null) return false] : PsiType:void
UBinaryExpression (operator = ==) [other == null] : PsiType:boolean
USimpleNameReferenceExpression (identifier = other) [other] : PsiType:Object
ULiteralExpression (value = null) [null] : PsiType:Void
UReturnExpression [return false] : PsiType:Void
ULiteralExpression (value = false) [false] : PsiType:boolean
UIfExpression [if (java.lang.Object != Foo) return false] : PsiType:void
UBinaryExpression (operator = !=) [java.lang.Object != Foo] : PsiType:boolean
UClassLiteralExpression [java.lang.Object] : PsiType:KClass<? extends Object>
UClassLiteralExpression [Foo] : PsiType:KClass<? extends Foo>
UReturnExpression [return false] : PsiType:Void
ULiteralExpression (value = false) [false] : PsiType:boolean
UReturnExpression [return s == (other as Foo).s] : PsiType:Void
UBinaryExpression (operator = ==) [s == (other as Foo).s] : PsiType:boolean
USimpleNameReferenceExpression (identifier = s) [s] : PsiType:String
UQualifiedReferenceExpression [(other as Foo).s] : PsiType:String
UParenthesizedExpression [(other as Foo)] : PsiType:Foo
UBinaryExpressionWithType [other as Foo] : PsiType:Foo
USimpleNameReferenceExpression (identifier = other) [other] : PsiType:Object
UTypeReferenceExpression (name = Foo) [Foo]
USimpleNameReferenceExpression (identifier = s) [s] : PsiType:String
UMethod (name = getS) [public final fun getS() : java.lang.String = UastEmptyExpression]
UMethod (name = Foo) [public fun Foo(@org.jetbrains.annotations.NotNull s: java.lang.String) = UastEmptyExpression]
UParameter (name = s) [@org.jetbrains.annotations.NotNull var s: java.lang.String]
UAnnotation (fqName = org.jetbrains.annotations.NotNull) [@org.jetbrains.annotations.NotNull]

32
plugins/uast-kotlin-fir/testData/type/classLiteral.types.fir.txt vendored Normal file
View File

@@ -0,0 +1,32 @@
UFile (package = ) [public final class ClassLiteralKt {...]
UClass (name = ClassLiteralKt) [public final class ClassLiteralKt {...}]
UField (name = stringClass) [private static final var stringClass: kotlin.reflect.KClass<java.lang.String> = java.lang.String]
UClassLiteralExpression [java.lang.String] : PsiType:KClass<String>
UMethod (name = getStringClass) [public static final fun getStringClass() : kotlin.reflect.KClass<java.lang.String> = UastEmptyExpression]
UMethod (name = box) [public static final fun box() : java.lang.String {...}]
UBlockExpression [{...}] : PsiType:Void
[!] UnknownKotlinExpression (PROPERTY) [[!] UnknownKotlinExpression (PROPERTY)]
[!] UnknownKotlinExpression (PROPERTY) [[!] UnknownKotlinExpression (PROPERTY)]
UReturnExpression [return if ([!] UnknownKotlinExpression (BINARY_EXPRESSION)) "OK" else xClass] : PsiType:Void
UIfExpression [if ([!] UnknownKotlinExpression (BINARY_EXPRESSION)) "OK" else xClass] : PsiType:String
[!] UnknownKotlinExpression (BINARY_EXPRESSION) [[!] UnknownKotlinExpression (BINARY_EXPRESSION)]
ULiteralExpression (value = "OK") ["OK"] : PsiType:String
USimpleNameReferenceExpression (identifier = xClass) [xClass] : PsiType:KClass<? extends CharSequence>
UClass (name = Foo) [public final class Foo {...}]
UField (name = s) [private final var s: java.lang.String]
UMethod (name = Foo) [public fun Foo(s: java.lang.String) = UastEmptyExpression]
UParameter (name = s) [var s: java.lang.String]
UMethod (name = getS) [public final fun getS() : java.lang.String = UastEmptyExpression]
UMethod (name = equals) [public fun equals(other: java.lang.Object) : boolean {...}]
UParameter (name = other) [var other: java.lang.Object]
UBlockExpression [{...}] : PsiType:Void
UIfExpression [if ([!] UnknownKotlinExpression (BINARY_EXPRESSION)) return false] : PsiType:Unit
[!] UnknownKotlinExpression (BINARY_EXPRESSION) [[!] UnknownKotlinExpression (BINARY_EXPRESSION)]
UReturnExpression [return false] : PsiType:Void
ULiteralExpression (value = false) [false] : PsiType:boolean
UIfExpression [if ([!] UnknownKotlinExpression (BINARY_EXPRESSION)) return false] : PsiType:Unit
[!] UnknownKotlinExpression (BINARY_EXPRESSION) [[!] UnknownKotlinExpression (BINARY_EXPRESSION)]
UReturnExpression [return false] : PsiType:Void
ULiteralExpression (value = false) [false] : PsiType:boolean
UReturnExpression [return [!] UnknownKotlinExpression (BINARY_EXPRESSION)] : PsiType:Void
[!] UnknownKotlinExpression (BINARY_EXPRESSION) [[!] UnknownKotlinExpression (BINARY_EXPRESSION)]

5
plugins/uast-kotlin-fir/tests/org/jetbrains/uast/test/kotlin/FE1UastTypesTestGenerated.java generated
View File

@@ -44,6 +44,11 @@ public class FE1UastTypesTestGenerated extends AbstractFE1UastTypesTest {
runTest("plugins/uast-kotlin-fir/testData/type/arrayGetAssignMultiIndex.kt");
}
@TestMetadata("classLiteral.kt")
public void testClassLiteral() throws Exception {
runTest("plugins/uast-kotlin-fir/testData/type/classLiteral.kt");
}
@TestMetadata("typeCast.kt")
public void testTypeCast() throws Exception {
runTest("plugins/uast-kotlin-fir/testData/type/typeCast.kt");

30
plugins/uast-kotlin-fir/tests/org/jetbrains/uast/test/kotlin/FirUastResolveApiTest.kt
View File

@@ -6,14 +6,19 @@
package org.jetbrains.uast.test.kotlin
import com.intellij.psi.PsiClass
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiField
import com.intellij.psi.PsiMethod
import com.intellij.testFramework.TestDataPath
import org.jetbrains.kotlin.psi.*
import org.jetbrains.kotlin.test.JUnit3RunnerWithInners
import org.jetbrains.kotlin.test.TestMetadata
import org.jetbrains.uast.UCallableReferenceExpression
import org.jetbrains.uast.UElement
import org.jetbrains.uast.UFile
import org.jetbrains.uast.test.common.kotlin.asRefNames
import org.jetbrains.uast.test.env.kotlin.AbstractFirUastTest
import org.jetbrains.uast.visitor.UastVisitor
import org.junit.Assert
import org.junit.runner.RunWith
import java.lang.IllegalStateException
@@ -36,6 +41,31 @@ class FirUastResolveApiTest : AbstractFirUastTest() {
// Bogus
}
@TestMetadata("MethodReference.kt")
fun testMethodReference() {
doCheck("plugins/uast-kotlin/testData/MethodReference.kt") { _, uFile ->
val facade = uFile.findFacade()
?: throw IllegalStateException("No facade found at ${uFile.asRefNames()}")
// val x = Foo::bar
val x = facade.fields.single()
var barReference: PsiElement? = null
x.accept(object : UastVisitor {
override fun visitElement(node: UElement): Boolean {
return false
}
override fun visitCallableReferenceExpression(node: UCallableReferenceExpression): Boolean {
barReference = node.resolve()
return false
}
})
Assert.assertNotNull("Foo::bar is not resolved", barReference)
Assert.assertTrue("Foo::bar is not a function", barReference is KtNamedFunction)
Assert.assertEquals("Foo.bar", (barReference as KtNamedFunction).fqName?.asString())
}
}
@TestMetadata("Imports.kt")
fun testImports() {
doCheck("plugins/uast-kotlin/testData/Imports.kt") { _, uFile ->

5
plugins/uast-kotlin-fir/tests/org/jetbrains/uast/test/kotlin/FirUastTypesTestGenerated.java generated
View File

@@ -44,6 +44,11 @@ public class FirUastTypesTestGenerated extends AbstractFirUastTypesTest {
runTest("plugins/uast-kotlin-fir/testData/type/arrayGetAssignMultiIndex.kt");
}
@TestMetadata("classLiteral.kt")
public void testClassLiteral() throws Exception {
runTest("plugins/uast-kotlin-fir/testData/type/classLiteral.kt");
}
@TestMetadata("typeCast.kt")
public void testTypeCast() throws Exception {
runTest("plugins/uast-kotlin-fir/testData/type/typeCast.kt");

14
plugins/uast-kotlin-fir/tests/org/jetbrains/uast/test/kotlin/utils.kt Normal file
View File

@@ -0,0 +1,14 @@
/*
* 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.test.kotlin
import org.jetbrains.kotlin.asJava.classes.KtLightClassForFacade
import org.jetbrains.uast.UClass
import org.jetbrains.uast.UFile
internal fun UFile.findFacade(): UClass? {
return classes.find { it.sourcePsi is KtLightClassForFacade }
}

8
plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/KotlinUastResolveProviderService.kt
View File

@@ -9,6 +9,7 @@ import com.intellij.psi.PsiElement
import com.intellij.psi.PsiType
import org.jetbrains.kotlin.codegen.state.KotlinTypeMapper
import org.jetbrains.kotlin.config.LanguageVersionSettings
import org.jetbrains.kotlin.psi.KtDoubleColonExpression
import org.jetbrains.kotlin.psi.KtElement
import org.jetbrains.kotlin.psi.KtExpression
import org.jetbrains.kotlin.psi.KtTypeReference
@@ -38,6 +39,13 @@ interface KotlinUastResolveProviderService : BaseKotlinUastResolveProviderServic
return ktTypeReference.toPsiType(source)
}
override fun getDoubleColonReceiverType(ktDoubleColonExpression: KtDoubleColonExpression, source: UElement): PsiType? {
val ktType =
ktDoubleColonExpression.analyze()[BindingContext.DOUBLE_COLON_LHS, ktDoubleColonExpression.receiverExpression]?.type
?: return null
return ktType.toPsiType(source, ktDoubleColonExpression, boxed = true)
}
override fun getExpressionType(uExpression: UExpression): PsiType? {
val ktElement = uExpression.sourcePsi as? KtExpression ?: return null
val ktType = ktElement.analyze()[BindingContext.EXPRESSION_TYPE_INFO, ktElement]?.type ?: return null

41
plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUClassLiteralExpression.kt
View File

@@ -1,41 +0,0 @@
/*
* Copyright 2010-2016 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jetbrains.uast.kotlin
import org.jetbrains.kotlin.psi.KtClassLiteralExpression
import org.jetbrains.kotlin.resolve.BindingContext.DOUBLE_COLON_LHS
import org.jetbrains.uast.DEFAULT_EXPRESSION_TYPES_LIST
import org.jetbrains.uast.UClassLiteralExpression
import org.jetbrains.uast.UElement
import org.jetbrains.uast.UExpression
class KotlinUClassLiteralExpression(
override val sourcePsi: KtClassLiteralExpression,
givenParent: UElement?
) : KotlinAbstractUExpression(givenParent), UClassLiteralExpression, KotlinUElementWithType {
override val type by lz {
val ktType = sourcePsi.analyze()[DOUBLE_COLON_LHS, sourcePsi.receiverExpression]?.type ?: return@lz null
ktType.toPsiType(this, sourcePsi, boxed = true)
}
override val expression: UExpression?
get() {
if (type != null) return null
val receiverExpression = sourcePsi.receiverExpression ?: return null
return KotlinConverter.convertExpression(receiverExpression, this, DEFAULT_EXPRESSION_TYPES_LIST)
}
}