mcprep_data_refresh.py

#!/usr/local/bin/python3
# Tool to pull down material names from jmc2obj and Mineways

import json
import os
import shutil
import sys
import urllib.request
import xml.etree.ElementTree as ET
import zipfile

MINEWAYS_URL = "https://raw.githubusercontent.com/erich666/Mineways/master/Win/tiles.h"

# unused (here) mapping of block name IDs to materials/textures
# JMC_URL = "https://raw.githubusercontent.com/jmc2obj/j-mc-2-obj/master/conf/blocks.conf"

# could have multiple mappings, just use the latest here
JMC_1_13 = "https://raw.githubusercontent.com/jmc2obj/j-mc-2-obj/master/conf/texsplit_1.13.conf"


def save_file_str(url):
	"""Save to temp location next to script."""
	request = urllib.request.Request(url)
	result = urllib.request.urlopen(request)
	result_string = result.read()
	result.close()
	return result_string.decode()


def get_jmc2obj_list():
	"""Get the list of material names for jmc2obj"""
	raw_str = save_file_str(JMC_URL)
	root = ET.fromstring(raw_str)
	outlist = {}
	for block in root:
		mats = block.findall("materials")
		for itm in mats:
			materials = itm.text.split()
			outlist.update({sub.strip():None for sub in materials})
		# if mats:
		# 	outlist.update({itm.text:None for itm in mats})
	return outlist


def get_jmc2obj_mapping():
	"""Download the original texture mapping from the jmc2obj source code"""
	raw_str = save_file_str(JMC_1_13)
	root = ET.fromstring(raw_str)
	outlist = {}
	ln = len('assets/minecraft/textures/')

	for tex in root:
		src_tex = tex.get('name')
		jmc_tex = tex.getchildren()[0].get('name')

		if not src_tex.endswith('.png'):
			print("Not a png: "+src_tex)
			raise Exception("Src file is not a png")

		# note that block is not plural in this conf file, but is in tex packs
		if src_tex.startswith('assets/minecraft/textures/block/'):
			outlist[jmc_tex] = os.path.basename(src_tex)[:-4]
		elif src_tex.startswith('assets/minecraft/textures/'):
			if jmc_tex in outlist:
				continue # don't duplicate for non blocks
			outlist[jmc_tex] = src_tex[ln:-4]

		# so, if it stars with "assets/minecraft/textures/block/",
		# then truncate to just the basename
		# else, join one level up (ie entity instead of block, etc)
	return outlist

def jmc2obj_extras():
	"""Known additional mappings for jmc2obj"""
	outlist = {
		"vines":"vine",

		# removed ~ Jan 15 2021,
		# added back to maintain longer backwards compatibility
		"cake_inside": "cake_inner",
		"cauldron_feet": "cauldron_bottom",
		"cauldron_inside": "cauldron_inner",
		"door_wood_bottom": "oak_door_bottom",
		"door_wood_top": "oak_door_top",
		"enderchest": "entity/chest/ender",
		"largechest": "entity/chest/normal_double",
		"largechest_trapped": "entity/chest/trapped_double",
		"plank_acacia": "acacia_planks",
		"plank_birch": "birch_planks",
		"plank_dark_oak": "dark_oak_planks",
		"plank_jungle": "jungle_planks",
		"plank_oak": "oak_planks",
		"plank_spruce": "spruce_planks",
		"quartz_bottom": "quartz_block_bottom",
		"quartz_side": "quartz_block_side",
		"quartz_side_chiseled": "chiseled_quartz_block",
		"quartz_side_lines": "quartz_pillar",
		"quartz_top": "quartz_block_top",
		"quartz_top_chiseled": "chiseled_quartz_block_top",
		"quartz_top_lines": "quartz_pillar_top",
		"red_sandstone_carved": "chiseled_red_sandstone",
		"sandstone_side": "sandstone",
		"sandstone_side_carved": "chiseled_sandstone",
		"stone_brick_circle": "chiseled_stone_bricks",
		"stone_brick_cracked": "cracked_stone_bricks",

	}
	return outlist

