def match_language(lang: str, log_dir: Path) -> str:
logged_langs = set(
[util.get_file_language(l) for l in log_dir.glob("*texts*")])
if lang in logged_langs:
return lang
elif lang[2:4] in [l[2:4] for l in logged_langs]:
return [l for l in logged_langs if l[2:4] == lang[2:4]][0]
else:
return [l for l in LANGUAGES if l in logged_langs][0]
def generate_diff(self, mod_dir: Path, modded_files: List[Union[str,
Path]]):
print("Checking for modified languages...")
languages = {
util.get_file_language(file)
for file in modded_files
if (isinstance(file, Path) and "Bootup_" in file.name
and "Graphic" not in file.name)
}
if not languages:
return None
util.vprint(f'Languages: {",".join(languages)}')
language_map = {}
save_langs = (LANGUAGES if self._options.get("all_langs", False) else
[util.get_settings("lang")])
for lang in save_langs:
if lang in languages:
language_map[lang] = lang
elif lang[2:4] in [l[2:4] for l in languages]:
language_map[lang] = [
l for l in languages if l[2:4] == lang[2:4]
][0]
else:
language_map[lang] = [l for l in LANGUAGES
if l in languages][0]
util.vprint(f"Language map:")
util.vprint(language_map)
language_diffs = {}
for language in set(language_map.values()):
print(f"Logging text changes for {language}...")
language_diffs[language] = diff_language(
mod_dir / util.get_content_path() / "Pack" /
f"Bootup_{language}.pack",
pool=self._pool,
)
return {
save_lang: language_diffs[map_lang]
for save_lang, map_lang in language_map.items()
}
def generate_master_rstb(verbose: bool = False):
"""Merges all installed RSTB modifications"""
print('Merging RSTB changes...')
if (util.get_master_modpack_dir() / 'logs' / 'master-rstb.log').exists():
(util.get_master_modpack_dir() / 'logs' / 'master-rstb.log').unlink()
table = get_stock_rstb()
rstb_values = {}
for mod in util.get_installed_mods():
rstb_values.update(get_mod_rstb_values(mod))
if (util.get_master_modpack_dir() / 'logs' / 'rstb.log').exists():
rstb_values.update(get_mod_rstb_values(util.get_master_modpack_dir()))
if (util.get_master_modpack_dir() / 'logs' / 'map.log').exists():
rstb_values.update(get_mod_rstb_values(
util.get_master_modpack_dir(), log_name='map.log'))
table, rstb_changes = merge_rstb(table, rstb_values)
for change in rstb_changes:
if not change[1] or (change[1] and verbose):
print(change[0])
for bootup_pack in util.get_master_modpack_dir().glob('content/Pack/Bootup_*.pack'):
lang = util.get_file_language(bootup_pack)
if table.is_in_table(f'Message/Msg_{lang}.product.sarc'):
table.delete_entry(f'Message/Msg_{lang}.product.sarc')
rstb_path = util.get_master_modpack_dir() / 'content' / 'System' / 'Resource' / \
'ResourceSizeTable.product.srsizetable'
if not rstb_path.exists():
rstb_path.parent.mkdir(parents=True, exist_ok=True)
with rstb_path.open('wb') as r_file:
with io.BytesIO() as buf:
table.write(buf, True)
r_file.write(util.compress(buf.getvalue()))
rstb_log = util.get_master_modpack_dir() / 'logs' / 'master-rstb.log'
rstb_log.parent.mkdir(parents=True, exist_ok=True)
with rstb_log.open('w', encoding='utf-8') as r_file:
r_file.write('\n'.join([change[0].strip() for change in rstb_changes]))
def generate_diff(self, mod_dir: Path, modded_files: List[Union[str, Path]]):
print("Checking for modified languages...")
mod_langs = {
util.get_file_language(file)
for file in modded_files
if (
isinstance(file, Path)
and "Bootup_" in file.name
and "Graphic" not in file.name
)
}
if not mod_langs:
return None
util.vprint(f'Languages: {",".join(mod_langs)}')
# find a user lang for each mod lang
language_map = map_languages(mod_langs, util.get_user_languages())
util.vprint("Language map:")
util.vprint(language_map)
language_diffs = {}
for mod_lang, user_lang in language_map.items():
print(f"Logging text changes for {user_lang}...")
mod_pack = (
mod_dir / util.get_content_path() / "Pack" / f"Bootup_{mod_lang}.pack"
)
if not user_lang == mod_lang:
mod_pack = swap_region(mod_pack, user_lang)
ref_pack = util.get_game_file(f"Pack/Bootup_{user_lang}.pack")
language_diffs[user_lang] = rsext.mergers.texts.diff_language(
str(mod_pack), str(ref_pack), user_lang[2:4] != mod_lang[2:4]
)
if not user_lang == mod_lang:
mod_pack.unlink()
return language_diffs
def get_user_languages() -> set:
langs = set()
for file in (util.get_update_dir() / "Pack").glob("Bootup_????.pack"):
langs.add(util.get_file_language(file.name))
return langs