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

# --------------------------------------------------------------------------- #
#                                                                             #
#    Plugin 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/>.    #
#                                                                             #
# --------------------------------------------------------------------------- #

phone_short_format = '%s%s%s-%s%s-%s%s'
phone_ru_format = '+%s (%s%s%s) ' + phone_short_format
phone_ua_format = '+%s%s (%s%s%s) ' + phone_short_format

def phonecode(type, jid, nick, text):
    if len(text):
        def_info,country = L('Not found!','%s/%s'%(jid,nick)),L('Unknown','%s/%s'%(jid,nick))
        if text[0] == '+': text = text[1:]
        if text.isdigit():
            if text[0] == '7':
                if len(text) > 1 and text[1] == '9':
                    country = L('Russia, Mobile','%s/%s'%(jid,nick))
                    if len(text) >= 5:
                        dc, pn = text[1:4], text[4:]
                        if len(pn) < 7: pn = pn.ljust(7, '0')
                        elif len(pn) > 7: pn = pn[:7]
                        result = cur_execute_fetchall('select * from def_ru_mobile where def=%s and defbegin<=%s and defend>=%s', (dc,pn,pn))
                        if result:
                            def pfs(t): return phone_short_format % tuple('%07d' % t)
                            def_info = '\n'.join(['%s, %s (%s) %s..%s, %s ' % (res[0],res[1],res[2],pfs(res[3]),pfs(res[4]),res[5]) for res in result])
                    else: def_info = L('Too many results!','%s/%s'%(jid,nick))
                else:
                    country = L('Russia','%s/%s'%(jid,nick))
                    if len(text) < 11: text = text.ljust(12, '2')
                    elif len(text) > 11: text = text[:11]
                    text = text[:7] + 'x'*4
                    pos = 7
                    while pos:
                        result = cur_execute_fetchall('select * from def_ru_stat where phone ilike %s', ('%s%%'%text,))
                        if result:
                            def_info = []
                            for res in result:
                                res = list(res)
                                while '' in res: res.remove('')
                                res[0] = phone_ru_format % tuple(res[0])
                                def_info += [', '.join(res)]
                            def_info = '\n'.join(def_info)
                            break
                        else:
                            text = list(text)
                            text[pos] = 'x'
                            text = ''.join(text)
                            pos -= 1
            elif text[:3] == '380':
                country = L('Ukraine','%s/%s'%(jid,nick))
                if len(text) < 12: text = text.ljust(12, '2')
                elif len(text) > 12: text = text[:12]
                text = text[:8]+'x'*4
                pos = 8
                while pos > 2:
                    result = cur_execute_fetchall('select * from def_ua_stat where phone ilike %s', ('%s%%'%text,))
                    if result:
                        def_info = []
                        for res in result:
                            res = list(res)
                            while '' in res: res.remove('')
                            res[0] = phone_ua_format % tuple(res[0])
                            def_info += [', '.join(res)]
                        def_info = '\n'.join(def_info)
                        break
                    else:
                        text = list(text)
                        text[pos] = 'x'
                        text = ''.join(text)
                        pos -= 1
            if country == L('Unknown','%s/%s'%(jid,nick)): result = def_info
            elif def_info == L('Not found!','%s/%s'%(jid,nick)): result = L('Country: %s, Phone number not found','%s/%s'%(jid,nick)) % (country,)
            else: result = L('Country: %s\n%s','%s/%s'%(jid,nick)) % (country,def_info)
        else:
            text = '%s%%'%text.lower().capitalize()
            def_info = []
            result = cur_execute_fetchall('select * from def_ru_stat where city ilike %s', (text,))
            if result:
                di = []
                for res in result:
                    res = list(res)
                    while '' in res: res.remove('')
                    res[0] = phone_ru_format % tuple(res[0])
                    di += [', '.join(res)]
                di = '\n'.join(di)
                def_info += [L('Country: %s\n%s','%s/%s'%(jid,nick)) % (L('Russia','%s/%s'%(jid,nick)),di)]
            result = cur_execute_fetchall('select * from def_ua_stat where city ilike %s', (text,))
            if result:
                di = []
                for res in result:
                    res = list(res)
                    while '' in res: res.remove('')
                    res[0] = phone_ua_format % tuple(res[0])
                    di += [', '.join(res)]
            def_info += [L('Country: %s\n%s','%s/%s'%(jid,nick)) % (L('Ukraine','%s/%s'%(jid,nick)),di)]
            if def_info: result = '\n\n'.join(def_info)
            else: result = L('Not found!','%s/%s'%(jid,nick))
        msg = result
    else: msg = L('What?','%s/%s'%(jid,nick))
    send_msg(type, jid, nick, msg)

global execute

execute = [(3, 'phone', phonecode, 2, 'Information about telephone city code, DEF code or search code for city.')]