# def jmc2mc(name, vanilla):
# 	"""Function that attemtps to map jmc2obj texture name to canonical"""
# 	if name in vanilla:
# 		return name

# 	# general used vars for checks
# 	sub = name.split("_")

# 	def flipterm(name):
# 		name = name.replace("lower", "bottom")
# 		return name.replace("upper", "top")

# 	# just reverse the two phrases
# 	if len(sub) == 2:
# 		reverse = sub[-1] + "_" + sub[0]
# 		if reverse in vanilla:
# 			return reverse
# 		reverse = flipterm(reverse)
# 		if reverse in vanilla:
# 			return reverse

# 	# `door_birch_lower` to `birch_door_bottom`
# 	# or `door_dark_oak_upper` to `dark_oak_door_top`
# 	if "door" in name:
# 		if len(sub)>=3:
# 			recon = "{}_door_{}".format(
# 				"_".join(sub[1:-1]),
# 				sub[-1])
# 			if recon in vanilla:
# 				return recon
# 			termflip = flipterm(recon)
# 			if termflip in vanilla:
# 				return termflip

# 	# flip around e.g. `wool_light_blue` to `light_blue_wool`,
# 	# also applies to concrete and carpet
# 	prefix_to_suffix = "_".join(sub[1:-1]) + "_" + sub[-1]
# 	if prefix_to_suffix in vanilla:
# 		return prefix_to_suffix
# 	ps_termflip = flipterm(prefix_to_suffix)
# 	if ps_termflip in vanilla:
# 		return ps_termflip

# 	# e.g. `log_dark_oak_top` to `dark_oak_log_top`

# 	# if "carpet" in sub:
# 	# 	wool = name.replace("carpet", "wool")
# 	# 	if wool in vanilla:
# 	# 		return wool
# 	if "carpet" in name:
# 		wool = reverse.replace("carpet", "wool")
# 		if wool in vanilla:
# 			return wool

# 	if "glass" in name:
# 		name = name.replace("glass", "stained_glass")


# 	# for those we REALLY can't generalize, hard code mapping
# 	hmap = {
# 		"":""
# 	}
# 	if name in hmap and hmap[name] in vanilla:
# 		return hmap[name]

# 	if name+"_top" in vanilla:
# 		return name+"_top"

# 	# started with over 390/646, downed it to:
# 	return None


def get_mineways_list(vanilla):
	"""Get the list of material names for Mineways"""
	raw_str = save_file_str(MINEWAYS_URL) # for individual file-based mats
	outlist = {}

	# This is super low-level page parsing, will break if source code changes
	textblock = raw_str.split("} gTilesTable[TOTAL_TILES] = {")[1]
	textblock = textblock.split("};")[0]

	mats = []
	for line in textblock.split('\n'):
		if not ',' in line or not line.strip().startswith('{'):
			continue
		prename = line.split(',')[4] # fourth item in {0,0,0,L"block"}
		name = prename.split('"')[1] # go from  ' L"lectern_sides"'  to 'lectern_sides'
		presubname = line.split(',')[4]
		presubname = presubname.split('"')[1]

		# special, bespoke names like MWO_double_chest_top_left should be ignored
		if presubname.lower().startswith("mwo_") or presubname.lower().startswith("mw_"):
			continue

		if presubname:
			# print("YES!: ", presubname)
			name_sub = name+"_"+presubname
			if name_sub in vanilla:
				outlist[name_sub] = name_sub
				#mats.append(name_sub)
			elif name in vanilla:
				outlist[name] = name
				#mats.append(name)
			else:
				outlist[name.lower()] = name
				#mats.append(name.lower())
		else:
			if name in vanilla:
				outlist[name] = name
				#mats.append(name)
			else:
				outlist[name.lower()] = name
				#mats.append(name.lower()) # make it none?
	return outlist

