eleksir/isida-4
1
Fork 0
Code Activity
Files
4643f7ea21487f2dbbfc5653797685eaf44b8160
isida-4/plugins/muc.py
diSabler e5295eb994 add: initial AI support
2014-05-01 14:07:44 +04:00

252 lines
14 KiB
Python
Raw Blame History

#!/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/>. #
# #
# --------------------------------------------------------------------------- #
visitors_list = {}
visitors_list_lock = False
# -------------- affiliation -----------------
def global_ban(type, jid, nick, text):
text = text.lower()
hroom = getRoom(jid)
al = get_level(jid,nick)[0]
if al == 9: af = 'owner'
else: af = get_affiliation(jid,nick)
if af != 'owner': msg = L('This command available only for conference owner!','%s/%s'%(jid,nick))
elif text == 'show' and al == 9:
ir = cur_execute_fetchone('select * from ignore_ban;')
if len(ir): msg = '%s\n%s' % (L('Global ban is off in:','%s/%s'%(jid,nick)),'\n'.join([t[0] for t in ir]))
else: msg = L('Global ban enable without limits!','%s/%s'%(jid,nick))
elif text == 'del' and af == 'owner':
if cur_execute_fetchone('select * from ignore_ban where room=%s' % (hroom,)): msg = L('Conference %s already deleted from global ban list!','%s/%s'%(jid,nick)) % hroom
else:
cur_execute('insert inro ignore_ban values (%s)' % (hroom,))
msg = L('Conference %s has been deleted from global ban list!','%s/%s'%(jid,nick)) % hroom
elif text == 'add' and af == 'owner':
if cur_execute_fetchone('select * from ignore_ban where room=%s' % (hroom,)):
cur_execute('delete from ignore_ban where room=%s' % (hroom,))
msg = L('Conference %s has been added from global ban list!','%s/%s'%(jid,nick)) % hroom
else: msg = L('Conference %s already exist in global ban list!','%s/%s'%(jid,nick)) % hroom
else:
if al == 9:
if cur_execute_fetchone('select * from ignore_ban where room=%s' % (hroom,)): msg = L('Your conference will be ignored for global ban!','%s/%s'%(jid,nick))
elif '@' not in text or '.' not in text: msg = L('I need jid!','%s/%s'%(jid,nick))
else:
reason = L('banned global by %s from %s','%s/%s'%(jid,nick)) % (nick, jid)
br = [t[0] for t in cur_execute_fetchall("select room from conference where split_part(room,'/',1) not in (select room from ignore_ban as igb);")]
for tmp in br:
i = xmpp.Node('iq', {'id': get_id(), 'type': 'set', 'to':getRoom(tmp)}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'affiliation':'outcast', 'jid':unicode(text)},[xmpp.Node('reason',{},reason)])])])
sender(i)
time.sleep(0.1)
msg = L('jid %s has been banned in %s conferences.','%s/%s'%(jid,nick)) % (text, len(br))
else: msg = L('Command temporary blocked!','%s/%s'%(jid,nick))
send_msg(type, jid, nick, msg)
def muc_tempo_ban(type, jid, nick, text):
text,mute = text.strip().split('\n'),False
if text:
cmd, par = text[0].lower().split(' ',1) if ' ' in text[0] else (text[0].lower().split(' ',1)[0],'')
if cmd == 'show':
if not par: par = '%'
tb = cur_execute_fetchall('select jid,time from tmp_ban where room=%s and jid ilike %s',(jid,par))
if tb: msg = L('Found: %s','%s/%s'%(jid,nick)) % '\n%s' % '\n'.join([['%s\t%s' % (ub[0],un_unix(ub[1]-int(time.time()),'%s/%s'%(jid,nick))),'%s\t< %s' % (ub[0],un_unix(GT('schedule_time'),'%s/%s'%(jid,nick)))][ub[1] < int(time.time())] for ub in tb])
else: msg = L('Not found.','%s/%s'%(jid,nick))
elif cmd == 'del':
if par:
tb = cur_execute_fetchall('select jid,time from tmp_ban where room=%s and jid ilike %s',(jid,par))
if tb:
msg = L('Removed: %s','%s/%s'%(jid,nick)) % '\n%s' % '\n'.join([['%s\t%s' % (ub[0],un_unix(ub[1]-int(time.time()),'%s/%s'%(jid,nick))),'%s\t< %s' % (ub[0],un_unix(GT('schedule_time'),'%s/%s'%(jid,nick)))][ub[1] < int(time.time())] for ub in tb])
for ub in tb: sender(xmpp.Node('iq', {'id': get_id(), 'type': 'set', 'to':jid}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'affiliation':'none', 'jid':getRoom(unicode(ub[0]))},[])])]))
cur_execute('delete from tmp_ban where room=%s and jid ilike %s',(jid,par))
else: msg = L('Not found.','%s/%s'%(jid,nick))
else: msg = L('What?','%s/%s'%(jid,nick))
else:
try: ban_time = int(text[1][:-1]) * {'s':1, 'm':60, 'h':3600, 'd':86400}[text[1][-1].lower()]
except: ban_time = None
if ban_time:
reason = ' '.join(text[2:])
if not reason: reason = L('No reason!','%s/%s'%(jid,nick))
reason = L('ban on %s since %s because: %s','%s/%s'%(jid,nick)) % (un_unix(ban_time,'%s/%s'%(jid,nick)), timeadd(tuple(time.localtime())), reason)
who,msg = text[0],L('done','%s/%s'%(jid,nick))
try: whojid = getRoom([t[4] for t in megabase if t[0]==jid and t[1]==who][0])
except: whojid = None
if not whojid:
fnd = cur_execute_fetchall('select jid from age where room=%s and (nick=%s or jid=%s) group by jid',(jid,who,who))
if len(fnd) == 1: whojid = getRoom(fnd[0][0])
elif len(fnd) > 1: msg = L('I seen different people with this nick!','%s/%s'%(jid,nick))
else: msg,whojid = L('I don\'n know %s, and use as is!','%s/%s'%(jid,nick)) % who,who
if whojid:
sender(xmpp.Node('iq', {'id': get_id(), 'type': 'set', 'to':jid}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'affiliation':'outcast', 'jid':unicode(whojid)},[xmpp.Node('reason',{},reason)])])]))
was_banned = cur_execute_fetchone('select time from tmp_ban where room=%s and jid=%s;',(jid,whojid))
if was_banned:
cur_execute('delete from tmp_ban where room=%s and jid=%s;',(jid,whojid))
ban_tm = was_banned[0]-int(time.time())
if ban_tm > 0: ban_nm = L('old ban time is %s','%s/%s'%(jid,nick)) % un_unix(ban_tm,'%s/%s'%(jid,nick))
else: ban_nm = L('just passed!','%s/%s'%(jid,nick))
msg = L('Updated: %s','%s/%s'%(jid,nick)) % ban_nm
cur_execute('insert into tmp_ban values (%s,%s,%s)',(jid,whojid,ban_time+int(time.time())))
if who == nick: mute = True
else: msg = L('Time format error!','%s/%s'%(jid,nick))
else: msg = L('What?','%s/%s'%(jid,nick))
if not mute: send_msg(type, jid, nick, msg)
def muc_ban(type, jid, nick,text): muc_affiliation(type, jid, nick, text, 'outcast',0)
def muc_banjid(type, jid, nick,text): muc_affiliation(type, jid, nick, text, 'outcast',1)
def muc_none(type, jid, nick,text): muc_affiliation(type, jid, nick, text, 'none',0)
def muc_nonejid(type, jid, nick,text): muc_affiliation(type, jid, nick, text, 'none',1)
def muc_member(type, jid, nick,text): muc_affiliation(type, jid, nick, text, 'member',0)
def muc_memberjid(type, jid, nick,text): muc_affiliation(type, jid, nick, text, 'member',1)
def muc_affiliation(type, jid, nick, text, aff, is_jid):
nowname = get_xnick(jid)
xtype = get_xtype(jid)
if xtype == 'owner':
if is_start: return
else: send_msg(type, jid, nick, L('Command is locked!','%s/%s'%(jid,nick)))
elif len(text):
skip = None
if '\n' in text: who, reason = text.split('\n',1)[0], text.split('\n',1)[1]
else: who, reason = text, []
if reason: reason = [xmpp.Node('reason',{},reason)]
whojid = [unicode(get_level(jid,who)[1]),who][is_jid]
if whojid != 'None': sender(xmpp.Node('iq', {'id': get_id(), 'type': 'set', 'to':jid}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'affiliation':aff, 'jid':whojid},reason)])]))
else: send_msg(type, jid, nick, L('I don\'t know %s','%s/%s'%(jid,nick)) % who)
else: send_msg(type, jid, nick, L('What?','%s/%s'%(jid,nick)))
def muc_ban_past(type, jid, nick,text): muc_affiliation_past(type, jid, nick, text, 'outcast')
def muc_none_past(type, jid, nick,text): muc_affiliation_past(type, jid, nick, text, 'none')
def muc_member_past(type, jid, nick,text): muc_affiliation_past(type, jid, nick, text, 'member')
def muc_affiliation_past(type, jid, nick, text, aff):
nowname = get_xnick(jid)
xtype = get_xtype(jid)
if xtype == 'owner':
if is_start: return
else: msg, text = L('Command is locked!','%s/%s'%(jid,nick)), ''
else: msg = L('What?','%s/%s'%(jid,nick))
if len(text):
skip = None
if '\n' in text: who, reason = text.split('\n',1)[0], text.split('\n',1)[1]
else: who, reason = text, []
if reason: reason = [xmpp.Node('reason',{},reason)]
fnd = cur_execute_fetchall('select jid from age where room=%s and (nick=%s or jid=%s) group by jid',(jid,who,who))
if len(fnd) == 1: msg, whojid = L('done','%s/%s'%(jid,nick)), getRoom(unicode(fnd[0][0]))
elif len(fnd) > 1:
whojid = getRoom(get_level(jid,who)[1])
if whojid != 'None': msg = L('done','%s/%s'%(jid,nick))
else: msg, skip = L('I seen some peoples with this nick. Get more info!','%s/%s'%(jid,nick)), True
else:
msg = L('I don\'n know %s, and use as is!','%s/%s'%(jid,nick)) % who
whojid = who
else: skip = True
if skip: send_msg(type, jid, nick, msg)
else:
i = xmpp.Node('iq', {'id': get_id(), 'type': 'set', 'to':jid}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'affiliation':aff, 'jid':unicode(whojid)},reason)])])
sender(i)
send_msg(type, jid, nick, msg)
def muc_kick(type, jid, nick, text): muc_role(type, jid, nick, text, 'none',1)
def muc_participant(type, jid, nick, text): muc_role(type, jid, nick, text, 'participant',1)
def muc_visitor(type, jid, nick, text): muc_role(type, jid, nick, text, 'visitor',1)
def muc_moderator(type, jid, nick, text): muc_role(type, jid, nick, text, 'moderator',1)
def muc_role(type, jid, nick, text, role, unused):
nowname = get_xnick(jid)
xtype = get_xtype(jid)
if xtype == 'owner':
if is_start: return
else: send_msg(type, jid, nick, L('Command is locked!','%s/%s'%(jid,nick)))
elif len(text):
skip = None
if '\n' in text: who, reason = text.split('\n',1)[0], text.split('\n',1)[1]
else: who, reason = text, []
if reason: reason = [xmpp.Node('reason',{},reason)]
whojid = unicode(get_level(jid,who)[1])
if whojid != 'None': sender(xmpp.Node('iq', {'id': get_id(), 'type': 'set', 'to':jid}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'role':role, 'nick':who},reason)])]))
else: send_msg(type, jid, nick, L('I don\'t know %s','%s/%s'%(jid,nick)) % who)
else: send_msg(type, jid, nick, L('What?','%s/%s'%(jid,nick)))
def check_unban():
tt = int(time.time())
ul = cur_execute_fetchall('select room,jid from tmp_ban where time<%s;',(tt,))
if ul:
for t in ul: sender(xmpp.Node('iq', {'id': get_id(), 'type': 'set', 'to':t[0]}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'affiliation':'none', 'jid':getRoom(unicode(t[1]))},[])])]))
cur_execute('delete from tmp_ban where time<%s;',(tt,))
def visitors_list_lock_wait():
while visitors_list_lock: time.sleep(0.05)
return True
def check_visitor():
global visitors_list, visitors_list_lock
visitors_list_lock = visitors_list_lock_wait()
ITT = int(time.time())
for t in visitors_list:
room = t.split('/')[0]
VISITOR_ACT = get_config(getRoom(room),'visitor_action')
if VISITOR_ACT != 'off' and ITT > visitors_list[t]:
reason = [xmpp.Node('reason',{},L('Too long was without voice!',t))]
if VISITOR_ACT == 'kick': sender(xmpp.Node('iq', {'id': get_id(), 'type': 'set', 'to':room}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'role':'none', 'nick':t.split('/',1)[1]},reason)])]))
else: sender(xmpp.Node('iq', {'id': get_id(), 'type': 'set', 'to':room}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'affiliation':'outcast', 'jid':get_jid_by_nick(t.split('/',1))},reason)])]))
visitors_list_lock = False
def visitor_presence(room,jid,nick,type,mass):
global visitors_list, visitors_list_lock
if getRoom(jid) == getRoom(Settings['jid']): return
if type == 'error': return
elif type == 'unavailable':
visitors_list_lock = visitors_list_lock_wait()
try: visitors_list.pop('%s/%s' % (room,nick))
except: pass
visitors_list_lock = False
if mass[1] == 'visitor':
visitors_list_lock = visitors_list_lock_wait()
visitors_list['%s/%s' % (room,nick)] = int(time.time()) + get_config_int(getRoom(room),'visitor_action_time')
visitors_list_lock = False
global execute, timer, presence_control
timer = [check_unban,check_visitor]
presence_control = [visitor_presence]
execute = [(7, 'ban_past', muc_ban_past, 2, 'Ban user.'),
(7, 'ban', muc_ban, 2, 'Ban user.'),
(7, 'banjid', muc_banjid, 2, 'Ban user by jid.'),
(7, 'unban', muc_none, 2, 'Unban user.'),
(7, 'unbanjid', muc_nonejid, 2, 'Unban user by jid.'),
(7, 'tban', muc_tempo_ban, 2, 'Temporary ban.\ntban show|del [jid] - show/del temporary bans\ntban nick\ntimeD|H|M|S\nreason - ban nick for time because reason.'),
(7, 'none_past', muc_none_past, 2, 'Remove user affiliation.'),
(7, 'none', muc_none, 2, 'Remove user affiliation.'),
(7, 'nonejid', muc_nonejid, 2, 'Remove user affiliation.'),
(7, 'member_past', muc_member_past, 2, 'Get member affiliation.'),
(7, 'member', muc_member, 2, 'Get member affiliation.'),
(7, 'memberjid', muc_memberjid, 2, 'Get member affiliation.'),
(7, 'kick', muc_kick, 2, 'Kick user.'),
(7, 'participant', muc_participant, 2, 'Change role to participant.'),
(7, 'visitor', muc_visitor, 2, 'Revoke voice.'),
(7, 'moderator', muc_moderator, 2, 'Grant moderator.'),
(8, 'global_ban', global_ban, 2, 'Global ban. Available only for confernce owner.\nglobal_ban del - remove conference from banlist,\nglobal_ban add - add conference into banlist,\nglobal_ban <jid> - ban jid in all rooms, where bot is admin.')]
View Git Blame Copy Permalink