package info.u_team.u_team_core.util.registry;

import java.util.function.Consumer;

import net.minecraftforge.common.MinecraftForge;
import net.minecraftforge.eventbus.api.IEventBus;
import net.minecraftforge.fml.javafmlmod.FMLJavaModLoadingContext;

public class BusRegistry {
	
	public static void registerMod(Consumer<IEventBus> consumer) {
		register(FMLJavaModLoadingContext.get().getModEventBus(), consumer);
	}
	
	public static void registerForge(Consumer<IEventBus> consumer) {
		register(MinecraftForge.EVENT_BUS, consumer);
	}
	
	public static void register(IEventBus bus, Consumer<IEventBus> consumer) {
		consumer.accept(bus);
	}
	
}

@@ -177,16 +177,81 @@
 entBus;%0A
+import net.minecraftforge.fml.common.Mod.EventBusSubscriber.Bus;%0A
 import n
@@ -590,32 +590,151 @@
 consumer);%0A%09%7D%0A%09%0A
+%09public static void registerForge(Bus bus, Consumer%3CIEventBus%3E consumer) %7B%0A%09%09register(bus.bus().get(), consumer);%0A%09%7D%0A%09%0A
 %09public static v

Add util method for the bus enum