def mineways_extras():
	"""Known additional mappings for Mineways"""
	outlist = {
		"Acacia_Door":"acacia_door_bottom",
		"Activator_Rail":"activator_rail",
		"Beacon":"beacon",
		# do NOT include Bamboo, it's completely off (overloaded w/ campfire)
		"Birch_Door":"birch_door_bottom",
		"Brewing_Stand":"brewing_stand_base", # maybe don't? since only for meshswap
		"Bookshelf":"bookshelf",
		"Bricks":"bricks",
		"Cactus":"cactus_side",
		"Command_Block":"chain_command_block",
		"Chain_Command_Block":"chain_command_block",
		"Carrots":"carrots_stage3",
		"Campfire":"campfire_log",
		"Chest":"entity/chest/normal",
		"MWO_chest_top":"entity/chest/normal",
		# "MWO_double_chest_top_right":"entity/chest/chest_double", ensures only placing once
		"MWO_double_chest_top_left":"entity/chest/normal_double",
		"Cobweb":"cobweb",
		"Crafting_Table":"crafting_table_top",
		"Crafting_Table__Cartography_Table":"cartography_table_top",
		"Crafting_Table__Fletching_Table":"fletching_table_top",
		"Crafting_Table__Smithing_Table":"smithing_table_top",
		"Dandelion":"dandelion",
		"Dark_Oak_Door":"dark_oak_door_bottom",
		"Dead_Bush":"dead_bush",
		"Detector_Rail":"detector_rail",
		"Enchanting_Table":"enchanting_table_top",
		"End_Rod":"end_rod",
		"Ender_Chest":"entity/chest/ender",
		"Furnace":"furnace_front_on", # assume on? meshswap implication
		"Furnace__Blast_Furnace":"blast_furnace_front_on", # assume on? meshswap implication
		"Furnace__Loom":"loom_top",
		"Furnace__Smoker":"smoker_front_on", # assume on? meshswap implication
		"Fire":"fire_0",
		"Glowstone":"glowstone",
		"Grass__Fern":"fern", # single block high
		"Grass__Tall_Grass":"grass", # ie tall grass
		"Glass":"glass",
		"Glass_Pane":"glass_pane_top",
		"Lily_Pad":"lily_pad",
		"Iron_Door":"iron_door_bottom",
		"Jack_o'Lantern":"jack_o_lantern",
		"Pumpkin":"carved_pumpkin",
		"Kelp":"kelp",
		# "Kelp__1":"",
		"Ladder":"ladder",
		"Lantern":"lantern",
		"Large_Flowers":"sunflower", # decide block
		"Large_Flowers__1":"",
		"Large_Flowers__2":"",
		"Large_Flowers__3":"large_fern_bottom",
		"Large_Flowers__4":"",
		"Large_Flowers__5":"",
		"Magma_Block":"magma",
		"Poppy":"poppy",
		"Poppy__Allium":"allium",
		"Poppy__Azure_Bluet":"azure_bluet",
		"Poppy__Blue_Orchid":"blue_orchid",
		"Poppy__Orange_Tulip":"orange_tulip",
		"Poppy__Oxeye_Daisy":"oxeye_daisy",
		"Poppy__Pink_Tulip":"pink_tulip",
		"Poppy__Red_Tulip":"red_tulip",
		"Poppy__White_Tulip":"white_tulip",
		"Poppy__Wither_Rose":"wither_rose",
		"Powered_Rail":"powered_rail",
		"Rail":"rail",
		"Redstone_Lamp_(active)":"redstone_lamp",
		"Redstone_Lamp_(inactive)":"redstone_lamp_off",
		"Redstone_Torch_(active)":"redstone_torch",
		"Redstone_Torch_(inactive)":"redstone_torch_off",
		"Sapling":"oak_sapling",
		"Sapling__Acacia_Sapling":"acacia_sapling",
		"Sapling__Birch_Sapling":"birch_sapling",
		"Sapling__Dark_Oak_Sapling":"dark_oak_sapling",
		"Sapling__Jungle_Sapling":"jungle_sapling",
		"Sapling__Spruce_Sapling":"spruce_sapling",
		"Spruce_Door":"spruce_door_bottom",
		"Seagrass":"tall_seagrass_bottom",
		"Sea_Pickle":"sea_pickle",
		"Sea_Lantern":"sea_lantern",
		"Sugar_Cane":"sugar_cane",
		"Stationary_Lava":"lava_still",
		"Stationary_Water":"water_still",
		# "Stained_Glass*":"",
		# "Stone_Bricks*":"",
		"Stone_Cutter":"stonecutter_top", # should be a meshswap item evetually
		"Sunflower":"sunflower_bottom",
		"Trapped_Chest":"entity/chest/normal",
		"Torch":"torch",
		"TNT":"tnt_top",
		"Vines":"vine",
		"Wheat":"wheat_stage7",
		"Wooden_Door":"oak_door_bottom",
		"Campfire":"campfire_log"
	}
	return outlist

