org.jetbrains.java.decompiler.struct.SingleFileContextSource Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of quiltflower Show documentation
Modern Java & JVM language decompiler aiming to be as accurate as possible, with an emphasis on output quality.
The newest version!
// Copyright 2000-2022 JetBrains s.r.o. and ForgeFlower contributors Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package org.jetbrains.java.decompiler.struct;

import org.jetbrains.java.decompiler.main.extern.IBytecodeProvider;
import org.jetbrains.java.decompiler.main.extern.IContextSource;
import org.jetbrains.java.decompiler.main.extern.IResultSaver;
import org.jetbrains.java.decompiler.util.DataInputFullStream;
import org.jetbrains.java.decompiler.util.InterpreterUtil;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;

// Only used for matching existing behavior, can be bad
class SingleFileContextSource implements IContextSource {
  private final File file;
  private final String qualifiedName;
  private final byte[] contents;

  @SuppressWarnings("deprecation")
  public SingleFileContextSource(final IBytecodeProvider legacyProvider, final File singleFile) throws IOException {
    this.file = singleFile;
    // A "fake" file could be provided via legacyProvider
    if (!singleFile.isFile() && legacyProvider == null) {
      this.contents = null;
      this.qualifiedName = null;
    } else {
      this.contents = legacyProvider == null ? InterpreterUtil.getBytes(singleFile) : legacyProvider.getBytecode(singleFile.getAbsolutePath(), null);

      if (this.contents != null && singleFile.getName().endsWith(CLASS_SUFFIX)) {
        try (final DataInputFullStream is = new DataInputFullStream(this.contents)) {
          var clazz = StructClass.create(is, false);
          this.qualifiedName = clazz.qualifiedName;
        }
      } else {
        this.qualifiedName = null;
      }
    }
  }

  @Override
  public String getName() {
    return "file " + this.file;
  }

  @Override
  public Entries getEntries() {
    if (this.contents == null) {
      return Entries.EMPTY;
    } else if (this.file.getName().endsWith(CLASS_SUFFIX)) {
      return new Entries(List.of(Entry.atBase(this.qualifiedName)), List.of(), List.of());
    } else {
      return new Entries(List.of(), List.of(), List.of(Entry.atBase(this.file.getName())));
    }
  }

  @Override
  public InputStream getInputStream(String resource) throws IOException {
    return new ByteArrayInputStream(this.contents);
  }

  @Override
  public IOutputSink createOutputSink(IResultSaver saver) {
    return new IOutputSink() {
      @Override
      public void close() throws IOException {
      }

      @Override
      public void begin() {
      }

      @Override
      public void acceptOther(String path) {
        saver.copyFile(file.getAbsolutePath(), "", path);
      }

      @Override
      public void acceptDirectory(String directory) {
        // not used
      }

      @Override
      public void acceptClass(String qualifiedName, String fileName, String content, int[] mapping) {
        saver.saveClassFile("", qualifiedName, file.getName().substring(0, file.getName().length() - CLASS_SUFFIX.length()) + ".java", content, mapping);
      }
    };
  }


}