org.jetbrains.kotlin.metadata.jvm.deserialization.JvmNameResolver.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler Show documentation
Kotlin Compiler
There is a newer version: 2.0.0
/*
 * Copyright 2000-2018 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.metadata.jvm.deserialization

import org.jetbrains.kotlin.metadata.deserialization.NameResolver
import org.jetbrains.kotlin.metadata.jvm.JvmProtoBuf
import org.jetbrains.kotlin.metadata.jvm.JvmProtoBuf.StringTableTypes.Record
import org.jetbrains.kotlin.metadata.jvm.JvmProtoBuf.StringTableTypes.Record.Operation.*

class JvmNameResolver(
    val types: JvmProtoBuf.StringTableTypes,
    val strings: Array
) : NameResolver {
    private val localNameIndices = types.localNameList.run { if (isEmpty()) emptySet() else toSet() }

    // Here we expand the 'range' field of the Record message for simplicity to a list of records
    // Note that as an optimization, range of each expanded record is equal to the original range, not 1. If correct ranges are needed,
    // please use the original record representation in [types.recordList].
    private val records: List = ArrayList().apply {
        val records = types.recordList
        this.ensureCapacity(records.size)
        for (record in records) {
            repeat(record.range) {
                this.add(record)
            }
        }
        this.trimToSize()
    }

    override fun getString(index: Int): String {
        val record = records[index]

        var string = when {
            record.hasString() -> record.string
            record.hasPredefinedIndex() && record.predefinedIndex in PREDEFINED_STRINGS.indices ->
                PREDEFINED_STRINGS[record.predefinedIndex]
            else -> strings[index]
        }

        if (record.substringIndexCount >= 2) {
            val (begin, end) = record.substringIndexList
            if (0 <= begin && begin <= end && end <= string.length) {
                string = string.substring(begin, end)
            }
        }

        if (record.replaceCharCount >= 2) {
            val (from, to) = record.replaceCharList
            string = string.replace(from.toChar(), to.toChar())
        }

        when (record.operation ?: NONE) {
            NONE -> {
                // Do nothing
            }
            INTERNAL_TO_CLASS_ID -> {
                string = string.replace('$', '.')
            }
            DESC_TO_CLASS_ID -> {
                if (string.length >= 2) {
                    string = string.substring(1, string.length - 1)
                }
                string = string.replace('$', '.')
            }
        }

        return string
    }

    override fun getQualifiedClassName(index: Int): String =
        getString(index)

    override fun isLocalClassName(index: Int): Boolean =
        index in localNameIndices

    companion object {
        // Simply "kotlin", but to avoid being renamed by namespace relocation (e.g., Shadow.relocate gradle plugin)
        private val kotlin = listOf('k', 'o', 't', 'l', 'i', 'n').joinToString(separator = "")

        val PREDEFINED_STRINGS = listOf(
            "$kotlin/Any",
            "$kotlin/Nothing",
            "$kotlin/Unit",
            "$kotlin/Throwable",
            "$kotlin/Number",

            "$kotlin/Byte", "$kotlin/Double", "$kotlin/Float", "$kotlin/Int",
            "$kotlin/Long", "$kotlin/Short", "$kotlin/Boolean", "$kotlin/Char",

            "$kotlin/CharSequence",
            "$kotlin/String",
            "$kotlin/Comparable",
            "$kotlin/Enum",

            "$kotlin/Array",
            "$kotlin/ByteArray", "$kotlin/DoubleArray", "$kotlin/FloatArray", "$kotlin/IntArray",
            "$kotlin/LongArray", "$kotlin/ShortArray", "$kotlin/BooleanArray", "$kotlin/CharArray",

            "$kotlin/Cloneable",
            "$kotlin/Annotation",

            "$kotlin/collections/Iterable", "$kotlin/collections/MutableIterable",
            "$kotlin/collections/Collection", "$kotlin/collections/MutableCollection",
            "$kotlin/collections/List", "$kotlin/collections/MutableList",
            "$kotlin/collections/Set", "$kotlin/collections/MutableSet",
            "$kotlin/collections/Map", "$kotlin/collections/MutableMap",
            "$kotlin/collections/Map.Entry", "$kotlin/collections/MutableMap.MutableEntry",

            "$kotlin/collections/Iterator", "$kotlin/collections/MutableIterator",
            "$kotlin/collections/ListIterator", "$kotlin/collections/MutableListIterator"
        )

        private val PREDEFINED_STRINGS_MAP = PREDEFINED_STRINGS.withIndex().associateBy({ it.value }, { it.index })

        fun getPredefinedStringIndex(string: String): Int? = PREDEFINED_STRINGS_MAP[string]
    }
}