kotlinx.reflect.lite.name.names.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kotlinx.reflect.lite Show documentation
Experimental lightweight library that replaces existing 'kotlin-reflect' implementation
The newest version!
/*
 * Copyright 2016-2022 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
 */

package kotlinx.reflect.lite.name

import kotlin.metadata.ClassName
import kotlin.metadata.isLocalClassName

internal typealias Name = String

internal data class FqName(val fqName: String) {
    val isRoot: Boolean get() = fqName.isEmpty()

    fun parent(): FqName = fqName.lastIndexOf('.').let { i ->
        if (i >= 0) FqName(fqName.substring(0, i)) else EMPTY
    }

    fun child(name: Name): FqName = FqName("$fqName.$name")

    fun shortName(): Name = fqName.substring(fqName.lastIndexOf('.') + 1)

    fun asString(): String = fqName

    override fun toString(): String = fqName

    companion object {
        @JvmField
        val EMPTY = FqName("")
    }
}

// Partially from: https://github.com/JetBrains/kotlin/blob/ea836fd46a1fef07d77c96f9d7e8d7807f793453/core/compiler.common/src/org/jetbrains/kotlin/name/ClassId.java#L34
internal data class ClassId(val packageFqName: FqName, val relativeClassName: FqName, val isLocal: Boolean = false) {
    constructor(packageFqName: FqName, relativeClassName: Name) : this(packageFqName, FqName(relativeClassName))

    constructor(className: ClassName) : this(
        FqName(className.substringBeforeLast('/', "").replace('/', '.')),
        FqName(className.substringAfterLast('/'))
    ) {
        require(!className.isLocalClassName()) { TODO("Local class names are not yet supported here: $className") }
    }

    val shortClassName: Name
        get() = relativeClassName.shortName()

    // TODO: investigate if call sites mean to use asJavaLookupFqName instead
    fun asSingleFqName(): FqName =
        if (packageFqName.isRoot) relativeClassName else FqName(packageFqName.asString() + "." + relativeClassName.asString())

    fun asJavaLookupFqName(): String =
        (if (packageFqName.isRoot) "" else (packageFqName.asString() + ".")) +
                relativeClassName.fqName.replace('.', '$')

    fun createNestedClassId(name: Name): ClassId =
        ClassId(packageFqName, relativeClassName.child(name), isLocal)

    fun getOuterClassId(): ClassId? =
        relativeClassName.parent().let {
            if (it.isRoot) null else ClassId(packageFqName, it, isLocal)
        }

    private fun asString(): String {
        return if (packageFqName.isRoot) relativeClassName.asString()
        else packageFqName.asString().replace('.', '/') + "/" + relativeClassName.asString()
    }

    fun asClassName(): ClassName {
        if (isLocal) TODO(asString())
        return asString()
    }

    override fun toString(): String = if (packageFqName.isRoot) "/" + asString() else asString()

    companion object {
        fun topLevel(topLevelFqName: FqName): ClassId =
            ClassId(topLevelFqName.parent(), FqName(topLevelFqName.shortName()))
    }
}