EntityLootTablesDataGenerator

package com.github.theonlytails.rubymod.datagen

import com.github.theonlytails.rubymod.registries.EntityTypeRegistry
import com.github.theonlytails.rubymod.registries.ItemRegistry
import net.minecraft.data.DataGenerator
import net.minecraft.data.LootTableProvider
import net.minecraft.entity.EntityType
import net.minecraft.loot.ItemLootEntry
import net.minecraft.loot.LootPool
import net.minecraft.loot.LootTable

class EntityLootTablesDataGenerator(private val generator: DataGenerator) : LootTableProvider(generator) {
    private val tables = hashMapOf<EntityType<*>, LootTable.Builder>()

    private fun addLootTables(loot: EntityLootTablesDataGenerator) {
        // Ruby Sheep
        loot.addLoot(EntityTypeRegistry.RUBY_SHEEP, LootTable.builder()
            .addLootPool(LootPool.builder()
                .addEntry(ItemLootEntry.builder(ItemRegistry.RUBY_WOOL_ITEM))
            )
            .addLootPool(LootPool.builder()
                .addEntry()
            )
        )
    }

    private fun addLoot(entityType: EntityType<*>, loot: LootTable.Builder) {
        tables[entityType] = loot
    }
}