Allow registering custom ColorResolvers (#8880)

src/main/java/net/minecraftforge/client/ColorResolverManager.java

@@ -0,0 +1,51 @@
+/*
+ * Copyright (c) Forge Development LLC and contributors
+ * SPDX-License-Identifier: LGPL-2.1-only
+ */
+
+package net.minecraftforge.client;
+
+import com.google.common.collect.ImmutableList;
+import net.minecraft.client.color.block.BlockTintCache;
+import net.minecraft.client.multiplayer.ClientLevel;
+import net.minecraft.world.level.ColorResolver;
+import net.minecraftforge.client.event.RegisterColorHandlersEvent;
+import net.minecraftforge.fml.ModLoader;
+import org.jetbrains.annotations.ApiStatus;
+
+import java.util.Map;
+
+/**
+ * Manager for custom {@link ColorResolver} instances, collected via {@link RegisterColorHandlersEvent.ColorResolvers}.
+ */
+public final class ColorResolverManager
+{
+
+    private static ImmutableList<ColorResolver> colorResolvers;
+
+    @ApiStatus.Internal
+    public static void init()
+    {
+        ImmutableList.Builder<ColorResolver> builder = ImmutableList.builder();
+        ModLoader.get().postEvent(new RegisterColorHandlersEvent.ColorResolvers(builder));
+        colorResolvers = builder.build();
+    }
+
+    /**
+     * Register a {@link BlockTintCache} for every registered {@link ColorResolver} into the given target map.
+     *
+     * @param level the level to use
+     * @param target the map to populate
+     */
+    public static void registerBlockTintCaches(ClientLevel level, Map<ColorResolver, BlockTintCache> target)
+    {
+        for (var resolver : colorResolvers)
+        {
+            target.put(resolver, new BlockTintCache(pos -> level.calculateBlockTint(pos, resolver)));
+        }
+    }
+
+    private ColorResolverManager()
+    {
+    }
+}

src/main/java/net/minecraftforge/client/event/RegisterColorHandlersEvent.java

@@ -5,10 +5,13 @@
 
 package net.minecraftforge.client.event;
 
+import com.google.common.collect.ImmutableList;
 import net.minecraft.client.color.block.BlockColor;
 import net.minecraft.client.color.block.BlockColors;
 import net.minecraft.client.color.item.ItemColor;
 import net.minecraft.client.color.item.ItemColors;
+import net.minecraft.core.BlockPos;
+import net.minecraft.world.level.ColorResolver;
 import net.minecraft.world.level.ItemLike;
 import net.minecraftforge.eventbus.api.Cancelable;
 import net.minecraftforge.eventbus.api.Event;
@@ -127,4 +130,26 @@ public abstract class RegisterColorHandlersEvent extends Event implements IModBu
             itemColors.register(itemColor, items);
         }
     }
+
+    /**
+     * Allows registration of custom {@link ColorResolver} implementations to be used with
+     * {@link net.minecraft.world.level.BlockAndTintGetter#getBlockTint(BlockPos, ColorResolver)}.
+     */
+    public static class ColorResolvers extends RegisterColorHandlersEvent
+    {
+        private final ImmutableList.Builder<ColorResolver> builder;
+
+        @ApiStatus.Internal
+        public ColorResolvers(ImmutableList.Builder<ColorResolver> builder)
+        {
+            this.builder = builder;
+        }
+
+        public void register(ColorResolver resolver)
+        {
+            this.builder.add(resolver);
+        }
+
+    }
+
 }