Move NameResolver to module 'metadata' and J2K

Also move NameResolverImpl to 'metadata' and JvmNameResolver to 'metadata.jvm'
2018-03-02 18:44:22 +01:00
parent 0ba42850f9
commit 88dea9aa70
45 changed files with 78 additions and 86 deletions
@@ -0,0 +1,115 @@
+/*
+ * Copyright 2000-2018 JetBrains s.r.o. 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<String>
+) : 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
+    val records: List<Record> = ArrayList<Record>().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 {
+        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]
+    }
+}