diff --git a/common/src/main/java/juuxel/adorn/lib/registry/Registered.java b/common/src/main/java/juuxel/adorn/lib/registry/Registered.java index edc517869..b018d7a97 100644 --- a/common/src/main/java/juuxel/adorn/lib/registry/Registered.java +++ b/common/src/main/java/juuxel/adorn/lib/registry/Registered.java @@ -1,9 +1,7 @@ package juuxel.adorn.lib.registry; -import net.minecraft.registry.RegistryKey; - import java.util.function.Supplier; +@FunctionalInterface public interface Registered extends Supplier { - RegistryKey key(); } diff --git a/common/src/main/java/juuxel/adorn/lib/registry/RegisteredMap.java b/common/src/main/java/juuxel/adorn/lib/registry/RegisteredMap.java index 9d0b11b15..ff751ca8b 100644 --- a/common/src/main/java/juuxel/adorn/lib/registry/RegisteredMap.java +++ b/common/src/main/java/juuxel/adorn/lib/registry/RegisteredMap.java @@ -10,7 +10,7 @@ import java.util.function.Supplier; import java.util.stream.Stream; -public final class RegisteredMap implements Supplier> { +public final class RegisteredMap implements Registered> { private final Map> map; private final Supplier> builtMap; diff --git a/fabric/src/main/java/juuxel/adorn/platform/fabric/RegistrarImpl.java b/fabric/src/main/java/juuxel/adorn/platform/fabric/RegistrarImpl.java index 135489d9d..f277b4ee6 100644 --- a/fabric/src/main/java/juuxel/adorn/platform/fabric/RegistrarImpl.java +++ b/fabric/src/main/java/juuxel/adorn/platform/fabric/RegistrarImpl.java @@ -4,7 +4,6 @@ import juuxel.adorn.lib.registry.Registered; import juuxel.adorn.lib.registry.Registrar; import net.minecraft.registry.Registry; -import net.minecraft.registry.RegistryKey; import java.util.ArrayList; import java.util.Iterator; @@ -21,20 +20,9 @@ public RegistrarImpl(Registry registry) { @Override public Registered register(String id, Supplier provider) { - var key = RegistryKey.of(registry.getKey(), AdornCommon.id(id)); - var registered = Registry.register(registry, key, provider.get()); + var registered = Registry.register(registry, AdornCommon.id(id), provider.get()); objects.add(registered); - return new Registered<>() { - @Override - public U get() { - return registered; - } - - @Override - public RegistryKey key() { - return key; - } - }; + return () -> registered; } @Override diff --git a/forge/src/main/java/juuxel/adorn/platform/forge/registrar/DeferredRegistrar.java b/forge/src/main/java/juuxel/adorn/platform/forge/registrar/DeferredRegistrar.java index 4982cd885..2600a262a 100644 --- a/forge/src/main/java/juuxel/adorn/platform/forge/registrar/DeferredRegistrar.java +++ b/forge/src/main/java/juuxel/adorn/platform/forge/registrar/DeferredRegistrar.java @@ -6,7 +6,6 @@ import net.minecraft.registry.Registry; import net.minecraft.registry.RegistryKey; import net.neoforged.bus.api.IEventBus; -import net.neoforged.neoforge.common.extensions.IHolderExtension; import net.neoforged.neoforge.registries.DeferredHolder; import net.neoforged.neoforge.registries.DeferredRegister; @@ -32,17 +31,7 @@ public void hook(IEventBus modBus) { public Registered register(String id, Supplier provider) { var registryObject = register.register(id, provider); objects.add(registryObject); - return new Registered<>() { - @Override - public RegistryKey key() { - return ((IHolderExtension) registryObject).getKey(); - } - - @Override - public U get() { - return registryObject.get(); - } - }; + return registryObject::get; } @Override