ConstructionWand/src/main/java/thetadev/constructionwand/basics/ReplacementRegistry.java

package thetadev.constructionwand.basics;

import net.minecraft.block.Block;
import net.minecraft.item.Item;
import net.minecraft.util.ResourceLocation;
import net.minecraftforge.registries.ForgeRegistries;
import thetadev.constructionwand.ConstructionWand;

import java.util.HashSet;
import java.util.Set;

public class ReplacementRegistry
{
	private static final HashSet<HashSet<Item>> replacements = new HashSet<>();

	public static void init() {
		for(Object key : ConfigServer.SIMILAR_BLOCKS.get()) {
			if(!(key instanceof String)) continue;
			HashSet<Item> set = new HashSet<>();

			for(String id : ((String)key).split(";")) {
				Item item = ForgeRegistries.ITEMS.getValue(new ResourceLocation(id));
				if(item == null) {
					ConstructionWand.LOGGER.warn("Replacement Registry: Could not find item "+id);
					continue;
				}
				set.add(item);
			}
			if(!set.isEmpty()) replacements.add(set);
		}
	}

	public static Set<Item> getMatchingSet(Item item) {
		HashSet<Item> res = new HashSet<>();

		for(HashSet<Item> set : replacements) {
			if(set.contains(item)) res.addAll(set);
		}
		res.remove(item);
		return res;
	}

	public static boolean matchBlocks(Block b1, Block b2) {
		if(b1 == b2) return true;

		for(HashSet<Item> set : replacements) {
			if(set.contains(b1.asItem()) && set.contains(b2.asItem())) return true;
		}
		return false;
	}
}