"""Poe as mensagens dos .translation importados do Godot 3 em ordem alfabetica.

O Translation guarda as mensagens num Map<StringName, StringName>, cuja ordem depende do endereco de memoria de cada
StringName, entao a mesma importacao gera arquivos com as mensagens em ordens diferentes (mesmo conteudo) e o .pck muda
a cada build. Aqui o PoolStringArray "messages" (pares chave/valor) e reordenado pela chave, o que nao muda o tamanho do
arquivo nem nenhum deslocamento.

  python3 -I sort_translations.py <pasta>   (todos os *.translation dela, recursivamente)
"""
import os, struct, sys


def u32(d, p):
    return struct.unpack('<I', d[p:p + 4])[0]


def fix(path):
    d = bytearray(open(path, 'rb').read())
    assert d[:4] == b'RSRC' and u32(d, 8) == 0 and u32(d, 12) == 3 and u32(d, 20) == 3, path  # little endian, Godot 3, formato 3
    p = 24
    p += 4 + u32(d, p)          # nome do tipo
    p += 8 + 14 * 4             # offset de metadados + reservado
    n = u32(d, p); p += 4
    names = []
    for _ in range(n):
        l = u32(d, p); p += 4
        names.append(bytes(d[p:p + l - 1]).decode()); p += l
    assert u32(d, p) == 0; p += 4   # sem recursos externos
    ni = u32(d, p); p += 4
    for _ in range(ni):
        p += 4 + u32(d, p) + 8
    p += 4 + u32(d, p)          # tipo do recurso principal
    pc = u32(d, p); p += 4
    for _ in range(pc):
        name = names[u32(d, p)]; p += 4
        vt = u32(d, p); p += 4
        if name == 'messages':
            assert vt == 34, vt
            cnt = u32(d, p); p += 4
            assert cnt % 2 == 0
            start = p
            items = []
            for _ in range(cnt):
                l = u32(d, p); p += 4
                s = bytes(d[p:p + l]); p += l
                items.append(s)
            pairs = sorted(zip(items[0::2], items[1::2]), key=lambda kv: kv[0])
            out = b''
            for ks, vs in pairs:
                out += struct.pack('<I', len(ks)) + ks + struct.pack('<I', len(vs)) + vs
            assert len(out) == p - start
            d[start:p] = out
            open(path, 'wb').write(d)
            return
        raise SystemExit('propriedade inesperada antes de messages: ' + name)
    raise SystemExit('sem messages em ' + path)


n = 0
for dp, _, fs in os.walk(sys.argv[1]):
    for f in sorted(fs):
        if f.endswith('.translation'):
            fix(os.path.join(dp, f)); n += 1
assert n, 'nenhum .translation'
print(f'{n} traducoes ordenadas')