Render the colors of the entities spawn egg in the filled sample take…

…r texture

src/client/java/top/offsetmonkey538/compactmobfarms/CompactMobFarmsClient.java

@@ -3,15 +3,18 @@
 import net.fabricmc.api.ClientModInitializer;
 import net.fabricmc.fabric.api.blockrenderlayer.v1.BlockRenderLayerMap;
 import net.fabricmc.fabric.api.client.networking.v1.ClientPlayNetworking;
+import net.fabricmc.fabric.api.client.rendering.v1.ColorProviderRegistry;
 import net.minecraft.client.gui.screen.ingame.HandledScreens;
 import net.minecraft.client.item.ModelPredicateProviderRegistry;
 import net.minecraft.client.render.RenderLayer;
 import net.minecraft.client.render.block.entity.BlockEntityRendererFactories;
 import net.minecraft.entity.EntityType;
+import net.minecraft.item.SpawnEggItem;
 import net.minecraft.registry.Registries;
 import top.offsetmonkey538.compactmobfarms.block.ModBlocks;
 import top.offsetmonkey538.compactmobfarms.block.entity.ModBlockEntityTypes;
 import top.offsetmonkey538.compactmobfarms.client.gui.screen.ingame.CompactMobFarmScreen;
+import top.offsetmonkey538.compactmobfarms.item.FilledSampleTakerItem;
 import top.offsetmonkey538.compactmobfarms.item.ModItems;
 import top.offsetmonkey538.compactmobfarms.item.SampleTakerItem;
 import top.offsetmonkey538.compactmobfarms.network.ModPackets;
@@ -29,7 +32,15 @@ public void onInitializeClient() {
 		BlockEntityRendererFactories.register(ModBlockEntityTypes.COMPACT_MOB_FARM, CompactMobFarmBlockEntityRenderer::new);
 		HandledScreens.register(ModScreenHandlers.COMPACT_MOB_FARM_SCREEN_HANDLER, CompactMobFarmScreen::new);
 
+		ColorProviderRegistry.ITEM.register((stack, tintIndex) -> {
+			if (tintIndex >= 2) return -1;
 
+			final SpawnEggItem spawnEgg = SpawnEggItem.forEntity(FilledSampleTakerItem.getSampledEntityType(stack));
+
+			if (spawnEgg == null) return -1;
+
+			return spawnEgg.getColor(tintIndex);
+		}, ModItems.FILLED_SAMPLE_TAKER);
 
 		ClientPlayNetworking.registerGlobalReceiver(ModPackets.GUI_ENTITY_CHANGED, (client, handler, buf, responseSender) -> {
 			if (!(client.currentScreen instanceof CompactMobFarmScreen screen)) return;

src/main/java/top/offsetmonkey538/compactmobfarms/datagen/ModModelProvider.java

@@ -32,7 +32,14 @@ public void generateBlockStateModels(BlockStateModelGenerator blockStateModelGen
     public void generateItemModels(ItemModelGenerator itemModelGenerator) {
         itemModelGenerator.register(ModItems.SPIRIT_BOTTLE, Models.GENERATED);
         itemModelGenerator.register(ModItems.SPAWNER_SHARD, Models.GENERATED);
-        itemModelGenerator.register(ModItems.FILLED_SAMPLE_TAKER, Models.GENERATED);
+        Models.GENERATED.upload(
+                ModelIds.getItemModelId(ModItems.FILLED_SAMPLE_TAKER),
+                TextureMap
+                        .layer0(ModItems.FILLED_SAMPLE_TAKER)
+                        .register(TextureKey.LAYER1, TextureMap.getSubId(ModItems.FILLED_SAMPLE_TAKER, "_layer_1"))
+                        .register(TextureKey.LAYER2, TextureMap.getSubId(ModItems.FILLED_SAMPLE_TAKER, "_overlay")),
+                itemModelGenerator.writer
+        );
 
         itemModelGenerator.writer.accept(ModelIds.getItemModelId(ModItems.COMPACT_MOB_FARM), new SimpleModelSupplier(ModelIds.getBlockModelId(ModBlocks.COMPACT_MOB_FARM)));
         Models.GENERATED.upload(ModelIds.getItemModelId(ModItems.SAMPLE_TAKER), TextureMap.layer0(ModItems.SAMPLE_TAKER), itemModelGenerator.writer, this::sampleTakerJsonGenerator);