import re

import glob

import requests

import os

import lxml.etree as ET # builtin library doesn't preserve comments

SETTINGS_PATH = "app/src/main/java/com/lagradost/cloudstream3/ui/settings/SettingsGeneral.kt"

START_MARKER = "/* begin language list */"

END_MARKER = "/* end language list */"

XML_NAME = "app/src/main/res/values-"

ISO_MAP_URL = "https://raw.githubusercontent.com/haliaeetus/iso-639/master/data/iso_639-1.min.json"

INDENT = " "*4

iso_map = requests.get(ISO_MAP_URL, timeout=300).json()

# Load settings file

src = open(SETTINGS_PATH, "r", encoding='utf-8').read()

before_src, rest = src.split(START_MARKER)

rest, after_src = rest.split(END_MARKER)

# Load already added langs

languages = {}

for lang in re.finditer(r'Triple\("(.*)", "(.*)", "(.*)"\)', rest):

flag, name, iso = lang.groups()

languages[iso] = (flag, name)

# Add not yet added langs

for folder in glob.glob(f"{XML_NAME}*"):

iso = folder[len(XML_NAME):]

if iso not in languages.keys():

entry = iso_map.get(iso.lower(),{'nativeName':iso})

languages[iso] = ("", entry['nativeName'].split(',')[0])

# Create triples

triples = []

for iso in sorted(languages.keys()):

flag, name = languages[iso]

triples.append(f'{INDENT}Triple("{flag}", "{name}", "{iso}"),')

# Update settings file

open(SETTINGS_PATH, "w+",encoding='utf-8').write(

before_src +

START_MARKER +

"\n" +

"\n".join(triples) +

"\n" +

END_MARKER +

after_src

)

# Go through each values.xml file and fix escaped \@string

for file in glob.glob(f"{XML_NAME}*/strings.xml"):

try:

tree = ET.parse(file)

for child in tree.getroot():

if not child.text:

continue

if child.text.startswith("\\@string/"):

print(f"[{file}] fixing {child.attrib['name']}")

child.text = child.text.replace("\\@string/", "@string/")

with open(file, 'wb') as fp:

fp.write(b'<?xml version="1.0" encoding="utf-8"?>\n')

tree.write(fp, encoding="utf-8", method="xml", pretty_print=True, xml_declaration=False)

# Remove trailing new line to be consistent with weblate

fp.seek(-1, os.SEEK_END)

fp.truncate()

except ET.ParseError as ex:

print(f"[{file}] {ex}")