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

# --------------------------------------------------------------------------- #
#                                                                             #
#    Plugin for iSida Jabber Bot                                              #
#    Copyright (C) Vit@liy <vitaliy@root.ua>                                  #
#                                                                             #
#    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/>.    #
#                                                                             #
# --------------------------------------------------------------------------- #

def sokr(type, jid, nick, text):
    target = ''
    text = text.strip()
    if not text: msg = L('What?','%s/%s'%(jid,nick))
    else:
        tmp_target = re.search('(\A| )\d+?(-\d+?)?(\Z| )', text)
        if tmp_target:
            target = tmp_target.group().strip()
            text = text.replace(target, ' ')
        text = reduce_spaces_all(text)
        if re.match('[a-zA-Z]+\Z', text):
            data = load_page('http://www.abbreviations.com/%s' % text)
            results = re.findall('<p class="desc">(.+?)</p>', data)
            results = [i for k,i in enumerate(results) if results.index(i) == k]
        else:
            data = load_page("http://www.sokr.ru/search/?", {'abbr': text.encode('utf-8'), 'abbr_exact': '1'})
            results = re.findall('em class="got_clear">.+?</em></a>.+?<p class="value">(.+?)</p>' , data, re.S)
        cr = len(results)
        if not results: msg = L('I don\'t know!','%s/%s'%(jid,nick))
        else:
            if not target:
                if cr == 1: target = '1'
                elif cr < 6:  target = '1-%s' % cr
                else: target = '1-5'
            try: n1 = n2 = int(target)
            except: n1, n2 = map(int, target.split('-'))
            if 0 < n1 <= n2 <= cr:
                msg = L('Total found %s matches. Result(s) %s:\n','%s/%s'%(jid,nick)) % (cr, target)
                msg += '\n'.join(['%s. %s' % (i[0]+n1, i[1].replace('\n',' ')) for i in enumerate(results[n1-1: n2])])
                msg = msg.replace('<br>', '').replace('\r','')
            else: msg = L('I don\'t know!','%s/%s'%(jid,nick))
    send_msg(type, jid, nick, msg)

global execute

execute = [(3, 'sokr', sokr, 2, 'Abbreviations.\nExamples: sokr abbr, sokr 6 abbr, sokr 3-7 abbr')]