import loottweaker.LootTweaker;
import loottweaker.vanilla.loot.LootTable;
import loottweaker.vanilla.loot.LootPool;
import loottweaker.vanilla.loot.Functions;

FreshwaterBiomes = ["Forest","Plains","Extreme Hills"];
AridBiomes = ["Desert"];
ArcticBiomes = ["Tundra","Taiga"];
SaltwaterBiomes = ["Beach","Ocean","Deep Ocean"];
TropicalBiomes = ["Jungle"];
BrackishBiomes = ["Swamp"];
BrackishBiomes += FreshwaterBiomes;
MushroomBiomes = ["Mushroom Island"];
MushroomBiomes += FreshwaterBiomes;
AllBiomes = FreshwaterBiomes + AridBiomes + ArcticBiomes + SaltwaterBiomes + TropicalBiomes + BrackishBiomes + MushroomBiomes;

fishArray = [
["Bluegill",<aquaculture:fish:0>,0.1,100,"Freshwater"],
["Perch",<aquaculture:fish:1>,0.1,100,"Freshwater"],
["Gar",<aquaculture:fish:2>,0.1,100,"Freshwater"],
["Bass",<aquaculture:fish:3>,0.1,100,"Freshwater"],
["Muskellunge",<aquaculture:fish:4>,0.1,100,"Freshwater"],
["Brown Trout",<aquaculture:fish:5>,0.1,100,"Freshwater"],
["Catfish",<aquaculture:fish:6>,0.1,100,"Freshwater"],
["Carp",<aquaculture:fish:7>,0.1,100,"Freshwater"],
["Blowfish",<aquaculture:fish:8>,0.1,100,"Saltwater"],
["Red Grouper",<aquaculture:fish:9>,0.1,100,"Saltwater"],
["Salmon",<aquaculture:fish:10>,0.1,100,"Saltwater"],
["Tuna",<aquaculture:fish:11>,0.1,100,"Saltwater"],
["Sword Fish",<aquaculture:fish:12>,0.1,100,"Saltwater"],
["Shark",<aquaculture:fish:13>,0.1,100,"Saltwater"],
["Whale",<aquaculture:fish:14>,0.1,100,"Saltwater"],
["Squid",<aquaculture:fish:15>,0.1,100,"Saltwater"],
["Jellyfish",<aquaculture:fish:16>,0.1,100,"Saltwater"],
["Frog",<aquaculture:fish:17>,0.1,100,"Swamp"],
["Turtle",<aquaculture:fish:18>,0.1,100,"Swamp"],
["Leech",<aquaculture:fish:19>,0.1,100,"Swamp"],
["Pirahna",<aquaculture:fish:20>,0.1,100,"Tropical"],
["Electric Eel",<aquaculture:fish:21>,0.1,100,"Tropical"],
["Tambaqui",<aquaculture:fish:22>,0.1,100,"Tropical"],
["Arapaima",<aquaculture:fish:23>,0.1,100,"Tropical"],
["Cod",<aquaculture:fish:24>,0.1,100,"Arctic"],
["Pollock",<aquaculture:fish:25>,0.1,100,"Arctic"],
["Herring",<aquaculture:fish:26>,0.1,100,"Arctic"],
["Halibut",<aquaculture:fish:27>,0.1,100,"Arctic"],
["Pink Salmon",<aquaculture:fish:28>,0.1,100,"Arctic"],
["Rainbow Trout",<aquaculture:fish:29>,0.1,100,"Arctic"],
["Blackfish",<aquaculture:fish:30>,0.1,100,"Arctic"],
["Capitaine",<aquaculture:fish:31>,0.1,100,"Arid"],
["Boulti",<aquaculture:fish:32>,0.1,100,"Arid"],
["Bagrid",<aquaculture:fish:33>,0.1,100,"Arid"],
["Syndontis",<aquaculture:fish:34>,0.1,100,"Arid"],
["Red Shrooma",<aquaculture:fish:35>,0.1,100,"Mushroom"],
["Brown Shrooma",<aquaculture:fish:36>,0.1,100,"Mushroom"],
["Goldfish",<aquaculture:fish:37>,0.1,100,"All"],
["Fish Bones",<aquaculture:fish:38>,0.1,100,"All"]
];


val fishing = LootTweaker.getTable("minecraft:gameplay/fishing/fish");
val fish = fishing.getPool("main");

for i in fishArray {
    var fishBiome = AllBiomes;
    if fishArray[i][4] = "Freshwater" {
        fishBiome = FreshwaterBiomes;
    }
    if fishArray[i][4] = "Saltwater" {
        fishBiome = SaltwaterBiomes;
    }
    if fishArray[i][4] = "Swamp" {
        fishBiome = BrackishBiomes;
    }
    if fishArray[i][4] = "Arid"{
        fishBiome = AridBiomes;
    }
    if fishArray[i][4] = "Arctic"{
        fishBiome = ArcticBiomes
    }
    if fishArray[i][4] = "Tropical"{
        fishBiome = TropicalBiomes
    }
    if fishArray[i][4] = "Mushroom"{
        fishBiome = MushroomBiomes
    }

    fish.addItemEntryHelper(fishArray[i][1], 75, 0,
    [
        Functions.zenscript(function(stack, random, context)
        {
            return stack * random.nextInt(fishArray[i][2], fishArray[i][3]);
        })
    ], [fishBiome]);
}