From 024ee5881b8907e6f303bb73b239bbfebabb5428 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florian=20N=C3=BCcke?= <florian@nuecke.de>
Date: Tue, 8 Feb 2022 13:55:19 +0100
Subject: [PATCH] Add JEI provider for differently configured items.

Includes computers, robots, hard drives, flash memory.
---
 .../integration/jei/ExtraItemsJEIPlugin.java  | 61 +++++++++++++++++++
 .../common/item/AbstractBlockDeviceItem.java  |  2 +-
 2 files changed, 62 insertions(+), 1 deletion(-)
 create mode 100644 src/main/java/li/cil/oc2/common/integration/jei/ExtraItemsJEIPlugin.java

diff --git a/src/main/java/li/cil/oc2/common/integration/jei/ExtraItemsJEIPlugin.java b/src/main/java/li/cil/oc2/common/integration/jei/ExtraItemsJEIPlugin.java
new file mode 100644
index 00000000..2acfdab5
--- /dev/null
+++ b/src/main/java/li/cil/oc2/common/integration/jei/ExtraItemsJEIPlugin.java
@@ -0,0 +1,61 @@
+/* SPDX-License-Identifier: MIT */
+
+package li.cil.oc2.common.integration.jei;
+
+import com.google.common.base.Strings;
+import li.cil.oc2.api.API;
+import li.cil.oc2.common.item.AbstractBlockDeviceItem;
+import li.cil.oc2.common.item.Items;
+import li.cil.oc2.common.util.ItemStackUtils;
+import li.cil.oc2.common.util.NBTUtils;
+import mezz.jei.api.IModPlugin;
+import mezz.jei.api.JeiPlugin;
+import mezz.jei.api.ingredients.subtypes.IIngredientSubtypeInterpreter;
+import mezz.jei.api.ingredients.subtypes.UidContext;
+import mezz.jei.api.registration.ISubtypeRegistration;
+import net.minecraft.nbt.CompoundTag;
+import net.minecraft.resources.ResourceLocation;
+import net.minecraft.world.item.ItemStack;
+
+import static li.cil.oc2.common.Constants.BLOCK_ENTITY_TAG_NAME_IN_ITEM;
+import static li.cil.oc2.common.Constants.ITEMS_TAG_NAME;
+
+@JeiPlugin
+public class ExtraItemsJEIPlugin implements IModPlugin {
+    @Override
+    public ResourceLocation getPluginUid() {
+        return new ResourceLocation(API.MOD_ID, "extra_items");
+    }
+
+    @Override
+    public void registerItemSubtypes(final ISubtypeRegistration registration) {
+        registration.registerSubtypeInterpreter(Items.COMPUTER.get(), new ComputerSubtypeInterpreter());
+        registration.registerSubtypeInterpreter(Items.ROBOT.get(), new RobotSubtypeInterpreter());
+        registration.registerSubtypeInterpreter(Items.HARD_DRIVE_CUSTOM.get(), new BlockDeviceSubtypeInterpreter());
+        registration.registerSubtypeInterpreter(Items.FLASH_MEMORY_CUSTOM.get(), new BlockDeviceSubtypeInterpreter());
+    }
+
+    private static final class ComputerSubtypeInterpreter implements IIngredientSubtypeInterpreter<ItemStack> {
+        @Override
+        public String apply(final ItemStack ingredient, final UidContext context) {
+            final CompoundTag itemsTag = NBTUtils.getChildTag(ingredient.getTag(), BLOCK_ENTITY_TAG_NAME_IN_ITEM, ITEMS_TAG_NAME);
+            return itemsTag.isEmpty() ? NONE : itemsTag.toString();
+        }
+    }
+
+    private static final class RobotSubtypeInterpreter implements IIngredientSubtypeInterpreter<ItemStack> {
+        @Override
+        public String apply(final ItemStack ingredient, final UidContext context) {
+            final CompoundTag itemsTag = NBTUtils.getChildTag(ingredient.getTag(), API.MOD_ID, ITEMS_TAG_NAME);
+            return itemsTag.isEmpty() ? NONE : itemsTag.toString();
+        }
+    }
+
+    private static final class BlockDeviceSubtypeInterpreter implements IIngredientSubtypeInterpreter<ItemStack> {
+        @Override
+        public String apply(final ItemStack ingredient, final UidContext context) {
+            final String registryName = ItemStackUtils.getModDataTag(ingredient).getString(AbstractBlockDeviceItem.DATA_TAG_NAME);
+            return Strings.isNullOrEmpty(registryName) ? NONE : registryName;
+        }
+    }
+}
diff --git a/src/main/java/li/cil/oc2/common/item/AbstractBlockDeviceItem.java b/src/main/java/li/cil/oc2/common/item/AbstractBlockDeviceItem.java
index 0343a751..69960807 100644
--- a/src/main/java/li/cil/oc2/common/item/AbstractBlockDeviceItem.java
+++ b/src/main/java/li/cil/oc2/common/item/AbstractBlockDeviceItem.java
@@ -15,7 +15,7 @@ import net.minecraft.world.item.ItemStack;
 import javax.annotation.Nullable;
 
 public abstract class AbstractBlockDeviceItem extends ModItem {
-    private static final String DATA_TAG_NAME = "data";
+    public static final String DATA_TAG_NAME = "data";
 
     ///////////////////////////////////////////////////////////////////