108 lines
3.1 KiB
Python
108 lines
3.1 KiB
Python
import sys
|
|
import os
|
|
from args import args
|
|
from assetcache import assetCache, assetGetCache
|
|
from assethelpers import getAssetRelativePath
|
|
import polib
|
|
import re
|
|
|
|
LANGUAGE_DATA = {}
|
|
LANGUAGE_KEYS = []
|
|
|
|
def processLanguageList():
|
|
# Language keys header data
|
|
headerKeys = "// Auto-generated language keys header file.\n"
|
|
headerKeys += "#pragma once\n"
|
|
headerKeys += "#include \"dusk.h\"\n\n"
|
|
|
|
# This is the desired chunk groups list.. if a language key STARTS with any
|
|
# of the keys in this list we would "like to" put it in that chunk group.
|
|
# If there is no match, or the list is full then we will add it to the next
|
|
# available chunk group (that isn't a 'desired' one).
|
|
desiredChunkGroups = {
|
|
'ui': 0
|
|
}
|
|
|
|
# Now, for each language key, create the header reference and index.
|
|
keyIndex = 0
|
|
languageKeyIndexes = {}
|
|
languageKeyChunk = {}
|
|
for key in LANGUAGE_KEYS:
|
|
headerKeys += f"#define {getLanguageVariableName(key)} {keyIndex}\n"
|
|
languageKeyIndexes[key] = keyIndex
|
|
keyIndex += 1
|
|
|
|
# Find desired chunk group
|
|
assignedChunk = None
|
|
for desiredKey in desiredChunkGroups:
|
|
if key.lower().startswith(desiredKey):
|
|
assignedChunk = desiredChunkGroups[desiredKey]
|
|
break
|
|
# If no desired chunk group matched, assign to -1
|
|
if assignedChunk is None:
|
|
assignedChunk = -1
|
|
languageKeyChunk[key] = assignedChunk
|
|
|
|
# Setup header.
|
|
for lang in LANGUAGE_DATA:
|
|
if key not in LANGUAGE_DATA[lang]:
|
|
print(f"Warning: Missing translation for key '{key}' in language '{lang}'")
|
|
sys.exit(1)
|
|
|
|
# Seal the header.
|
|
headerKeys += f"\n#define LANG_KEY_COUNT {len(LANGUAGE_KEYS)}\n"
|
|
|
|
# Now we can generate the language string chunks.
|
|
for lang in LANGUAGE_DATA:
|
|
langData = LANGUAGE_DATA[lang]
|
|
|
|
# Key = chunkIndex, value = chunkInfo
|
|
languageChunks = {}
|
|
for key in LANGUAGE_KEYS:
|
|
keyIndex = languageKeyIndexes[key]
|
|
chunkIndex = languageKeyChunk[key]
|
|
|
|
# Write out the language keys header file
|
|
outputFile = os.path.join(args.headers_dir, "locale", "language", "keys.h")
|
|
os.makedirs(os.path.dirname(outputFile), exist_ok=True)
|
|
with open(outputFile, "w") as f:
|
|
f.write(headerKeys)
|
|
|
|
def getLanguageVariableName(languageKey):
|
|
# Take the language key, prepend LANG_, uppercase, replace any non symbols
|
|
# with _
|
|
key = languageKey.strip().upper()
|
|
key = re.sub(r'[^A-Z0-9]', '_', key)
|
|
return f"LANG_{key}"
|
|
|
|
def processLanguage(asset):
|
|
cache = assetGetCache(asset['path'])
|
|
if cache is not None:
|
|
return cache
|
|
|
|
# Load PO File
|
|
po = polib.pofile(asset['path'])
|
|
|
|
langName = po.metadata.get('Language')
|
|
if langName not in LANGUAGE_DATA:
|
|
LANGUAGE_DATA[langName] = {}
|
|
|
|
for entry in po:
|
|
key = entry.msgid
|
|
val = entry.msgstr
|
|
|
|
if key not in LANGUAGE_KEYS:
|
|
LANGUAGE_KEYS.append(key)
|
|
|
|
if key not in LANGUAGE_DATA[langName]:
|
|
LANGUAGE_DATA[langName][key] = val
|
|
else:
|
|
print(f"Error: Duplicate translation key '{key}' in language '{langName}'")
|
|
sys.exit(1)
|
|
|
|
outLanguageData = {
|
|
'data': po,
|
|
'path': asset['path'],
|
|
'files': []
|
|
}
|
|
return assetCache(asset['path'], outLanguageData) |