Introduce ClassDataProvider

core/deserialization/src/org/jetbrains/kotlin/serialization/deserialization/ClassDataFinder.java

@@ -19,9 +19,8 @@ package org.jetbrains.kotlin.serialization.deserialization;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 import org.jetbrains.kotlin.name.ClassId;
-import org.jetbrains.kotlin.serialization.ClassData;
 
 public interface ClassDataFinder {
     @Nullable
-    ClassData findClassData(@NotNull ClassId classId);
+    ClassDataProvider findClassData(@NotNull ClassId classId);
 }

core/deserialization/src/org/jetbrains/kotlin/serialization/deserialization/ClassDataProvider.kt

@@ -0,0 +1,25 @@
+/*
+ * 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.kotlin.descriptors.SourceElement
+import org.jetbrains.kotlin.serialization.ClassData
+
+public class ClassDataProvider(
+        public val classData: ClassData,
+        public val sourceElement: SourceElement = SourceElement.NO_SOURCE
+)

core/deserialization/src/org/jetbrains/kotlin/serialization/deserialization/ClassDeserializer.kt

@@ -33,7 +33,11 @@ public class ClassDeserializer(private val components: DeserializationComponents
         val classId = key.classId
         components.fictitiousClassDescriptorFactory.createClass(classId)?.let { return it }
 
-        val classData = key.classData ?: components.classDataFinder.findClassData(classId) ?: return null
+        var classData = key.classData ?: run {
+            val classDataProvider = components.classDataFinder.findClassData(classId)
+            classDataProvider?.classData
+        } ?: return null
+
         val outerContext = if (classId.isNestedClass()) {
             val outerClass = deserializeClass(classId.getOuterClassId()) as? DeserializedClassDescriptor ?: return null
 

core/deserialization/src/org/jetbrains/kotlin/serialization/deserialization/ResourceLoadingClassDataFinder.kt

@@ -28,15 +28,16 @@ public open class ResourceLoadingClassDataFinder(
         private val serializedResourcePaths: SerializedResourcePaths,
         private val loadResource: (path: String) -> InputStream?
 ) : ClassDataFinder {
-    override fun findClassData(classId: ClassId): ClassData? {
+    override fun findClassData(classId: ClassId): ClassDataProvider? {
         val packageFragment = packageFragmentProvider.getPackageFragments(classId.getPackageFqName()).singleOrNull()
                                       as? DeserializedPackageFragment ?: return null
 
         val stream = loadResource(serializedResourcePaths.getClassMetadataPath(classId)) ?: return null
 
-        return ClassData(
+        val classData = ClassData(
                 packageFragment.nameResolver,
                 ProtoBuf.Class.parseFrom(stream, serializedResourcePaths.extensionRegistry)
         )
+        return ClassDataProvider(classData)
     }
 }