diff --git a/src/main/java/ru/dbotthepony/mc/otm/OverdriveThatMatters.java b/src/main/java/ru/dbotthepony/mc/otm/OverdriveThatMatters.java index 50512d13c..60474feb2 100644 --- a/src/main/java/ru/dbotthepony/mc/otm/OverdriveThatMatters.java +++ b/src/main/java/ru/dbotthepony/mc/otm/OverdriveThatMatters.java @@ -15,6 +15,7 @@ import net.minecraftforge.fml.javafmlmod.FMLJavaModLoadingContext; import org.apache.logging.log4j.LogManager; import org.apache.logging.log4j.Logger; import ru.dbotthepony.mc.otm.block.entity.blackhole.ExplosionQueue; +import ru.dbotthepony.mc.otm.capability.MatteryCapability; import ru.dbotthepony.mc.otm.capability.MatteryPlayerCapability; import ru.dbotthepony.mc.otm.capability.drive.DrivePool; import ru.dbotthepony.mc.otm.client.ClientEventHandlerKt; @@ -85,6 +86,7 @@ public final class OverdriveThatMatters { modBus.addListener(EventPriority.HIGHEST, this::setup); modBus.addListener(EventPriority.NORMAL, this::setupClient); + modBus.addListener(EventPriority.NORMAL, MatteryCapability::register); } private void setup(final FMLCommonSetupEvent event) { diff --git a/src/main/java/ru/dbotthepony/mc/otm/capability/MatteryCapability.java b/src/main/java/ru/dbotthepony/mc/otm/capability/MatteryCapability.java index 0e7c89898..df493e246 100644 --- a/src/main/java/ru/dbotthepony/mc/otm/capability/MatteryCapability.java +++ b/src/main/java/ru/dbotthepony/mc/otm/capability/MatteryCapability.java @@ -59,4 +59,15 @@ public class MatteryCapability { @Nonnull @NotNull public static final Capability CURIOS_ITEM = CapabilityManager.get(new CapabilityToken<>() {}); + + public static void register(RegisterCapabilitiesEvent event) { + event.register(IMatteryEnergyStorage.class); + event.register(MatteryPlayerCapability.class); + event.register(IMatterHandler.class); + event.register(IMatterGraphNode.class); + event.register(IPatternStorage.class); + event.register(IReplicationTaskProvider.class); + event.register(IMatteryDrive.class); + event.register(IStorageGraphNode.class); + } }