def split_underscore_mappings(mineways_dict):
	"""Returns the list, adding new items like Sapling__Spruce_Sapling to Spruce_Sapling"""
	return {itm.split("__")[-1]:mineways_dict[itm] for itm in mineways_dict
		if "__" in itm}

def mineways2mc(name, vanilla):
	"""Function that attemtps to map Mineways texture name to canonical"""
	if name in vanilla:
		return name
	return None


def get_vanilla_list(copy_file=False):
	"""Get the list of material names from vanilla Minecraft (local install)"""
	outlist = {}

	# OSX path
	path = os.path.join(
		os.path.expanduser('~'),
		"Library", "Application Support", "minecraft", "versions")
	if not os.path.isdir(path):
		raise Exception('Could not get vanilla path')

	versions = [ver for ver in os.listdir(path)
				if os.path.isdir(os.path.join(path, ver))]

	# turn into sortable tuples
	verion_tuples = []
	for ver in versions:
		temp = []
		for itm in ver.split('.'):
			try:
				temp.append(int(itm))
			except:
				break
		verion_tuples.append(tuple(temp))

	# parallel sort the list based on the generated tuple
	verion_tuples, versions = zip(*sorted(zip(verion_tuples, versions)))
	# print(versions)

	jarfile = None
	for i, ver in reversed(list(enumerate(verion_tuples))):
		ver_folder = os.path.join(path, versions[i])
		any_jar = [jar for jar in os.listdir(ver_folder)
			if jar.lower().endswith('.jar')]
		if any_jar:
			jarfile = os.path.join(path, versions[i], any_jar[0])
			break

	if not jarfile:
		raise Exception("Could not get most recent jar version")

	mcprep_resources = os.path.join("MCprep_addon", "MCprep_resources",
		"resourcepacks", "mcprep_default")
	prefix = 'assets/minecraft/textures/'
	mcp_subfolders = ["block", "entity", "environment", "item", "mob_effect",
		"models", "painting", "particle"]
	if copy_file:
		for sub in mcp_subfolders:
			checkpath = os.path.join(mcprep_resources, prefix, sub)
			if os.path.isdir(checkpath):
				print("Removing MCprep resources folder: "+sub)
				shutil.rmtree(checkpath)
			else:
				print("Error! Could not find "+checkpath)
		print("Removed MCprep resource folders, will copy over replacements")

	print("Got jar version: {}".format(os.path.basename(jarfile)))
	archive = zipfile.ZipFile(jarfile, 'r')

	for name in archive.namelist():
		if not (name.endswith('.png') or name.endswith('.mcmeta')):
			continue
		base = os.path.basename(name)[:-4]
		sub_mcp = name.startswith(prefix) and name.split(prefix)[1].split(os.sep)[0] in mcp_subfolders

		if copy_file is True and sub_mcp is True:
			# TODO: Further ensure subfolder is one of mcp_subfolders
			# copy file to MCprep resource directory
			new_path = os.path.join(mcprep_resources, name)
			os.makedirs(os.path.dirname(new_path), exist_ok=True)
			with archive.open(name) as zf, open(new_path, 'wb') as f:
				shutil.copyfileobj(zf, f)
			print("\tCopied "+name)

		# limit to textures only hereafter
		if not name.endswith('.png'):
			continue

		if name.startswith(prefix+"block"):
			outlist[base] = base
		elif base in outlist:
			continue # don't duplicate for non blocks textures
		elif name.startswith(prefix+"item"):
			continue # skip adding duplicative item mappings
		elif name.startswith(prefix): # at least in textures folder
			if 'lava' in name and 'particle' in name:
				continue # hack to avoid clash with jmc2obj lava:lava_still
			outlist[base] = name[len(prefix):-4]
		else:
			outlist[base] = None
			# print("Not in textures folder: "+name)
			# mostly just "realms" stuff

	archive.close()
	return outlist


