Add raid hero gifts for the fluix researcher villager

Fixes #8456

Author: Mithi83

src/generated/resources/data/ae2/loot_table/gameplay/hero_of_the_village/fluix_researcher_gifts.json

@@ -0,0 +1,20 @@
+{
+  "type": "minecraft:gift",
+  "pools": [
+    {
+      "bonus_rolls": 0.0,
+      "entries": [
+        {
+          "type": "minecraft:item",
+          "name": "ae2:certus_quartz_crystal"
+        },
+        {
+          "type": "minecraft:item",
+          "name": "ae2:fluix_crystal"
+        }
+      ],
+      "rolls": 1.0
+    }
+  ],
+  "random_sequence": "ae2:gameplay/hero_of_the_village/fluix_researcher_gifts"
+}

src/generated/resources/data/neoforge/data_maps/villager_profession/raid_hero_gifts.json

@@ -0,0 +1,7 @@
+{
+  "values": {
+    "ae2:fluix_researcher": {
+      "loot_table": "ae2:gameplay/hero_of_the_village/fluix_researcher_gifts"
+    }
+  }
+}

src/main/java/appeng/datagen/AE2DataGenerators.java

@@ -37,6 +37,7 @@
 import appeng.core.AppEng;
 import appeng.core.definitions.AEDamageTypes;
 import appeng.datagen.providers.advancements.AdvancementGenerator;
+import appeng.datagen.providers.datamaps.RaidHeroGiftsProvider;
 import appeng.datagen.providers.localization.LocalizationProvider;
 import appeng.datagen.providers.loot.AE2LootTableProvider;
 import appeng.datagen.providers.models.BlockModelProvider;
@@ -121,6 +122,9 @@ public static void onGatherData(GatherDataEvent event) {
         pack.addProvider(bindRegistries(QuartzCuttingRecipesProvider::new, registries));
         pack.addProvider(bindRegistries(UpgradeRecipes::new, registries));
 
+        // DataMaps
+        pack.addProvider(bindRegistries(RaidHeroGiftsProvider::new, registries));
+
         // Must run last
         pack.addProvider(packOutput -> localization);
     }

src/main/java/appeng/datagen/providers/datamaps/RaidHeroGiftsProvider.java

@@ -0,0 +1,25 @@
+package appeng.datagen.providers.datamaps;
+
+import static net.neoforged.neoforge.registries.datamaps.builtin.NeoForgeDataMaps.RAID_HERO_GIFTS;
+
+import java.util.concurrent.CompletableFuture;
+
+import net.minecraft.core.HolderLookup;
+import net.minecraft.data.PackOutput;
+import net.neoforged.neoforge.common.data.DataMapProvider;
+import net.neoforged.neoforge.registries.datamaps.builtin.RaidHeroGift;
+
+import appeng.datagen.providers.IAE2DataProvider;
+import appeng.init.InitVillager;
+
+public class RaidHeroGiftsProvider extends DataMapProvider implements IAE2DataProvider {
+    public RaidHeroGiftsProvider(PackOutput packOutput, CompletableFuture<HolderLookup.Provider> lookupProvider) {
+        super(packOutput, lookupProvider);
+    }
+
+    @Override
+    public void gather(HolderLookup.Provider provider) {
+        this.builder(RAID_HERO_GIFTS)
+                .add(InitVillager.ID, new RaidHeroGift(InitVillager.LOOT_TABLE_KEY), false);
+    }
+}

src/main/java/appeng/datagen/providers/loot/AE2LootTableProvider.java

@@ -15,7 +15,8 @@
 
 public class AE2LootTableProvider extends LootTableProvider {
     private static final List<SubProviderEntry> SUB_PROVIDERS = List.of(
-            new SubProviderEntry(BlockDropProvider::new, LootContextParamSets.BLOCK));
+            new SubProviderEntry(BlockDropProvider::new, LootContextParamSets.BLOCK),
+            new SubProviderEntry(RaidHeroGiftLootProvider::new, LootContextParamSets.GIFT));
 
     public AE2LootTableProvider(PackOutput packOutput, CompletableFuture<HolderLookup.Provider> provider) {
         super(packOutput, Set.of(), SUB_PROVIDERS, provider);

src/main/java/appeng/datagen/providers/loot/RaidHeroGiftLootProvider.java

@@ -0,0 +1,46 @@
+/*
+ * This file is part of Applied Energistics 2.
+ * Copyright (c) 2021, TeamAppliedEnergistics, All rights reserved.
+ *
+ * Applied Energistics 2 is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Applied Energistics 2 is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Applied Energistics 2.  If not, see <http://www.gnu.org/licenses/lgpl>.
+ */
+
+package appeng.datagen.providers.loot;
+
+import java.util.function.BiConsumer;
+
+import net.minecraft.core.HolderLookup;
+import net.minecraft.data.loot.LootTableSubProvider;
+import net.minecraft.resources.ResourceKey;
+import net.minecraft.world.level.storage.loot.LootPool;
+import net.minecraft.world.level.storage.loot.LootTable;
+import net.minecraft.world.level.storage.loot.entries.LootItem;
+
+import appeng.core.definitions.AEItems;
+import appeng.init.InitVillager;
+
+public class RaidHeroGiftLootProvider implements LootTableSubProvider {
+    public RaidHeroGiftLootProvider(HolderLookup.Provider lookupProvider) {
+    }
+
+    @Override
+    public void generate(BiConsumer<ResourceKey<LootTable>, LootTable.Builder> biConsumer) {
+        biConsumer.accept(
+                InitVillager.LOOT_TABLE_KEY,
+                LootTable.lootTable()
+                        .withPool(LootPool.lootPool()
+                                .add(LootItem.lootTableItem(AEItems.CERTUS_QUARTZ_CRYSTAL))
+                                .add(LootItem.lootTableItem(AEItems.FLUIX_CRYSTAL))));
+    }
+}

src/main/java/appeng/init/InitVillager.java

@@ -17,6 +17,7 @@
 import net.minecraft.world.entity.npc.VillagerTrades;
 import net.minecraft.world.item.Items;
 import net.minecraft.world.level.ItemLike;
+import net.minecraft.world.level.storage.loot.LootTable;
 import net.neoforged.neoforge.event.village.VillagerTradesEvent;
 
 import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
@@ -38,6 +39,9 @@ private InitVillager() {
     public static final VillagerProfession PROFESSION = new VillagerProfession(ID.toString(), e -> e.is(POI_KEY),
             e -> e.is(POI_KEY), ImmutableSet.of(), ImmutableSet.of(), SoundEvents.VILLAGER_WORK_LIBRARIAN);
 
+    public static final ResourceKey<LootTable> LOOT_TABLE_KEY = ResourceKey.create(Registries.LOOT_TABLE,
+            AppEng.makeId("gameplay/hero_of_the_village/fluix_researcher_gifts"));
+
     public static void initProfession(Registry<VillagerProfession> registry) {
         Registry.register(registry, ID, PROFESSION);
     }