diff --git a/src/main/java/com/fusionflux/portalcubed/data/PortalCubedDatagen.java b/src/main/java/com/fusionflux/portalcubed/data/PortalCubedDatagen.java index 1b1ea34b..729e01ad 100644 --- a/src/main/java/com/fusionflux/portalcubed/data/PortalCubedDatagen.java +++ b/src/main/java/com/fusionflux/portalcubed/data/PortalCubedDatagen.java @@ -4,8 +4,8 @@ import net.fabricmc.fabric.api.datagen.v1.FabricDataGenerator; public class PortalCubedDatagen implements DataGeneratorEntrypoint { - @Override - public void onInitializeDataGenerator(FabricDataGenerator generator) { - generator.createPack().addProvider(PortalCubedEntityLoot::new); - } + @Override + public void onInitializeDataGenerator(FabricDataGenerator generator) { + generator.createPack().addProvider(PortalCubedEntityLoot::new); + } } diff --git a/src/main/java/com/fusionflux/portalcubed/data/PortalCubedEntityLoot.java b/src/main/java/com/fusionflux/portalcubed/data/PortalCubedEntityLoot.java index ca26229f..ad2bd682 100644 --- a/src/main/java/com/fusionflux/portalcubed/data/PortalCubedEntityLoot.java +++ b/src/main/java/com/fusionflux/portalcubed/data/PortalCubedEntityLoot.java @@ -1,14 +1,9 @@ package com.fusionflux.portalcubed.data; -import java.util.HashMap; -import java.util.Map; -import java.util.function.BiConsumer; - import com.fusionflux.portalcubed.entity.PortalCubedEntities; import com.fusionflux.portalcubed.items.PortalCubedItems; import net.fabricmc.fabric.api.datagen.v1.FabricDataOutput; import net.fabricmc.fabric.api.datagen.v1.provider.SimpleFabricLootTableProvider; - import net.minecraft.resources.ResourceLocation; import net.minecraft.world.entity.EntityType; import net.minecraft.world.level.ItemLike; @@ -20,58 +15,62 @@ import net.minecraft.world.level.storage.loot.functions.CopyNameFunction.NameSource; import net.minecraft.world.level.storage.loot.parameters.LootContextParamSets; +import java.util.HashMap; +import java.util.Map; +import java.util.function.BiConsumer; + public class PortalCubedEntityLoot extends SimpleFabricLootTableProvider { - private final Map tables = new HashMap<>(); + private final Map tables = new HashMap<>(); - public PortalCubedEntityLoot(FabricDataOutput output) { - super(output, LootContextParamSets.ENTITY); - } + public PortalCubedEntityLoot(FabricDataOutput output) { + super(output, LootContextParamSets.ENTITY); + } - @Override - public void generate(BiConsumer out) { - buildLootTables(); - tables.forEach(out); - } + @Override + public void generate(BiConsumer out) { + buildLootTables(); + tables.forEach(out); + } - public void buildLootTables() { - keepName(PortalCubedEntities.STORAGE_CUBE, PortalCubedItems.STORAGE_CUBE); - keepName(PortalCubedEntities.COMPANION_CUBE, PortalCubedItems.COMPANION_CUBE); - keepName(PortalCubedEntities.REDIRECTION_CUBE, PortalCubedItems.REDIRECTION_CUBE); - keepName(PortalCubedEntities.SCHRODINGER_CUBE, PortalCubedItems.SCHRODINGER_CUBE); - keepName(PortalCubedEntities.RADIO, PortalCubedItems.RADIO); - keepName(PortalCubedEntities.OLD_AP_CUBE, PortalCubedItems.OLD_AP_CUBE); - keepName(PortalCubedEntities.PORTAL_1_COMPANION_CUBE, PortalCubedItems.PORTAL_1_COMPANION_CUBE); - keepName(PortalCubedEntities.PORTAL_1_STORAGE_CUBE, PortalCubedItems.PORTAL_1_STORAGE_CUBE); - keepName(PortalCubedEntities.BEANS, PortalCubedItems.BEANS); - keepName(PortalCubedEntities.MUG, PortalCubedItems.MUG); - keepName(PortalCubedEntities.JUG, PortalCubedItems.JUG); - keepName(PortalCubedEntities.COMPUTER, PortalCubedItems.COMPUTER); - keepName(PortalCubedEntities.CHAIR, PortalCubedItems.CHAIR); - keepName(PortalCubedEntities.LIL_PINEAPPLE, PortalCubedItems.LIL_PINEAPPLE); - keepName(PortalCubedEntities.HOOPY, PortalCubedItems.HOOPY); - keepName(PortalCubedEntities.CORE_FRAME, PortalCubedItems.CORE_FRAME); - keepName(PortalCubedEntities.ANGER_CORE, PortalCubedItems.ANGER_CORE); - keepName(PortalCubedEntities.MORALITY_CORE, PortalCubedItems.MORALITY_CORE); - keepName(PortalCubedEntities.CAKE_CORE, PortalCubedItems.CAKE_CORE); - keepName(PortalCubedEntities.CURIOSITY_CORE, PortalCubedItems.CURIOSITY_CORE); - keepName(PortalCubedEntities.SPACE_CORE, PortalCubedItems.SPACE_CORE); - keepName(PortalCubedEntities.FACT_CORE, PortalCubedItems.FACT_CORE); - keepName(PortalCubedEntities.ADVENTURE_CORE, PortalCubedItems.ADVENTURE_CORE); - keepName(PortalCubedEntities.ENERGY_PELLET, PortalCubedItems.ENERGY_PELLET); - keepName(PortalCubedEntities.TURRET, PortalCubedItems.TURRET); - } + public void buildLootTables() { + keepName(PortalCubedEntities.STORAGE_CUBE, PortalCubedItems.STORAGE_CUBE); + keepName(PortalCubedEntities.COMPANION_CUBE, PortalCubedItems.COMPANION_CUBE); + keepName(PortalCubedEntities.REDIRECTION_CUBE, PortalCubedItems.REDIRECTION_CUBE); + keepName(PortalCubedEntities.SCHRODINGER_CUBE, PortalCubedItems.SCHRODINGER_CUBE); + keepName(PortalCubedEntities.RADIO, PortalCubedItems.RADIO); + keepName(PortalCubedEntities.OLD_AP_CUBE, PortalCubedItems.OLD_AP_CUBE); + keepName(PortalCubedEntities.PORTAL_1_COMPANION_CUBE, PortalCubedItems.PORTAL_1_COMPANION_CUBE); + keepName(PortalCubedEntities.PORTAL_1_STORAGE_CUBE, PortalCubedItems.PORTAL_1_STORAGE_CUBE); + keepName(PortalCubedEntities.BEANS, PortalCubedItems.BEANS); + keepName(PortalCubedEntities.MUG, PortalCubedItems.MUG); + keepName(PortalCubedEntities.JUG, PortalCubedItems.JUG); + keepName(PortalCubedEntities.COMPUTER, PortalCubedItems.COMPUTER); + keepName(PortalCubedEntities.CHAIR, PortalCubedItems.CHAIR); + keepName(PortalCubedEntities.LIL_PINEAPPLE, PortalCubedItems.LIL_PINEAPPLE); + keepName(PortalCubedEntities.HOOPY, PortalCubedItems.HOOPY); + keepName(PortalCubedEntities.CORE_FRAME, PortalCubedItems.CORE_FRAME); + keepName(PortalCubedEntities.ANGER_CORE, PortalCubedItems.ANGER_CORE); + keepName(PortalCubedEntities.MORALITY_CORE, PortalCubedItems.MORALITY_CORE); + keepName(PortalCubedEntities.CAKE_CORE, PortalCubedItems.CAKE_CORE); + keepName(PortalCubedEntities.CURIOSITY_CORE, PortalCubedItems.CURIOSITY_CORE); + keepName(PortalCubedEntities.SPACE_CORE, PortalCubedItems.SPACE_CORE); + keepName(PortalCubedEntities.FACT_CORE, PortalCubedItems.FACT_CORE); + keepName(PortalCubedEntities.ADVENTURE_CORE, PortalCubedItems.ADVENTURE_CORE); + keepName(PortalCubedEntities.ENERGY_PELLET, PortalCubedItems.ENERGY_PELLET); + keepName(PortalCubedEntities.TURRET, PortalCubedItems.TURRET); + } - public void add(EntityType type, Builder builder) { - tables.put(type.getDefaultLootTable(), builder); - } + public void add(EntityType type, Builder builder) { + tables.put(type.getDefaultLootTable(), builder); + } - public void keepName(EntityType type, ItemLike item) { - add(type, LootTable.lootTable() - .withPool(LootPool.lootPool() - .add(LootItem.lootTableItem(item) - .apply(CopyNameFunction.copyName(NameSource.THIS)) - ) - ) - ); - } + public void keepName(EntityType type, ItemLike item) { + add(type, LootTable.lootTable() + .withPool(LootPool.lootPool() + .add(LootItem.lootTableItem(item) + .apply(CopyNameFunction.copyName(NameSource.THIS)) + ) + ) + ); + } }