ChunkPopulateEvent help

Okay, so I have this code:

Code:java

@EventHandler
public void onGen(ChunkPopulateEvent e){
BlockState[] tileEnts = e.getChunk().getTileEntities();
for (BlockState state : tileEnts) {
if (state.getType() != Material.CHEST)
continue;
//A chest has been generated, what would you like to do?
}
}

I know that this code checks if a chest has been generated from the ChunkPopulateEvent, but I want to know how to add an item to that chest. Basically add custom dungeon loot to that chest.

 

I've never used ChunkPopulateEvent and so I have no idea what getTileEntities() does.

Code:

@EventHandler
public void onGen(ChunkPopulateEvent event) {
    BlockState[] tileEnts = event.getChunk().getTileEntities();
    for (BlockState state : tileEnts) {
        if (state != null && state.getType() == Material.CHEST) {
            Chest chest = (Chest) state;
            chest.getBlockInventory().add(new ItemStack(Material.DIAMOND));
        }
    }
}

 

That's all I needed. Thank you!

EDIT: For anyone using this code, the line

Code:java

chest.getBlockInventory().add(new ItemStack(Material.DIAMOND));

needs to be

Code:java

chest.getBlockInventory().addItem(new ItemStack(Material.DIAMOND));