public class TutorialChestLootTableProvider extends SimpleFabricLootTableProvider {
  public TutorialChestLootTableProvider(FabricDataOutput output, CompletableFuture<RegistryWrapper.WrapperLookup> registryLookup) {
    super(output, registryLookup, LootContextTypes.CHEST);
  }

  public static final RegistryKey<LootTable> TEST_CHEST = RegistryKey.of(RegistryKeys.LOOT_TABLE, Identifier.of("tutorial", "test_chest"));

  @Override
  public void accept(BiConsumer<RegistryKey<LootTable>, LootTable.Builder> lootTableBiConsumer) {
    lootTableBiConsumer.accept(TEST_CHEST, LootTable.builder()
        .pool(LootPool.builder().rolls(ConstantLootNumberProvider.create(1.0F))
            .with(ItemEntry.builder(Items.DIAMOND)
                .apply(SetCountLootFunction.builder(ConstantLootNumberProvider.create(1.0F))))
            .with(ItemEntry.builder(Items.DIAMOND_SWORD)))
    );
  }
}