nj2k/src/org/jetbrains/kotlin/nj2k/symbols/JKFieldSymbol.kt

/*
 * Copyright 2010-2019 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.kotlin.nj2k.symbols


import com.intellij.psi.PsiVariable
import org.jetbrains.kotlin.j2k.ast.Nullability
import org.jetbrains.kotlin.nj2k.tree.JKVariable
import org.jetbrains.kotlin.nj2k.types.*
import org.jetbrains.kotlin.psi.KtCallableDeclaration
import org.jetbrains.kotlin.psi.KtEnumEntry
import org.jetbrains.kotlin.psi.psiUtil.containingClass

sealed class JKFieldSymbol : JKSymbol {
    abstract val fieldType: JKType?
}

class JKUniverseFieldSymbol(override val typeFactory: JKTypeFactory) : JKFieldSymbol(), JKUniverseSymbol<JKVariable> {
    override val fieldType: JKType
        get() = target.type.type

    override lateinit var target: JKVariable
}

class JKMultiverseFieldSymbol(
    override val target: PsiVariable,
    override val typeFactory: JKTypeFactory
) : JKFieldSymbol(), JKMultiverseSymbol<PsiVariable> {
    override val fieldType: JKType
        get() = typeFactory.fromPsiType(target.type)
}

class JKMultiversePropertySymbol(
    override val target: KtCallableDeclaration,
    override val typeFactory: JKTypeFactory
) : JKFieldSymbol(), JKMultiverseKtSymbol<KtCallableDeclaration> {
    override val fieldType: JKType?
        get() = target.typeReference?.toJK(typeFactory)
}

class JKMultiverseKtEnumEntrySymbol(
    override val target: KtEnumEntry,
    override val typeFactory: JKTypeFactory
) : JKFieldSymbol(), JKMultiverseKtSymbol<KtEnumEntry> {
    override val fieldType: JKType?
        get() = target.containingClass()?.let { klass ->
            JKClassType(
                symbolProvider.provideDirectSymbol(klass) as? JKClassSymbol ?: return@let null,
                emptyList(),
                Nullability.NotNull
            )
        }
}

class JKUnresolvedField(
    override val target: String,
    override val typeFactory: JKTypeFactory
) : JKFieldSymbol(), JKUnresolvedSymbol {
    override val fieldType: JKType
        get() = typeFactory.types.nullableAny
}