84 lines
3.5 KiB
Java
84 lines
3.5 KiB
Java
|
package net.minecraft.data.registries;
|
||
|
|
|
||
|
|
import com.google.gson.JsonElement;
|
||
|
|
import com.mojang.serialization.DynamicOps;
|
||
|
|
import com.mojang.serialization.Encoder;
|
||
|
|
import com.mojang.serialization.JsonOps;
|
||
|
|
import com.mojang.serialization.DataResult.Error;
|
||
|
|
import java.nio.file.Path;
|
||
|
|
import java.util.Optional;
|
||
|
|
import java.util.concurrent.CompletableFuture;
|
||
|
|
import net.minecraft.core.Holder;
|
||
|
|
import net.minecraft.core.HolderLookup;
|
||
|
|
import net.minecraft.core.Registry;
|
||
|
|
import net.minecraft.data.CachedOutput;
|
||
|
|
import net.minecraft.data.DataProvider;
|
||
|
|
import net.minecraft.data.PackOutput;
|
||
|
|
import net.minecraft.resources.RegistryDataLoader;
|
||
|
|
import net.minecraft.resources.ResourceKey;
|
||
|
|
|
||
|
|
public class RegistriesDatapackGenerator implements DataProvider {
|
||
|
|
private final PackOutput output;
|
||
|
|
private final CompletableFuture<HolderLookup.Provider> registries;
|
||
|
|
|
||
|
|
public RegistriesDatapackGenerator(PackOutput p_256643_, CompletableFuture<HolderLookup.Provider> p_255780_) {
|
||
|
|
this.registries = p_255780_;
|
||
|
|
this.output = p_256643_;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public CompletableFuture<?> run(CachedOutput p_255785_) {
|
||
|
|
return this.registries
|
||
|
|
.thenCompose(
|
||
|
|
p_325918_ -> {
|
||
|
|
DynamicOps<JsonElement> dynamicops = p_325918_.createSerializationContext(JsonOps.INSTANCE);
|
||
|
|
return CompletableFuture.allOf(
|
||
|
|
RegistryDataLoader.WORLDGEN_REGISTRIES
|
||
|
|
.stream()
|
||
|
|
.flatMap(p_256552_ -> this.dumpRegistryCap(p_255785_, p_325918_, dynamicops, (RegistryDataLoader.RegistryData<?>)p_256552_).stream())
|
||
|
|
.toArray(CompletableFuture[]::new)
|
||
|
|
);
|
||
|
|
}
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
private <T> Optional<CompletableFuture<?>> dumpRegistryCap(
|
||
|
|
CachedOutput p_256502_, HolderLookup.Provider p_256492_, DynamicOps<JsonElement> p_256000_, RegistryDataLoader.RegistryData<T> p_256449_
|
||
|
|
) {
|
||
|
|
ResourceKey<? extends Registry<T>> resourcekey = p_256449_.key();
|
||
|
|
return p_256492_.lookup(resourcekey)
|
||
|
|
.map(
|
||
|
|
p_358460_ -> {
|
||
|
|
PackOutput.PathProvider packoutput$pathprovider = this.output.createRegistryElementsPathProvider(resourcekey);
|
||
|
|
return CompletableFuture.allOf(
|
||
|
|
p_358460_.listElements()
|
||
|
|
.map(
|
||
|
|
p_256105_ -> dumpValue(
|
||
|
|
packoutput$pathprovider.json(p_256105_.key().location()),
|
||
|
|
p_256502_,
|
||
|
|
p_256000_,
|
||
|
|
p_256449_.elementCodec(),
|
||
|
|
p_256105_.value()
|
||
|
|
)
|
||
|
|
)
|
||
|
|
.toArray(CompletableFuture[]::new)
|
||
|
|
);
|
||
|
|
}
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
private static <E> CompletableFuture<?> dumpValue(
|
||
|
|
Path p_255678_, CachedOutput p_256438_, DynamicOps<JsonElement> p_256127_, Encoder<E> p_255938_, E p_256590_
|
||
|
|
) {
|
||
|
|
return p_255938_.encodeStart(p_256127_, p_256590_)
|
||
|
|
.mapOrElse(
|
||
|
|
p_341074_ -> DataProvider.saveStable(p_256438_, p_341074_, p_255678_),
|
||
|
|
p_341071_ -> CompletableFuture.failedFuture(new IllegalStateException("Couldn't generate file '" + p_255678_ + "': " + p_341071_.message()))
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public final String getName() {
|
||
|
|
return "Registries";
|
||
|
|
}
|
||
|
|
}
|