#!/usr/bin/python

import re, sys
from lxml import etree
import mygit, xwconfig


# Take an English strings.xml file and another, "join" them on the
# name of each string, and then produce an array that's a mapping of
# English to the other. Get ride of extra whitespace etc in the
# English strings so they're identical to how an Android app displays
# them.

english = 'res/values/strings.xml'
other_f = 'res_src/values-%s/strings.xml'

def readIDs(rDotJava):
    ids = {}
    start = re.compile('\s*public static final class string {\s*')
    end = re.compile('\s*}\s*')
    entry = re.compile('\s*public static final int (\S+)=(0x.*);\s*')
    inLine = False
    for line in rDotJava.splitlines():
        if inLine:
            if end.match(line):
                break
            else:
                match = entry.match(line)
                if match:
                    name = match.group(1)
                    value = int(match.group(2), 16)
                    ids[name] = value
        elif start.match(line):
            inLine = True
    return ids

def asMap( repo, rev, path, ids ):
    map = {}
    data = repo.cat( path, rev )
    doc = etree.fromstring( data )
    for elem in doc.iter():
        if 'string' == elem.tag:
            text = elem.text
            if text:
                # print 'text before:', text
                text = " ".join(re.split('\s+', text)) \
                          .replace("\\'", "'") \
                          .replace( '\\"', '"' )
                # print 'text after:', text
                name = elem.get('name')
                id = ids[name]
                map[id] = text
    return map

# Build from the most recent revisions of the english and locale
# strings.xml files that are compatible (haven't changed since)
# stringsHash on the R.java file.  For now, just get what matches,
# assuming that all are updated with the same commit -- which they
# aren't.
def getXlationFor( repo, rDotJava, locale, stringsHash ):
    ids = readIDs(rDotJava)
    eng = asMap( repo, stringsHash, english, ids )
    other = asMap( repo, stringsHash, other_f % (locale), ids )
    result = []
    for key in eng.keys():
        if key in other:
            result.append( { 'id' : key, 'loc' : other[key] } )
    return result, stringsHash

def main():
    repo = mygit.GitRepo( xwconfig.k_REPOPATH )
    hash = '33a83b0e2fcf062f4f640ccab0785b2d2b439542'
    rDotJava = repo.cat( 'R.java', hash )
    data = getXlationFor( repo, rDotJava, 'ca_PS', hash )
    print data
    data = getXlationFor( repo, rDotJava, 'ba_CK', hash )
    print data

##############################################################################
if __name__ == '__main__':
    main()