import json

with open('minecraft_data/registries.json') as f:
    ITEMS = json.load(f)['minecraft:item']['entries']

BEDS = [
    'white_bed',
    'orange_bed',
    'magenta_bed',
    'light_blue_bed',
    'yellow_bed',
    'lime_bed',
    'pink_bed',
    'gray_bed',
    'light_gray_bed',
    'cyan_bed',
    'purple_bed',
    'blue_bed',
    'brown_bed',
    'green_bed',
    'red_bed',
    'black_bed',
]

SHOVELS = [
    'wooden_shovel',
    'stone_shovel',
    'golden_shovel',
    'iron_shovel',
    'diamond_shovel',
    'netherite_shovel',
]

SAPLINGS = [
    'oak_sapling',
    'spruce_sapling',
    'birch_sapling',
    'jungle_sapling',
    'acacia_sapling',
    'dark_oak_sapling',
]

BED_IDS = set()
for item_name in BEDS:
    BED_IDS.add(ITEMS['minecraft:'+item_name]['protocol_id'])

SHOVEL_IDS = set()
for item_name in SHOVELS:
    SHOVEL_IDS.add(ITEMS['minecraft:'+item_name]['protocol_id'])

SAPLING_IDS = set()
for item_name in SAPLINGS:
    SAPLING_IDS.add(ITEMS['minecraft:'+item_name]['protocol_id'])

ITEM_NAMES = {}
for item_name, item in ITEMS.items():
    ITEM_NAMES[ITEMS[item_name]['protocol_id']] = item_name.replace('minecraft:', '')

def get_id(name):
    return ITEMS['minecraft:' + name]['protocol_id']

CHEST_ID = get_id('chest')
GAPPLE_ID = get_id('enchanted_golden_apple')
SAND_ID = get_id('sand')
NETHERWART_ID = get_id('nether_wart')

NEEDED_ITEMS = BED_IDS | set([CHEST_ID])
WANTED_ITEMS = SAPLING_IDS | set([NETHERWART_ID])