def vanilla_overrides(vanilla_map):
	"""go through and create the mapping with special overrides"""
	outlist = vanilla_map.copy()
	overrides = {
		"fire":"fire_0",
		"Campfire":"campfire_log"
	}
	outlist.update(overrides)
	return outlist


def get_current_json(backup=False):
	"""Returns filepath of current json"""
	suffix = "" if not backup else "backup"
	filepath = os.path.join(
		"MCprep_addon", "MCprep_resources",
		"mcprep_data_update"+suffix+".json")
	return filepath


def read_base_mapping():
	"""Read in the existing mcprep_data_update.json file shipped with MCprep"""
	filepath = "mcprep_data_base.json"
	if not os.path.isfile(filepath):
		raise Exception("File missing: "+filepath)

	with open(filepath, 'r') as rawfile:
		data = json.load(rawfile)
	return data

def get_cannon_block_mappping():
	"""Returns a dict of the Block (not material) names to texturepack map"""

	# only include those special cases that need overrides,
	# used in meshswap for display name
	outlist = {
		"entity/chest/normal":"chest",
		"entity/chest/normal_double":"chest_double",
		"fire_0":"fire"
	}
	return outlist


def run_all(auto=False):
	"""Execute getting all of the texture lists and to compare."""

	data = {"blocks": {}}

	# loads in existing resource json, for truly hand-stated things
	# ie reflection, emission, etc
	base_override = read_base_mapping()

	if auto:
		xin = "n"
	else:
		xin = input("Copy latest vanilla textures to MCprep? (y): ")

	# consider also passing in pref to get specific version, for old names
	vanilla = get_vanilla_list(xin=="y")
	vanilla_map = vanilla_overrides(vanilla) # don't need to return as pass ref?

	# load material lists from online blobs
	jmc = get_jmc2obj_mapping()
	mineways = get_mineways_list(vanilla)

	# load the hard-coded lists
	# hard_coded = ["reflective", "water", "emit", "desaturated",
	# 	"animated_mineways_index", "solid", "metallic"]
	# for setname in hard_coded:
	# 	data["blocks"] = {}
	# 	data["blocks"][setname] = existing["blocks"].get(setname)

	# now update the main blocks with the known mappings only
	# already exact, but could do cross check for any current MC canonical missing
	data["blocks"]["block_mapping_jmc"] = jmc
	data["blocks"]["block_mapping_jmc"].update(jmc2obj_extras())

	# data["blocks"]["block_mapping_jmc"] = {
	# 	mat:jmc2mc(mat, vanilla) for mat in jmc
	# 	if jmc2mc(mat, vanilla) is not None}
	data["blocks"]["block_mapping_mineways"] = mineways
	data["blocks"]["block_mapping_mineways"].update(mineways_extras())
	data["blocks"]["block_mapping_mineways"].update(
		split_underscore_mappings(data["blocks"]["block_mapping_mineways"]))

	data["blocks"]["block_mapping_mc"] = vanilla_map
	data["blocks"]["canon_mapping_block"] = get_cannon_block_mappping()
	# data["blocks"]["block_mapping_mineways"] = {
	# 	mat:mineways2mc(mat, vanilla) for mat in mineways
	# 	if mineways2mc(mat, vanilla) is not None}

	data["blocks"].update(base_override["blocks"])

	vanilla_blocks = [vanilla[itm] for itm in vanilla
			if itm is not None
			and vanilla[itm] is not None
			and ("/" not in vanilla[itm] or vanilla[itm].startswith("blocks"))]

	bl = data["blocks"]
	print_str = """Found the following:
	{jmc} jmc2obj blocks
	{mine} Mineways blocks
	{MC} mc textures,
	{BLK} are blocks
	""".format(
		jmc=len(bl["block_mapping_jmc"]),
		mine=len(bl["block_mapping_mineways"]),
		MC=len(vanilla),
		BLK=len(vanilla_blocks))
	print(print_str)
	# same for the other two..

	# Goal: generate, best we can, the actual mapping file to use for materials

	# save the output
	fileout = "mcprep_data_update_staging.json"
	fileout = os.path.abspath(fileout)
	with open(fileout, "w") as dmp:
		json.dump(data, dmp, indent="\t", sort_keys=True)
	print("Output file:")
	print(fileout)

	if auto:
		xin = "y"
	else:
		xin = input("Show missing jmc2obj mappings? (y): ")
	if xin == "y":
		print("PRINTING jmc2obj MATERIALS WITH NO VANILLA MAPPING")
		miss = [itm for itm in jmc
			if os.path.basename(jmc[itm]) not in vanilla]
		for itm in sorted(miss):
			print("\t"+itm)
		print("Total miss {}, versus {} total".format(
			len(miss), len(jmc)))

		# now, find how many vanilla textures are NOT in jmc2obj
		print("PRINTING vanilla MATERIALS NOT IN JMC2OBJ")
		jmc_v = [jmc[itm] for itm in jmc]
		miss = [vanilla[itm] for itm in vanilla
			if itm not in jmc_v
			and vanilla[itm] is not None
			and ("/" not in vanilla[itm] or vanilla[itm].startswith("blocks"))]
		for itm in sorted(miss):
			print("\t"+itm)
		print("Total miss {}, versus {} (blocks only)".format(
			len(miss), len(vanilla_blocks)))

	if auto:
		xin = "y"
	else:
		xin = input("Show missing Mineways mappings? (y): ")
	if xin == "y":
		print("PRINTING mineways MISSED MATERIALS")
		miss = [itm for itm in mineways
			if not mineways[itm]
			or os.path.basename(mineways[itm]) not in vanilla]
		for itm in sorted(miss):
			print("\t"+itm)
		print("\tTotal miss {}, versus {} total".format(
			len(miss), len(mineways)))

		# now, find how many vanilla textures are NOT in jmc2obj
		print("PRINTING vanilla MATERIALS NOT IN MINEWAYS")
		minew_v = [mineways[itm] for itm in mineways]
		miss = [vanilla[itm] for itm in vanilla
			if itm not in minew_v
			and vanilla[itm] is not None
			and ("/" not in vanilla[itm] or vanilla[itm].startswith("blocks"))]
		for itm in sorted(miss):
			print("\t"+itm)
		print("Total miss {}, versus {} (blocks only)".format(
			len(miss), len(vanilla_blocks)))

	if auto:
		xin = "y"
	else:
		xin = input("Replace current mapping file? (y): ")
	if xin == "y":
		main_file = get_current_json()
		bup_file = get_current_json(backup=True)
		if os.path.isfile(bup_file):
			os.remove(bup_file)
		os.rename(fileout, main_file)
		print("Replaced file: "+main_file)


if __name__ == '__main__':
	if "-auto" in sys.argv:
		run_all(auto=True)
	else:
		run_all()