Getting fireworks with effects

Okay, I have it so when a dispenser launches a firework, the firework gets replaced so it has infinite fireworks. The only problem is that it doesn't replace fireworks that have effects and colors in them. PLEASE HELP!

Code:

package com.endlessshadow;
 
import org.bukkit.Material;
import org.bukkit.block.Dispenser;
import org.bukkit.event.EventHandler;
import org.bukkit.event.Listener;
import org.bukkit.event.block.BlockDispenseEvent;
import org.bukkit.inventory.Inventory;
import org.bukkit.inventory.ItemStack;
 
public class DispenseListener implements Listener {
public EndlessFireworks plugin;
       
        public DispenseListener(EndlessFireworks plugin) {
                this.plugin = plugin;
        }
 
        @EventHandler
        public void onDispense(BlockDispenseEvent evt){
       ItemStack FIREWORK = new ItemStack(Material.FIREWORK, 1);
       Dispenser dispenser = (Dispenser) evt.getBlock().getState();
       Inventory inv = dispenser.getInventory();
       if(inv.contains(FIREWORK)){
       inv.addItem(evt.getItem());
       }
        }
}

 

Instead of doing inv.contains(ItemStack), you could try inv.contains(Material.FIREWORK) instead. This should return true for any firework, regardless of effects but I'm not 100% certain.

 

I'll try it

It worked! Thanks!

EDIT by Moderator: merged posts, please use the edit button instead of double posting.

 

Glad I could help