From cf1cf25cb4c4ec25745b7ae698e8cece4d070e90 Mon Sep 17 00:00:00 2001
From: Evgeny Gerashchenko <Evgeny.Gerashchenko@jetbrains.com>
Date: Thu, 13 Feb 2014 20:39:53 +0400
Subject: [PATCH] Rendering class or package proto when JPS test fails.

Original commit: 5c8253c09954d99916ae6d5d12f24c39c0a48621
---
 jps/jps-plugin/jps-plugin.iml                 |  4 ++
 .../jet/jps/build/classFilesComparison.kt     | 40 +++++++++++++++----
 2 files changed, 37 insertions(+), 7 deletions(-)

diff --git a/jps/jps-plugin/jps-plugin.iml b/jps/jps-plugin/jps-plugin.iml
index 37590a836e4..1b4d326d32d 100644
--- a/jps/jps-plugin/jps-plugin.iml
+++ b/jps/jps-plugin/jps-plugin.iml
@@ -16,6 +16,10 @@
     <orderEntry type="library" scope="TEST" name="idea-full" level="project" />
     <orderEntry type="module" module-name="backend" scope="TEST" />
     <orderEntry type="module" module-name="descriptors" scope="TEST" />
+    <orderEntry type="module" module-name="frontend.java" scope="TEST" />
+    <orderEntry type="module" module-name="descriptor.loader.java" scope="TEST" />
+    <orderEntry type="module" module-name="serialization.java" scope="TEST" />
+    <orderEntry type="module" module-name="serialization" scope="TEST" />
     <orderEntry type="module" module-name="compiler-tests" scope="TEST" />
   </component>
 </module>
diff --git a/jps/jps-plugin/test/org/jetbrains/jet/jps/build/classFilesComparison.kt b/jps/jps-plugin/test/org/jetbrains/jet/jps/build/classFilesComparison.kt
index 8cd431c2296..6af1c8a34d2 100644
--- a/jps/jps-plugin/test/org/jetbrains/jet/jps/build/classFilesComparison.kt
+++ b/jps/jps-plugin/test/org/jetbrains/jet/jps/build/classFilesComparison.kt
@@ -28,10 +28,6 @@ import com.google.common.hash.Hashing
 import com.intellij.openapi.util.io.FileUtil
 import com.google.common.collect.Sets
 import java.util.HashSet
-import com.intellij.openapi.vfs.local.CoreLocalFileSystem
-import org.jetbrains.jet.lang.resolve.kotlin.VirtualFileKotlinClass
-import org.jetbrains.jet.storage.LockBasedStorageManager
-import org.jetbrains.jet.lang.resolve.kotlin.header.ReadKotlinClassHeaderAnnotationVisitor
 import org.jetbrains.jet.lang.resolve.kotlin.header.KotlinClassHeader
 import org.jetbrains.jet.descriptors.serialization.BitEncoding
 import org.jetbrains.jet.descriptors.serialization.DebugJavaProtoBuf
@@ -39,6 +35,7 @@ import com.google.protobuf.ExtensionRegistry
 import java.io.ByteArrayInputStream
 import org.jetbrains.jet.descriptors.serialization.DebugProtoBuf
 import java.util.Arrays
+import org.jetbrains.jet.lang.resolve.kotlin.VirtualFileKotlinClass
 
 fun File.hash() = Files.hash(this, Hashing.crc32())
 
@@ -114,13 +111,42 @@ fun assertEqualDirectories(expected: File, actual: File, ignore: (File) -> Boole
 
 fun classFileToString(classFile: File): String {
     val out = StringWriter()
-    val traceVisitor = TraceClassVisitor(PrintWriter(out))
-    ClassReader(classFile.readBytes()).accept(traceVisitor, 0)
 
-    // TODO serialize protobuf
+    val classBytes = classFile.readBytes()
+
+    val traceVisitor = TraceClassVisitor(PrintWriter(out))
+    ClassReader(classBytes).accept(traceVisitor, 0)
+
+    val classHeader = VirtualFileKotlinClass.readClassHeader(classBytes)
+
+    val annotationDataEncoded = classHeader?.annotationData
+    if (annotationDataEncoded != null) {
+        ByteArrayInputStream(BitEncoding.decodeBytes(annotationDataEncoded)).use {
+            input ->
+
+        out.write("\n------ simpleNames proto -----\n${DebugProtoBuf.SimpleNameTable.parseDelimitedFrom(input)}")
+        out.write("\n------ qualifiedNames proto -----\n${DebugProtoBuf.QualifiedNameTable.parseDelimitedFrom(input)}")
+
+        when (classHeader!!.kind) {
+            KotlinClassHeader.Kind.PACKAGE_FACADE ->
+                out.write("\n------ package proto -----\n${DebugProtoBuf.Package.parseFrom(input, getExtensionRegistry())}")
+
+                KotlinClassHeader.Kind.CLASS ->
+                    out.write("\n------ class proto -----\n${DebugProtoBuf.Class.parseFrom(input, getExtensionRegistry())}")
+                else -> throw IllegalStateException()
+            }
+        }
+    }
+
     return out.toString()
 }
 
+fun getExtensionRegistry(): ExtensionRegistry {
+    val registry = ExtensionRegistry.newInstance()!!
+    DebugJavaProtoBuf.registerAllExtensions(registry)
+    return registry
+}
+
 fun fileToStringRepresentation(file: File): String {
     return when {
         file.name.endsWith(".class") -> {