Share Kotlin metadata strings

base/src/main/java/proguard/classfile/util/StringSharer.java

@@ -22,14 +22,18 @@
 import proguard.classfile.attribute.visitor.AttributeVisitor;
 import proguard.classfile.constant.*;
 import proguard.classfile.constant.visitor.ConstantVisitor;
+import proguard.classfile.kotlin.KotlinClassKindMetadata;
+import proguard.classfile.kotlin.KotlinMetadata;
+import proguard.classfile.kotlin.visitor.KotlinMetadataVisitor;
 import proguard.classfile.visitor.ClassVisitor;
 
 /**
  * This {@link ClassVisitor} shares strings in the class files that it visits.
  *
  * @author Eric Lafortune
  */
-public class StringSharer implements ClassVisitor, ConstantVisitor, AttributeVisitor {
+public class StringSharer
+    implements ClassVisitor, ConstantVisitor, AttributeVisitor, KotlinMetadataVisitor {
   // A fields acting as an argument for the visitor methods.
   private String name;
   private String type;
@@ -50,6 +54,9 @@ public void visitProgramClass(ProgramClass programClass) {
     // Replace attribute name strings in the constant pool by internalized
     // strings.
     programClass.attributesAccept(this);
+
+    // Replace strings in Kotlin metadata.
+    programClass.kotlinMetadataAccept(this);
   }
 
   @Override
@@ -73,6 +80,9 @@ public void visitLibraryClass(LibraryClass libraryClass) {
         }
       }
     }
+
+    // Replace strings in Kotlin metadata.
+    libraryClass.kotlinMetadataAccept(this);
   }
 
   // Implementations for ConstantVisitor.
@@ -154,4 +164,25 @@ public void visitAnyAttribute(Clazz clazz, Attribute attribute) {
     name = attribute.getAttributeName(clazz).intern();
     clazz.constantPoolEntryAccept(attribute.u2attributeNameIndex, this);
   }
+
+  // Implementations for KotlinMetadataVisitor.
+
+  public void visitAnyKotlinMetadata(Clazz clazz, KotlinMetadata kotlinMetadata) {}
+
+  public void visitKotlinClassMetadata(
+      Clazz clazz, KotlinClassKindMetadata kotlinClassKindMetadata) {
+    if (kotlinClassKindMetadata.referencedClass != null) {
+      kotlinClassKindMetadata.className = kotlinClassKindMetadata.referencedClass.getName();
+    }
+
+    if (kotlinClassKindMetadata.referencedCompanionClass != null) {
+      kotlinClassKindMetadata.companionObjectName =
+          kotlinClassKindMetadata.referencedCompanionField.getName(clazz);
+    }
+
+    if (kotlinClassKindMetadata.anonymousObjectOriginClass != null) {
+      kotlinClassKindMetadata.anonymousObjectOriginName =
+          kotlinClassKindMetadata.anonymousObjectOriginClass.getName();
+    }
+  }
 }