99 lines
2.7 KiB
Python
99 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: UTF-8 -*-
|
|
import os
|
|
import json
|
|
import re
|
|
import sys
|
|
import copy
|
|
import shutil
|
|
from fluent.syntax import parse, serialize
|
|
from fluent.syntax.ast import Message, TextElement, Identifier, Pattern, Junk
|
|
|
|
# clone an existing ftl string as a new key, optionally modifying it
|
|
# eg:
|
|
# $ python duplicate-string.py /path/to/templates/media-check.ftl window-title check-media-action
|
|
|
|
ftl_filename, old_key, new_key = sys.argv[1:]
|
|
|
|
# # split up replacements
|
|
# replacements = []
|
|
# for repl in repls.split(","):
|
|
# if not repl:
|
|
# continue
|
|
# replacements.append(repl.split("="))
|
|
|
|
# add file as prefix to key
|
|
prefix = os.path.splitext(os.path.basename(ftl_filename))[0]
|
|
old_key = f"{prefix}-{old_key}"
|
|
new_key = f"{prefix}-{new_key}"
|
|
|
|
# def transform_string(msg):
|
|
# for (old, new) in replacements:
|
|
# msg = msg.replace(old, f"{new}")
|
|
# # strip leading/trailing whitespace
|
|
# return msg.strip()
|
|
|
|
|
|
def dupe_key(fname, old, new):
|
|
if not os.path.exists(fname):
|
|
return
|
|
|
|
with open(fname) as file:
|
|
orig = file.read()
|
|
obj = parse(orig)
|
|
for ent in obj.body:
|
|
if isinstance(ent, Junk):
|
|
raise Exception(f"file had junk! {fname} {ent}")
|
|
|
|
# locate the existing key
|
|
for item in obj.body:
|
|
if isinstance(item, Message):
|
|
if item.id.name == old:
|
|
item2 = copy.deepcopy(item)
|
|
item2.id.name = new
|
|
obj.body.append(item2)
|
|
break
|
|
# print(item.id.name)
|
|
# print(item.value.elements)
|
|
# for e in item.value.elements:
|
|
# print(e.value)
|
|
|
|
modified = serialize(obj, with_junk=True)
|
|
# escape leading dots
|
|
modified = re.sub(r"(?ms)^( +)\.", '\\1{"."}', modified)
|
|
|
|
# ensure the resulting serialized file is valid by parsing again
|
|
obj = parse(modified)
|
|
for ent in obj.body:
|
|
if isinstance(ent, Junk):
|
|
raise Exception(f"introduced junk! {fname} {ent}")
|
|
|
|
# it's ok, write it out
|
|
with open(fname, "w") as file:
|
|
file.write(modified)
|
|
|
|
|
|
i18ndir = os.path.join(os.path.dirname(ftl_filename), "..")
|
|
langs = os.listdir(i18ndir)
|
|
|
|
for lang in langs:
|
|
if lang == "template":
|
|
# template
|
|
ftl_path = ftl_filename
|
|
else:
|
|
# translation
|
|
ftl_path = ftl_filename.replace("templates", lang)
|
|
ftl_dir = os.path.dirname(ftl_path)
|
|
|
|
if not os.path.exists(ftl_dir):
|
|
continue
|
|
|
|
dupe_key(ftl_path, old_key, new_key)
|
|
|
|
# copy file from repo into src
|
|
srcdir = os.path.join(i18ndir, "..", "..")
|
|
src_filename = os.path.join(srcdir, os.path.basename(ftl_filename))
|
|
shutil.copy(ftl_filename, src_filename)
|
|
|
|
print("done")
|