Create/src/main/java/com/simibubi/create/AllInteractionBehaviours.java

package com.simibubi.create;

import com.simibubi.create.content.contraptions.components.deployer.DeployerMovingInteraction;
import com.simibubi.create.content.contraptions.components.structureMovement.interaction.LeverMovingInteraction;
import com.simibubi.create.content.contraptions.components.structureMovement.MovingInteractionBehaviour;

import net.minecraft.block.Block;
import net.minecraft.block.Blocks;
import net.minecraft.util.ResourceLocation;

import javax.annotation.Nullable;

import java.util.HashMap;
import java.util.function.Supplier;

public class AllInteractionBehaviours {
	private static final HashMap<ResourceLocation, Supplier<MovingInteractionBehaviour>> INTERACT_BEHAVIOURS = new HashMap<>();

	public static void addInteractionBehaviour (ResourceLocation loc, Supplier<MovingInteractionBehaviour> behaviour) {
		if (INTERACT_BEHAVIOURS.containsKey(loc)) {
			Create.LOGGER.warn("Interaction behaviour for " + loc.toString() + " was overridden");
		}
		INTERACT_BEHAVIOURS.put(loc, behaviour);
	}

	public static void addInteractionBehavioiur (Block block, Supplier<MovingInteractionBehaviour> behaviour) {
		addInteractionBehaviour(block.getRegistryName(), behaviour);
	}

	@Nullable
	public static MovingInteractionBehaviour of (ResourceLocation loc) {
		return (INTERACT_BEHAVIOURS.get(loc) == null) ? null : INTERACT_BEHAVIOURS.get(loc).get();
	}

	@Nullable
	public static MovingInteractionBehaviour of (Block block) {
		return of(block.getRegistryName());
	}

	public static boolean contains (Block block) {
		return INTERACT_BEHAVIOURS.containsKey(block.getRegistryName());
	}

	static void register () {
		addInteractionBehaviour(Blocks.LEVER.getRegistryName(), LeverMovingInteraction::new);
		addInteractionBehaviour(AllBlocks.DEPLOYER.getId(), DeployerMovingInteraction::new);
	}
}