import java.util.Arrays;
import java.util.HashSet;
import java.util.Optional;
import java.util.Set;

public class Main {
    static MiningNode[] VANILLA_NODES = new MiningNode[4];
    static {
        var base = new MiningNode(Optional.empty(), Set.of());
        var stone = new MiningNode(Optional.of(new BlockTag("needs_stone_tool")), set(base));
        var iron = new MiningNode(Optional.of(new BlockTag("needs_iron_tool")), set(stone));
        var diamond = new MiningNode(Optional.of(new BlockTag("needs_diamond_tool")), set(iron));
        VANILLA_NODES[0] = base;
        VANILLA_NODES[1] = stone;
        VANILLA_NODES[2] = iron;
        VANILLA_NODES[3] = diamond;
    }

    @SafeVarargs
    static <T> Set<T> set(T... elements) {
        return new HashSet<>(Arrays.asList(elements));
    }

    public static void main(String[] args) {
        var myLevel1 = new MiningNode(Optional.of(new BlockTag("needs_my_tool_level_1")), set(VANILLA_NODES[1]));
        var myLevel2 = new MiningNode(Optional.of(new BlockTag("needs_my_tool_level_2")), set(myLevel1));
        var myLevel3 = new MiningNode(Optional.of(new BlockTag("needs_my_tool_level_3")), set(myLevel2));
        VANILLA_NODES[2].predecessors().add(myLevel3);
        System.out.println(VANILLA_NODES[1].contains(new BlockTag("needs_my_tool_level_2")));
        System.out.println(VANILLA_NODES[2].contains(new BlockTag("needs_my_tool_level_2")));
    }

    record ToolMaterial(
            int miningLevel,
            int itemDurability,
            float miningSpeed,
            int attackDamage,
            Optional<MiningNode> miningNode
    ) {
        MiningNode getNode() {
            return miningNode.orElseGet(() -> VANILLA_NODES[miningLevel]);
        }
    }

    record MiningNode(
            Optional<BlockTag> tag,
            Set<MiningNode> predecessors
    ) {
        boolean contains(BlockTag otherTag) {
            return (tag.isPresent() && tag.get().equals(otherTag)) || predecessors.stream().anyMatch(predecessor -> predecessor.contains(otherTag));
        }
    }

    record BlockTag(String id) {}
}