Move common deserialization components to :core:metadata and :core:metadata.jvm

2020-09-02 14:42:32 +03:00
parent cdab00fdc8
commit 5d4d07066b
8 changed files with 3 additions and 2 deletions
@@ -0,0 +1,37 @@
+/*
+ * 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
+
+import org.jetbrains.kotlin.metadata.ProtoBuf
+import org.jetbrains.kotlin.protobuf.ExtensionRegistryLite
+import org.jetbrains.kotlin.protobuf.GeneratedMessageLite.GeneratedExtension
+
+open class SerializerExtensionProtocol(
+    val extensionRegistry: ExtensionRegistryLite,
+    val packageFqName: GeneratedExtension<ProtoBuf.Package, Int>,
+    val constructorAnnotation: GeneratedExtension<ProtoBuf.Constructor, List<ProtoBuf.Annotation>>,
+    val classAnnotation: GeneratedExtension<ProtoBuf.Class, List<ProtoBuf.Annotation>>,
+    val functionAnnotation: GeneratedExtension<ProtoBuf.Function, List<ProtoBuf.Annotation>>,
+    val propertyAnnotation: GeneratedExtension<ProtoBuf.Property, List<ProtoBuf.Annotation>>,
+    val propertyGetterAnnotation: GeneratedExtension<ProtoBuf.Property, List<ProtoBuf.Annotation>>,
+    val propertySetterAnnotation: GeneratedExtension<ProtoBuf.Property, List<ProtoBuf.Annotation>>,
+    val enumEntryAnnotation: GeneratedExtension<ProtoBuf.EnumEntry, List<ProtoBuf.Annotation>>,
+    val compileTimeValue: GeneratedExtension<ProtoBuf.Property, ProtoBuf.Annotation.Argument.Value>,
+    val parameterAnnotation: GeneratedExtension<ProtoBuf.ValueParameter, List<ProtoBuf.Annotation>>,
+    val typeAnnotation: GeneratedExtension<ProtoBuf.Type, List<ProtoBuf.Annotation>>,
+    val typeParameterAnnotation: GeneratedExtension<ProtoBuf.TypeParameter, List<ProtoBuf.Annotation>>
+)
@@ -0,0 +1,18 @@
+/*
+ * Copyright 2010-2020 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.serialization.deserialization
+
+import org.jetbrains.kotlin.descriptors.SourceElement
+import org.jetbrains.kotlin.metadata.ProtoBuf
+import org.jetbrains.kotlin.metadata.deserialization.BinaryVersion
+import org.jetbrains.kotlin.metadata.deserialization.NameResolver
+
+data class ClassData(
+    val nameResolver: NameResolver,
+    val classProto: ProtoBuf.Class,
+    val metadataVersion: BinaryVersion,
+    val sourceElement: SourceElement
+)
@@ -0,0 +1,26 @@
+/*
+ * 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.annotations.Nullable;
+import org.jetbrains.kotlin.name.ClassId;
+
+public interface ClassDataFinder {
+    @Nullable
+    ClassData findClassData(@NotNull ClassId classId);
+}
@@ -0,0 +1,17 @@
+/*
+ * 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.serialization.deserialization
+
+import org.jetbrains.kotlin.metadata.deserialization.NameResolver
+import org.jetbrains.kotlin.name.ClassId
+import org.jetbrains.kotlin.name.Name
+
+fun NameResolver.getClassId(index: Int): ClassId {
+    return ClassId.fromString(getQualifiedClassName(index), isLocalClassName(index))
+}
+
+fun NameResolver.getName(index: Int): Name =
+    Name.guessByFirstCharacter(getString(index))
@@ -0,0 +1,42 @@
+/*
+ * Copyright 2010-2016 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.kotlin.descriptors.SourceElement
+import org.jetbrains.kotlin.metadata.ProtoBuf
+import org.jetbrains.kotlin.metadata.deserialization.BinaryVersion
+import org.jetbrains.kotlin.metadata.deserialization.NameResolver
+import org.jetbrains.kotlin.name.ClassId
+
+class ProtoBasedClassDataFinder(
+    proto: ProtoBuf.PackageFragment,
+    private val nameResolver: NameResolver,
+    private val metadataVersion: BinaryVersion,
+    private val classSource: (ClassId) -> SourceElement = { SourceElement.NO_SOURCE }
+) : ClassDataFinder {
+    private val classIdToProto =
+        proto.class_List.associateBy { klass ->
+            nameResolver.getClassId(klass.fqName)
+        }
+
+    val allClassIds: Collection<ClassId> get() = classIdToProto.keys
+
+    override fun findClassData(classId: ClassId): ClassData? {
+        val classProto = classIdToProto[classId] ?: return null
+        return ClassData(nameResolver, classProto, metadataVersion, classSource(classId))
+    }
+}
@@ -0,0 +1,40 @@
+/*
+ * Copyright 2010-2020 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.serialization.deserialization.builtins
+
+import org.jetbrains.kotlin.metadata.builtins.BuiltInsProtoBuf
+import org.jetbrains.kotlin.name.FqName
+import org.jetbrains.kotlin.protobuf.ExtensionRegistryLite
+import org.jetbrains.kotlin.serialization.SerializerExtensionProtocol
+
+object BuiltInSerializerProtocol : SerializerExtensionProtocol(
+    ExtensionRegistryLite.newInstance().apply(BuiltInsProtoBuf::registerAllExtensions),
+    BuiltInsProtoBuf.packageFqName,
+    BuiltInsProtoBuf.constructorAnnotation,
+    BuiltInsProtoBuf.classAnnotation,
+    BuiltInsProtoBuf.functionAnnotation,
+    BuiltInsProtoBuf.propertyAnnotation,
+    BuiltInsProtoBuf.propertyGetterAnnotation,
+    BuiltInsProtoBuf.propertySetterAnnotation,
+    BuiltInsProtoBuf.enumEntryAnnotation,
+    BuiltInsProtoBuf.compileTimeValue,
+    BuiltInsProtoBuf.parameterAnnotation,
+    BuiltInsProtoBuf.typeAnnotation,
+    BuiltInsProtoBuf.typeParameterAnnotation
+) {
+    const val BUILTINS_FILE_EXTENSION = "kotlin_builtins"
+
+    fun getBuiltInsFilePath(fqName: FqName): String =
+        fqName.asString().replace('.', '/') + "/" + getBuiltInsFileName(
+            fqName
+        )
+
+    fun getBuiltInsFileName(fqName: FqName): String =
+        shortName(fqName) + "." + BUILTINS_FILE_EXTENSION
+
+    private fun shortName(fqName: FqName): String =
+        if (fqName.isRoot) "default-package" else fqName.shortName().asString()
+}