Make iron golems drop lots of coins, make villagers with professions …

…drop a small amount
SamB440 · Nov 4, 2023 · c9c3c7b · c9c3c7b
1 parent 3971cab
commit c9c3c7b
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/main/java/com/convallyria/taleofkingdoms/common/item/ItemHelper.java b/src/main/java/com/convallyria/taleofkingdoms/common/item/ItemHelper.java
@@ -4,8 +4,11 @@
 import com.convallyria.taleofkingdoms.common.item.common.ItemCoin;
 import net.minecraft.entity.Entity;
 import net.minecraft.entity.mob.Monster;
+import net.minecraft.entity.passive.IronGolemEntity;
+import net.minecraft.entity.passive.VillagerEntity;
 import net.minecraft.item.Item;
 import net.minecraft.server.MinecraftServer;
+import net.minecraft.village.VillagerProfession;
 
 import java.util.function.Consumer;
 
@@ -27,9 +30,16 @@ public static boolean isHostileEntity(Entity entityLiving) {
      * @param entity entity to drop coins for
      */
     public static void dropCoins(Entity entity) {
+        Consumer<MinecraftServer> dropCoins = server -> dropItem(ItemRegistry.ITEMS.get(ItemRegistry.TOKItem.COIN), 1, entity);
         if (isHostileEntity(entity)) {
-            Consumer<MinecraftServer> dropCoins = server -> dropItem(ItemRegistry.ITEMS.get(ItemRegistry.TOKItem.COIN), 1, entity);
             TaleOfKingdoms.getAPI().getScheduler().repeatN(dropCoins, 25, 0, 1);
+        } else if (entity instanceof IronGolemEntity) {
+            TaleOfKingdoms.getAPI().getScheduler().repeatN(dropCoins, 45, 0, 1);
+        } else if (entity instanceof VillagerEntity villagerEntity) {
+            final VillagerProfession profession = villagerEntity.getVillagerData().getProfession();
+            if (!profession.id().equals("nitwit") && !profession.id().equals("none")) {
+                TaleOfKingdoms.getAPI().getScheduler().repeatN(dropCoins, 10, 0, 1);
+            }
         }
     }