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

# cron * * * * * \n command

def time_cron(type, jid, nick, text):
	ar = text.split(' ',1)[0].lower()
	try: par = text.split(' ',1)[1]
	except: par = ''
	if ar == 'show': msg = time_cron_show(jid,nick,par)
	elif ar == 'del': msg = time_cron_del(jid,nick,par)
	else: msg = time_cron_add(text,jid,nick)
	send_msg(type, jid, nick, msg)

def time_cron_show(jid,nick,ar):
	al = get_level(jid,nick)[0]
	if ar:
		if al == 9 and ar.lower().split()[0] in ['all','global','total']: room = '%'
		else: room = jid
	else: room = jid
	c = cur_execute_fetchall('select * from cron where room ilike %s order by room, time',(room,))
	if c:
		tmp, idx = [], 1
		for t in c:
			if t[2] == '\n': mode = '[silent]'
			elif not t[2]: mode = '[anonim]'
			else: mode = ''
			if room=='%': msg = '%s|%s' % (t[0],disp_time(t[3],'%s/%s'%(jid,nick)))
			else: msg = '%s. %s' % (idx,disp_time(t[3],'%s/%s'%(jid,nick)))
			if mode: msg = '%s %s' % (msg,mode)
			if t[4]: msg += ' [%s]' % t[4]
			msg += ' -> %s' % t[5]
			tmp.append(msg)
			idx += 1
		return L('Cron rules:\n%s','%s/%s'%(jid,nick)) % '\n'.join(tmp)
	else: return L('There is no cron rules.','%s/%s'%(jid,nick))

def time_cron_add(ar,jid,nick):
	try: cron_time, cron_cmd = ar.split('\n',1)
	except: return L('Not enough parameters!','%s/%s'%(jid,nick))
	try:
		SM,RM,NM = 'silent','once','anonim'
		CMD = [SM,RM,NM]
		ct = cron_time.lower().split()
		rm,sm,nm = RM in ct,SM in ct,NM in ct
		if rm or sm or nm:
			for t in CMD:
				try: ct.remove(t)
				except: pass
		ct = ' '.join(ct)
		if rm: cron_time,repeat_time = ct,''
		else: cron_time,repeat_time = ct,ct
		next_time = crontab.CronTab(cron_time).next() + time.time()
	except: return L('Error in time format!','%s/%s'%(jid,nick))
	lvl,rj = get_level(jid,nick)
	amm,tcmd = -1,cron_cmd.split(' ')[0]
	for tmp in comms:
		if tmp[1] == tcmd:
			amm = tmp[0]
			break
	if amm < 0: return L('Command not found: %s','%s/%s'%(jid,nick)) % tcmd
	elif amm > lvl: return L('Not allowed launch: %s','%s/%s'%(jid,nick)) % tcmd
	else:
		if sm: nick = '\n'
		elif nm: nick = ''
		cur_execute('insert into cron values (%s,%s,%s,%s,%s,%s,%s)', (jid,getRoom(rj),nick,next_time,repeat_time,cron_cmd,lvl))
		return '%s -> %s' % (disp_time(next_time,'%s/%s'%(jid,nick)),cron_cmd)

def time_cron_del(jid,nick,ar):
	al = get_level(jid,nick)[0]
	if al == 9 and ar.lower() == 'all':
		cur_execute('delete from cron where room=%s',(jid,))
		return L('All cron rules removed!','%s/%s'%(jid,nick))
	elif not ar: return L('Need choise record number.','%s/%s'%(jid,nick))
	elif not ar.isdigit(): return L('Record ID is numeric.','%s/%s'%(jid,nick))
	else:
		c = cur_execute_fetchall('select * from cron where room ilike %s order by room, time',(jid,))
		try: rec = c[int(ar)-1]
		except: return L('Record #%s not found!','%s/%s'%(jid,nick)) % ar
		cur_execute('delete from cron where room=%s and jid=%s and nick=%s and time=%s and repeat=%s and command=%s and level=%s',rec)
		msg = disp_time(rec[3],'%s/%s'%(jid,nick))
		if rec[2] == '\n': mode = '[silent]'
		elif not rec[2]: mode = '[anonim]'
		else: mode = ''
		if mode: msg = '%s %s' % (msg,mode)
		if rec[4]: msg += ' [%s]' % rec[4]
		msg += ' -> %s' % rec[5]
		return L('Removed: %s','%s/%s'%(jid,nick)) % msg

def cron_action():
	itt = int(time.time())
	c = cur_execute_fetchall('select distinct * from cron where %s >= time',(itt,))
	if c:
		cur_execute('delete from cron where %s >= time',(itt,))
		for t in c:
			if t[4]:
				tm = crontab.CronTab(t[4]).next() + time.time()
				m = list(t[:3]) + [tm] + list(t[4:7])
				cur_execute('insert into cron values (%s,%s,%s,%s,%s,%s,%s)', m)
			tmp = cur_execute_fetchone('select room from conference where room ilike %s',('%s/%%'%t[0],))
			if not tmp:
				pprint('Can\'t execute by cron: %s in %s' % (t[5].split()[0],t[0]),'red')
				return
			else: nowname = getResourse(tmp[0])
			if t[2] == '\n': tmp_nick,tmp_type = '%s_cron_%d' % (nowname,time.time()),'chat'
			else: tmp_nick,tmp_type = t[2],'groupchat'
			com_parser(t[6], nowname, tmp_type, t[0], tmp_nick, t[5], Settings['jid'])

global execute, timer

timer = [cron_action]

execute = [(7, 'cron', time_cron, 2, 'Execute command by cron. Used unix-type time format.\ncron [once] [anonim|silent] * * * * *\ncommand')]