package com.titanicmc.titanic.registry;

import com.titanicmc.titanic.TitanicMod;
import com.titanicmc.titanic.blockentity.BoilerBlockEntity;
import com.titanicmc.titanic.blockentity.HelmBlockEntity;
import com.titanicmc.titanic.blockentity.LiftingFrameBlockEntity;
import com.titanicmc.titanic.blockentity.RepairPanelBlockEntity;
import net.minecraft.world.level.block.entity.BlockEntityType;
import net.minecraftforge.registries.DeferredRegister;
import net.minecraftforge.registries.ForgeRegistries;
import net.minecraftforge.registries.RegistryObject;

public class ModBlockEntities {

    public static final DeferredRegister<BlockEntityType<?>> BLOCK_ENTITIES =
            DeferredRegister.create(ForgeRegistries.BLOCK_ENTITY_TYPES, TitanicMod.MODID);

    public static final RegistryObject<BlockEntityType<LiftingFrameBlockEntity>> LIFTING_FRAME =
            BLOCK_ENTITIES.register("lifting_frame", () -> BlockEntityType.Builder.of(
                    LiftingFrameBlockEntity::new, ModBlocks.LIFTING_FRAME.get()).build(null));

    public static final RegistryObject<BlockEntityType<RepairPanelBlockEntity>> REPAIR_PANEL =
            BLOCK_ENTITIES.register("repair_panel", () -> BlockEntityType.Builder.of(
                    RepairPanelBlockEntity::new, ModBlocks.REPAIR_PANEL.get()).build(null));

    public static final RegistryObject<BlockEntityType<HelmBlockEntity>> HELM =
            BLOCK_ENTITIES.register("helm", () -> BlockEntityType.Builder.of(
                    HelmBlockEntity::new, ModBlocks.HELM.get()).build(null));

    public static final RegistryObject<BlockEntityType<BoilerBlockEntity>> BOILER =
            BLOCK_ENTITIES.register("boiler", () -> BlockEntityType.Builder.of(
                    BoilerBlockEntity::new, ModBlocks.BOILER.get()).build(null));
}