Extract interface out of NameResolver
This commit is contained in:
+3
-3
@@ -20,7 +20,7 @@ import com.google.protobuf.ExtensionRegistryLite
|
|||||||
import org.jetbrains.kotlin.serialization.ClassData
|
import org.jetbrains.kotlin.serialization.ClassData
|
||||||
import org.jetbrains.kotlin.serialization.PackageData
|
import org.jetbrains.kotlin.serialization.PackageData
|
||||||
import org.jetbrains.kotlin.serialization.ProtoBuf
|
import org.jetbrains.kotlin.serialization.ProtoBuf
|
||||||
import org.jetbrains.kotlin.serialization.deserialization.NameResolver
|
import org.jetbrains.kotlin.serialization.deserialization.NameResolverImpl
|
||||||
import java.io.ByteArrayInputStream
|
import java.io.ByteArrayInputStream
|
||||||
|
|
||||||
public object JvmProtoBufUtil {
|
public object JvmProtoBufUtil {
|
||||||
@@ -37,7 +37,7 @@ public object JvmProtoBufUtil {
|
|||||||
@JvmStatic
|
@JvmStatic
|
||||||
public fun readClassDataFrom(bytes: ByteArray): ClassData {
|
public fun readClassDataFrom(bytes: ByteArray): ClassData {
|
||||||
val input = ByteArrayInputStream(bytes)
|
val input = ByteArrayInputStream(bytes)
|
||||||
val nameResolver = NameResolver.read(input)
|
val nameResolver = NameResolverImpl.read(input)
|
||||||
val classProto = ProtoBuf.Class.parseFrom(input, EXTENSION_REGISTRY)
|
val classProto = ProtoBuf.Class.parseFrom(input, EXTENSION_REGISTRY)
|
||||||
return ClassData(nameResolver, classProto)
|
return ClassData(nameResolver, classProto)
|
||||||
}
|
}
|
||||||
@@ -49,7 +49,7 @@ public object JvmProtoBufUtil {
|
|||||||
@JvmStatic
|
@JvmStatic
|
||||||
public fun readPackageDataFrom(bytes: ByteArray): PackageData {
|
public fun readPackageDataFrom(bytes: ByteArray): PackageData {
|
||||||
val input = ByteArrayInputStream(bytes)
|
val input = ByteArrayInputStream(bytes)
|
||||||
val nameResolver = NameResolver.read(input)
|
val nameResolver = NameResolverImpl.read(input)
|
||||||
val packageProto = ProtoBuf.Package.parseFrom(input, EXTENSION_REGISTRY)
|
val packageProto = ProtoBuf.Package.parseFrom(input, EXTENSION_REGISTRY)
|
||||||
return PackageData(nameResolver, packageProto)
|
return PackageData(nameResolver, packageProto)
|
||||||
}
|
}
|
||||||
|
|||||||
+1
-1
@@ -37,7 +37,7 @@ public abstract class DeserializedPackageFragment(
|
|||||||
private val loadResource: (path: String) -> InputStream?
|
private val loadResource: (path: String) -> InputStream?
|
||||||
) : PackageFragmentDescriptorImpl(module, fqName) {
|
) : PackageFragmentDescriptorImpl(module, fqName) {
|
||||||
|
|
||||||
val nameResolver = NameResolver.read(
|
val nameResolver = NameResolverImpl.read(
|
||||||
loadResource(serializedResourcePaths.getStringTableFilePath(fqName))
|
loadResource(serializedResourcePaths.getStringTableFilePath(fqName))
|
||||||
?: loadResourceSure(serializedResourcePaths.fallbackPaths.getStringTableFilePath(fqName))
|
?: loadResourceSure(serializedResourcePaths.fallbackPaths.getStringTableFilePath(fqName))
|
||||||
)
|
)
|
||||||
|
|||||||
+5
-72
@@ -20,84 +20,17 @@ import org.jetbrains.annotations.NotNull;
|
|||||||
import org.jetbrains.kotlin.name.ClassId;
|
import org.jetbrains.kotlin.name.ClassId;
|
||||||
import org.jetbrains.kotlin.name.FqName;
|
import org.jetbrains.kotlin.name.FqName;
|
||||||
import org.jetbrains.kotlin.name.Name;
|
import org.jetbrains.kotlin.name.Name;
|
||||||
import org.jetbrains.kotlin.serialization.ProtoBuf;
|
|
||||||
import org.jetbrains.kotlin.utils.UtilsPackage;
|
|
||||||
|
|
||||||
import java.io.IOException;
|
public interface NameResolver {
|
||||||
import java.io.InputStream;
|
|
||||||
import java.util.LinkedList;
|
|
||||||
|
|
||||||
import static org.jetbrains.kotlin.serialization.ProtoBuf.QualifiedNameTable.QualifiedName;
|
|
||||||
|
|
||||||
public class NameResolver {
|
|
||||||
@NotNull
|
@NotNull
|
||||||
public static NameResolver read(@NotNull InputStream in) {
|
String getString(int index);
|
||||||
try {
|
|
||||||
ProtoBuf.StringTable simpleNames = ProtoBuf.StringTable.parseDelimitedFrom(in);
|
|
||||||
ProtoBuf.QualifiedNameTable qualifiedNames = ProtoBuf.QualifiedNameTable.parseDelimitedFrom(in);
|
|
||||||
return new NameResolver(simpleNames, qualifiedNames);
|
|
||||||
}
|
|
||||||
catch (IOException e) {
|
|
||||||
throw UtilsPackage.rethrow(e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private final ProtoBuf.StringTable strings;
|
|
||||||
private final ProtoBuf.QualifiedNameTable qualifiedNames;
|
|
||||||
|
|
||||||
public NameResolver(
|
|
||||||
@NotNull ProtoBuf.StringTable strings,
|
|
||||||
@NotNull ProtoBuf.QualifiedNameTable qualifiedNames
|
|
||||||
) {
|
|
||||||
this.strings = strings;
|
|
||||||
this.qualifiedNames = qualifiedNames;
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
@NotNull
|
||||||
public String getString(int index) {
|
Name getName(int index);
|
||||||
return strings.getString(index);
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
@NotNull
|
||||||
public Name getName(int index) {
|
ClassId getClassId(int index);
|
||||||
return Name.guess(strings.getString(index));
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
@NotNull
|
||||||
public ClassId getClassId(int index) {
|
FqName getFqName(int index);
|
||||||
LinkedList<String> packageFqName = new LinkedList<String>();
|
|
||||||
LinkedList<String> relativeClassName = new LinkedList<String>();
|
|
||||||
boolean local = false;
|
|
||||||
|
|
||||||
while (index != -1) {
|
|
||||||
QualifiedName proto = qualifiedNames.getQualifiedName(index);
|
|
||||||
String shortName = strings.getString(proto.getShortName());
|
|
||||||
switch (proto.getKind()) {
|
|
||||||
case CLASS:
|
|
||||||
relativeClassName.addFirst(shortName);
|
|
||||||
break;
|
|
||||||
case PACKAGE:
|
|
||||||
packageFqName.addFirst(shortName);
|
|
||||||
break;
|
|
||||||
case LOCAL:
|
|
||||||
relativeClassName.addFirst(shortName);
|
|
||||||
local = true;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
index = proto.getParentQualifiedName();
|
|
||||||
}
|
|
||||||
|
|
||||||
return new ClassId(FqName.fromSegments(packageFqName), FqName.fromSegments(relativeClassName), local);
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
public FqName getFqName(int index) {
|
|
||||||
QualifiedName qualifiedName = qualifiedNames.getQualifiedName(index);
|
|
||||||
Name shortName = getName(qualifiedName.getShortName());
|
|
||||||
if (!qualifiedName.hasParentQualifiedName()) {
|
|
||||||
return FqName.topLevel(shortName);
|
|
||||||
}
|
|
||||||
return getFqName(qualifiedName.getParentQualifiedName()).child(shortName);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
+107
@@ -0,0 +1,107 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2015 JetBrains s.r.o.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package org.jetbrains.kotlin.serialization.deserialization;
|
||||||
|
|
||||||
|
import org.jetbrains.annotations.NotNull;
|
||||||
|
import org.jetbrains.kotlin.name.ClassId;
|
||||||
|
import org.jetbrains.kotlin.name.FqName;
|
||||||
|
import org.jetbrains.kotlin.name.Name;
|
||||||
|
import org.jetbrains.kotlin.serialization.ProtoBuf;
|
||||||
|
import org.jetbrains.kotlin.utils.UtilsPackage;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.util.LinkedList;
|
||||||
|
|
||||||
|
import static org.jetbrains.kotlin.serialization.ProtoBuf.QualifiedNameTable.QualifiedName;
|
||||||
|
|
||||||
|
public class NameResolverImpl implements NameResolver {
|
||||||
|
@NotNull
|
||||||
|
public static NameResolverImpl read(@NotNull InputStream in) {
|
||||||
|
try {
|
||||||
|
ProtoBuf.StringTable simpleNames = ProtoBuf.StringTable.parseDelimitedFrom(in);
|
||||||
|
ProtoBuf.QualifiedNameTable qualifiedNames = ProtoBuf.QualifiedNameTable.parseDelimitedFrom(in);
|
||||||
|
return new NameResolverImpl(simpleNames, qualifiedNames);
|
||||||
|
}
|
||||||
|
catch (IOException e) {
|
||||||
|
throw UtilsPackage.rethrow(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private final ProtoBuf.StringTable strings;
|
||||||
|
private final ProtoBuf.QualifiedNameTable qualifiedNames;
|
||||||
|
|
||||||
|
public NameResolverImpl(
|
||||||
|
@NotNull ProtoBuf.StringTable strings,
|
||||||
|
@NotNull ProtoBuf.QualifiedNameTable qualifiedNames
|
||||||
|
) {
|
||||||
|
this.strings = strings;
|
||||||
|
this.qualifiedNames = qualifiedNames;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
@NotNull
|
||||||
|
public String getString(int index) {
|
||||||
|
return strings.getString(index);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
@NotNull
|
||||||
|
public Name getName(int index) {
|
||||||
|
return Name.guess(strings.getString(index));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
@NotNull
|
||||||
|
public ClassId getClassId(int index) {
|
||||||
|
LinkedList<String> packageFqName = new LinkedList<String>();
|
||||||
|
LinkedList<String> relativeClassName = new LinkedList<String>();
|
||||||
|
boolean local = false;
|
||||||
|
|
||||||
|
while (index != -1) {
|
||||||
|
QualifiedName proto = qualifiedNames.getQualifiedName(index);
|
||||||
|
String shortName = strings.getString(proto.getShortName());
|
||||||
|
switch (proto.getKind()) {
|
||||||
|
case CLASS:
|
||||||
|
relativeClassName.addFirst(shortName);
|
||||||
|
break;
|
||||||
|
case PACKAGE:
|
||||||
|
packageFqName.addFirst(shortName);
|
||||||
|
break;
|
||||||
|
case LOCAL:
|
||||||
|
relativeClassName.addFirst(shortName);
|
||||||
|
local = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
index = proto.getParentQualifiedName();
|
||||||
|
}
|
||||||
|
|
||||||
|
return new ClassId(FqName.fromSegments(packageFqName), FqName.fromSegments(relativeClassName), local);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
@NotNull
|
||||||
|
public FqName getFqName(int index) {
|
||||||
|
QualifiedName qualifiedName = qualifiedNames.getQualifiedName(index);
|
||||||
|
Name shortName = getName(qualifiedName.getShortName());
|
||||||
|
if (!qualifiedName.hasParentQualifiedName()) {
|
||||||
|
return FqName.topLevel(shortName);
|
||||||
|
}
|
||||||
|
return getFqName(qualifiedName.getParentQualifiedName()).child(shortName);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -20,7 +20,7 @@ import org.jetbrains.kotlin.descriptors.FunctionDescriptor
|
|||||||
import org.jetbrains.kotlin.serialization.ProtoBuf
|
import org.jetbrains.kotlin.serialization.ProtoBuf
|
||||||
import org.jetbrains.kotlin.serialization.deserialization.DeserializationContext
|
import org.jetbrains.kotlin.serialization.deserialization.DeserializationContext
|
||||||
import org.jetbrains.kotlin.serialization.deserialization.MemberDeserializer
|
import org.jetbrains.kotlin.serialization.deserialization.MemberDeserializer
|
||||||
import org.jetbrains.kotlin.serialization.deserialization.NameResolver
|
import org.jetbrains.kotlin.serialization.deserialization.NameResolverImpl
|
||||||
import org.jetbrains.kotlin.serialization.jvm.BitEncoding
|
import org.jetbrains.kotlin.serialization.jvm.BitEncoding
|
||||||
import org.jetbrains.kotlin.serialization.jvm.JvmProtoBufUtil
|
import org.jetbrains.kotlin.serialization.jvm.JvmProtoBufUtil
|
||||||
import kotlin.jvm.internal.KotlinCallable
|
import kotlin.jvm.internal.KotlinCallable
|
||||||
@@ -37,7 +37,7 @@ import kotlin.reflect.jvm.internal.getOrCreateModule
|
|||||||
public fun <R> Function<R>.reflect(): KFunction<R>? {
|
public fun <R> Function<R>.reflect(): KFunction<R>? {
|
||||||
val callable = javaClass.getAnnotation(KotlinCallable::class.java) ?: return null
|
val callable = javaClass.getAnnotation(KotlinCallable::class.java) ?: return null
|
||||||
val input = BitEncoding.decodeBytes(callable.data).inputStream()
|
val input = BitEncoding.decodeBytes(callable.data).inputStream()
|
||||||
val nameResolver = NameResolver.read(input)
|
val nameResolver = NameResolverImpl.read(input)
|
||||||
val proto = ProtoBuf.Callable.parseFrom(input, JvmProtoBufUtil.EXTENSION_REGISTRY)
|
val proto = ProtoBuf.Callable.parseFrom(input, JvmProtoBufUtil.EXTENSION_REGISTRY)
|
||||||
val moduleData = javaClass.getOrCreateModule()
|
val moduleData = javaClass.getOrCreateModule()
|
||||||
val context = DeserializationContext(moduleData.deserialization, nameResolver, moduleData.module,
|
val context = DeserializationContext(moduleData.deserialization, nameResolver, moduleData.module,
|
||||||
|
|||||||
+2
-1
@@ -30,6 +30,7 @@ import org.jetbrains.kotlin.idea.decompiler.textBuilder.DirectoryBasedKotlinJava
|
|||||||
import org.jetbrains.kotlin.name.FqName
|
import org.jetbrains.kotlin.name.FqName
|
||||||
import org.jetbrains.kotlin.psi.JetFile
|
import org.jetbrains.kotlin.psi.JetFile
|
||||||
import org.jetbrains.kotlin.serialization.deserialization.NameResolver
|
import org.jetbrains.kotlin.serialization.deserialization.NameResolver
|
||||||
|
import org.jetbrains.kotlin.serialization.deserialization.NameResolverImpl
|
||||||
import org.jetbrains.kotlin.serialization.js.KotlinJavascriptSerializedResourcePaths
|
import org.jetbrains.kotlin.serialization.js.KotlinJavascriptSerializedResourcePaths
|
||||||
import org.jetbrains.kotlin.serialization.js.toClassData
|
import org.jetbrains.kotlin.serialization.js.toClassData
|
||||||
import org.jetbrains.kotlin.serialization.js.toPackageData
|
import org.jetbrains.kotlin.serialization.js.toPackageData
|
||||||
@@ -57,7 +58,7 @@ public open class KotlinJavaScriptStubBuilder : ClsStubBuilder() {
|
|||||||
val stringsFile = moduleDirectory.findFileByRelativePath(stringsFileName)
|
val stringsFile = moduleDirectory.findFileByRelativePath(stringsFileName)
|
||||||
assert(stringsFile != null) { "strings file not found: $stringsFileName" }
|
assert(stringsFile != null) { "strings file not found: $stringsFileName" }
|
||||||
|
|
||||||
val nameResolver = NameResolver.read(ByteArrayInputStream(stringsFile!!.contentsToByteArray(false)))
|
val nameResolver = NameResolverImpl.read(ByteArrayInputStream(stringsFile!!.contentsToByteArray(false)))
|
||||||
val components = createStubBuilderComponents(file, packageFqName, nameResolver)
|
val components = createStubBuilderComponents(file, packageFqName, nameResolver)
|
||||||
|
|
||||||
if (isPackageHeader) {
|
if (isPackageHeader) {
|
||||||
|
|||||||
+1
-1
@@ -38,7 +38,7 @@ public class KotlinJavaScriptDeserializerForDecompiler(
|
|||||||
val stringsFileName = KotlinJavascriptSerializedResourcePaths.getStringTableFilePath(directoryPackageFqName)
|
val stringsFileName = KotlinJavascriptSerializedResourcePaths.getStringTableFilePath(directoryPackageFqName)
|
||||||
val stringsFile = moduleDirectory.findFileByRelativePath(stringsFileName)
|
val stringsFile = moduleDirectory.findFileByRelativePath(stringsFileName)
|
||||||
assert(stringsFile != null) { "strings file not found: $stringsFileName" }
|
assert(stringsFile != null) { "strings file not found: $stringsFileName" }
|
||||||
NameResolver.read(ByteArrayInputStream(stringsFile!!.contentsToByteArray(false)))
|
NameResolverImpl.read(ByteArrayInputStream(stringsFile!!.contentsToByteArray(false)))
|
||||||
}
|
}
|
||||||
|
|
||||||
private val metaFileFinder = DirectoryBasedKotlinJavaScriptMetaFileFinder(packageDirectory, directoryPackageFqName, nameResolver)
|
private val metaFileFinder = DirectoryBasedKotlinJavaScriptMetaFileFinder(packageDirectory, directoryPackageFqName, nameResolver)
|
||||||
|
|||||||
Reference in New Issue
Block a user