From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001 From: Jake Potrebic Date: Wed, 2 Mar 2022 13:33:08 -0800 Subject: [PATCH] Add PaperRegistry PaperRegistry is a server-backed impl of bukkit's Registry interface diff --git a/src/main/java/io/papermc/paper/registry/PaperRegistry.java b/src/main/java/io/papermc/paper/registry/PaperRegistry.java new file mode 100644 index 0000000000000000000000000000000000000000..7c265d27da034986be73921d35bf08ae250b42f3 --- /dev/null +++ b/src/main/java/io/papermc/paper/registry/PaperRegistry.java @@ -0,0 +1,167 @@ +package io.papermc.paper.registry; + +import com.google.common.base.Preconditions; +import com.google.common.base.Suppliers; +import net.minecraft.core.Holder; +import net.minecraft.core.Registry; +import net.minecraft.core.RegistryAccess; +import net.minecraft.resources.ResourceKey; +import net.minecraft.resources.ResourceLocation; +import net.minecraft.server.MinecraftServer; +import org.bukkit.Keyed; +import org.bukkit.NamespacedKey; +import org.bukkit.craftbukkit.util.CraftNamespacedKey; +import org.checkerframework.checker.nullness.qual.NonNull; +import org.checkerframework.checker.nullness.qual.Nullable; +import org.checkerframework.framework.qual.DefaultQualifier; + +import java.util.Collections; +import java.util.HashMap; +import java.util.Iterator; +import java.util.Map; +import java.util.Objects; +import java.util.Optional; +import java.util.concurrent.ConcurrentHashMap; +import java.util.function.Consumer; +import java.util.function.Supplier; + +@DefaultQualifier(NonNull.class) +public abstract class PaperRegistry implements org.bukkit.Registry { + + @SuppressWarnings("FieldMayBeFinal") // non-final for testing + private static Supplier REGISTRY_ACCESS = Suppliers.memoize(() -> MinecraftServer.getServer().registryAccess()); + private static final Map, PaperRegistry> INTERNAL_REGISTRIES = new HashMap<>(); + public static final Map, PaperRegistry> REGISTRIES = Collections.unmodifiableMap(INTERNAL_REGISTRIES); + private static final Map, PaperRegistry> REGISTRY_BY_API_CLASS = new HashMap<>(); + private static final Map>, PaperRegistry> REGISTRY_BY_RES_KEY = new HashMap<>(); + + private boolean registered; + private final RegistryKey registryKey; + private final Supplier> registry; + private final Map cache = new ConcurrentHashMap<>(); + private final Map> resourceKeyCache = new ConcurrentHashMap<>(); + + public PaperRegistry(RegistryKey registryKey) { + this.registryKey = registryKey; + this.registry = Suppliers.memoize(() -> REGISTRY_ACCESS.get().registryOrThrow(this.registryKey.resourceKey())); + } + + @Override + public @Nullable API get(NamespacedKey key) { + return this.cache.computeIfAbsent(key, k -> { + final @Nullable MINECRAFT nms = this.registry.get().get(CraftNamespacedKey.toMinecraft(k)); + if (nms != null) { + return this.convertToApi(k, nms); + } + return null; + }); + } + + public abstract @Nullable API convertToApi(NamespacedKey key, MINECRAFT nms); + + public API convertToApiOrThrow(ResourceLocation resourceLocation, MINECRAFT nms) { + return Objects.requireNonNull(this.convertToApi(resourceLocation, nms), resourceLocation + " has a null api representation"); + } + + public @Nullable API convertToApi(ResourceLocation resourceLocation, MINECRAFT nms) { + return this.convertToApi(CraftNamespacedKey.fromMinecraft(resourceLocation), nms); + } + + public API convertToApiOrThrow(Holder nmsHolder) { + return Objects.requireNonNull(this.convertToApi(nmsHolder), nmsHolder + " has a null api representation"); + } + + public @Nullable API convertToApi(Holder nmsHolder) { + final Optional> key = nmsHolder.unwrapKey(); + if (nmsHolder.isBound() && key.isPresent()) { + return this.convertToApi(key.get().location(), nmsHolder.value()); + } else if (!nmsHolder.isBound() && key.isPresent()) { + return this.convertToApi(key.get().location(), this.registry.get().getOrThrow(key.get())); + } else if (nmsHolder.isBound() && key.isEmpty()) { + final @Nullable ResourceLocation loc = this.registry.get().getKey(nmsHolder.value()); + if (loc != null) { + return this.convertToApi(loc, nmsHolder.value()); + } + } + throw new IllegalStateException("Cannot convert " + nmsHolder + " to an API type in: " + this.registryKey); + } + + public void convertToApi(Iterable> holders, Consumer apiConsumer, boolean throwOnNull) { + for (Holder holder : holders) { + final @Nullable API api = this.convertToApi(holder); + if (api == null && throwOnNull) { + throw new NullPointerException(holder + " has a null api representation"); + } else if (api != null) { + apiConsumer.accept(api); + } + } + } + + public MINECRAFT getMinecraftValue(API apiValue) { + return this.registry.get().getOptional(CraftNamespacedKey.toMinecraft(apiValue.getKey())).orElseThrow(); + } + + public Holder getMinecraftHolder(API apiValue) { + return this.registry.get().getHolderOrThrow(this.resourceKeyCache.computeIfAbsent(apiValue.getKey(), key -> ResourceKey.create(this.registryKey.resourceKey(), CraftNamespacedKey.toMinecraft(key)))); + } + + @Override + public Iterator iterator() { + return this.registry.get().keySet().stream().map(key -> this.get(CraftNamespacedKey.fromMinecraft(key))).iterator(); + } + + public void clearCache() { + this.cache.clear(); + } + + public void register() { + if (this.registered) { + throw new IllegalStateException("Already registered: " + this.registryKey.apiClass()); + } + INTERNAL_REGISTRIES.put(this.registryKey, this); + REGISTRY_BY_API_CLASS.put(this.registryKey.apiClass(), this); + REGISTRY_BY_RES_KEY.put(this.registryKey.resourceKey(), this); + this.registered = true; + } + + @Override + public boolean equals(@Nullable Object o) { + if (this == o) return true; + if (o == null || !PaperRegistry.class.isAssignableFrom(o.getClass())) return false; + PaperRegistry that = (PaperRegistry) o; + return this.registryKey.equals(that.registryKey); + } + + @Override + public int hashCode() { + return Objects.hash(this.registryKey); + } + + protected static Supplier> registryFor(ResourceKey> registryKey) { + return Suppliers.memoize(() -> REGISTRY_ACCESS.get().registryOrThrow(registryKey)); + } + + public static void clearCaches() { + for (PaperRegistry registry : INTERNAL_REGISTRIES.values()) { + registry.clearCache(); + } + } + + @SuppressWarnings("unchecked") + public static PaperRegistry getRegistry(Class classOfT) { + Preconditions.checkArgument(REGISTRY_BY_API_CLASS.containsKey(classOfT), "No registry for that type"); + return (PaperRegistry) REGISTRY_BY_API_CLASS.get(classOfT); + } + + @SuppressWarnings("unchecked") + public static PaperRegistry getRegistry(ResourceKey> resourceKey) { + Preconditions.checkArgument(REGISTRY_BY_RES_KEY.containsKey(resourceKey)); + return (PaperRegistry) REGISTRY_BY_RES_KEY.get(resourceKey); + } + + @SuppressWarnings("unchecked") + public static PaperRegistry getRegistry(RegistryKey registryKey) { + Preconditions.checkArgument(INTERNAL_REGISTRIES.containsKey(registryKey)); + return (PaperRegistry) INTERNAL_REGISTRIES.get(registryKey); + } +} diff --git a/src/main/java/io/papermc/paper/registry/RegistryKey.java b/src/main/java/io/papermc/paper/registry/RegistryKey.java new file mode 100644 index 0000000000000000000000000000000000000000..6f39e343147803e15e7681c993b8797a629702e7 --- /dev/null +++ b/src/main/java/io/papermc/paper/registry/RegistryKey.java @@ -0,0 +1,8 @@ +package io.papermc.paper.registry; + +import net.minecraft.core.Registry; +import net.minecraft.resources.ResourceKey; +import org.bukkit.Keyed; + +public record RegistryKey(Class apiClass, ResourceKey> resourceKey) { +} diff --git a/src/main/java/net/minecraft/server/MinecraftServer.java b/src/main/java/net/minecraft/server/MinecraftServer.java index 17e0425a520bd95074bf34c262c36c5603266ea7..f05f5d13f23599dbb95a7062ebbb41334ec21f14 100644 --- a/src/main/java/net/minecraft/server/MinecraftServer.java +++ b/src/main/java/net/minecraft/server/MinecraftServer.java @@ -2020,6 +2020,7 @@ public abstract class MinecraftServer extends ReentrantBlockableEventLoop Registry registryFor(Class classOfT) { + return io.papermc.paper.registry.PaperRegistry.getRegistry(classOfT); + } // Paper end /** diff --git a/src/test/java/org/bukkit/support/AbstractTestingBase.java b/src/test/java/org/bukkit/support/AbstractTestingBase.java index e73a9a957cd55bf838e301ed531295162f2cfb89..23435a5914a29ac4f87ad0f34ce83842422df49f 100644 --- a/src/test/java/org/bukkit/support/AbstractTestingBase.java +++ b/src/test/java/org/bukkit/support/AbstractTestingBase.java @@ -39,6 +39,15 @@ public abstract class AbstractTestingBase { MultiPackResourceManager resourceManager = new MultiPackResourceManager(PackType.SERVER_DATA, Collections.singletonList(new VanillaPackResources(ServerPacksSource.BUILT_IN_METADATA, "minecraft"))); // add tags and loot tables for unit tests REGISTRY_CUSTOM = RegistryAccess.builtinCopy().freeze(); + // Paper start + try { + java.lang.reflect.Field field = io.papermc.paper.registry.PaperRegistry.class.getDeclaredField("REGISTRY_ACCESS"); + field.trySetAccessible(); + field.set(null, com.google.common.base.Suppliers.ofInstance(REGISTRY_CUSTOM)); + } catch (ReflectiveOperationException ex) { + throw new IllegalStateException("Could not reflectively set RegistryAccess in PaperRegistry", ex); + } + // Paper end // Register vanilla pack DATA_PACK = ReloadableServerResources.loadResources(resourceManager, REGISTRY_CUSTOM, Commands.CommandSelection.DEDICATED, 0, MoreExecutors.directExecutor(), MoreExecutors.directExecutor()).join(); // Bind tags