Allow searching by mekasuit piece/mekatool in EMI to display supported modules (mekanism/Mekanism-Feature-Requests#688)

Author: pupnewfster

src/datagen/generated/mekanism/.cache/c2007283a832f570d2167eeffdb50ecb2f811654

@@ -11,8 +11,8 @@
 import mekanism.generators.client.integration.GeneratorsAliases;
 import mekanism.generators.common.MekanismGenerators;
 import mekanism.generators.common.registries.GeneratorsBlocks;
-import mekanism.generators.common.registries.GeneratorsFluids;
 import mekanism.generators.common.registries.GeneratorsChemicals;
+import mekanism.generators.common.registries.GeneratorsFluids;
 import mekanism.generators.common.registries.GeneratorsItems;
 import net.minecraft.core.HolderLookup;
 import net.minecraft.data.PackOutput;
@@ -29,6 +29,7 @@ protected void addAliases(HolderLookup.Provider lookupProvider) {
         addChemicalAliases();
         addMultiblockAliases();
         addAliases(GeneratorsBlocks.GAS_BURNING_GENERATOR, GeneratorsAliases.GBG_ETHENE, GeneratorsAliases.GBG_ETHYLENE);
+        addModuleAliases(GeneratorsItems.ITEMS);
     }
 
     private void addChemicalAliases() {

src/datagen/generated/mekanism/assets/emi/aliases/mekanism.json

@@ -23,13 +23,16 @@
 import java.util.function.Function;
 import mekanism.api.SerializationConstants;
 import mekanism.api.annotations.NothingNullByDefault;
+import mekanism.api.gear.IModuleHelper;
 import mekanism.api.providers.IChemicalProvider;
 import mekanism.api.providers.IFluidProvider;
 import mekanism.api.text.IHasTranslationKey;
 import mekanism.client.recipe_viewer.emi.ChemicalEmiStack;
 import mekanism.common.DataGenSerializationConstants;
+import mekanism.common.content.gear.IModuleItem;
 import mekanism.common.integration.MekanismHooks;
 import mekanism.common.lib.collection.HashList;
+import mekanism.common.registration.impl.ItemDeferredRegister;
 import net.minecraft.core.HolderLookup;
 import net.minecraft.data.CachedOutput;
 import net.minecraft.data.DataProvider;
@@ -38,7 +41,9 @@
 import net.minecraft.data.PackOutput.Target;
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.util.ExtraCodecs;
+import net.minecraft.world.item.Item;
 import net.minecraft.world.level.ItemLike;
+import net.neoforged.neoforge.registries.DeferredHolder;
 
 @NothingNullByDefault
 public abstract class BaseEmiAliasProvider implements DataProvider {
@@ -125,6 +130,18 @@ protected void addAliases(List<EmiIngredient> stacks, IHasTranslationKey... alia
         }
     }
 
+    protected void addModuleAliases(ItemDeferredRegister items) {
+        for (DeferredHolder<Item, ? extends Item> entry : items.getEntries()) {
+            if (entry.get() instanceof IModuleItem module) {
+                addAliases(entry.get(), IModuleHelper.INSTANCE.getSupported(module.getModuleData())
+                      .stream()
+                      .map(item -> (IHasTranslationKey) item::getDescriptionId)
+                      .toArray(IHasTranslationKey[]::new)
+                );
+            }
+        }
+    }
+
     @Override
     public String getName() {
         return "EMI Alias Provider: " + modid;

src/datagen/generated/mekanismgenerators/.cache/94507507da148bcdf0ef759163aca13369d21ac5

@@ -12,8 +12,8 @@
 import mekanism.common.item.block.ItemBlockEnergyCube;
 import mekanism.common.registration.impl.BlockRegistryObject;
 import mekanism.common.registries.MekanismBlocks;
-import mekanism.common.registries.MekanismFluids;
 import mekanism.common.registries.MekanismChemicals;
+import mekanism.common.registries.MekanismFluids;
 import mekanism.common.registries.MekanismItems;
 import mekanism.common.resource.IResource;
 import mekanism.common.tier.FactoryTier;
@@ -77,6 +77,8 @@ private void addUnitAliases() {
         addAliases(MekanismItems.MODULE_MOTORIZED_SERVO, getTranslationKey(Enchantments.SWIFT_SNEAK));
         addAliases(MekanismItems.MODULE_LOCOMOTIVE_BOOSTING, MobEffects.MOVEMENT_SPEED.value()::getDescriptionId);
         addAliases(MekanismItems.MODULE_SOUL_SURFER, getTranslationKey(Enchantments.SOUL_SPEED));
+
+        addModuleAliases(MekanismItems.ITEMS);
     }
 
     private IHasTranslationKey getTranslationKey(ResourceKey<Enchantment> enchantmentKey) {

src/datagen/generated/mekanismgenerators/assets/emi/aliases/mekanismgenerators.json

@@ -12,8 +12,10 @@
 import java.lang.reflect.Constructor;
 import java.lang.reflect.Method;
 import java.nio.file.Path;
+import java.util.ArrayList;
 import java.util.EnumMap;
 import java.util.LinkedHashMap;
+import java.util.List;
 import java.util.Set;
 import java.util.concurrent.CompletableFuture;
 import java.util.function.Function;
@@ -37,10 +39,15 @@
 import net.minecraft.data.DataProvider;
 import net.minecraft.data.PackOutput;
 import net.neoforged.bus.api.SubscribeEvent;
+import net.neoforged.fml.DeferredWorkQueue;
+import net.neoforged.fml.ModContainer;
+import net.neoforged.fml.ModList;
 import net.neoforged.fml.common.EventBusSubscriber;
 import net.neoforged.fml.config.ConfigTracker;
 import net.neoforged.fml.config.ModConfig;
 import net.neoforged.fml.event.config.ModConfigEvent;
+import net.neoforged.fml.event.lifecycle.InterModEnqueueEvent;
+import net.neoforged.fml.event.lifecycle.InterModProcessEvent;
 import net.neoforged.fml.util.ObfuscationReflectionHelper;
 import net.neoforged.neoforge.common.data.ExistingFileHelper;
 import net.neoforged.neoforge.data.event.GatherDataEvent;
@@ -72,6 +79,7 @@ private MekanismDataGenerator() {
     @SubscribeEvent
     public static void gatherData(GatherDataEvent event) {
         bootstrapConfigs(Mekanism.MODID);
+        bootstrapIMC();
         DataGenerator gen = event.getGenerator();
         PackOutput output = gen.getPackOutput();
         ExistingFileHelper existingFileHelper = event.getExistingFileHelper();
@@ -125,6 +133,24 @@ public static void bootstrapConfigs(String modid) {
         }
     }
 
+    private static void bootstrapIMC() {
+        List<ModContainer> mods = new ArrayList<>();
+        DeferredWorkQueue enqueueIMC = new DeferredWorkQueue("IMC Bootstrap: Enqueue IMC");
+        for (ModContainer mod : ModList.get().getSortedMods()) {
+            //Handle all our modules
+            if (mod.getModId().startsWith(Mekanism.MODID)) {
+                mods.add(mod);
+                mod.getEventBus().post(new InterModEnqueueEvent(mod, enqueueIMC));
+            }
+        }
+        enqueueIMC.runTasks();
+        DeferredWorkQueue processIMC = new DeferredWorkQueue("IMC Bootstrap: Process IMC");
+        for (ModContainer mod : mods) {
+            mod.getEventBus().post(new InterModProcessEvent(mod, processIMC));
+        }
+        processIMC.runTasks();
+    }
+
     /**
      * Basically a copy of {@link DataProvider#saveStable(CachedOutput, JsonElement, Path)} but it takes a consumer of the output stream instead of serializes json using
      * GSON. Use it to write arbitrary files.