From 606046e719c05680b7bccc8be5ed62d7f3af5161 Mon Sep 17 00:00:00 2001
From: modmuss50 <modmuss50@gmail.com>
Date: Wed, 21 Jul 2021 23:21:47 +0100
Subject: [PATCH] Ensure files are closed and outputs cleaned up when
 remapping.

---
 .../net/fabricmc/loom/build/JarRemapper.java  | 30 ++++++-----
 .../net/fabricmc/loom/util/CloseableList.java | 52 +++++++++++++++++++
 2 files changed, 70 insertions(+), 12 deletions(-)
 create mode 100644 src/main/java/net/fabricmc/loom/util/CloseableList.java

diff --git a/src/main/java/net/fabricmc/loom/build/JarRemapper.java b/src/main/java/net/fabricmc/loom/build/JarRemapper.java
index 231040b..47158bc 100644
--- a/src/main/java/net/fabricmc/loom/build/JarRemapper.java
+++ b/src/main/java/net/fabricmc/loom/build/JarRemapper.java
@@ -25,6 +25,7 @@
 package net.fabricmc.loom.build;
 
 import java.io.IOException;
+import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -37,6 +38,7 @@ import java.util.function.BiFunction;
 import org.gradle.api.Action;
 import org.objectweb.asm.commons.Remapper;
 
+import net.fabricmc.loom.util.CloseableList;
 import net.fabricmc.stitch.util.Pair;
 import net.fabricmc.tinyremapper.IMappingProvider;
 import net.fabricmc.tinyremapper.InputTag;
@@ -89,22 +91,26 @@ public class JarRemapper {
 			remapper.readInputsAsync(tag, data.input);
 		}
 
-		List<OutputConsumerPath> outputConsumers = new ArrayList<>();
+		//noinspection MismatchedQueryAndUpdateOfCollection
+		try (CloseableList<OutputConsumerPath> outputConsumers = new CloseableList<>()) {
+			for (RemapData data : remapData) {
+				OutputConsumerPath outputConsumer = new OutputConsumerPath.Builder(data.output).build();
+				outputConsumers.add(outputConsumer);
 
-		for (RemapData data : remapData) {
-			OutputConsumerPath outputConsumer = new OutputConsumerPath.Builder(data.output).build();
-			outputConsumers.add(outputConsumer);
+				outputConsumer.addNonClassFiles(data.input);
 
-			outputConsumer.addNonClassFiles(data.input);
+				data.processAccessWidener(remapper.getRemapper());
+				remapper.apply(outputConsumer, data.tag);
+			}
 
-			data.processAccessWidener(remapper.getRemapper());
-			remapper.apply(outputConsumer, data.tag);
-		}
+			remapper.finish();
+		} catch (Exception e) {
+			for (RemapData data : remapData) {
+				// Cleanup bad outputs
+				Files.deleteIfExists(data.output);
+			}
 
-		remapper.finish();
-
-		for (OutputConsumerPath outputConsumer : outputConsumers) {
-			outputConsumer.close();
+			throw new IOException("Failed to remap %s files".formatted(remapData.size()), e);
 		}
 
 		remapData.forEach(RemapData::complete);
diff --git a/src/main/java/net/fabricmc/loom/util/CloseableList.java b/src/main/java/net/fabricmc/loom/util/CloseableList.java
new file mode 100644
index 0000000..3d8ebf0
--- /dev/null
+++ b/src/main/java/net/fabricmc/loom/util/CloseableList.java
@@ -0,0 +1,52 @@
+/*
+ * This file is part of fabric-loom, licensed under the MIT License (MIT).
+ *
+ * Copyright (c) 2021 FabricMC
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+package net.fabricmc.loom.util;
+
+import java.io.Closeable;
+import java.io.IOException;
+import java.util.ArrayList;
+
+public class CloseableList<T extends Closeable> extends ArrayList<T> implements Closeable {
+	@Override
+	public void close() throws IOException {
+		IOException exception = null;
+
+		for (T t : this) {
+			try {
+				t.close();
+			} catch (IOException e) {
+				if (exception == null) {
+					exception = new IOException("Failed to close list");
+				}
+
+				exception.addSuppressed(e);
+			}
+		}
+
+		if (exception != null) {
+			throw exception;
+		}
+	}
+}