com.google.devtools.ksp.impl.symbol.kotlin.resolved.KSClassifierReferenceResolvedImpl.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of symbol-processing-cmdline Show documentation

Symbol processing for K/N and command line

There is a newer version: 2.1.0-RC2-1.0.28

Show newest version

package com.google.devtools.ksp.impl.symbol.kotlin.resolved

import com.google.devtools.ksp.common.IdKeyPair
import com.google.devtools.ksp.common.IdKeyTriple
import com.google.devtools.ksp.common.KSObjectCache
import com.google.devtools.ksp.symbol.*
import org.jetbrains.kotlin.analysis.api.types.KaClassType
import org.jetbrains.kotlin.analysis.api.types.KaClassTypeQualifier
import org.jetbrains.kotlin.analysis.api.types.KaTypeParameterType

class KSClassifierReferenceResolvedImpl private constructor(
    private val ktType: KaClassType,
    private val index: Int,
    override val parent: KSTypeReference?
) : KSClassifierReference {
    companion object :
        KSObjectCache, KSClassifierReferenceResolvedImpl>() {
        fun getCached(ktType: KaClassType, index: Int, parent: KSTypeReference?) =
            cache.getOrPut(IdKeyTriple(ktType, index, parent)) {
                KSClassifierReferenceResolvedImpl(ktType, index, parent)
            }
    }

    private val classifierReference: KaClassTypeQualifier
        get() = ktType.qualifiers[index]

    override val qualifier: KSClassifierReference? by lazy {
        if (index == 0) {
            null
        } else {
            getCached(ktType, index - 1, parent)
        }
    }

    override fun referencedName(): String {
        return classifierReference.name.asString()
    }

    override val typeArguments: List by lazy {
        classifierReference.typeArguments.map { KSTypeArgumentResolvedImpl.getCached(it, this) }
    }

    override val origin: Origin = parent?.origin ?: Origin.SYNTHETIC

    override val location: Location
        get() = parent?.location ?: NonExistLocation

    override fun toString(): String {
        return referencedName() + if (typeArguments.isNotEmpty()) "<${
        typeArguments.joinToString(", ") { it.toString() }
        }>" else ""
    }
}

class KSClassifierParameterImpl private constructor(
    private val ktType: KaTypeParameterType,
    override val parent: KSTypeReference?
) : KSClassifierReference {
    companion object : KSObjectCache, KSClassifierParameterImpl>() {
        fun getCached(ktType: KaTypeParameterType, parent: KSTypeReference?) =
            KSClassifierParameterImpl.cache.getOrPut(IdKeyPair(ktType, parent)) {
                KSClassifierParameterImpl(ktType, parent)
            }
    }

    override val qualifier: KSClassifierReference? = null

    override fun referencedName(): String {
        return ktType.name.asString()
    }

    override val typeArguments: List
        get() = emptyList()
    override val origin: Origin
        get() = parent?.origin ?: Origin.SYNTHETIC
    override val location: Location
        get() = parent?.location ?: NonExistLocation

    override fun toString(): String {
        return referencedName()
    }
}