231 lines
9.8 KiB
Java
231 lines
9.8 KiB
Java
/*
|
|
* Minecraft Forge
|
|
* Copyright (c) 2016-2019.
|
|
*
|
|
* 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.fml.loading.FMLConfig;
|
|
import net.minecraftforge.forgespi.language.ModFileScanData;
|
|
import net.minecraftforge.fml.loading.FMLLoader;
|
|
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.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.*;
|
|
import java.util.function.BiConsumer;
|
|
import java.util.function.Consumer;
|
|
import java.util.function.Function;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.Stream;
|
|
|
|
import static net.minecraftforge.fml.Logging.LOADING;
|
|
|
|
/**
|
|
* 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 ForkJoinPool modLoadingThreadPool;
|
|
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));
|
|
final int loadingThreadCount = FMLConfig.loadingThreadCount();
|
|
LOGGER.debug(LOADING, "Using {} threads for parallel mod-loading", loadingThreadCount);
|
|
modLoadingThreadPool = new ForkJoinPool(loadingThreadCount, ModList::newForkJoinWorkerThread, null, false);
|
|
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;
|
|
}
|
|
|
|
static LifecycleEventProvider.EventHandler<LifecycleEventProvider.LifecycleEvent, Consumer<List<ModLoadingException>>, Executor, Runnable> inlineDispatcher = (event, errors, executor, ticker) -> ModList.get().dispatchSynchronousEvent(event, errors, executor, ticker);
|
|
|
|
static LifecycleEventProvider.EventHandler<LifecycleEventProvider.LifecycleEvent, Consumer<List<ModLoadingException>>, Executor, Runnable> parallelDispatcher = (event, errors, executor, ticker) -> ModList.get().dispatchParallelEvent(event, errors, executor, ticker);
|
|
|
|
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);
|
|
}
|
|
|
|
private void dispatchSynchronousEvent(LifecycleEventProvider.LifecycleEvent lifecycleEvent, final Consumer<List<ModLoadingException>> errorHandler, final Executor executor, final Runnable ticker) {
|
|
LOGGER.debug(LOADING, "Dispatching synchronous event {}", lifecycleEvent);
|
|
executor.execute(ticker);
|
|
FMLLoader.getLanguageLoadingProvider().forEach(lp->lp.consumeLifecycleEvent(()->lifecycleEvent));
|
|
this.mods.stream().forEach(m->m.transitionState(lifecycleEvent, errorHandler));
|
|
FMLLoader.getLanguageLoadingProvider().forEach(lp->lp.consumeLifecycleEvent(()->lifecycleEvent));
|
|
}
|
|
private void dispatchParallelEvent(LifecycleEventProvider.LifecycleEvent lifecycleEvent, final Consumer<List<ModLoadingException>> errorHandler, final Executor executor, final Runnable ticker) {
|
|
LOGGER.debug(LOADING, "Dispatching parallel event {}", lifecycleEvent);
|
|
FMLLoader.getLanguageLoadingProvider().forEach(lp->lp.consumeLifecycleEvent(()->lifecycleEvent));
|
|
DeferredWorkQueue.clear();
|
|
try
|
|
{
|
|
final ForkJoinTask<?> parallelTask = modLoadingThreadPool.submit(() -> this.mods.parallelStream().forEach(m -> m.transitionState(lifecycleEvent, errorHandler)));
|
|
while (ticker != null && !parallelTask.isDone()) {
|
|
executor.execute(ticker);
|
|
}
|
|
parallelTask.get();
|
|
}
|
|
catch (InterruptedException | ExecutionException e)
|
|
{
|
|
LOGGER.error(LOADING, "Encountered an exception during parallel processing - sleeping 10 seconds to wait for jobs to finish", e);
|
|
errorHandler.accept(Collections.singletonList(new UncaughtModLoadingException(lifecycleEvent.fromStage(), e)));
|
|
modLoadingThreadPool.awaitQuiescence(10, TimeUnit.SECONDS);
|
|
if (!modLoadingThreadPool.isQuiescent()) {
|
|
LOGGER.fatal(LOADING, "The parallel pool has failed to quiesce correctly, forcing a shutdown. There is something really wrong here");
|
|
modLoadingThreadPool.shutdownNow();
|
|
throw new RuntimeException("Forge played \"STOP IT NOW MODS!\" - it was \"NOT VERY EFFECTIVE\"");
|
|
}
|
|
}
|
|
DeferredWorkQueue.runTasks(lifecycleEvent.fromStage(), errorHandler);
|
|
FMLLoader.getLanguageLoadingProvider().forEach(lp->lp.consumeLifecycleEvent(()->lifecycleEvent));
|
|
}
|
|
|
|
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).
|
|
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);
|
|
}
|
|
}
|
|
}
|