mirror of
https://github.com/Aurorastation/Aurora.3.git
synced 2025-12-14 12:12:03 +00:00
1d335dd9e9
Adds TG's mapmerge2, a new TGM-compatible merger that handles things like dictionary expansion much better than the old tool, as well as supporting a git merge hook for map files. This PR does not remove the old merger as this one is more annoying to setup.
128 lines
3.8 KiB
Python
128 lines
3.8 KiB
Python
# Common code for the frontend interface of map tools
|
|
import sys
|
|
import os
|
|
import pathlib
|
|
import shutil
|
|
from collections import namedtuple
|
|
|
|
Settings = namedtuple('Settings', ['map_folder', 'tgm'])
|
|
MapsToRun = namedtuple('MapsToRun', ['files', 'indices'])
|
|
|
|
def string_to_num(s):
|
|
try:
|
|
return int(s)
|
|
except ValueError:
|
|
return -1
|
|
|
|
def read_settings():
|
|
# discover map folder if needed
|
|
try:
|
|
map_folder = os.environ['MAPROOT']
|
|
except KeyError:
|
|
map_folder = '_maps/'
|
|
for _ in range(8):
|
|
if os.path.exists(map_folder):
|
|
break
|
|
map_folder = os.path.join('..', map_folder)
|
|
else:
|
|
map_folder = None
|
|
|
|
# assume TGM is True by default
|
|
tgm = os.environ.get('TGM', "1") == "1"
|
|
|
|
return Settings(map_folder, tgm)
|
|
|
|
def pretty_path(settings, path_str):
|
|
if settings.map_folder:
|
|
return path_str[len(os.path.commonpath([settings.map_folder, path_str]))+1:]
|
|
else:
|
|
return path_str
|
|
|
|
def prompt_maps(settings, verb):
|
|
if not settings.map_folder:
|
|
print("Could not autodetect the _maps folder, set MAPROOT")
|
|
exit(1)
|
|
|
|
list_of_files = list()
|
|
for root, directories, filenames in os.walk(settings.map_folder):
|
|
for filename in [f for f in filenames if f.endswith(".dmm")]:
|
|
list_of_files.append(pathlib.Path(root, filename))
|
|
|
|
last_dir = ""
|
|
for i, this_file in enumerate(list_of_files):
|
|
this_dir = this_file.parent
|
|
if last_dir != this_dir:
|
|
print("--------------------------------")
|
|
last_dir = this_dir
|
|
print("[{}]: {}".format(i, pretty_path(settings, str(this_file))))
|
|
|
|
print("--------------------------------")
|
|
in_list = input("List the maps you want to " + verb + " (example: 1,3-5,12):\n")
|
|
in_list = in_list.replace(" ", "")
|
|
in_list = in_list.split(",")
|
|
|
|
valid_indices = list()
|
|
for m in in_list:
|
|
index_range = m.split("-")
|
|
if len(index_range) == 1:
|
|
index = string_to_num(index_range[0])
|
|
if index >= 0 and index < len(list_of_files):
|
|
valid_indices.append(index)
|
|
elif len(index_range) == 2:
|
|
index0 = string_to_num(index_range[0])
|
|
index1 = string_to_num(index_range[1])
|
|
if index0 >= 0 and index0 <= index1 and index1 < len(list_of_files):
|
|
valid_indices.extend(range(index0, index1 + 1))
|
|
|
|
return MapsToRun(list_of_files, valid_indices)
|
|
|
|
def process(settings, verb, *, modify=True, backup=None):
|
|
if backup is None:
|
|
backup = modify # by default, backup when we modify
|
|
assert modify or not backup # doesn't make sense to backup when not modifying
|
|
|
|
if len(sys.argv) > 1:
|
|
maps = sys.argv[1:]
|
|
else:
|
|
maps = prompt_maps(settings, verb)
|
|
maps = [str(maps.files[i]) for i in maps.indices]
|
|
print()
|
|
|
|
if not maps:
|
|
print("No maps selected.")
|
|
return
|
|
|
|
if modify:
|
|
print(f"Maps WILL{'' if settings.tgm else ' NOT'} be converted to tgm.")
|
|
if backup:
|
|
print("Backups will be created with a \".before\" extension.")
|
|
else:
|
|
print("Warning: backups are NOT being taken.")
|
|
|
|
print(f"\nWill {verb} these maps:")
|
|
for path_str in maps:
|
|
print(pretty_path(settings, path_str))
|
|
|
|
try:
|
|
confirm = input(f"\nPress Enter to {verb}...\n")
|
|
except KeyboardInterrupt:
|
|
confirm = "^C"
|
|
if confirm != "":
|
|
print(f"\nAborted.")
|
|
return
|
|
|
|
for path_str in maps:
|
|
print(f' - {pretty_path(settings, path_str)}')
|
|
|
|
if backup:
|
|
shutil.copyfile(path_str, path_str + ".before")
|
|
|
|
try:
|
|
yield path_str
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
else:
|
|
print("Succeeded.")
|
|
|
|
print("\nFinished.")
|