Jake Potrebic
61 lines
2.1 KiB
Java
61 lines
2.1 KiB
Java
package io.papermc.paper.registry;
|
|
|
|
import io.papermc.paper.registry.set.NamedRegistryKeySetImpl;
|
|
import io.papermc.paper.registry.tag.Tag;
|
|
import io.papermc.paper.registry.tag.TagKey;
|
|
import java.util.Collection;
|
|
import java.util.function.Predicate;
|
|
import net.minecraft.core.HolderSet;
|
|
import net.minecraft.core.registries.BuiltInRegistries;
|
|
import org.bukkit.Keyed;
|
|
import org.bukkit.Particle;
|
|
import org.bukkit.Registry;
|
|
import org.bukkit.entity.EntityType;
|
|
import org.bukkit.potion.PotionType;
|
|
import org.jspecify.annotations.NullMarked;
|
|
|
|
@NullMarked
|
|
public class PaperSimpleRegistry<T extends Enum<T> & Keyed, M> extends Registry.SimpleRegistry<T> {
|
|
|
|
static Registry<EntityType> entityType() {
|
|
return new PaperSimpleRegistry<>(EntityType.class, entity -> entity != EntityType.UNKNOWN, BuiltInRegistries.ENTITY_TYPE);
|
|
}
|
|
|
|
static Registry<Particle> particleType() {
|
|
return new PaperSimpleRegistry<>(Particle.class, BuiltInRegistries.PARTICLE_TYPE);
|
|
}
|
|
|
|
static Registry<PotionType> potion() {
|
|
return new PaperSimpleRegistry<>(PotionType.class, BuiltInRegistries.POTION);
|
|
}
|
|
|
|
private final net.minecraft.core.Registry<M> nmsRegistry;
|
|
|
|
protected PaperSimpleRegistry(final Class<T> type, final net.minecraft.core.Registry<M> nmsRegistry) {
|
|
super(type);
|
|
this.nmsRegistry = nmsRegistry;
|
|
}
|
|
|
|
public PaperSimpleRegistry(final Class<T> type, final Predicate<T> predicate, final net.minecraft.core.Registry<M> nmsRegistry) {
|
|
super(type, predicate);
|
|
this.nmsRegistry = nmsRegistry;
|
|
}
|
|
|
|
@Override
|
|
public boolean hasTag(final TagKey<T> key) {
|
|
final net.minecraft.tags.TagKey<M> nmsKey = PaperRegistries.toNms(key);
|
|
return this.nmsRegistry.get(nmsKey).isPresent();
|
|
}
|
|
|
|
@Override
|
|
public Tag<T> getTag(final TagKey<T> key) {
|
|
final HolderSet.Named<M> namedHolderSet = this.nmsRegistry.get(PaperRegistries.toNms(key)).orElseThrow();
|
|
return new NamedRegistryKeySetImpl<>(key, namedHolderSet);
|
|
}
|
|
|
|
@Override
|
|
public Collection<Tag<T>> getTags() {
|
|
return this.nmsRegistry.getTags().<Tag<T>>map(NamedRegistryKeySetImpl::new).toList();
|
|
}
|
|
}
|