org.jetbrains.java.decompiler.main.plugins.JarPluginLoader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vineflower Show documentation

Modern Java & JVM language decompiler aiming to be as accurate as possible, with an emphasis on output quality.

The newest version!

package org.jetbrains.java.decompiler.main.plugins;

import java.io.*;
import java.net.URI;
import java.nio.file.*;
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;

// Finds plugins included in the main Vineflower Jar via Jar-In-Jar
public class JarPluginLoader {
  static List> PLUGIN_CLASSES = new ArrayList<>();

  public static void init() {
    try {
      File myFile = new File(JarPluginLoader.class.getProtectionDomain().getCodeSource().getLocation().toURI());

      // Ensure we are running out of a file
      if (myFile.exists() && !myFile.isDirectory() && myFile.getPath().endsWith(".jar")) {
        URI uri = URI.create("jar:" + myFile.toURI());

        Map env = new HashMap<>();

        FileSystem zipfs = FileSystems.newFileSystem(uri, env);

        Path pluginsDir = zipfs.getPath("META-INF", "plugins");
        if (!Files.exists(pluginsDir)) {
          zipfs.close();
          return;
        }

        try (Stream pluginStream = Files.list(pluginsDir)) {
          List plugins = pluginStream.collect(Collectors.toList());

          for (Path pluginJar : plugins) {
            FileSystem pluginfs = FileSystems.newFileSystem(pluginJar, (ClassLoader) null);

            Path file = pluginfs.getPath("META-INF", "services", "org.jetbrains.java.decompiler.api.plugin.Plugin");
            if (!Files.exists(file)) {
              pluginfs.close();
              continue;
            }

            String pluginClass = Files.readString(file);

            InJarClassLoader loader = new InJarClassLoader(pluginfs, JarPluginLoader.class.getClassLoader());

            Class clazz = Class.forName(pluginClass, true, loader);
            PLUGIN_CLASSES.add(clazz);
          }
        }
      }

    } catch (Exception e) {
      // TODO: use decompiler logger, but it's not ready yet. potentially set up a logger defer system?
//      DecompilerContext.getLogger().writeMessage("Couldn't load plugins!", IFernflowerLogger.Severity.INFO, e);
      e.printStackTrace();
    }
  }
}