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

watch_time = time.time()
watch_count = 0
watch_reset = True
watch_last_activity = {}
watch_activity = {}

def connect_watch():
    global iq_request, watch_time, game_over, watch_count, bot_exit_type, watch_reset, plugins_reload
    if GT('watcher_self_ping') and (time.time() - watch_time) > GT('watch_size'):
        watch_time = time.time()
        watch_count += 1
        watch_reset = True
        iqid = get_id()
        i = xmpp.Node('iq', {'id': iqid, 'type': 'get', 'to':selfjid}, payload = [xmpp.Node('ping', {'xmlns': xmpp.NS_URN_PING},[])])
        iq_request[iqid]=(time.time(),watcher_reset,['chat',god,'',''],xmpp.NS_URN_PING)
        sender(i)
        to = GT('timeout') - 10
        if to <= 10: to = 600
        while to > 0 and not game_over and not plugins_reload:
            to -= 1
            time.sleep(1)
        if game_over or plugins_reload: return
        if watch_reset:
            pprint('Restart by watcher\'s timeout!','red')
            bot_exit_type, game_over = 'restart', True
            time.sleep(2)

def watch_room_activity():
    if not GT('watcher_room_activity'): return
    global watch_last_activity
    to = int(time.time())-GT('watch_activity_timeout')
    cnf = cur_execute_fetchall('select room from conference;')
    if cnf:
        for t in cnf:
            tmp = t[0]
            try: cw = watch_last_activity[getRoom(tmp)]
            except: cw = to
            if cw < to:
                watch_last_activity[getRoom(tmp)] = int(time.time())
                text = tmp
                if '\n' in text: text,passwd = text.split('\n',2)
                else: passwd = ''
                zz = join(text, passwd)
                while str(zz)[:3] == '409':
                    time.sleep(1)
                    text += '_'
                    zz = join(text, passwd)
                time.sleep(1)
                pprint('Low activity! Try rejoin into %s' % text,'white')

def watcher_reset(a,b,c,d,e):
    global watch_reset
    watch_reset = None

def c_watcher(type, jid, nick): send_msg(type, jid, nick, L('Timeout for ask: %s | Timeout for answer: %s | Last ask: %s | Total checks: %s','%s/%s'%(jid,nick)) % (GT('watch_size'),GT('timeout'),un_unix(int(time.time() - watch_time),'%s/%s'%(jid,nick)),watch_count))

def activity_watch(type, jid, nick):
    rooms = cur_execute_fetchall("select split_part(room,'/',1) from conference;")
    cnf = [[watch_activity[t],t] for t in watch_activity.keys()]
    cnf.sort(reverse=1)
    tt = int(time.time())
    msg = '\n'.join(['%s - %s' % (t[1],un_unix(tt-t[0],'%s/%s'%(jid,nick))) for t in cnf])
    rooms = ['%s - %s' % (t[0],L('Unknown','%s/%s'%(jid,nick))) for t in rooms if t[0] not in watch_activity.keys()]
    if rooms: msg = '%s\n%s' % (msg,'\n'.join(rooms))
    msg = L('Last activity in room(s):\n%s','%s/%s'%(jid,nick)) % msg
    if type == 'groupchat':
        send_msg('chat', jid, nick, msg)
        msg = L('Send for you in private','%s/%s'%(jid,nick))
    send_msg(type, jid, nick, msg)

def connect_watch_uni(room,jid,nick,type,mass):
    global watch_last_activity
    if jid != 'None':
        gr = getRoom(room)
        watch_last_activity[gr] = int(time.time())
        if jid != Settings['jid']: watch_activity[gr] = int(time.time())

global execute, timer

if GT('iq_version_enable'): timer = [connect_watch,watch_room_activity]
presence_control = [connect_watch_uni]
message_control = [connect_watch_uni]

execute = [(6,'watcher',c_watcher,1,'Connection activity control.'),
           (9,'activity',activity_watch,1,'Show activity of conferences.')]