From 011668365ecd74301a1c564533768a0e7cf56f18 Mon Sep 17 00:00:00 2001
From: shedaniel <daniel@shedaniel.me>
Date: Mon, 29 May 2023 14:04:48 +0800
Subject: Close #1391

---
 .../me/shedaniel/rei/plugin/client/entry/ItemEntryDefinition.java  | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'runtime/src/main/java')

diff --git a/runtime/src/main/java/me/shedaniel/rei/plugin/client/entry/ItemEntryDefinition.java b/runtime/src/main/java/me/shedaniel/rei/plugin/client/entry/ItemEntryDefinition.java
index b41884343..a866cff8e 100644
--- a/runtime/src/main/java/me/shedaniel/rei/plugin/client/entry/ItemEntryDefinition.java
+++ b/runtime/src/main/java/me/shedaniel/rei/plugin/client/entry/ItemEntryDefinition.java
@@ -63,6 +63,7 @@ import net.minecraft.network.chat.Component;
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.tags.TagKey;
 import net.minecraft.world.inventory.tooltip.TooltipComponent;
+import net.minecraft.world.item.BlockItem;
 import net.minecraft.world.item.Item;
 import net.minecraft.world.item.ItemStack;
 import org.jetbrains.annotations.Nullable;
@@ -215,7 +216,11 @@ public class ItemEntryDefinition implements EntryDefinition<ItemStack>, EntrySer
     
     @Override
     public Stream<? extends TagKey<?>> getTagsFor(EntryStack<ItemStack> entry, ItemStack value) {
-        return value.getTags();
+        Stream<? extends TagKey<?>> tags = value.getTags();
+        if (value.getItem() instanceof BlockItem blockItem) {
+            tags = Stream.concat(tags, blockItem.getBlock().builtInRegistryHolder().tags());
+        }
+        return tags;
     }
     
     @Environment(EnvType.CLIENT)
-- 
cgit