package com.TheRPGAdventurer;

import static com.TheRPGAdventurer.RealmOfTheDragonsLootTables.RegistrationHandler.create;
import com.TheRPGAdventurer.RealmOfTheDragons;
import java.util.HashSet;
import java.util.Set;
import net.minecraft.util.ResourceLocation;
import net.minecraft.world.storage.loot.LootTable;
import net.minecraft.world.storage.loot.LootTableList;

public class RealmOfTheDragonsLootTables {

	public static final ResourceLocation ENTITIES_DRAGON_AMETHYST = create("amethsyt");
	public static final ResourceLocation ENTITIES_DRAGON_GARNET = create("garnet");
	public static final ResourceLocation ENTITIES_DRAGON_JADE = create("jade");
	public static final ResourceLocation ENTITIES_DRAGON_RUBY = create("ruby");
	public static final ResourceLocation ENTITIES_DRAGON_SAPPHIRE = create("sapphire");
	public static final ResourceLocation ENTITIIES_DRAGON = create("dragon");

	/**
	 * Register this mod's {@link LootTable}s.
	 */
	public static void registerLootTables() {
		RegistrationHandler.LOOT_TABLES.forEach(LootTableList::register);
	}

	public static class RegistrationHandler {

		/**
		 * Stores the IDs of this mod's {@link LootTable}s.
		 */
		private static final Set<ResourceLocation> LOOT_TABLES = new HashSet<>();

		/**
		 * Create a {@link LootTable} ID.
		 *
		 * @param id The ID of the LootTable without the testmod3: prefix
		 * @return The ID of the LootTable
		 */
		protected static ResourceLocation create(String id) {
			final ResourceLocation lootTable = new ResourceLocation(RealmOfTheDragons.MODID, id);
			RegistrationHandler.LOOT_TABLES.add(lootTable);
			return lootTable;
		}
	}
}