Support injecting userdev specific files

Author: LexManos

src/main/java/net/minecraftforge/mcmaven/impl/repo/forge/InjectTask.java

@@ -22,15 +22,13 @@
 public final class InjectTask implements Task {
     private final File build;
     private final Artifact name;
-    private final Cache cache;
     private final Patcher patcher;
     private final Mappings mappings;
     private final Task task;
 
     InjectTask(File build, Cache cache, Artifact name, Patcher patcher, Task input, Mappings mappings) {
         this.build = mappings.getFolder(build);
         this.name = name;
-        this.cache = cache;
         this.patcher = patcher;
         this.mappings = mappings;
         this.task = this.injectData(input);
@@ -53,7 +51,7 @@ public String name() {
 
     private Task injectData(Task input) {
         return Task.named("injectData[" + this.name.getName() + "][" + mappings + ']',
-            Task.deps(input),
+            Task.deps(input, this.patcher.filterBinaryInjections()),
             () -> injectDataImpl(input, new File(this.build, "injected.jar"))
         );
     }
@@ -65,13 +63,10 @@ private File injectDataImpl(Task inputTask, File outputJar) {
         cache.add("recompiled", recompiledJar);
 
         var universals = new ArrayList<File>();
-        for (var p : this.patcher.getStack()) {
-            if (p.config.universal != null) {
-                var universal = this.cache.maven().download(Artifact.from(p.config.universal));
-                universals.add(universal);
-                cache.add(universal);
-            }
-        }
+        var injectTask = this.patcher.filterBinaryInjections();
+        var inject = injectTask != null ? injectTask.execute() : null;
+        if (inject != null)
+            cache.add("inject", inject);
 
         if (Mavenizer.checkCache(outputJar, cache))
             return outputJar;
@@ -80,8 +75,10 @@ private File injectDataImpl(Task inputTask, File outputJar) {
             var jars = new ArrayList<>(universals);
 
             jars.add(recompiledJar);
+            if (inject != null)
+                jars.add(inject);
 
-            FileUtils.mergeJars(outputJar, true, (file, name) -> file == recompiledJar || !name.endsWith(".class"), jars.toArray(File[]::new));
+            FileUtils.mergeJars(outputJar, true, (file, name) -> (file == recompiledJar || file == inject) || !name.endsWith(".class"), jars.toArray(File[]::new));
         } catch (IOException e) {
             return Util.sneak(e);
         }

src/main/java/net/minecraftforge/mcmaven/impl/repo/forge/Patcher.java

@@ -5,9 +5,11 @@
 package net.minecraftforge.mcmaven.impl.repo.forge;
 
 import java.io.File;
+import java.io.FileInputStream;
 import java.io.FileNotFoundException;
 import java.io.FileOutputStream;
 import java.io.IOException;
+import java.nio.charset.StandardCharsets;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashMap;
@@ -19,7 +21,10 @@
 import java.util.function.Predicate;
 import java.util.function.Supplier;
 import java.util.stream.Collectors;
+import java.util.zip.ZipEntry;
 import java.util.zip.ZipFile;
+import java.util.zip.ZipInputStream;
+import java.util.zip.ZipOutputStream;
 
 import io.codechicken.diffpatch.cli.PatchOperation;
 import io.codechicken.diffpatch.util.LogLevel;
@@ -70,6 +75,7 @@ public class Patcher implements Supplier<Task> {
     private final Task downloadSources;
     private final Task predecomp;
     private final Task last;
+    private final Task filterBinaryInjections;
 
     /**
      * Creates a new Patcher for the given Forge repo.
@@ -162,6 +168,15 @@ public class Patcher implements Supplier<Task> {
         }
 
         this.last = last;
+
+        var binaryInjections = new ArrayList<Patcher>();
+        for (var p : this.getStack()) {
+            if (p.config.inject != null)
+                binaryInjections.add(p);
+        }
+
+        this.filterBinaryInjections = binaryInjections.isEmpty() ? null :
+            Task.named("filterBinaryInjections[" + this.name.getName() + ']', this::filterBinaryInjectionsImpl);
     }
 
     private RuntimeException except(String message) {
@@ -215,6 +230,14 @@ public MCPSide getMCPSide() {
         return this.mcpSide == null ? this.parent.getMCPSide() : this.mcpSide;
     }
 
+    public String getName() {
+        return this.name.getName();
+    }
+
+    public Artifact getArtifact() {
+        return this.name;
+    }
+
     public String getDataHash() {
         return this.dataHash;
     }
@@ -675,4 +698,71 @@ private File injectSourcesImpl(Task inputTask, File output) {
         cache.save();
         return output;
     }
+
+    public @Nullable Task filterBinaryInjections() {
+        return this.filterBinaryInjections;
+    }
+
+    private File filterBinaryInjectionsImpl() {
+        var output = new File(this.build, "binary-injections.jar");
+        var cache = Util.cache(output);
+        cache.addKnown("data", this.getDataHash());
+        for (var p : getStack(false)) {
+            if (p.config.inject != null)
+                cache.addKnown("parent-" + p.getName(), p.getDataHash());
+        }
+
+        if (Mavenizer.checkCache(output, cache))
+            return output;
+
+        if (output.getParentFile() != null)
+            output.getParentFile().mkdirs();
+
+        try (var zos = new ZipOutputStream(new FileOutputStream(output))) {
+            var servicesLists = new HashMap<String, List<String>>();
+            var seen = new HashSet<String>();
+            for (var patcher : this.getStack()) {
+                try (var zin = new ZipInputStream(new FileInputStream(this.data))) {
+                    ZipEntry entry;
+                    var prefix = config.inject;
+                    while ((entry = zin.getNextEntry()) != null) {
+                        if (!entry.getName().startsWith(prefix) || entry.getName().length() <= prefix.length())
+                            continue;
+
+                        String name = entry.getName().substring(prefix.length());
+
+                        if (name.startsWith("META-INF/services/") && !entry.isDirectory()) {
+                            var existing = servicesLists.computeIfAbsent(name, _ -> new ArrayList<>());
+                            if (existing.size() > 0) {
+                                existing.add("");
+                                existing.add("# " + patcher.getArtifact());
+                            }
+                            existing.add(new String(zin.readAllBytes(), StandardCharsets.UTF_8));
+                        } else if (seen.add(name)) {
+                            var _new = new ZipEntry(name);
+                            _new.setTime(0);
+                            zos.putNextEntry(_new);
+                            zin.transferTo(zos);
+                        }
+                    }
+                }
+            }
+
+            for(var kv : servicesLists.entrySet()) {
+                String name = kv.getKey();
+                ZipEntry _new = new ZipEntry(name);
+                _new.setTime(0);
+                zos.putNextEntry(_new);
+                for (var line : kv.getValue()) {
+                    zos.write(line.getBytes(StandardCharsets.UTF_8));
+                    zos.write('\n');
+                }
+            }
+        } catch (IOException e) {
+            return Util.sneak(e);
+        }
+
+        cache.save();
+        return output;
+    }
 }