feat(plugins): allow to load classes using input stream or byte array in jadx-input plugin (#1457)

jadx-plugins/jadx-java-input/src/test/java/jadx/plugins/input/java/CustomLoadTest.java

@@ -0,0 +1,129 @@
+package jadx.plugins.input.java;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import jadx.api.JadxArgs;
+import jadx.api.JadxDecompiler;
+import jadx.api.plugins.input.data.ILoadResult;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.jupiter.api.Assertions.fail;
+
+class CustomLoadTest {
+
+	private JadxDecompiler jadx;
+
+	@BeforeEach
+	void init() {
+		jadx = new JadxDecompiler(new JadxArgs());
+	}
+
+	@AfterEach
+	void close() {
+		jadx.close();
+	}
+
+	@Test
+	void loadFiles() {
+		List<Path> files = Stream.of("HelloWorld.class", "HelloWorld$HelloInner.class")
+				.map(this::getSample)
+				.collect(Collectors.toList());
+		ILoadResult loadResult = JavaInputPlugin.loadClassFiles(files);
+		loadDecompiler(loadResult);
+		assertThat(jadx.getClassesWithInners())
+				.hasSize(2)
+				.satisfiesOnlyOnce(cls -> assertThat(cls.getName()).isEqualTo("HelloWorld"))
+				.satisfiesOnlyOnce(cls -> assertThat(cls.getName()).isEqualTo("HelloInner"));
+	}
+
+	@Test
+	void loadFromInputStream() throws IOException {
+		String fileName = "HelloWorld$HelloInner.class";
+		try (InputStream in = Files.newInputStream(getSample(fileName))) {
+			ILoadResult loadResult = JavaInputPlugin.loadFromInputStream(in, fileName);
+			loadDecompiler(loadResult);
+			assertThat(jadx.getClassesWithInners())
+					.hasSize(1)
+					.satisfiesOnlyOnce(cls -> assertThat(cls.getName()).isEqualTo("HelloWorld$HelloInner"));
+
+			System.out.println(jadx.getClassesWithInners().get(0).getCode());
+		}
+	}
+
+	@Test
+	void loadSingleClass() throws IOException {
+		String fileName = "HelloWorld.class";
+		byte[] content = Files.readAllBytes(getSample(fileName));
+		ILoadResult loadResult = JavaInputPlugin.loadSingleClass(content, fileName);
+		loadDecompiler(loadResult);
+		assertThat(jadx.getClassesWithInners())
+				.hasSize(1)
+				.satisfiesOnlyOnce(cls -> assertThat(cls.getName()).isEqualTo("HelloWorld"));
+
+		System.out.println(jadx.getClassesWithInners().get(0).getCode());
+	}
+
+	@Test
+	void load() {
+		ILoadResult loadResult = JavaInputPlugin.load(loader -> {
+			List<JavaClassReader> inputs = new ArrayList<>(2);
+			try {
+				String hello = "HelloWorld.class";
+				byte[] content = Files.readAllBytes(getSample(hello));
+				inputs.add(loader.loadClass(content, hello));
+
+				String helloInner = "HelloWorld$HelloInner.class";
+				InputStream in = Files.newInputStream(getSample(helloInner));
+				inputs.addAll(loader.loadInputStream(in, helloInner));
+			} catch (Exception e) {
+				fail(e);
+			}
+			return inputs;
+		});
+		loadDecompiler(loadResult);
+		assertThat(jadx.getClassesWithInners())
+				.hasSize(2)
+				.satisfiesOnlyOnce(cls -> assertThat(cls.getName()).isEqualTo("HelloWorld"))
+				.satisfiesOnlyOnce(cls -> {
+					assertThat(cls.getName()).isEqualTo("HelloInner");
+					assertThat(cls.getCode()).isEqualTo(""); // no code for moved inner class
+				});
+
+		assertThat(jadx.getClasses())
+				.hasSize(1)
+				.satisfiesOnlyOnce(cls -> assertThat(cls.getName()).isEqualTo("HelloWorld"))
+				.satisfiesOnlyOnce(cls -> assertThat(cls.getInnerClasses()).hasSize(1)
+						.satisfiesOnlyOnce(inner -> assertThat(inner.getName()).isEqualTo("HelloInner")));
+
+		jadx.getClassesWithInners().forEach(cls -> System.out.println(cls.getCode()));
+	}
+
+	public void loadDecompiler(ILoadResult load) {
+		try {
+			jadx.addCustomLoad(load);
+			jadx.load();
+		} catch (Exception e) {
+			fail(e);
+		}
+	}
+
+	public Path getSample(String name) {
+		try {
+			return Paths.get(ClassLoader.getSystemResource("samples/" + name).toURI());
+		} catch (Exception e) {
+			return fail(e);
+		}
+	}
+}