239 lines
9.2 KiB
Java
239 lines
9.2 KiB
Java
/*
|
|
* Minecraft Forge
|
|
* Copyright (c) 2016-2020.
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation version 2.1
|
|
* of the License.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
package net.minecraftforge.fml;
|
|
|
|
import net.minecraftforge.eventbus.api.Event;
|
|
import net.minecraftforge.fml.event.lifecycle.IModBusEvent;
|
|
import net.minecraftforge.forgespi.language.ModFileScanData;
|
|
import net.minecraftforge.fml.loading.moddiscovery.ModFile;
|
|
import net.minecraftforge.fml.loading.moddiscovery.ModFileInfo;
|
|
import net.minecraftforge.fml.loading.moddiscovery.ModInfo;
|
|
import org.apache.logging.log4j.LogManager;
|
|
import org.apache.logging.log4j.Logger;
|
|
|
|
import java.util.AbstractMap;
|
|
import java.util.ArrayList;
|
|
import java.util.Arrays;
|
|
import java.util.Collection;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Objects;
|
|
import java.util.Optional;
|
|
import java.util.concurrent.CompletableFuture;
|
|
import java.util.concurrent.CompletionException;
|
|
import java.util.concurrent.CompletionStage;
|
|
import java.util.concurrent.Executor;
|
|
import java.util.concurrent.ForkJoinPool;
|
|
import java.util.concurrent.ForkJoinWorkerThread;
|
|
import java.util.function.BiConsumer;
|
|
import java.util.function.BiFunction;
|
|
import java.util.function.Consumer;
|
|
import java.util.function.Function;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.Stream;
|
|
|
|
/**
|
|
* Master list of all mods - game-side version. This is classloaded in the game scope and
|
|
* can dispatch game level events as a result.
|
|
*/
|
|
public class ModList
|
|
{
|
|
private static Logger LOGGER = LogManager.getLogger();
|
|
private static ModList INSTANCE;
|
|
private final List<ModFileInfo> modFiles;
|
|
private final List<ModInfo> sortedList;
|
|
private final Map<String, ModFileInfo> fileById;
|
|
private List<ModContainer> mods;
|
|
private Map<String, ModContainer> indexedMods;
|
|
private List<ModFileScanData> modFileScanData;
|
|
|
|
private ModList(final List<ModFile> modFiles, final List<ModInfo> sortedList)
|
|
{
|
|
this.modFiles = modFiles.stream().map(ModFile::getModFileInfo).map(ModFileInfo.class::cast).collect(Collectors.toList());
|
|
this.sortedList = sortedList.stream().
|
|
map(ModInfo.class::cast).
|
|
collect(Collectors.toList());
|
|
this.fileById = this.modFiles.stream().map(ModFileInfo::getMods).flatMap(Collection::stream).
|
|
map(ModInfo.class::cast).
|
|
collect(Collectors.toMap(ModInfo::getModId, ModInfo::getOwningFile));
|
|
CrashReportExtender.registerCrashCallable("Mod List", this::crashReport);
|
|
}
|
|
|
|
private String getModContainerState(String modId) {
|
|
return getModContainerById(modId).map(ModContainer::getCurrentState).map(Object::toString).orElse("NONE");
|
|
}
|
|
|
|
private String fileToLine(ModFile mf) {
|
|
return mf.getFileName() + " " + mf.getModInfos().get(0).getDisplayName() + " " +
|
|
mf.getModInfos().stream().map(mi -> mi.getModId() + "@" + mi.getVersion() + " " +
|
|
getModContainerState(mi.getModId())).collect(Collectors.joining(", ", "{", "}"));
|
|
}
|
|
private String crashReport() {
|
|
return "\n"+applyForEachModFile(this::fileToLine).collect(Collectors.joining("\n\t\t", "\t\t", ""));
|
|
}
|
|
|
|
public static ModList of(List<ModFile> modFiles, List<ModInfo> sortedList)
|
|
{
|
|
INSTANCE = new ModList(modFiles, sortedList);
|
|
return INSTANCE;
|
|
}
|
|
|
|
public static ModList get() {
|
|
return INSTANCE;
|
|
}
|
|
|
|
private static ForkJoinWorkerThread newForkJoinWorkerThread(ForkJoinPool pool) {
|
|
ForkJoinWorkerThread thread = ForkJoinPool.defaultForkJoinWorkerThreadFactory.newThread(pool);
|
|
thread.setName("modloading-worker-" + thread.getPoolIndex());
|
|
// The default sets it to the SystemClassloader, so copy the current one.
|
|
thread.setContextClassLoader(Thread.currentThread().getContextClassLoader());
|
|
return thread;
|
|
}
|
|
|
|
public List<ModFileInfo> getModFiles()
|
|
{
|
|
return modFiles;
|
|
}
|
|
|
|
public ModFileInfo getModFileById(String modid)
|
|
{
|
|
return this.fileById.get(modid);
|
|
}
|
|
|
|
<T extends Event & IModBusEvent> Function<Executor, CompletableFuture<List<Throwable>>> futureVisitor(
|
|
final ModLoadingStage.EventGenerator<T> eventGenerator,
|
|
final BiFunction<ModLoadingStage, Throwable, ModLoadingStage> stateChange) {
|
|
return executor -> gather(
|
|
this.mods.stream()
|
|
.map(mod -> ModContainer.buildTransitionHandler(mod, eventGenerator, stateChange, executor))
|
|
.collect(Collectors.toList()))
|
|
.thenComposeAsync(ModList::completableFutureFromExceptionList, executor);
|
|
}
|
|
static CompletionStage<List<Throwable>> completableFutureFromExceptionList(List<? extends Map.Entry<?, Throwable>> t) {
|
|
if (t.stream().noneMatch(e->e.getValue()!=null)) {
|
|
return CompletableFuture.completedFuture(Collections.emptyList());
|
|
} else {
|
|
final List<Throwable> throwables = t.stream().filter(e -> e.getValue() != null).map(Map.Entry::getValue).collect(Collectors.toList());
|
|
CompletableFuture<List<Throwable>> cf = new CompletableFuture<>();
|
|
final RuntimeException accumulator = new RuntimeException();
|
|
cf.completeExceptionally(accumulator);
|
|
throwables.forEach(exception -> {
|
|
if (exception instanceof CompletionException) {
|
|
exception = exception.getCause();
|
|
}
|
|
if (exception.getSuppressed().length!=0) {
|
|
Arrays.stream(exception.getSuppressed()).forEach(accumulator::addSuppressed);
|
|
} else {
|
|
accumulator.addSuppressed(exception);
|
|
}
|
|
});
|
|
return cf;
|
|
}
|
|
}
|
|
|
|
static <V> CompletableFuture<List<Map.Entry<V, Throwable>>> gather(List<? extends CompletableFuture<? extends V>> futures) {
|
|
List<Map.Entry<V, Throwable>> list = new ArrayList<>(futures.size());
|
|
CompletableFuture<?>[] results = new CompletableFuture[futures.size()];
|
|
futures.forEach(future -> {
|
|
int i = list.size();
|
|
list.add(null);
|
|
results[i] = future.whenComplete((result, exception) -> list.set(i, new AbstractMap.SimpleImmutableEntry<>(result, exception)));
|
|
});
|
|
return CompletableFuture.allOf(results).handle((r, th)->null).thenApply(res -> list);
|
|
}
|
|
|
|
void setLoadedMods(final List<ModContainer> modContainers)
|
|
{
|
|
this.mods = modContainers;
|
|
this.indexedMods = modContainers.stream().collect(Collectors.toMap(ModContainer::getModId, Function.identity()));
|
|
}
|
|
|
|
@SuppressWarnings("unchecked")
|
|
public <T> Optional<T> getModObjectById(String modId)
|
|
{
|
|
return getModContainerById(modId).map(ModContainer::getMod).map(o -> (T) o);
|
|
}
|
|
|
|
public Optional<? extends ModContainer> getModContainerById(String modId)
|
|
{
|
|
return Optional.ofNullable(this.indexedMods.get(modId));
|
|
}
|
|
|
|
public Optional<? extends ModContainer> getModContainerByObject(Object obj)
|
|
{
|
|
return mods.stream().filter(mc -> mc.getMod() == obj).findFirst();
|
|
}
|
|
|
|
public List<ModInfo> getMods()
|
|
{
|
|
return this.sortedList;
|
|
}
|
|
|
|
public boolean isLoaded(String modTarget)
|
|
{
|
|
return this.indexedMods.containsKey(modTarget);
|
|
}
|
|
|
|
public int size()
|
|
{
|
|
return mods.size();
|
|
}
|
|
|
|
public List<ModFileScanData> getAllScanData()
|
|
{
|
|
if (modFileScanData == null)
|
|
{
|
|
modFileScanData = this.sortedList.stream().
|
|
map(ModInfo::getOwningFile).
|
|
filter(Objects::nonNull).
|
|
map(ModFileInfo::getFile).
|
|
distinct().
|
|
map(ModFile::getScanResult).
|
|
collect(Collectors.toList());
|
|
}
|
|
return modFileScanData;
|
|
|
|
}
|
|
|
|
public void forEachModFile(Consumer<ModFile> fileConsumer)
|
|
{
|
|
modFiles.stream().map(ModFileInfo::getFile).forEach(fileConsumer);
|
|
}
|
|
|
|
public <T> Stream<T> applyForEachModFile(Function<ModFile, T> function) {
|
|
return modFiles.stream().map(ModFileInfo::getFile).map(function);
|
|
}
|
|
|
|
public void forEachModContainer(BiConsumer<String, ModContainer> modContainerConsumer) {
|
|
indexedMods.forEach(modContainerConsumer);
|
|
}
|
|
|
|
public <T> Stream<T> applyForEachModContainer(Function<ModContainer, T> function) {
|
|
return indexedMods.values().stream().map(function);
|
|
}
|
|
|
|
private static class UncaughtModLoadingException extends ModLoadingException {
|
|
public UncaughtModLoadingException(ModLoadingStage stage, Throwable originalException) {
|
|
super(null, stage, "fml.modloading.uncaughterror", originalException);
|
|
}
|
|
}
|
|
}
|