#!/usr/bin/python
# -*- coding: utf-8 -*-

# --------------------------------------------------------------------------- #
#                                                                             #
#    Locale generator for iSida Jabber Bot                                    #
#    Copyright (C) diSabler <dsy@dsy.name>                                    #
#                                                                             #
#    This program is free software: you can redistribute it and/or modify     #
#    it under the terms of the GNU General Public License as published by     #
#    the Free Software Foundation, either version 3 of the License, or        #
#    (at your option) any later version.                                      #
#                                                                             #
#    This program is distributed in the hope that it will be useful,          #
#    but WITHOUT ANY WARRANTY; without even the implied warranty of           #
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the            #
#    GNU General Public License for more details.                             #
#                                                                             #
#    You should have received a copy of the GNU General Public License        #
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.    #
#                                                                             #
# --------------------------------------------------------------------------- #

import os,re,time,sys

sys.path = ['../../lib'] + sys.path

def L(t): return t

def GT(t):
    if t in ['wot_appid','disco_exclude','lfm_api']: return ''
    elif t in ['html_logs_enable']: return 0
    elif t == 'wot_region': return 'RU'
    else: return None

def readfile(filename):
    fp = file(filename)
    data = fp.read()
    fp.close()
    return data

def writefile(filename, data):
    fp = file(filename, 'wb')
    fp.write(data)
    fp.close()

def regenerate(ptl,locale_name):
    global execute,exec_yes,comms
    def L(text):
        if not len(text): return text
        try: return locales[text]
        except: return text
    path_to_locale = '%s/%s' % (ptl,locale_name)
    root_files = os.listdir(path_to_source)
    plugins_files = os.listdir(path_to_plugins)
    plugins_files.sort()
    file_list,localez = [],[]
    regex = r'L\(u?\'(.*?)\'[,|\)]'

    for tmp in root_files:
        if tmp[-3:] == '.py' and tmp[0] not in ['.','_']: file_list.append('%s/%s' % (path_to_source,tmp))

    for tmp in plugins_files:
        if tmp[-3:] == '.py' and tmp[0] not in ['.','_']: file_list.append('%s/%s' % (path_to_plugins,tmp))

    locales = {}

    if os.path.isfile(path_to_locale):
        lf = readfile(path_to_locale).decode('UTF').replace('\r','').split('\n')
        for c in lf:
            if ('#' not in c[:3]) and len(c) and '\t' in c: locales[c.split('\t',1)[0]] = c.split('\t',1)[1]
    else: print 'Locale not found!'

    result = locale_header + '\n\n'

    error_count = 0

    for tmp in file_list:
        execute,exec_yes,comms = [],[],[]
        if 'plugin' in tmp: execfile(tmp,globals())
        body = str(readfile(tmp)).replace('\\\'','\'').replace('\\\"','\"').replace('\\\\','\\')
        tm = re.findall(regex,body,re.S+re.U)
        for t in execute+exec_yes+comms: tm.append(t[4].replace('\n','\\n'))
        tm2 = re.findall('# translate: (.*)',body)
        if tm2: tm2 = tm2[0].split(',')
        if tm or tm2:
            for tmp2 in tm:
                if tmp2 not in localez:
                    localez.append(tmp2)
                    tm2.append(tmp2)
            if tm2:
                result += '%s%s\n' % (file_mark,'/'.join(tmp.split('/')[2:]))
                for tmp2 in tm2:
                    tr = L(tmp2)
                    if tmp2 == tr:
                        error_count += 1
                        loc = locale_mark
                    else: loc = tr
                    result += '%s\t%s\n' % (tmp2,loc)
                result += '\n'

    execfile(path_to_mainpy,globals())
    tm2 = []
    for t in config_prefs.keys():
        if config_prefs[t][0] not in localez:
            localez.append(config_prefs[t][0])
            tm2.append(config_prefs[t][0])
        if config_prefs[t][1] not in localez:
            localez.append(config_prefs[t][1])
            tm2.append(config_prefs[t][1])
        
    for t in config_groups:
        if t[0] not in localez: 
            localez.append(t[0])
            tm2.append(t[0])
        
    for t in owner_prefs.keys():
        if owner_prefs[t][0] not in localez: 
            localez.append(owner_prefs[t][0])
            tm2.append(owner_prefs[t][0])
        
    for t in owner_groups:
        if t[0] not in localez:
            localez.append(t[0])
            tm2.append(t[0])

    if tm2:
        result += '%s%s\n' % (file_mark,'Generic')
        for tmp2 in tm2:
            tr = L(tmp2)
            if tmp2 == tr:
                error_count += 1
                loc = locale_mark
            else: loc = tr
            result += '%s\t%s\n' % (tmp2,loc)
        result += '\n'
            
    result += '# write locale file: %s' % locale_name
    result += '\n# total: %s' % len(localez)

    msg = 'write locale file: %s' % locale_name
    if error_count:
        msg += '\nmissed translations: %s\n' % error_count
        result += '\n# missed translations: %s\n' % error_count
    else:
        msg += '\nregenarated without mistakes!\n'
        result += '\n# regenarated without mistakes!\n'
        
    return msg, result, len(localez)

path_to_source = '../..'
path_to_locale = '%s/data/locales/' % path_to_source
path_to_plugins = '%s/plugins' % path_to_source
path_to_mainpy = '%s/main.py' % path_to_plugins

smiles_dirs = []
data_folder = path_to_source + '/data/%s'
loc_folder     = data_folder % 'locales/%s.txt'
log_folder     = data_folder % 'conflogs/%s'
public_log = log_folder % 'chatlogs/%s'
system_log = log_folder % 'syslogs/%s'
gactmessage = []
base_type = 'pgsql'
    
locale_mark = 'NO_TRANSLATE'
file_mark = '# file: '
locale_header = '''#---------------------------------------------
#      Autogenerate for locales
#    (c) Disabler Production Lab.
#---------------------------------------------'''

smiles_dirs_case = logs_css_path = ''
execute,exec_yes,comms = [],[],[]

if __name__ == "__main__" :
    print '\n',locale_header.replace('#',''),'\n'
    lf = os.listdir(path_to_locale)
    for tmp in lf:
        if tmp[-4:] == '.txt':
            msg, result, lz = regenerate(path_to_locale,tmp)
            writefile(tmp,'%s\n' % result.encode('utf-8'))
            print msg
    print 'Total found definitions: %s\n' % lz
    print 'Done!'