diff --git a/patches/api/Add-Lifecycle-Event-system.patch b/patches/api/Add-Lifecycle-Event-system.patch index e9c0558c60..a2163839ff 100644 --- a/patches/api/Add-Lifecycle-Event-system.patch +++ b/patches/api/Add-Lifecycle-Event-system.patch @@ -455,15 +455,19 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + +import io.papermc.paper.plugin.lifecycle.event.LifecycleEvent; +import io.papermc.paper.plugin.lifecycle.event.LifecycleEventOwner; ++import java.util.Optional; +import java.util.ServiceLoader; +import org.jetbrains.annotations.ApiStatus; + +@ApiStatus.Internal +interface LifecycleEventTypeProvider { + -+ LifecycleEventTypeProvider PROVIDER = ServiceLoader.load(LifecycleEventTypeProvider.class) -+ .findFirst() -+ .orElseThrow(); ++ Optional INSTANCE = ServiceLoader.load(LifecycleEventTypeProvider.class) ++ .findFirst(); ++ ++ static LifecycleEventTypeProvider provider() { ++ return INSTANCE.orElseThrow(); ++ } + + LifecycleEventType.Monitorable monitor(String name, Class ownerType); + @@ -494,33 +498,33 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + + // + @ApiStatus.Internal -+ private static LifecycleEventType.Monitorable plugin(final String name) { ++ static LifecycleEventType.Monitorable plugin(final String name) { + return monitor(name, Plugin.class); + } + + @ApiStatus.Internal -+ private static LifecycleEventType.Prioritizable pluginPrioritized(final String name) { ++ static LifecycleEventType.Prioritizable pluginPrioritized(final String name) { + return prioritized(name, Plugin.class); + } + + @ApiStatus.Internal -+ private static LifecycleEventType.Monitorable bootstrap(final String name) { ++ static LifecycleEventType.Monitorable bootstrap(final String name) { + return monitor(name, BootstrapContext.class); + } + + @ApiStatus.Internal -+ private static LifecycleEventType.Prioritizable bootstrapPrioritized(final String name) { ++ static LifecycleEventType.Prioritizable bootstrapPrioritized(final String name) { + return prioritized(name, BootstrapContext.class); + } + + @ApiStatus.Internal -+ private static LifecycleEventType.Monitorable monitor(final String name, final Class ownerType) { -+ return LifecycleEventTypeProvider.PROVIDER.monitor(name, ownerType); ++ static LifecycleEventType.Monitorable monitor(final String name, final Class ownerType) { ++ return LifecycleEventTypeProvider.provider().monitor(name, ownerType); + } + + @ApiStatus.Internal -+ private static LifecycleEventType.Prioritizable prioritized(final String name, final Class ownerType) { -+ return LifecycleEventTypeProvider.PROVIDER.prioritized(name, ownerType); ++ static LifecycleEventType.Prioritizable prioritized(final String name, final Class ownerType) { ++ return LifecycleEventTypeProvider.provider().prioritized(name, ownerType); + } + // + diff --git a/patches/api/Brigadier-based-command-API.patch b/patches/api/Brigadier-based-command-API.patch index a62ac355ea..eee696886b 100644 --- a/patches/api/Brigadier-based-command-API.patch +++ b/patches/api/Brigadier-based-command-API.patch @@ -1587,8 +1587,8 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + .findFirst(); + + static VanillaArgumentProvider provider() { -+ return PROVIDER.orElseThrow(); -+ } ++ return PROVIDER.orElseThrow(); ++ } + + ArgumentType entity(); + @@ -1942,7 +1942,7 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + // @ApiStatus.Internal - private static LifecycleEventType.Monitorable plugin(final String name) { + static LifecycleEventType.Monitorable plugin(final String name) { diff --git a/src/main/java/org/bukkit/command/Command.java b/src/main/java/org/bukkit/command/Command.java index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644 --- a/src/main/java/org/bukkit/command/Command.java diff --git a/patches/api/Registry-Modification-API.patch b/patches/api/Registry-Modification-API.patch index 706d082045..824a5a9fd9 100644 --- a/patches/api/Registry-Modification-API.patch +++ b/patches/api/Registry-Modification-API.patch @@ -788,6 +788,11 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 +@ApiStatus.Internal +@DefaultQualifier(NonNull.class) +record TagKeyImpl(RegistryKey registryKey, Key key) implements TagKey { ++ ++ @Override ++ public String toString() { ++ return "#" + this.key + " (in " + this.registryKey + ")"; ++ } +} diff --git a/src/main/java/org/bukkit/Registry.java b/src/main/java/org/bukkit/Registry.java index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644 diff --git a/patches/api/Tag-Lifecycle-Events.patch b/patches/api/Tag-Lifecycle-Events.patch new file mode 100644 index 0000000000..fc513a13c7 --- /dev/null +++ b/patches/api/Tag-Lifecycle-Events.patch @@ -0,0 +1,406 @@ +From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001 +From: Jake Potrebic +Date: Thu, 20 Jun 2024 09:40:53 -0700 +Subject: [PATCH] Tag Lifecycle Events + + +diff --git a/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEventTypeProvider.java b/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEventTypeProvider.java +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644 +--- a/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEventTypeProvider.java ++++ b/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEventTypeProvider.java +@@ -0,0 +0,0 @@ interface LifecycleEventTypeProvider { + LifecycleEventType.Monitorable monitor(String name, Class ownerType); + + LifecycleEventType.Prioritizable prioritized(String name, Class ownerType); ++ ++ TagEventTypeProvider tagProvider(); + } +diff --git a/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEvents.java b/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEvents.java +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644 +--- a/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEvents.java ++++ b/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEvents.java +@@ -0,0 +0,0 @@ public final class LifecycleEvents { + */ + public static final LifecycleEventType.Prioritizable> COMMANDS = prioritized("commands", LifecycleEventOwner.class); + ++ /** ++ * These events are for registering tags to the server's tag system. You can register a handler for these events ++ * only in {@link io.papermc.paper.plugin.bootstrap.PluginBootstrap#bootstrap(BootstrapContext)}. ++ */ ++ public static final TagEventTypeProvider TAGS = LifecycleEventTypeProvider.provider().tagProvider(); ++ + // + @ApiStatus.Internal + static LifecycleEventType.Monitorable plugin(final String name) { +diff --git a/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/TagEventTypeProvider.java b/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/TagEventTypeProvider.java +new file mode 100644 +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 +--- /dev/null ++++ b/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/TagEventTypeProvider.java +@@ -0,0 +0,0 @@ ++package io.papermc.paper.plugin.lifecycle.event.types; ++ ++import io.papermc.paper.plugin.bootstrap.BootstrapContext; ++import io.papermc.paper.plugin.lifecycle.event.registrar.ReloadableRegistrarEvent; ++import io.papermc.paper.registry.RegistryKey; ++import io.papermc.paper.tag.PostFlattenTagRegistrar; ++import io.papermc.paper.tag.PreFlattenTagRegistrar; ++import org.checkerframework.checker.nullness.qual.NonNull; ++import org.jetbrains.annotations.ApiStatus; ++ ++/** ++ * Provides event types for tag registration. ++ * ++ * @see PreFlattenTagRegistrar ++ * @see PostFlattenTagRegistrar ++ */ ++@ApiStatus.Experimental ++@ApiStatus.NonExtendable ++public interface TagEventTypeProvider { ++ ++ /** ++ * Get a prioritizable, reloadable registrar event for tags before they are flattened. ++ * ++ * @param registryKey the registry key for the tag type ++ * @return the registry event type ++ * @param the type of value in the tag ++ * @see PreFlattenTagRegistrar ++ */ ++ LifecycleEventType.@NonNull Prioritizable>> preFlatten(@NonNull RegistryKey registryKey); ++ ++ /** ++ * Get a prioritizable, reloadable registrar event for tags after they are flattened. ++ * ++ * @param registryKey the registry key for the tag type ++ * @return the registry event type ++ * @param the type of value in the tag ++ * @see PostFlattenTagRegistrar ++ */ ++ LifecycleEventType.@NonNull Prioritizable>> postFlatten(@NonNull RegistryKey registryKey); ++} +diff --git a/src/main/java/io/papermc/paper/tag/PostFlattenTagRegistrar.java b/src/main/java/io/papermc/paper/tag/PostFlattenTagRegistrar.java +new file mode 100644 +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 +--- /dev/null ++++ b/src/main/java/io/papermc/paper/tag/PostFlattenTagRegistrar.java +@@ -0,0 +0,0 @@ ++package io.papermc.paper.tag; ++ ++import io.papermc.paper.plugin.lifecycle.event.registrar.Registrar; ++import io.papermc.paper.registry.RegistryKey; ++import io.papermc.paper.registry.TypedKey; ++import io.papermc.paper.registry.tag.TagKey; ++import java.util.Collection; ++import java.util.Map; ++import org.checkerframework.checker.nullness.qual.NonNull; ++import org.jetbrains.annotations.ApiStatus; ++import org.jetbrains.annotations.Contract; ++import org.jetbrains.annotations.Unmodifiable; ++ ++/** ++ * Registrar for tags after they have been flattened. Flattened ++ * tags are tags which have any nested tags resolved to the tagged ++ * values the nested tags point to. This registrar, being a post-flatten ++ * registrar, allows for modification after that flattening has happened, when ++ * tags only point to individual entries and not other nested tags. ++ *

++ * An example of a custom enchant being registered to the vanilla ++ * {@code #minecraft:in_enchanting_table} tag. ++ *

{@code
++ * class YourBootstrapClass implements PluginBootstrap {
++ *
++ *     @Override
++ *     public void bootstrap(@NotNull BootstrapContext context) {
++ *         LifecycleEventManager manager = context.getLifecycleManager();
++ *         manager.registerEventHandler(LifecycleEvents.TAGS.postFlatten(RegistryKey.ENCHANTMENT), event -> {
++ *             final PostFlattenTagRegistrar registrar = event.registrar();
++ *             registrar.addToTag(
++ *                 EnchantmentTagKeys.IN_ENCHANTING_TABLE,
++ *                 Set.of(CUSTOM_ENCHANT)
++ *             );
++ *         });
++ *     }
++ * }
++ * }
++ * ++ * @param the type of value in the tag ++ * @see PreFlattenTagRegistrar ++ */ ++@ApiStatus.Experimental ++@ApiStatus.NonExtendable ++public interface PostFlattenTagRegistrar extends Registrar { ++ ++ /** ++ * Get the registry key for this tag registrar. ++ * ++ * @return the registry key ++ */ ++ @NonNull RegistryKey registryKey(); ++ ++ /** ++ * Get a copy of all tags currently held in this registrar. ++ * ++ * @return an immutable map of all tags ++ */ ++ @Contract(value = "-> new", pure = true) ++ @Unmodifiable @NonNull Map, Collection>> getAllTags(); ++ ++ /** ++ * Checks if this registrar has a tag with the given key. ++ * ++ * @param tagKey the key to check for ++ * @return true if the tag exists, false otherwise ++ */ ++ @Contract(pure = true) ++ boolean hasTag(@NonNull TagKey tagKey); ++ ++ /** ++ * Get the tag with the given key. Use {@link #hasTag(TagKey)} to check ++ * if a tag exists first. ++ * ++ * @param tagKey the key of the tag to get ++ * @return an immutable list of tag entries ++ * @throws java.util.NoSuchElementException if the tag does not exist ++ * @see #hasTag(TagKey) ++ */ ++ @Contract(value = "_ -> new", pure = true) ++ @Unmodifiable @NonNull Collection> getTag(@NonNull TagKey tagKey); ++ ++ /** ++ * Adds values to the given tag. If the tag does not exist, it will be created. ++ * ++ * @param tagKey the key of the tag to add to ++ * @param values the values to add ++ * @see #setTag(TagKey, Collection) ++ */ ++ @Contract(mutates = "this") ++ void addToTag(@NonNull TagKey tagKey, @NonNull Collection> values); ++ ++ /** ++ * Sets the values of the given tag. If the tag does not exist, it will be created. ++ * If the tag does exist, it will be overwritten. ++ * ++ * @param tagKey the key of the tag to set ++ * @param values the values to set ++ * @see #addToTag(TagKey, Collection) ++ */ ++ @Contract(mutates = "this") ++ void setTag(@NonNull TagKey tagKey, @NonNull Collection> values); ++} +diff --git a/src/main/java/io/papermc/paper/tag/PreFlattenTagRegistrar.java b/src/main/java/io/papermc/paper/tag/PreFlattenTagRegistrar.java +new file mode 100644 +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 +--- /dev/null ++++ b/src/main/java/io/papermc/paper/tag/PreFlattenTagRegistrar.java +@@ -0,0 +0,0 @@ ++package io.papermc.paper.tag; ++ ++import io.papermc.paper.plugin.lifecycle.event.registrar.Registrar; ++import io.papermc.paper.registry.RegistryKey; ++import io.papermc.paper.registry.tag.TagKey; ++import java.util.Collection; ++import java.util.List; ++import java.util.Map; ++import org.checkerframework.checker.nullness.qual.NonNull; ++import org.jetbrains.annotations.ApiStatus; ++import org.jetbrains.annotations.Contract; ++import org.jetbrains.annotations.Unmodifiable; ++ ++/** ++ * Registrar for tags before they are flattened. Flattened ++ * tags are tags which have any nested tags resolved to the tagged ++ * values the nested tags point to. This registrar, being a pre-flatten ++ * registrar, allows for modification before that flattening has happened, when ++ * tags both point to individual entries and other nested tags. ++ *

++ * An example of a tag being created in a pre-flatten registrar: ++ *

{@code
++ * class YourBootstrapClass implements PluginBootstrap {
++ *
++ *     @Override
++ *     public void bootstrap(@NotNull BootstrapContext context) {
++ *         LifecycleEventManager manager = context.getLifecycleManager();
++ *         manager.registerEventHandler(LifecycleEvents.TAGS.preFlatten(RegistryKey.ITEM), event -> {
++ *             final PreFlattenTagRegistrar registrar = event.registrar();
++ *             registrar.setTag(AXE_PICKAXE, Set.of(
++ *                 TagEntry.tagEntry(ItemTypeTagKeys.PICKAXES),
++ *                 TagEntry.tagEntry(ItemTypeTagKeys.AXES)
++ *             ));
++ *         });
++ *     }
++ * }
++ * }
++ * ++ * @param the type of value in the tag ++ * @see PostFlattenTagRegistrar ++ */ ++@ApiStatus.Experimental ++@ApiStatus.NonExtendable ++public interface PreFlattenTagRegistrar extends Registrar { ++ ++ /** ++ * Get the registry key for this tag registrar. ++ * ++ * @return the registry key ++ */ ++ @NonNull RegistryKey registryKey(); ++ ++ /** ++ * Get a copy of all tags currently held in this registrar. ++ * ++ * @return an immutable map of all tags ++ */ ++ @Contract(value = "-> new", pure = true) ++ @Unmodifiable @NonNull Map, Collection>> getAllTags(); ++ ++ /** ++ * Checks if this registrar has a tag with the given key. ++ * ++ * @param tagKey the key to check for ++ * @return true if the tag exists, false otherwise ++ */ ++ @Contract(pure = true) ++ boolean hasTag(@NonNull TagKey tagKey); ++ ++ /** ++ * Get the tag with the given key. Use {@link #hasTag(TagKey)} to check ++ * if a tag exists first. ++ * ++ * @param tagKey the key of the tag to get ++ * @return an immutable list of tag entries ++ * @throws java.util.NoSuchElementException if the tag does not exist ++ * @see #hasTag(TagKey) ++ */ ++ @Contract(value = "_ -> new", pure = true) ++ @Unmodifiable @NonNull List> getTag(@NonNull TagKey tagKey); ++ ++ /** ++ * Adds entries to the given tag. If the tag does not exist, it will be created. ++ * ++ * @param tagKey the key of the tag to add to ++ * @param entries the entries to add ++ * @see #setTag(TagKey, Collection) ++ */ ++ @Contract(mutates = "this") ++ void addToTag(@NonNull TagKey tagKey, @NonNull Collection> entries); ++ ++ /** ++ * Sets the entries of the given tag. If the tag does not exist, it will be created. ++ * If the tag does exist, it will be overwritten. ++ * ++ * @param tagKey the key of the tag to set ++ * @param entries the entries to set ++ * @see #addToTag(TagKey, Collection) ++ */ ++ @Contract(mutates = "this") ++ void setTag(@NonNull TagKey tagKey, @NonNull Collection> entries); ++} +diff --git a/src/main/java/io/papermc/paper/tag/TagEntry.java b/src/main/java/io/papermc/paper/tag/TagEntry.java +new file mode 100644 +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 +--- /dev/null ++++ b/src/main/java/io/papermc/paper/tag/TagEntry.java +@@ -0,0 +0,0 @@ ++package io.papermc.paper.tag; ++ ++import io.papermc.paper.registry.TypedKey; ++import io.papermc.paper.registry.tag.TagKey; ++import net.kyori.adventure.key.Keyed; ++import org.checkerframework.checker.nullness.qual.NonNull; ++import org.jetbrains.annotations.ApiStatus; ++import org.jetbrains.annotations.Contract; ++ ++/** ++ * An entry is a pre-flattened tag. Represents ++ * either an individual registry entry or a whole tag. ++ * ++ * @param the type of value in the tag ++ * @see PreFlattenTagRegistrar ++ */ ++@ApiStatus.Experimental ++@ApiStatus.NonExtendable ++public interface TagEntry extends Keyed { ++ ++ /** ++ * Create required tag entry for a single value. ++ * ++ * @param entryKey the key of the entry ++ * @return a new tag entry for a value ++ * @param the type of value ++ */ ++ @Contract(value = "_ -> new", pure = true) ++ static @NonNull TagEntry valueEntry(final @NonNull TypedKey entryKey) { ++ return valueEntry(entryKey, true); ++ } ++ ++ /** ++ * Create tag entry for a single value. ++ * ++ * @param entryKey the key of the entry ++ * @param isRequired if this entry is required (see {@link #isRequired()}) ++ * @return a new tag entry for a value ++ * @param the type of value ++ */ ++ @Contract(value = "_, _ -> new", pure = true) ++ static @NonNull TagEntry valueEntry(final @NonNull TypedKey entryKey, final boolean isRequired) { ++ return new TagEntryImpl<>(entryKey.key(), false, isRequired); ++ } ++ ++ /** ++ * Create a required tag entry for a nested tag. ++ * ++ * @param tagKey they key for the tag ++ * @return a new tag entry for a tag ++ * @param the type of value ++ */ ++ @Contract(value = "_ -> new", pure = true) ++ static @NonNull TagEntry tagEntry(final @NonNull TagKey tagKey) { ++ return tagEntry(tagKey, true); ++ } ++ ++ /** ++ * Create a tag entry for a nested tag. ++ * ++ * @param tagKey they key for the tag ++ * @param isRequired if this entry is required (see {@link #isRequired()}) ++ * @return a new tag entry for a tag ++ * @param the type of value ++ */ ++ @Contract(value = "_, _ -> new", pure = true) ++ static @NonNull TagEntry tagEntry(final @NonNull TagKey tagKey, final boolean isRequired) { ++ return new TagEntryImpl<>(tagKey.key(), true, isRequired); ++ } ++ ++ /** ++ * Returns if this entry represents a tag. ++ * ++ * @return true if this entry is a tag, false if it is an individual entry ++ */ ++ @Contract(pure = true) ++ boolean isTag(); ++ ++ /** ++ * Returns if this entry is required. If an entry is required, ++ * the value or tag must exist on the server in order for the tag ++ * to load correctly. A missing value will prevent the tag holding ++ * that missing value from being created. ++ * ++ * @return true if this entry is required, false if it is optional ++ */ ++ @Contract(pure = true) ++ boolean isRequired(); ++} +diff --git a/src/main/java/io/papermc/paper/tag/TagEntryImpl.java b/src/main/java/io/papermc/paper/tag/TagEntryImpl.java +new file mode 100644 +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 +--- /dev/null ++++ b/src/main/java/io/papermc/paper/tag/TagEntryImpl.java +@@ -0,0 +0,0 @@ ++package io.papermc.paper.tag; ++ ++import net.kyori.adventure.key.Key; ++import org.jetbrains.annotations.ApiStatus; ++ ++@ApiStatus.Internal ++record TagEntryImpl(Key key, boolean isTag, boolean isRequired) implements TagEntry { ++} diff --git a/patches/server/Add-Lifecycle-Event-system.patch b/patches/server/Add-Lifecycle-Event-system.patch index 99c6e0f131..e975dc169a 100644 --- a/patches/server/Add-Lifecycle-Event-system.patch +++ b/patches/server/Add-Lifecycle-Event-system.patch @@ -468,6 +468,8 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + } + } + ++ public abstract boolean hasHandlers(); ++ + public abstract void forEachHandler(E event, Consumer> consumer, Predicate> predicate); + + public abstract void removeMatching(Predicate> predicate); @@ -505,7 +507,7 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 +public final class LifecycleEventTypeProviderImpl implements LifecycleEventTypeProvider { + + public static LifecycleEventTypeProviderImpl instance() { -+ return (LifecycleEventTypeProviderImpl) LifecycleEventTypeProvider.PROVIDER; ++ return (LifecycleEventTypeProviderImpl) LifecycleEventTypeProvider.provider(); + } + + @Override @@ -550,6 +552,11 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + } + + @Override ++ public boolean hasHandlers() { ++ return !this.handlers.isEmpty(); ++ } ++ ++ @Override + public MonitorLifecycleEventHandlerConfigurationImpl newHandler(final LifecycleEventHandler handler) { + return new MonitorLifecycleEventHandlerConfigurationImpl<>(handler, this); + } @@ -652,6 +659,11 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + } + + @Override ++ public boolean hasHandlers() { ++ return !this.handlers.isEmpty(); ++ } ++ ++ @Override + protected void register(final O owner, final AbstractLifecycleEventHandlerConfiguration config) { + Preconditions.checkArgument(config instanceof PrioritizedLifecycleEventHandlerConfigurationImpl, "Configuration must be a PrioritizedLifecycleEventHandlerConfiguration"); + this.handlers.add(new RegisteredHandler<>(owner, config)); diff --git a/patches/server/Add-registry-entry-and-builders.patch b/patches/server/Add-registry-entry-and-builders.patch index 883e3d59ff..29d82487bd 100644 --- a/patches/server/Add-registry-entry-and-builders.patch +++ b/patches/server/Add-registry-entry-and-builders.patch @@ -96,7 +96,7 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + protected HolderSet exclusiveWith = HolderSet.empty(); // Paper added default to empty. + + // Effects -+ protected final DataComponentMap effects; ++ protected DataComponentMap effects; + + protected final Conversions conversions; + diff --git a/patches/server/Registry-Modification-API.patch b/patches/server/Registry-Modification-API.patch index ce9e8f6d8b..300c4c5717 100644 --- a/patches/server/Registry-Modification-API.patch +++ b/patches/server/Registry-Modification-API.patch @@ -134,12 +134,12 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 +import org.checkerframework.checker.nullness.qual.Nullable; +import org.intellij.lang.annotations.Subst; + -+public final class PaperRegistryListenerManager { ++public class PaperRegistryListenerManager { + + public static final PaperRegistryListenerManager INSTANCE = new PaperRegistryListenerManager(); + -+ public final RegistryEventMap valueAddHooks = new RegistryEventMap("value add"); -+ public final RegistryEventMap freezeHooks = new RegistryEventMap("freeze"); ++ public final RegistryEventMap valueAddEventTypes = new RegistryEventMap("value add"); ++ public final RegistryEventMap freezeEventTypes = new RegistryEventMap("freeze"); + + private PaperRegistryListenerManager() { + } @@ -200,7 +200,7 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + ) { + Preconditions.checkState(LaunchEntryPointHandler.INSTANCE.hasEntered(Entrypoint.BOOTSTRAPPER), registry.key() + " tried to run modification listeners before bootstrappers have been called"); // verify that bootstrappers have been called + final @Nullable RegistryEntryInfo entry = PaperRegistries.getEntry(registry.key()); -+ if (!RegistryEntry.Modifiable.isModifiable(entry) || !this.valueAddHooks.hasHooks(entry.apiKey())) { ++ if (!RegistryEntry.Modifiable.isModifiable(entry) || !this.valueAddEventTypes.hasHandlers(entry.apiKey())) { + return registerMethod.register((WritableRegistry) registry, key, nms, registrationInfo); + } + final RegistryEntry.Modifiable modifiableEntry = RegistryEntry.Modifiable.asModifiable(entry); @@ -217,7 +217,7 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + final RegistrationInfo registrationInfo, + final Conversions conversions + ) { -+ if (!RegistryEntry.Modifiable.isModifiable(entry) || !this.valueAddHooks.hasHooks(entry.apiKey())) { ++ if (!RegistryEntry.Modifiable.isModifiable(entry) || !this.valueAddEventTypes.hasHandlers(entry.apiKey())) { + registry.register(key, builder.build(), registrationInfo); + return; + } @@ -237,7 +237,7 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + @Subst("namespace:key") final ResourceLocation beingAdded = key.location(); + @SuppressWarnings("PatternValidation") final TypedKey typedKey = TypedKey.create(entry.apiKey(), Key.key(beingAdded.getNamespace(), beingAdded.getPath())); + final RegistryEntryAddEventImpl event = entry.createEntryAddEvent(typedKey, builder, conversions); -+ LifecycleEventRunner.INSTANCE.callEvent(this.valueAddHooks.getHook(entry.apiKey()), event); ++ LifecycleEventRunner.INSTANCE.callEvent(this.valueAddEventTypes.getEventType(entry.apiKey()), event); + if (oldNms != null) { + ((MappedRegistry) registry).clearIntrusiveHolder(oldNms); + } @@ -261,27 +261,27 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + + public > void runFreezeListeners(final ResourceKey> resourceKey, final Conversions conversions) { + final @Nullable RegistryEntryInfo entry = PaperRegistries.getEntry(resourceKey); -+ if (!RegistryEntry.Addable.isAddable(entry) || !this.freezeHooks.hasHooks(entry.apiKey())) { ++ if (!RegistryEntry.Addable.isAddable(entry) || !this.freezeEventTypes.hasHandlers(entry.apiKey())) { + return; + } + final RegistryEntry.Addable writableEntry = RegistryEntry.Addable.asAddable(entry); + final WritableCraftRegistry writableRegistry = PaperRegistryAccess.instance().getWritableRegistry(entry.apiKey()); + final RegistryFreezeEventImpl event = writableEntry.createFreezeEvent(writableRegistry, conversions); -+ LifecycleEventRunner.INSTANCE.callEvent(this.freezeHooks.getHook(entry.apiKey()), event); ++ LifecycleEventRunner.INSTANCE.callEvent(this.freezeEventTypes.getEventType(entry.apiKey()), event); + } + + public > RegistryEntryAddEventType getRegistryValueAddEventType(final RegistryEventProvider type) { + if (!RegistryEntry.Modifiable.isModifiable(PaperRegistries.getEntry(type.registryKey()))) { + throw new IllegalArgumentException(type.registryKey() + " does not support RegistryEntryAddEvent"); + } -+ return this.valueAddHooks.getOrCreate(type, RegistryEntryAddEventTypeImpl::new); ++ return this.valueAddEventTypes.getOrCreate(type.registryKey(), RegistryEntryAddEventTypeImpl::new); + } + + public > LifecycleEventType.Prioritizable> getRegistryFreezeEventType(final RegistryEventProvider type) { + if (!RegistryEntry.Addable.isAddable(PaperRegistries.getEntry(type.registryKey()))) { + throw new IllegalArgumentException(type.registryKey() + " does not support RegistryFreezeEvent"); + } -+ return this.freezeHooks.getOrCreate(type, RegistryLifecycleEventType::new); ++ return this.freezeEventTypes.getOrCreate(type.registryKey(), RegistryLifecycleEventType::new); + } +} diff --git a/src/main/java/io/papermc/paper/registry/WritableCraftRegistry.java b/src/main/java/io/papermc/paper/registry/WritableCraftRegistry.java @@ -702,9 +702,10 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 +package io.papermc.paper.registry.event; + +import io.papermc.paper.plugin.bootstrap.BootstrapContext; ++import io.papermc.paper.plugin.lifecycle.event.LifecycleEvent; +import io.papermc.paper.plugin.lifecycle.event.LifecycleEventRunner; ++import io.papermc.paper.plugin.lifecycle.event.types.AbstractLifecycleEventType; +import io.papermc.paper.plugin.lifecycle.event.types.LifecycleEventType; -+import io.papermc.paper.registry.RegistryBuilder; +import io.papermc.paper.registry.RegistryKey; +import java.util.HashMap; +import java.util.Map; @@ -713,7 +714,7 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + +public final class RegistryEventMap { + -+ private final Map, LifecycleEventType, ?>> hooks = new HashMap<>(); ++ private final Map, LifecycleEventType> eventTypes = new HashMap<>(); + private final String name; + + public RegistryEventMap(final String name) { @@ -721,25 +722,26 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + } + + @SuppressWarnings("unchecked") -+ public , E extends RegistryEvent, ET extends LifecycleEventType> ET getOrCreate(final RegistryEventProvider type, final BiFunction, ? super String, ET> eventTypeCreator) { -+ final ET registerHook; -+ if (this.hooks.containsKey(type.registryKey())) { -+ registerHook = (ET) this.hooks.get(type.registryKey()); ++ public > ET getOrCreate(final RegistryKey registryKey, final BiFunction, ? super String, ET> eventTypeCreator) { ++ final ET eventType; ++ if (this.eventTypes.containsKey(registryKey)) { ++ eventType = (ET) this.eventTypes.get(registryKey); + } else { -+ registerHook = eventTypeCreator.apply(type, this.name); -+ LifecycleEventRunner.INSTANCE.addEventType(registerHook); -+ this.hooks.put(type.registryKey(), registerHook); ++ eventType = eventTypeCreator.apply(registryKey, this.name); ++ LifecycleEventRunner.INSTANCE.addEventType(eventType); ++ this.eventTypes.put(registryKey, eventType); + } -+ return registerHook; ++ return eventType; + } + + @SuppressWarnings("unchecked") -+ public > LifecycleEventType getHook(final RegistryKey registryKey) { -+ return (LifecycleEventType) Objects.requireNonNull(this.hooks.get(registryKey), "No hook for " + registryKey); ++ public LifecycleEventType getEventType(final RegistryKey registryKey) { ++ return (LifecycleEventType) Objects.requireNonNull(this.eventTypes.get(registryKey), "No hook for " + registryKey); + } + -+ public boolean hasHooks(final RegistryKey registryKey) { -+ return this.hooks.containsKey(registryKey); ++ public boolean hasHandlers(final RegistryKey registryKey) { ++ final AbstractLifecycleEventType type = ((AbstractLifecycleEventType) this.eventTypes.get(registryKey)); ++ return type != null && type.hasHandlers(); + } + +} @@ -830,15 +832,15 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 +import io.papermc.paper.plugin.lifecycle.event.handler.LifecycleEventHandler; +import io.papermc.paper.plugin.lifecycle.event.types.PrioritizableLifecycleEventType; +import io.papermc.paper.registry.RegistryBuilder; ++import io.papermc.paper.registry.RegistryKey; +import io.papermc.paper.registry.event.RegistryEntryAddEvent; -+import io.papermc.paper.registry.event.RegistryEventProvider; +import java.util.function.Consumer; +import java.util.function.Predicate; + +public class RegistryEntryAddEventTypeImpl> extends PrioritizableLifecycleEventType, RegistryEntryAddConfiguration> implements RegistryEntryAddEventType { + -+ public RegistryEntryAddEventTypeImpl(final RegistryEventProvider type, final String eventName) { -+ super(type.registryKey() + " / " + eventName, BootstrapContext.class); ++ public RegistryEntryAddEventTypeImpl(final RegistryKey registryKey, final String eventName) { ++ super(registryKey + " / " + eventName, BootstrapContext.class); + } + + @Override @@ -914,14 +916,13 @@ index 0000000000000000000000000000000000000000..00000000000000000000000000000000 + +import io.papermc.paper.plugin.bootstrap.BootstrapContext; +import io.papermc.paper.plugin.lifecycle.event.types.PrioritizableLifecycleEventType; -+import io.papermc.paper.registry.RegistryBuilder; ++import io.papermc.paper.registry.RegistryKey; +import io.papermc.paper.registry.event.RegistryEvent; -+import io.papermc.paper.registry.event.RegistryEventProvider; + -+public final class RegistryLifecycleEventType, E extends RegistryEvent> extends PrioritizableLifecycleEventType.Simple { ++public final class RegistryLifecycleEventType> extends PrioritizableLifecycleEventType.Simple { + -+ public RegistryLifecycleEventType(final RegistryEventProvider type, final String eventName) { -+ super(type.registryKey() + " / " + eventName, BootstrapContext.class); ++ public RegistryLifecycleEventType(final RegistryKey registryKey, final String eventName) { ++ super(registryKey + " / " + eventName, BootstrapContext.class); + } +} diff --git a/src/main/java/io/papermc/paper/registry/legacy/DelayedRegistry.java b/src/main/java/io/papermc/paper/registry/legacy/DelayedRegistry.java diff --git a/patches/server/Tag-Lifecycle-Events.patch b/patches/server/Tag-Lifecycle-Events.patch new file mode 100644 index 0000000000..1e786da93e --- /dev/null +++ b/patches/server/Tag-Lifecycle-Events.patch @@ -0,0 +1,537 @@ +From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001 +From: Jake Potrebic +Date: Thu, 20 Jun 2024 09:40:57 -0700 +Subject: [PATCH] Tag Lifecycle Events + +== AT == +public net/minecraft/tags/TagEntry id +public net/minecraft/tags/TagEntry tag +public net/minecraft/tags/TagEntry required + +diff --git a/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEventTypeProviderImpl.java b/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEventTypeProviderImpl.java +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644 +--- a/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEventTypeProviderImpl.java ++++ b/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/LifecycleEventTypeProviderImpl.java +@@ -0,0 +0,0 @@ public final class LifecycleEventTypeProviderImpl implements LifecycleEventTypeP + return (LifecycleEventTypeProviderImpl) LifecycleEventTypeProvider.provider(); + } + ++ private final PaperTagEventTypeProvider provider = new PaperTagEventTypeProvider(); ++ + @Override + public LifecycleEventType.Monitorable monitor(final String name, final Class ownerType) { + return LifecycleEventRunner.INSTANCE.addEventType(new MonitorableLifecycleEventType<>(name, ownerType)); +@@ -0,0 +0,0 @@ public final class LifecycleEventTypeProviderImpl implements LifecycleEventTypeP + public LifecycleEventType.Prioritizable prioritized(final String name, final Class ownerType) { + return LifecycleEventRunner.INSTANCE.addEventType(new PrioritizableLifecycleEventType.Simple<>(name, ownerType)); + } ++ ++ @Override ++ public PaperTagEventTypeProvider tagProvider() { ++ return this.provider; ++ } + } +diff --git a/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/PaperTagEventTypeProvider.java b/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/PaperTagEventTypeProvider.java +new file mode 100644 +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 +--- /dev/null ++++ b/src/main/java/io/papermc/paper/plugin/lifecycle/event/types/PaperTagEventTypeProvider.java +@@ -0,0 +0,0 @@ ++package io.papermc.paper.plugin.lifecycle.event.types; ++ ++import io.papermc.paper.plugin.bootstrap.BootstrapContext; ++import io.papermc.paper.plugin.lifecycle.event.registrar.ReloadableRegistrarEvent; ++import io.papermc.paper.registry.RegistryKey; ++import io.papermc.paper.tag.PaperTagListenerManager; ++import io.papermc.paper.tag.PostFlattenTagRegistrar; ++import io.papermc.paper.tag.PreFlattenTagRegistrar; ++import org.checkerframework.checker.nullness.qual.NonNull; ++import org.checkerframework.framework.qual.DefaultQualifier; ++ ++@DefaultQualifier(NonNull.class) ++public class PaperTagEventTypeProvider implements TagEventTypeProvider { ++ ++ @Override ++ public PrioritizableLifecycleEventType.Simple>> preFlatten(final RegistryKey registryKey) { ++ return PaperTagListenerManager.INSTANCE.getPreFlattenType(registryKey); ++ } ++ ++ @Override ++ public PrioritizableLifecycleEventType.Simple>> postFlatten(final RegistryKey registryKey) { ++ return PaperTagListenerManager.INSTANCE.getPostFlattenType(registryKey); ++ } ++} +diff --git a/src/main/java/io/papermc/paper/tag/PaperPostFlattenTagRegistrar.java b/src/main/java/io/papermc/paper/tag/PaperPostFlattenTagRegistrar.java +new file mode 100644 +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 +--- /dev/null ++++ b/src/main/java/io/papermc/paper/tag/PaperPostFlattenTagRegistrar.java +@@ -0,0 +0,0 @@ ++package io.papermc.paper.tag; ++ ++import com.google.common.collect.Collections2; ++import com.google.common.collect.ImmutableMap; ++import io.papermc.paper.adventure.PaperAdventure; ++import io.papermc.paper.plugin.bootstrap.BootstrapContext; ++import io.papermc.paper.plugin.lifecycle.event.registrar.PaperRegistrar; ++import io.papermc.paper.registry.RegistryKey; ++import io.papermc.paper.registry.TypedKey; ++import io.papermc.paper.registry.tag.TagKey; ++import java.util.ArrayList; ++import java.util.Collection; ++import java.util.Collections; ++import java.util.List; ++import java.util.Map; ++import java.util.NoSuchElementException; ++import java.util.Optional; ++import java.util.function.Function; ++import net.minecraft.resources.ResourceLocation; ++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; ++ ++@SuppressWarnings("BoundedWildcard") ++@DefaultQualifier(NonNull.class) ++public class PaperPostFlattenTagRegistrar implements PaperRegistrar, PostFlattenTagRegistrar { ++ ++ public final Map> tags; ++ private final Function> fromIdConverter; ++ private final Function toIdConverter; ++ private final RegistryKey registryKey; ++ ++ public PaperPostFlattenTagRegistrar( ++ final Map> tags, ++ final TagEventConfig config ++ ) { ++ this.tags = tags; ++ this.fromIdConverter = config.fromIdConverter(); ++ this.toIdConverter = config.toIdConverter(); ++ this.registryKey = config.apiRegistryKey(); ++ } ++ ++ @Override ++ public void setCurrentContext(final @Nullable BootstrapContext owner) { ++ } ++ ++ @Override ++ public RegistryKey registryKey() { ++ return this.registryKey; ++ } ++ ++ @Override ++ public Map, Collection>> getAllTags() { ++ final ImmutableMap.Builder, Collection>> tags = ImmutableMap.builderWithExpectedSize(this.tags.size()); ++ for (final Map.Entry> entry : this.tags.entrySet()) { ++ final TagKey key = TagKey.create(this.registryKey, CraftNamespacedKey.fromMinecraft(entry.getKey())); ++ tags.put(key, this.convert(entry.getValue())); ++ } ++ return tags.build(); ++ } ++ ++ private Collection> convert(final Collection nms) { ++ return Collections.unmodifiableCollection( ++ Collections2.transform(nms, m -> this.convert(this.toIdConverter.apply(m))) ++ ); ++ } ++ ++ private TypedKey convert(final ResourceLocation location) { ++ return TypedKey.create(this.registryKey, CraftNamespacedKey.fromMinecraft(location)); ++ } ++ ++ private M convert(final TypedKey key) { ++ final Optional optional = this.fromIdConverter.apply(PaperAdventure.asVanilla(key.key())); ++ if (optional.isEmpty()) { ++ throw new IllegalArgumentException(key + " doesn't exist"); ++ } ++ return optional.get(); ++ } ++ ++ @Override ++ public boolean hasTag(final TagKey tagKey) { ++ return this.tags.containsKey(PaperAdventure.asVanilla(tagKey.key())); ++ } ++ ++ private Collection getNmsTag(final TagKey tagKey, final boolean create) { ++ final ResourceLocation vanillaKey = PaperAdventure.asVanilla(tagKey.key()); ++ Collection tag = this.tags.get(vanillaKey); ++ if (tag == null) { ++ if (create) { ++ tag = this.tags.computeIfAbsent(vanillaKey, k -> new ArrayList<>()); ++ } else { ++ throw new NoSuchElementException("Tag " + tagKey + " is not present"); ++ } ++ } ++ return tag; ++ } ++ ++ @Override ++ public Collection> getTag(final TagKey tagKey) { ++ return this.convert(this.getNmsTag(tagKey, false)); ++ } ++ ++ @Override ++ public void addToTag(final TagKey tagKey, final Collection> values) { ++ final Collection nmsTag = new ArrayList<>(this.getNmsTag(tagKey, true)); ++ for (final TypedKey key : values) { ++ nmsTag.add(this.convert(key)); ++ } ++ this.tags.put(PaperAdventure.asVanilla(tagKey.key()), nmsTag); ++ } ++ ++ @Override ++ public void setTag(final TagKey tagKey, final Collection> values) { ++ final List newList = List.copyOf(Collections2.transform(values, this::convert)); ++ this.tags.put(PaperAdventure.asVanilla(tagKey.key()), newList); ++ } ++} +diff --git a/src/main/java/io/papermc/paper/tag/PaperPreFlattenTagRegistrar.java b/src/main/java/io/papermc/paper/tag/PaperPreFlattenTagRegistrar.java +new file mode 100644 +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 +--- /dev/null ++++ b/src/main/java/io/papermc/paper/tag/PaperPreFlattenTagRegistrar.java +@@ -0,0 +0,0 @@ ++package io.papermc.paper.tag; ++ ++import com.google.common.base.Preconditions; ++import com.google.common.collect.Collections2; ++import com.google.common.collect.ImmutableMap; ++import com.google.common.collect.Lists; ++import io.papermc.paper.adventure.PaperAdventure; ++import io.papermc.paper.plugin.bootstrap.BootstrapContext; ++import io.papermc.paper.plugin.lifecycle.event.registrar.PaperRegistrar; ++import io.papermc.paper.registry.RegistryKey; ++import io.papermc.paper.registry.tag.TagKey; ++import java.util.ArrayList; ++import java.util.Collection; ++import java.util.Collections; ++import java.util.HashMap; ++import java.util.List; ++import java.util.Map; ++import java.util.NoSuchElementException; ++import net.minecraft.resources.ResourceLocation; ++import net.minecraft.tags.TagLoader; ++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; ++ ++@SuppressWarnings("BoundedWildcard") ++@DefaultQualifier(NonNull.class) ++public class PaperPreFlattenTagRegistrar implements PaperRegistrar, PreFlattenTagRegistrar { ++ ++ public final Map> tags; ++ private final RegistryKey registryKey; ++ ++ private @Nullable BootstrapContext owner; ++ ++ public PaperPreFlattenTagRegistrar( ++ final Map> tags, ++ final TagEventConfig config ++ ) { ++ this.tags = new HashMap<>(tags); ++ this.registryKey = config.apiRegistryKey(); ++ } ++ ++ @Override ++ public void setCurrentContext(final @Nullable BootstrapContext owner) { ++ this.owner = owner; ++ } ++ ++ @Override ++ public RegistryKey registryKey() { ++ return this.registryKey; ++ } ++ ++ @Override ++ public Map, Collection>> getAllTags() { ++ final ImmutableMap.Builder, Collection>> builder = ImmutableMap.builderWithExpectedSize(this.tags.size()); ++ for (final Map.Entry> entry : this.tags.entrySet()) { ++ final TagKey key = TagKey.create(this.registryKey, CraftNamespacedKey.fromMinecraft(entry.getKey())); ++ builder.put(key, convert(entry.getValue())); ++ } ++ return builder.build(); ++ } ++ ++ private static List> convert(final List nmsEntries) { ++ return Collections.unmodifiableList(Lists.transform(nmsEntries, PaperPreFlattenTagRegistrar::convert)); ++ } ++ ++ private static io.papermc.paper.tag.TagEntry convert(final TagLoader.EntryWithSource nmsEntry) { ++ return new TagEntryImpl<>(CraftNamespacedKey.fromMinecraft(nmsEntry.entry().id), nmsEntry.entry().tag, nmsEntry.entry().required); ++ } ++ ++ private TagLoader.EntryWithSource convert(final TagEntry entry) { ++ Preconditions.checkState(this.owner != null, "Owner is not set"); ++ final ResourceLocation vanilla = PaperAdventure.asVanilla(entry.key()); ++ final net.minecraft.tags.TagEntry nmsEntry; ++ if (entry.isTag()) { ++ if (entry.isRequired()) { ++ nmsEntry = net.minecraft.tags.TagEntry.tag(vanilla); ++ } else { ++ nmsEntry = net.minecraft.tags.TagEntry.optionalTag(vanilla); ++ } ++ } else { ++ if (entry.isRequired()) { ++ nmsEntry = net.minecraft.tags.TagEntry.element(vanilla); ++ } else { ++ nmsEntry = net.minecraft.tags.TagEntry.optionalElement(vanilla); ++ } ++ } ++ return new TagLoader.EntryWithSource(nmsEntry, this.owner.getPluginMeta().getDisplayName()); ++ } ++ ++ @Override ++ public boolean hasTag(final TagKey tagKey) { ++ return this.tags.containsKey(PaperAdventure.asVanilla(tagKey.key())); ++ } ++ ++ private List getNmsTag(final TagKey tagKey, boolean create) { ++ final ResourceLocation vanillaKey = PaperAdventure.asVanilla(tagKey.key()); ++ List tag = this.tags.get(vanillaKey); ++ if (tag == null) { ++ if (create) { ++ tag = this.tags.computeIfAbsent(vanillaKey, k -> new ArrayList<>()); ++ } else { ++ throw new NoSuchElementException("Tag " + tagKey + " is not present"); ++ } ++ } ++ return tag; ++ } ++ ++ @Override ++ public List> getTag(final TagKey tagKey) { ++ return convert(this.getNmsTag(tagKey, false)); ++ } ++ ++ @Override ++ public void addToTag(final TagKey tagKey, final Collection> entries) { ++ final List nmsTag = new ArrayList<>(this.getNmsTag(tagKey, true)); ++ for (final TagEntry tagEntry : entries) { ++ nmsTag.add(this.convert(tagEntry)); ++ } ++ this.tags.put(PaperAdventure.asVanilla(tagKey.key()), nmsTag); ++ } ++ ++ @Override ++ public void setTag(final TagKey tagKey, final Collection> entries) { ++ final List newList = List.copyOf(Collections2.transform(entries, this::convert)); ++ this.tags.put(PaperAdventure.asVanilla(tagKey.key()), newList); ++ } ++} +diff --git a/src/main/java/io/papermc/paper/tag/PaperTagListenerManager.java b/src/main/java/io/papermc/paper/tag/PaperTagListenerManager.java +new file mode 100644 +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 +--- /dev/null ++++ b/src/main/java/io/papermc/paper/tag/PaperTagListenerManager.java +@@ -0,0 +0,0 @@ ++package io.papermc.paper.tag; ++ ++import io.papermc.paper.plugin.bootstrap.BootstrapContext; ++import io.papermc.paper.plugin.lifecycle.event.LifecycleEventRunner; ++import io.papermc.paper.plugin.lifecycle.event.registrar.ReloadableRegistrarEvent; ++import io.papermc.paper.plugin.lifecycle.event.types.AbstractLifecycleEventType; ++import io.papermc.paper.plugin.lifecycle.event.types.PrioritizableLifecycleEventType; ++import io.papermc.paper.registry.PaperRegistries; ++import io.papermc.paper.registry.RegistryKey; ++import io.papermc.paper.registry.event.RegistryEventMap; ++import java.util.Collection; ++import java.util.List; ++import java.util.Map; ++import net.minecraft.core.Holder; ++import net.minecraft.core.Registry; ++import net.minecraft.resources.ResourceLocation; ++import net.minecraft.tags.TagLoader; ++import org.checkerframework.checker.nullness.qual.NonNull; ++import org.checkerframework.checker.nullness.qual.Nullable; ++import org.checkerframework.framework.qual.DefaultQualifier; ++ ++@DefaultQualifier(NonNull.class) ++public class PaperTagListenerManager { ++ ++ public static final String PRE_FLATTEN_EVENT_NAME = "pre-flatten"; ++ public static final String POST_FLATTEN_EVENT_NAME = "post-flatten"; ++ ++ public static final PaperTagListenerManager INSTANCE = new PaperTagListenerManager(); ++ ++ public final RegistryEventMap preFlatten = new RegistryEventMap(PRE_FLATTEN_EVENT_NAME); ++ public final RegistryEventMap postFlatten = new RegistryEventMap(POST_FLATTEN_EVENT_NAME); ++ ++ private PaperTagListenerManager() { ++ } ++ ++ public Map> firePreFlattenEvent( ++ final Map> initial, ++ final @Nullable TagEventConfig config ++ ) { ++ if (config == null || config.preFlatten() == null || !config.preFlatten().hasHandlers()) { ++ return initial; ++ } ++ final PaperPreFlattenTagRegistrar registrar = new PaperPreFlattenTagRegistrar<>(initial, config); ++ LifecycleEventRunner.INSTANCE.callReloadableRegistrarEvent( ++ config.preFlatten(), ++ registrar, ++ BootstrapContext.class, ++ config.cause() ++ ); ++ return Map.copyOf(registrar.tags); ++ } ++ ++ public Map> firePostFlattenEvent( ++ final Map> initial, ++ final @Nullable TagEventConfig config ++ ) { ++ if (config == null || config.postFlatten() == null || !config.postFlatten().hasHandlers()) { ++ return initial; ++ } ++ final PaperPostFlattenTagRegistrar registrar = new PaperPostFlattenTagRegistrar<>(initial, config); ++ LifecycleEventRunner.INSTANCE.callReloadableRegistrarEvent( ++ config.postFlatten(), ++ registrar, ++ BootstrapContext.class, ++ config.cause() ++ ); ++ return Map.copyOf(registrar.tags); ++ } ++ ++ public @Nullable TagEventConfig, B> createEventConfig(final Registry registry, final ReloadableRegistrarEvent.Cause cause) { ++ if (PaperRegistries.getEntry(registry.key()) == null) { ++ // TODO probably should be able to modify every registry ++ return null; ++ } ++ final RegistryKey registryKey = PaperRegistries.registryFromNms(registry.key()); ++ @Nullable AbstractLifecycleEventType>, ?> preFlatten = null; ++ if (this.preFlatten.hasHandlers(registryKey)) { ++ preFlatten = (AbstractLifecycleEventType>, ?>) this.preFlatten.>>getEventType(registryKey); ++ } ++ @Nullable AbstractLifecycleEventType>, ?> postFlatten = null; ++ if (this.postFlatten.hasHandlers(registryKey)) { ++ postFlatten = (AbstractLifecycleEventType>, ?>) this.postFlatten.>>getEventType(registryKey); ++ } ++ return new TagEventConfig<>( ++ preFlatten, ++ postFlatten, ++ cause, ++ registry::getHolder, ++ h -> ((Holder.Reference) h).key().location(), ++ PaperRegistries.registryFromNms(registry.key()) ++ ); ++ } ++ ++ public PrioritizableLifecycleEventType.Simple>> getPreFlattenType(final RegistryKey registryKey) { ++ return this.preFlatten.getOrCreate(registryKey, (ignored, name) -> { ++ return new PrioritizableLifecycleEventType.Simple<>(name, BootstrapContext.class); ++ }); ++ } ++ ++ public PrioritizableLifecycleEventType.Simple>> getPostFlattenType(final RegistryKey registryKey) { ++ return this.postFlatten.getOrCreate(registryKey, (ignored, name) -> { ++ return new PrioritizableLifecycleEventType.Simple<>(name, BootstrapContext.class); ++ }); ++ } ++} +diff --git a/src/main/java/io/papermc/paper/tag/TagEventConfig.java b/src/main/java/io/papermc/paper/tag/TagEventConfig.java +new file mode 100644 +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 +--- /dev/null ++++ b/src/main/java/io/papermc/paper/tag/TagEventConfig.java +@@ -0,0 +0,0 @@ ++package io.papermc.paper.tag; ++ ++import io.papermc.paper.plugin.bootstrap.BootstrapContext; ++import io.papermc.paper.plugin.lifecycle.event.registrar.ReloadableRegistrarEvent; ++import io.papermc.paper.plugin.lifecycle.event.types.AbstractLifecycleEventType; ++import io.papermc.paper.registry.RegistryKey; ++import java.util.Optional; ++import java.util.function.Function; ++import net.minecraft.resources.ResourceLocation; ++import org.checkerframework.checker.nullness.qual.NonNull; ++import org.checkerframework.checker.nullness.qual.Nullable; ++import org.checkerframework.framework.qual.DefaultQualifier; ++ ++@DefaultQualifier(NonNull.class) ++public record TagEventConfig( ++ @Nullable AbstractLifecycleEventType>, ?> preFlatten, ++ @Nullable AbstractLifecycleEventType>, ?> postFlatten, ++ ReloadableRegistrarEvent.Cause cause, ++ Function> fromIdConverter, ++ Function toIdConverter, ++ RegistryKey apiRegistryKey ++) { ++} +diff --git a/src/main/java/net/minecraft/server/ServerFunctionLibrary.java b/src/main/java/net/minecraft/server/ServerFunctionLibrary.java +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644 +--- a/src/main/java/net/minecraft/server/ServerFunctionLibrary.java ++++ b/src/main/java/net/minecraft/server/ServerFunctionLibrary.java +@@ -0,0 +0,0 @@ public class ServerFunctionLibrary implements PreparableReloadListener { + return null; + }).join()); + this.functions = builder.build(); +- this.tags = this.tagsLoader.build((Map>)intermediate.getFirst()); ++ this.tags = this.tagsLoader.build((Map>)intermediate.getFirst(), null); // Paper - command function tags are not implemented yet + }, + applyExecutor + ); +diff --git a/src/main/java/net/minecraft/tags/TagLoader.java b/src/main/java/net/minecraft/tags/TagLoader.java +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644 +--- a/src/main/java/net/minecraft/tags/TagLoader.java ++++ b/src/main/java/net/minecraft/tags/TagLoader.java +@@ -0,0 +0,0 @@ public class TagLoader { + return list.isEmpty() ? Either.right(builder.build()) : Either.left(list); + } + +- public Map> build(Map> tags) { ++ // Paper start - fire tag registrar events ++ public Map> build(Map> tags, @Nullable io.papermc.paper.tag.TagEventConfig eventConfig) { ++ tags = io.papermc.paper.tag.PaperTagListenerManager.INSTANCE.firePreFlattenEvent(tags, eventConfig); ++ // Paper end - fire tag registrar events + final Map> map = Maps.newHashMap(); + TagEntry.Lookup lookup = new TagEntry.Lookup() { + @Nullable +@@ -0,0 +0,0 @@ public class TagLoader { + ) + .ifRight(resolvedEntries -> map.put(id, (Collection)resolvedEntries)) + ); +- return map; ++ return io.papermc.paper.tag.PaperTagListenerManager.INSTANCE.firePostFlattenEvent(map, eventConfig); // Paper - fire tag registrar events + } + +- public Map> loadAndBuild(ResourceManager manager) { +- return this.build(this.load(manager)); ++ // Paper start - fire tag registrar events ++ public Map> loadAndBuild(ResourceManager manager, @Nullable io.papermc.paper.tag.TagEventConfig eventConfig) { ++ return this.build(this.load(manager), eventConfig); ++ // Paper end - fire tag registrar events + } + + public static record EntryWithSource(TagEntry entry, String source) { +diff --git a/src/main/java/net/minecraft/tags/TagManager.java b/src/main/java/net/minecraft/tags/TagManager.java +index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644 +--- a/src/main/java/net/minecraft/tags/TagManager.java ++++ b/src/main/java/net/minecraft/tags/TagManager.java +@@ -0,0 +0,0 @@ public class TagManager implements PreparableReloadListener { + ) { + List>> list = this.registryAccess + .registries() +- .map(registry -> this.createLoader(manager, prepareExecutor, (RegistryAccess.RegistryEntry)registry)) ++ .map(registry -> this.createLoader(manager, prepareExecutor, (RegistryAccess.RegistryEntry)registry, applyExecutor instanceof net.minecraft.server.MinecraftServer ? io.papermc.paper.plugin.lifecycle.event.registrar.ReloadableRegistrarEvent.Cause.INITIAL : io.papermc.paper.plugin.lifecycle.event.registrar.ReloadableRegistrarEvent.Cause.RELOAD)) // Paper - add registrar event cause + .toList(); + return CompletableFuture.allOf(list.toArray(CompletableFuture[]::new)) + .thenCompose(synchronizer::wait) +@@ -0,0 +0,0 @@ public class TagManager implements PreparableReloadListener { + + private CompletableFuture> createLoader( + ResourceManager resourceManager, Executor prepareExecutor, RegistryAccess.RegistryEntry requirement ++ , io.papermc.paper.plugin.lifecycle.event.registrar.ReloadableRegistrarEvent.Cause cause // Paper - add registrar event cause + ) { + ResourceKey> resourceKey = requirement.key(); + Registry registry = requirement.value(); + TagLoader> tagLoader = new TagLoader<>(registry::getHolder, Registries.tagsDirPath(resourceKey)); +- return CompletableFuture.supplyAsync(() -> new TagManager.LoadResult<>(resourceKey, tagLoader.loadAndBuild(resourceManager)), prepareExecutor); ++ // Paper start - fire tag registrar events ++ final io.papermc.paper.tag.TagEventConfig, ?> config = io.papermc.paper.tag.PaperTagListenerManager.INSTANCE.createEventConfig(registry, cause); ++ return CompletableFuture.supplyAsync(() -> new TagManager.LoadResult<>(resourceKey, tagLoader.loadAndBuild(resourceManager, config)), prepareExecutor); ++ // Paper end - fire tag registrar events + } + + public static record LoadResult(ResourceKey> key, Map>> tags) {