proguard.resources.kotlinmodule.visitor.KotlinModulePrinter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of proguard-core Show documentation
ProGuardCORE is a free library to read, analyze, modify, and write Java class files.
There is a newer version: 9.1.6
/*
 * ProGuardCORE -- library to process Java bytecode.
 *
 * Copyright (c) 2002-2020 Guardsquare NV
 *
 * 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 proguard.resources.kotlinmodule.visitor;

import static proguard.classfile.kotlin.visitor.KotlinMetadataPrinter.DEFAULT_MISSING_REF_INDICATOR;

import java.io.PrintWriter;
import java.util.Map;
import proguard.resources.file.visitor.*;
import proguard.resources.kotlinmodule.*;

/** Print Kotlin modules. */
public class KotlinModulePrinter
    implements ResourceFileVisitor, KotlinModuleVisitor, KotlinModulePackageVisitor {
  private static final String INDENTATION = "  ";

  private final PrintWriter pw;

  private int indentation;

  private String missingRefIndicator;

  public KotlinModulePrinter() {
    this(new PrintWriter(System.out, true), DEFAULT_MISSING_REF_INDICATOR);
  }

  public KotlinModulePrinter(PrintWriter pw, String missingRefIndicator) {
    this.pw = pw;
    this.missingRefIndicator = missingRefIndicator;
  }

  // Implementations for ResourceFileVisitor.

  @Override
  public void visitKotlinModule(KotlinModule kotlinModule) {
    println("_____________________________________________________________________");
    println("[MODL] " + kotlinModule.name + " from file(" + kotlinModule.fileName + ")");

    indent();
    kotlinModule.modulePackagesAccept(this);
    outdent();
  }

  public void visitKotlinModulePackage(
      KotlinModule kotlinModule, KotlinModulePackage kotlinModulePart) {
    println("[MPKG] \"" + kotlinModulePart.fqName + "\"");

    indent();
    for (int i = 0; i < kotlinModulePart.fileFacadeNames.size(); i++) {
      String fileFacadeName = kotlinModulePart.fileFacadeNames.get(i);
      println(
          "[FFAC] "
              + hasRefIndicator(kotlinModulePart.referencedFileFacades.get(i))
              + fileFacadeName);
    }

    kotlinModulePart.multiFileClassParts.values().stream()
        .distinct()
        .forEachOrdered(
            (multiFileFacadeName) -> {
              println("[MFAC] " + multiFileFacadeName);

              indent();
              kotlinModulePart.multiFileClassParts.entrySet().stream()
                  .filter(e -> e.getValue().equals(multiFileFacadeName))
                  .map(Map.Entry::getKey)
                  .forEachOrdered(
                      multiFilePartName ->
                          println(
                              "[MPRT] "
                                  + hasRefIndicator(
                                      kotlinModulePart.referencedMultiFileParts.get(
                                          multiFilePartName))
                                  + multiFilePartName));
              outdent();
            });

    outdent();
  }

  // Small utility methods.

  private void indent() {
    indentation++;
  }

  private void outdent() {
    indentation--;
  }

  private void println(String string) {
    print(string);
    println();
  }

  private void print(String string) {
    for (int index = 0; index < indentation; index++) {
      pw.print(INDENTATION);
    }

    pw.print(string);
  }

  private void println() {
    pw.println();
  }

  private String hasRefIndicator(Object arg) {
    return arg == null ? this.missingRefIndicator : "";
  }
}