proguard.resources.kotlinmodule.io.KotlinModuleWriter 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-2023 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.io;

import java.io.IOException;
import java.io.OutputStream;
import java.util.function.BiConsumer;
import kotlin.metadata.jvm.JvmMetadataVersion;
import kotlin.metadata.jvm.KmModule;
import kotlin.metadata.jvm.KmPackageParts;
import kotlin.metadata.jvm.KotlinModuleMetadata;
import proguard.classfile.util.ClassUtil;
import proguard.resources.file.visitor.ResourceFileVisitor;
import proguard.resources.kotlinmodule.KotlinModule;

/**
 * @author James Hamilton
 */
public class KotlinModuleWriter implements ResourceFileVisitor {
  private final OutputStream outputStream;
  private final BiConsumer errorHandler;

  public KotlinModuleWriter(OutputStream outputStream) {
    this(null, outputStream);
  }

  public KotlinModuleWriter(
      BiConsumer errorHandler, OutputStream outputStream) {
    this.errorHandler = errorHandler;
    this.outputStream = outputStream;
  }

  @Override
  public void visitKotlinModule(KotlinModule kotlinModule) {
    try {
      KmModule kmModule = new KmModule();

      kotlinModule.modulePackagesAccept(
          (module, modulePackage) ->
              kmModule
                  .getPackageParts()
                  .put(
                      ClassUtil.externalClassName(modulePackage.fqName),
                      new KmPackageParts(
                          modulePackage.fileFacadeNames, modulePackage.multiFileClassParts)));

      // TODO: Support module optional annotations in our model.
      // kmModule.getOptionalAnnotationClasses();

      byte[] transformedBytes =
          new KotlinModuleMetadata(
                  kmModule,
                  kotlinModule.version.canBeWritten()
                      ? new JvmMetadataVersion(kotlinModule.version.toArray())
                      : JvmMetadataVersion.LATEST_STABLE_SUPPORTED)
              .write();
      outputStream.write(transformedBytes);
    } catch (IOException e) {
      if (this.errorHandler != null) {
        this.errorHandler.accept(kotlinModule, e.getMessage());
      } else {
        throw new RuntimeException("Error while writing module file", e);
      }
    }
  }
}