Merge cell/upgraded/inventory item processing into one processor

src/main/java/gripe/_90/appliede/emc/StorageCellProcessor.java → src/main/java/gripe/_90/appliede/emc/InventoryItemProcessor.java

@@ -6,6 +6,9 @@
 
 import appeng.api.stacks.AEItemKey;
 import appeng.api.storage.StorageCells;
+import appeng.api.upgrades.IUpgradeableItem;
+import appeng.items.contents.NetworkToolMenuHost;
+import appeng.items.tools.NetworkToolItem;
 
 import gripe._90.appliede.AppliedE;
 
@@ -16,27 +19,49 @@
 
 @SuppressWarnings("unused")
 @NBTProcessor
-public class StorageCellProcessor implements INBTProcessor {
+public class InventoryItemProcessor implements INBTProcessor {
     @Override
     public String getName() {
-        return "AE2CellProcessor";
+        return "AE2InventoryItemProcessor";
     }
 
     @Override
     public String getDescription() {
-        return "(AppliedE) Calculates EMC value of Applied Energistics 2 item cells depending on contents.";
+        return "(AppliedE) Calculates EMC value of Applied Energistics 2 inventory items such as cells and wireless terminals.";
     }
 
     @Override
     public long recalculateEMC(@NotNull ItemInfo itemInfo, long currentEmc) throws ArithmeticException {
-        var cell = StorageCells.getCellInventory(itemInfo.createStack(), null);
+        if (itemInfo.getItem() instanceof NetworkToolItem) {
+            var stack = itemInfo.createStack();
+            var bigEmc = BigInteger.valueOf(currentEmc);
+            var inventory = new NetworkToolMenuHost(null, -1, stack, null).getInventory();
 
-        if (cell == null) {
+            for (var item : inventory) {
+                var itemEmc = IEMCProxy.INSTANCE.getValue(item);
+                bigEmc = bigEmc.add(BigInteger.valueOf(itemEmc).multiply(BigInteger.valueOf(item.getCount())));
+            }
+
+            return AppliedE.clampedLong(bigEmc);
+        }
+
+        if (!(itemInfo.getItem() instanceof IUpgradeableItem upgradeable)) {
             return currentEmc;
         }
 
+        var stack = itemInfo.createStack();
         var bigEmc = BigInteger.valueOf(currentEmc);
 
+        for (var upgrade : upgradeable.getUpgrades(stack)) {
+            bigEmc = bigEmc.add(BigInteger.valueOf(IEMCProxy.INSTANCE.getValue(upgrade)));
+        }
+
+        var cell = StorageCells.getCellInventory(itemInfo.createStack(), null);
+
+        if (cell == null) {
+            return AppliedE.clampedLong(bigEmc);
+        }
+
         for (var key : cell.getAvailableStacks()) {
             if (key.getKey() instanceof AEItemKey item) {
                 var keyEmc = IEMCProxy.INSTANCE.getValue(item.toStack());