LootRegisty.java

package joshie.mariculture.modules.fishery.loot;

import joshie.mariculture.core.util.annotation.MCEvents;
import net.minecraft.util.ResourceLocation;
import net.minecraft.world.storage.loot.LootEntry;
import net.minecraft.world.storage.loot.LootEntryTable;
import net.minecraft.world.storage.loot.LootPool;
import net.minecraft.world.storage.loot.conditions.LootCondition;
import net.minecraftforge.event.LootTableLoadEvent;
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
import net.minecraftforge.fml.relauncher.ReflectionHelper;

import static joshie.mariculture.core.lib.MaricultureInfo.MODID;

@MCEvents(modules = "fishery")
public class LootRegisty {
    public static final LootRegisty INSTANCE = new LootRegisty();
    private static final String FISHING = "gameplay/fishing/";
    private static final String[] LOOT_TABLES = new String[]{ FISHING + "fish", FISHING + "junk", FISHING + "treasure" };

    @SubscribeEvent
    public void onLootLoading(LootTableLoadEvent event) {
        if (event.getName().toString().equals("minecraft:gameplay/fishing")) {
            LootPool pool = event.getTable().getPool("main");
            if (pool != null) {
                for (String name : LOOT_TABLES) {
                    LootEntry entry = pool.getEntry("minecraft:" + name);
                    if (entry != null) {
                        pool.addEntry(getEntry(MODID + "_" + name.replace(FISHING, ""), name, getVanillaQuality(entry), getVanillaWeight(entry)));
                    }
                }
            }
        }
    }

    private int getVanillaQuality(LootEntry entry) {
        return ReflectionHelper.getPrivateValue(LootEntry.class, entry, "quality", "field_186365_d");
    }

    private int getVanillaWeight(LootEntry entry) {
        return ReflectionHelper.getPrivateValue(LootEntry.class, entry, "weight", "field_186364_c");
    }

    private LootEntryTable getEntry(String unique, String name, int quality, int weight) {
        return new LootEntryTable(new ResourceLocation(MODID, name), weight, quality, new LootCondition[0], unique);
    }
}