2020-06-27 19:18:32 +00:00
|
|
|
/*
|
|
|
|
* This file is part of fabric-loom, licensed under the MIT License (MIT).
|
|
|
|
*
|
2021-07-10 20:50:53 +00:00
|
|
|
* Copyright (c) 2020 FabricMC
|
2020-06-27 19:18:32 +00:00
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*/
|
|
|
|
|
2020-12-24 20:58:30 +00:00
|
|
|
package net.fabricmc.loom.build;
|
2020-06-27 19:18:32 +00:00
|
|
|
|
|
|
|
import java.io.IOException;
|
2021-07-21 22:21:47 +00:00
|
|
|
import java.nio.file.Files;
|
2020-06-27 19:18:32 +00:00
|
|
|
import java.nio.file.Path;
|
|
|
|
import java.util.ArrayList;
|
|
|
|
import java.util.Arrays;
|
|
|
|
import java.util.HashSet;
|
|
|
|
import java.util.List;
|
|
|
|
import java.util.Set;
|
|
|
|
import java.util.function.BiConsumer;
|
|
|
|
import java.util.function.BiFunction;
|
|
|
|
|
2020-12-21 21:02:39 +00:00
|
|
|
import org.gradle.api.Action;
|
2020-06-27 19:18:32 +00:00
|
|
|
import org.objectweb.asm.commons.Remapper;
|
|
|
|
|
2021-07-21 22:21:47 +00:00
|
|
|
import net.fabricmc.loom.util.CloseableList;
|
2020-06-27 19:18:32 +00:00
|
|
|
import net.fabricmc.stitch.util.Pair;
|
|
|
|
import net.fabricmc.tinyremapper.IMappingProvider;
|
|
|
|
import net.fabricmc.tinyremapper.InputTag;
|
|
|
|
import net.fabricmc.tinyremapper.OutputConsumerPath;
|
|
|
|
import net.fabricmc.tinyremapper.TinyRemapper;
|
|
|
|
|
|
|
|
public class JarRemapper {
|
|
|
|
private final List<IMappingProvider> mappingProviders = new ArrayList<>();
|
|
|
|
private final Set<Path> classPath = new HashSet<>();
|
|
|
|
private final List<RemapData> remapData = new ArrayList<>();
|
2020-12-21 21:02:39 +00:00
|
|
|
private List<Action<TinyRemapper.Builder>> remapOptions;
|
2020-06-27 19:18:32 +00:00
|
|
|
|
|
|
|
public void addMappings(IMappingProvider mappingProvider) {
|
|
|
|
mappingProviders.add(mappingProvider);
|
|
|
|
}
|
|
|
|
|
|
|
|
public void addToClasspath(Path... paths) {
|
|
|
|
classPath.addAll(Arrays.asList(paths));
|
|
|
|
}
|
|
|
|
|
|
|
|
public RemapData scheduleRemap(Path input, Path output) {
|
|
|
|
RemapData data = new RemapData(input, output);
|
|
|
|
remapData.add(data);
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
|
|
|
|
public void remap() throws IOException {
|
|
|
|
TinyRemapper.Builder remapperBuilder = TinyRemapper.newRemapper();
|
|
|
|
mappingProviders.forEach(remapperBuilder::withMappings);
|
|
|
|
|
2020-12-21 21:02:39 +00:00
|
|
|
if (remapOptions != null) {
|
|
|
|
for (Action<TinyRemapper.Builder> remapOption : remapOptions) {
|
|
|
|
remapOption.execute(remapperBuilder);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-06-27 19:18:32 +00:00
|
|
|
TinyRemapper remapper = remapperBuilder.build();
|
|
|
|
|
|
|
|
Path[] remapClasspath = classPath.stream()
|
|
|
|
.filter(path ->
|
|
|
|
remapData.stream().noneMatch(remapData -> remapData.input.equals(path))
|
|
|
|
)
|
|
|
|
.toArray(Path[]::new);
|
|
|
|
|
|
|
|
remapper.readClassPathAsync(remapClasspath);
|
|
|
|
|
|
|
|
for (RemapData data : remapData) {
|
|
|
|
InputTag tag = remapper.createInputTag();
|
|
|
|
data.tag = tag;
|
|
|
|
remapper.readInputsAsync(tag, data.input);
|
|
|
|
}
|
|
|
|
|
2021-07-21 22:21:47 +00:00
|
|
|
//noinspection MismatchedQueryAndUpdateOfCollection
|
|
|
|
try (CloseableList<OutputConsumerPath> outputConsumers = new CloseableList<>()) {
|
|
|
|
for (RemapData data : remapData) {
|
|
|
|
OutputConsumerPath outputConsumer = new OutputConsumerPath.Builder(data.output).build();
|
|
|
|
outputConsumers.add(outputConsumer);
|
2020-06-27 19:18:32 +00:00
|
|
|
|
2021-07-21 22:21:47 +00:00
|
|
|
outputConsumer.addNonClassFiles(data.input);
|
2020-06-27 19:18:32 +00:00
|
|
|
|
2021-12-17 15:37:00 +00:00
|
|
|
data.processAccessWidener(remapper.getEnvironment().getRemapper());
|
2021-07-21 22:21:47 +00:00
|
|
|
remapper.apply(outputConsumer, data.tag);
|
|
|
|
}
|
2020-06-27 19:18:32 +00:00
|
|
|
|
2021-07-21 22:21:47 +00:00
|
|
|
remapper.finish();
|
|
|
|
} catch (Exception e) {
|
|
|
|
for (RemapData data : remapData) {
|
|
|
|
// Cleanup bad outputs
|
|
|
|
Files.deleteIfExists(data.output);
|
|
|
|
}
|
2020-06-27 19:18:32 +00:00
|
|
|
|
2021-07-21 22:21:47 +00:00
|
|
|
throw new IOException("Failed to remap %s files".formatted(remapData.size()), e);
|
2020-06-27 19:18:32 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
remapData.forEach(RemapData::complete);
|
|
|
|
}
|
|
|
|
|
2020-12-21 21:02:39 +00:00
|
|
|
public void addOptions(List<Action<TinyRemapper.Builder>> remapOptions) {
|
|
|
|
this.remapOptions = remapOptions;
|
|
|
|
}
|
|
|
|
|
2020-06-27 19:18:32 +00:00
|
|
|
public static class RemapData {
|
|
|
|
public final Path input;
|
|
|
|
public final Path output;
|
|
|
|
BiFunction<RemapData, Remapper, Pair<String, byte[]>> accesWidenerSupplier;
|
|
|
|
BiConsumer<RemapData, Pair<String, byte[]>> onComplete;
|
|
|
|
|
|
|
|
private InputTag tag;
|
|
|
|
private Pair<String, byte[]> accessWidener;
|
|
|
|
|
|
|
|
public RemapData(Path input, Path output) {
|
|
|
|
this.input = input;
|
|
|
|
this.output = output;
|
|
|
|
}
|
|
|
|
|
|
|
|
public RemapData complete(BiConsumer<RemapData, Pair<String, byte[]>> onComplete) {
|
|
|
|
this.onComplete = onComplete;
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
|
|
|
public RemapData supplyAccessWidener(BiFunction<RemapData, Remapper, Pair<String, byte[]>> beforeFinish) {
|
|
|
|
this.accesWidenerSupplier = beforeFinish;
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
|
|
|
private void complete() {
|
|
|
|
if (onComplete != null) {
|
|
|
|
onComplete.accept(this, accessWidener);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private void processAccessWidener(Remapper remapper) {
|
|
|
|
if (accesWidenerSupplier != null) {
|
|
|
|
accessWidener = accesWidenerSupplier.apply(this, remapper);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|