Files
urlbot-native/plugins.py

882 lines
21 KiB
Python
Raw Normal View History

#!/usr/bin/python3
# -*- coding: utf-8 -*-
if '__main__' == __name__:
print('''this is a plugin file, which is not meant to be executed''')
exit(-1)
import time, random, unicodedata, re, sys, urllib.request, json
2015-02-05 19:00:38 +01:00
import types
import traceback
2015-02-06 13:19:14 +01:00
import urllib.parse
from local_config import conf, set_conf
from common import *
2015-02-11 01:44:21 +01:00
from excuses import excuses
2014-11-09 16:52:22 +01:00
from urlbot import extract_title
from functools import wraps
2015-06-21 00:56:33 +02:00
ptypes_PARSE = 'parser'
ptypes_COMMAND = 'command'
ptypes = [ptypes_PARSE, ptypes_COMMAND]
2014-09-29 19:15:00 +02:00
joblist = []
plugins = {p : [] for p in ptypes}
2015-06-20 14:18:50 +02:00
got_hangup = False
def pluginfunction(name, desc, plugin_type, ratelimit_class = RATE_GLOBAL, enabled = True):
2015-02-09 03:46:17 +01:00
''' A decorator to make a plugin out of a function '''
if plugin_type not in ptypes:
2015-02-09 03:46:17 +01:00
raise TypeError('Illegal plugin_type: %s' % plugin_type)
def decorate(f):
f.is_plugin = True
2015-02-06 12:47:27 +01:00
f.is_enabled = enabled
f.plugin_name = name
f.plugin_desc = desc
f.plugin_type = plugin_type
f.ratelimit_class = ratelimit_class
return f
return decorate
2014-09-29 19:15:00 +02:00
def register_event(t, callback, args):
joblist.append((t, callback, args))
2015-02-09 03:46:17 +01:00
@pluginfunction('mental_ill', 'parse mental illness', ptypes_PARSE, ratelimit_class = RATE_NO_SILENCE | RATE_GLOBAL)
def parse_mental_ill(**args):
min_ill = 3
c = 0
flag = False
# return True for min_ill '!' in a row
for d in args['data']:
if '!' == d or '?' == d:
c += 1
else:
c = 0
if (min_ill <= c):
flag = True
break
if True == flag:
log.plugin('sent mental illness reply')
return {
'msg': '''Multiple exclamation/question marks are a sure sign of mental disease, with %s as a living example.''' % args['reply_user']
}
2015-02-09 03:46:17 +01:00
@pluginfunction('debbug', 'parse Debian bug numbers', ptypes_PARSE, ratelimit_class = RATE_NO_SILENCE | RATE_GLOBAL)
def parse_debbug(**args):
2014-11-09 16:52:22 +01:00
bugs = re.findall(r'#(\d{4,})', args['data'])
if not bugs:
return None
out = []
for b in bugs:
log.plugin('detected Debian bug #%s' % b)
url = 'https://bugs.debian.org/cgi-bin/bugreport.cgi?bug=%s' % b
status, title = extract_title(url)
if 0 == status:
out.append('Debian Bug: %s: %s' % (title, url))
elif 3 == status:
out.append('error for #%s: %s' % (b, title))
else:
log.plugin('parse_debbug(): unknown status %d' % status)
2014-11-09 16:52:22 +01:00
return {
'msg': out
2014-11-09 16:52:22 +01:00
}
2015-02-09 03:46:17 +01:00
@pluginfunction('cve', 'parse a CVE handle', ptypes_PARSE, ratelimit_class = RATE_NO_SILENCE | RATE_GLOBAL)
def parse_cve(**args):
2014-12-01 17:50:24 +01:00
cves = re.findall(r'(CVE-\d\d\d\d-\d+)', args['data'].upper())
if not cves:
return None
log.plugin('detected CVE handle')
2014-12-01 17:50:24 +01:00
return {
'msg': ['https://security-tracker.debian.org/tracker/%s' % c for c in cves]
2014-12-01 17:50:24 +01:00
}
2015-06-14 16:50:40 +02:00
@pluginfunction('dsa', 'parse a DSA handle', ptypes_PARSE, ratelimit_class = RATE_NO_SILENCE | RATE_GLOBAL)
def parse_dsa(**args):
dsas = re.findall(r'(DSA-\d\d\d\d-\d+)', args['data'].upper())
if not dsas:
return None
log.plugin('detected DSA handle')
2015-06-14 16:50:40 +02:00
return {
'msg': ['https://security-tracker.debian.org/tracker/%s' % d for d in dsas]
2015-06-14 16:50:40 +02:00
}
2015-02-09 03:46:17 +01:00
@pluginfunction('skynet', 'parse skynet', ptypes_PARSE)
def parse_skynet(**args):
if 'skynet' in args['data'].lower():
log.plugin('sent skynet reply')
return {
'msg': '''I'm an independent bot and have nothing to do with other artificial intelligence systems!'''
}
2015-06-17 10:53:44 +02:00
@pluginfunction('moin', 'parse moin/bye', ptypes_PARSE)
def parse_moin_bye(**args):
2015-06-20 15:17:11 +02:00
if args['reply_user'] in conf('moin-disabled-user'):
log.plugin('moin blacklist match')
2015-06-20 15:17:11 +02:00
return
2015-06-17 10:53:44 +02:00
moin = [
'Hi',
'Guten Morgen', 'Morgen',
2015-06-17 10:53:44 +02:00
'Moin',
2015-06-18 13:41:38 +02:00
'Tag', 'Tach',
2015-06-20 23:04:19 +02:00
'NAbend', 'Abend',
'Hallo', 'Hello'
2015-06-17 10:53:44 +02:00
]
bye = [
'Nacht', 'gN8', 'N8',
2015-06-17 10:53:44 +02:00
'bye',
]
for direction in [moin, bye]:
for d in direction:
words = re.split(r'\W+', args['data'])
# assumption: longer sentences are not greetings
if 3 < len(args['data'].split()):
continue
for w in words:
if d.lower() == w.lower():
if args['reply_user'] in conf('moin-modified-user'):
log.plugin('being "quiet" for %s' % w)
return {
'msg': '/me %s' % random.choice([
"doesn't say anything at all",
'whistles uninterested',
'just ignores this incident'
])
}
log.plugin('sent %s reply for %s' % (
'moin' if direction is moin else 'bye', w
))
return {
'msg': '''%s, %s''' % (
random.choice(direction),
args['reply_user']
)
}
2015-06-17 10:53:44 +02:00
2015-02-08 22:59:34 +01:00
@pluginfunction('latex', r'reacts on \LaTeX', ptypes_PARSE)
2015-02-09 03:46:17 +01:00
def parse_latex(**args):
2015-02-08 22:59:34 +01:00
if r'\LaTeX' in args['data']:
return {
'msg': '''LaTeX is way too complex for me, I'm happy with fmt(1)'''
}
2015-06-18 14:05:44 +02:00
@pluginfunction('me-action', 'reacts to /me.*%{bot_user}', ptypes_PARSE)
2015-02-22 20:15:23 +01:00
def parse_slash_me(**args):
if args['data'].lower().startswith('/me') and (conf('bot_user') in args['data'].lower()):
log.plugin('sent /me reply')
2015-02-22 20:15:23 +01:00
me_replys = [
'are you that rude to everybody?',
'oh, thank you...',
'do you really think that was nice?',
'that sounds very interesting...',
"excuse me, but I'm already late for an appointment"
]
return {
2015-06-18 13:51:50 +02:00
'msg': args['reply_user'] + ': %s' % random.choice(me_replys)
2015-02-22 20:15:23 +01:00
}
2015-02-08 22:59:34 +01:00
#@pluginfunction('dummy_parser', 'dummy_parser desc', ptypes_PARSE)
#def parse_skynet(**args):
# if 'dummy_parser' in args['data'].lower():
# log.plugin('dummy_parser triggered')
2015-02-08 22:59:34 +01:00
# return {
# 'msg': 'dummy_parser triggered'
# }
def data_parse_other(msg_obj):
data = msg_obj['body']
reply_user = msg_obj['mucnick']
for p in plugins[ptypes_PARSE]:
if ratelimit_exceeded(p.ratelimit_class):
continue
if not p.is_enabled:
continue
ret = p(reply_user=reply_user, data=data)
if None != ret:
if 'msg' in list(ret.keys()):
2014-09-27 05:51:18 +02:00
ratelimit_touch(RATE_CHAT)
send_reply(ret['msg'], msg_obj)
2015-02-09 03:46:17 +01:00
@pluginfunction('help', 'print help for a command or all known commands', ptypes_COMMAND)
def command_help(argv, **args):
command = argv[0]
what = argv[1] if len(argv) > 1 else None
2015-01-03 19:04:01 +01:00
if 'help' != command:
return
2014-09-27 16:15:01 +02:00
2015-01-03 19:04:01 +01:00
if None == what:
log.plugin('empty help request, sent all commands')
2015-02-08 22:37:27 +01:00
commands = args['cmd_list']
commands.sort()
parsers = args['parser_list']
parsers.sort()
2015-01-03 19:04:01 +01:00
return {
'msg': [
'%s: known commands: %s' % (
args['reply_user'], str(commands).strip('[]')
),
'known parsers: %s' % str(parsers).strip('[]')
]
}
flag = False
2015-06-18 14:09:29 +02:00
for p in plugins[ptypes_COMMAND] + plugins[ptypes_PARSE]:
if what == p.plugin_name:
flag = True
log.plugin('sent help for %s' % what)
return {
2015-06-18 14:09:29 +02:00
'msg': args['reply_user'] + ': help for %s %s: %s' % (
'parser' if p.plugin_type == ptypes_PARSE else 'command',
what, p.plugin_desc
2015-01-03 19:04:01 +01:00
)
}
if not flag:
log.plugin('no help found for %s' % what)
return {
'msg': args['reply_user'] + ': no such command: %s' % what
}
2015-02-09 03:46:17 +01:00
@pluginfunction('version', 'prints version', ptypes_COMMAND)
def command_version(argv, **args):
if 'version' != argv[0]:
return
log.plugin('sent version string')
return {
'msg': args['reply_user'] + (''': I'm running ''' + VERSION)
}
2015-02-09 03:46:17 +01:00
@pluginfunction('klammer', 'prints an anoying paper clip aka. Karl Klammer', ptypes_COMMAND)
def command_klammer(argv, **args):
if 'klammer' != argv[0]:
return
log.plugin('sent karl klammer')
return {
'msg': (
args['reply_user'] + ',',
r''' _, Was moechten''',
r'''( _\_ Sie tun?''',
r''' \0 O\ ''',
r''' \\ \\ [ ] ja ''',
r''' \`' ) [ ] noe''',
r''' `'' '''
)
}
2014-11-17 19:49:02 +01:00
2015-02-09 03:46:17 +01:00
@pluginfunction('unikot', 'prints an unicode string', ptypes_COMMAND)
def command_unicode(argv, **args):
if 'unikot' != argv[0]:
return
log.plugin('sent some unicode')
return {
'msg': (
args['reply_user'] + ''', here's some''',
'''┌────────┐''',
'''│Unicode!│''',
'''└────────┘'''
)
}
2015-02-09 03:46:17 +01:00
@pluginfunction('source', 'prints git URL', ptypes_COMMAND)
def command_source(argv, **args):
if not argv[0] in ('source', 'src'):
return
log.plugin('sent source URL')
return {
'msg': 'My source code can be found at %s' % conf('src-url')
}
2015-02-09 03:46:17 +01:00
@pluginfunction('dice', 'rolls a dice, optional N times', ptypes_COMMAND, ratelimit_class = RATE_INTERACTIVE)
def command_dice(argv, **args):
if 'dice' != argv[0]:
return
2014-12-20 17:02:26 +01:00
count = 0
try:
count = 1 if len(argv) < 2 else int(argv[1])
2014-12-20 17:02:26 +01:00
except ValueError as e:
return {
'msg': '%s: dice: error when parsing int(%s): %s' % (
args['reply_user'], argv[1], str(e)
2014-12-20 17:02:26 +01:00
)
}
if 0 >= count or 5 <= count:
return {
'msg': '%s: dice: invalid arguments (0 < N < 5)' % args['reply_user']
}
dice_char = ['', '', '', '', '', '', '']
msg = 'rolling %s for %s:' % (
'a dice' if 1 == count else '%d dices' % count, args['reply_user']
)
for i in range(count):
rnd = 0
if args['reply_user'] in conf('enhanced-random-user'):
rnd = 0 # this might confuse users. good.
log.plugin('sent random (enhanced)')
else:
rnd = random.randint(1, 6)
log.plugin('sent random')
# the \u200b chars ('ZERO WIDTH SPACE') avoid interpreting stuff as smileys
# by some strange clients
msg += ' %s (\u200b%d\u200b)' % (dice_char[rnd], rnd)
2014-12-20 17:02:26 +01:00
return {
'msg': msg
}
2015-02-09 03:46:17 +01:00
@pluginfunction('uptime', 'prints uptime', ptypes_COMMAND)
def command_uptime(argv, **args):
if 'uptime' != argv[0]:
return
u = int(conf('uptime') + time.time())
plural_uptime = 's'
plural_request = 's'
if 1 == u:
plural_uptime = ''
if 1 == conf('request_counter'):
plural_request = ''
log.plugin('sent statistics')
return {
'msg': args['reply_user'] + (''': happily serving for %d second%s, %d request%s so far.''' % (u, plural_uptime, conf('request_counter'), plural_request))
}
2015-02-09 03:46:17 +01:00
@pluginfunction('ping', 'sends pong', ptypes_COMMAND, ratelimit_class = RATE_INTERACTIVE)
def command_ping(argv, **args):
if 'ping' != argv[0]:
return
rnd = random.randint(0, 3) # 1:4
if 0 == rnd:
msg = args['reply_user'] + ''': peng (You're dead now.)'''
log.plugin('sent pong (variant)')
elif 1 == rnd:
msg = args['reply_user'] + ''': I don't like you, leave me alone.'''
log.plugin('sent pong (dontlike)')
else:
msg = args['reply_user'] + ''': pong'''
log.plugin('sent pong')
return {
'msg': msg
}
2015-02-09 03:46:17 +01:00
@pluginfunction('info', 'prints info message', ptypes_COMMAND)
def command_info(argv, **args):
if 'info' != argv[0]:
return
log.plugin('sent long info')
return {
'msg': args['reply_user'] + (''': I'm a bot, my job is to extract <title> tags from posted URLs. In case I'm annoying or for further questions, please talk to my master %s. I'm rate limited and shouldn't post more than %d messages per %d seconds. To make me exit immediately, highlight me with 'hangup' in the message (emergency only, please). For other commands, highlight me with 'help'.''' % (conf('bot_owner'), conf('hist_max_count'), conf('hist_max_time')))
}
2015-02-09 03:46:17 +01:00
@pluginfunction('teatimer', 'sets a tea timer to $1 or currently %d seconds' % conf('tea_steep_time'), ptypes_COMMAND)
def command_teatimer(argv, **args):
if 'teatimer' != argv[0]:
return
steep = conf('tea_steep_time')
2014-09-29 19:15:00 +02:00
if len(argv) > 1:
2014-10-29 13:12:41 +01:00
try:
steep = int(argv[1])
except Exception as e:
2014-10-29 13:12:41 +01:00
return {
'msg': args['reply_user'] + ': error when parsing int(%s): %s' % (
argv[1], str(e)
)
2014-10-29 13:12:41 +01:00
}
ready = time.time() + steep
try:
log.plugin('tea timer set to %s' % time.strftime('%F.%T', time.localtime(ready)))
except ValueError as e:
2014-09-29 19:15:00 +02:00
return {
'msg': args['reply_user'] + ': time format error: ' + str(e)
2014-09-29 19:15:00 +02:00
}
# FIXME: this is currently broken because the msg_obj gets modified by the very
# first reply and can't be reused to .reply() with another message
register_event(ready, send_reply, (args['reply_user'] + ': Your tea is ready!', args['msg_obj']))
return {
'msg': args['reply_user'] + ': Tea timer set to %s' % time.strftime(
'%F.%T', time.localtime(ready)
)
}
2015-02-09 03:46:17 +01:00
@pluginfunction('decode', 'prints the long description of an unicode character', ptypes_COMMAND)
def command_decode(argv, **args):
if 'decode' != argv[0]:
return
2015-05-31 22:02:06 +02:00
if len(argv) <= 1:
return {
'msg': args['reply_user'] + ': usage: decode {single character}'
}
char = argv[1]
char_esc = str(char.encode('unicode_escape'))[3:-1]
log.plugin('decode called for %s' % char)
try:
uni_name = unicodedata.name(char)
except Exception as e:
log.plugin('decode(%s) failed: %s' % (char, str(e)))
return {
'msg': args['reply_user'] + ": can't decode %s (%s): %s" % (char, char_esc, str(e))
}
return {
'msg': args['reply_user'] + ': %s (%s) is called "%s"' % (char, char_esc, uni_name)
}
2015-02-09 03:46:17 +01:00
@pluginfunction('show-blacklist', 'show the current URL blacklist, optionally filtered', ptypes_COMMAND)
def command_show_blacklist(argv, **args):
if 'show-blacklist' != argv[0]:
return
log.plugin('sent URL blacklist')
argv1 = None if len(argv) < 2 else argv[1]
return {
'msg': [
args['reply_user'] + ': URL blacklist%s: ' % (
'' if not argv1 else ' (limited to %s)' % argv1
)
] + [
b for b in conf('url_blacklist')
if not argv1 or argv1 in b
]
}
2015-02-09 03:46:17 +01:00
def usersetting_get(argv, args):
blob = conf_load()
arg_user = args['reply_user']
arg_key = argv[1]
if not arg_user in blob['user_pref']:
return {
'msg': args['reply_user'] + ': user key not found'
}
return {
'msg': args['reply_user'] + ': %s == %s' % (
arg_key,
'on' if blob['user_pref'][arg_user][arg_key] else 'off'
)
}
2015-02-09 03:46:17 +01:00
@pluginfunction('set', 'modify a user setting', ptypes_COMMAND)
def command_usersetting(argv, **args):
if 'set' != argv[0]:
return
settings = ['spoiler']
arg_user = args['reply_user']
arg_key = argv[1] if len(argv) > 1 else None
arg_val = argv[2] if len(argv) > 2 else None
if not arg_key in settings:
return {
'msg': args['reply_user'] + ': known settings: ' + (', '.join(settings))
}
if not arg_val in ['on', 'off', None]:
return {
'msg': args['reply_user'] + ': possible values for %s: on, off' % arg_key
}
if None == arg_val:
# display current value
return usersetting_get(argv, args)
if conf('persistent_locked'):
return {
'msg': args['reply_user'] + ''': couldn't get exclusive lock'''
}
set_conf('persistent_locked', True)
blob = conf_load()
if 'user_pref' not in blob:
blob['user_pref'] = {}
if not arg_user in blob['user_pref']:
blob['user_pref'][arg_user] = {}
blob['user_pref'][arg_user][arg_key] = 'on' == arg_val
conf_save(blob)
set_conf('persistent_locked', False)
# display value written to db
2015-02-09 03:46:17 +01:00
return usersetting_get(argv, args)
2015-02-09 03:46:17 +01:00
@pluginfunction('cake', 'displays a cake ASCII art', ptypes_COMMAND)
def command_cake(argv, **args):
if 'cake' != argv[0]:
2014-12-16 07:58:35 +01:00
return
cakes = [ "No cake for you!",
2015-02-06 15:13:03 +01:00
("The Enrichment Center is required to remind you "
"that you will be baked, and then there will be cake."),
"The cake is a lie!",
("This is your fault. I'm going to kill you. "
"And all the cake is gone. You don't even care, do you?"),
2015-02-06 15:13:03 +01:00
"Quit now and cake will be served immediately.",
("Enrichment Center regulations require both hands to be "
"empty before any cake..."),
("Uh oh. Somebody cut the cake. I told them to wait for "
"you, but they did it anyway. There is still some left, "
"though, if you hurry back."),
"I'm going to kill you, and all the cake is gone.",
"Who's gonna make the cake when I'm gone? You?" ]
2014-12-16 07:58:35 +01:00
return {
2015-02-22 20:15:23 +01:00
'msg': args['reply_user'] + ': %s' % (random.sample(cakes, 1)[0])
2014-12-16 07:58:35 +01:00
}
2015-02-09 03:46:17 +01:00
@pluginfunction('remember', 'remembers something', ptypes_COMMAND)
def command_remember(argv, **args):
if 'remember' != argv[0]:
return
log.plugin('remember plugin called')
if not len(argv) > 1:
return {
'msg': args['reply_user'] + ': invalid message'
}
to_remember = ' '.join(args['data'].split()[2:]) # this is a little dirty. A little lot
set_conf('data_remember', to_remember)
return {
'msg': args['reply_user'] + ': remembering ' + to_remember
}
2015-02-09 03:46:17 +01:00
@pluginfunction('recall', "recalls something previously 'remember'ed", ptypes_COMMAND)
def command_recall(argv, **args):
if 'recall' != argv[0]:
return
log.plugin('recall plugin called')
return {
'msg': args['reply_user'] + ': recalling %s' % conf('data_remember')
}
#TODO: send a hint if someone types plugin as command
2015-02-09 03:46:17 +01:00
@pluginfunction('plugin', "'disable' or 'enable' plugins", ptypes_COMMAND)
def command_plugin_activation(argv, **args):
command = argv[0]
plugin = argv[1] if len(argv) > 1 else None
2015-01-03 18:59:44 +01:00
if not command in ('enable', 'disable'):
return
log.plugin('plugin activation plugin called')
if None == plugin:
return {
'msg': args['reply_user'] + ': no plugin given'
}
elif command_plugin_activation.plugin_name == plugin:
return {
'msg': args['reply_user'] + ': not allowed'
}
for p in plugins[ptypes_COMMAND] + plugins[ptypes_PARSE]:
if p.plugin_name == plugin:
p.is_enabled = 'enable' == command
2015-01-03 18:59:44 +01:00
return {
2015-02-09 03:46:17 +01:00
'msg': args['reply_user'] + ': %sd %s' % (
2015-01-03 18:59:44 +01:00
command, plugin
)
}
return {
2015-01-03 18:59:44 +01:00
'msg': args['reply_user'] + ': unknown plugin %s' % plugin
}
2015-02-09 03:46:17 +01:00
@pluginfunction('wp-en', 'crawl the english Wikipedia', ptypes_COMMAND)
def command_wp_en(argv, **args):
if 'wp-en' != argv[0]:
return
if argv[0]:
argv[0] = 'wp'
2015-02-09 03:46:17 +01:00
return command_wp(argv, lang='en', **args)
2015-02-09 03:46:17 +01:00
@pluginfunction('wp', 'crawl the german Wikipedia', ptypes_COMMAND)
def command_wp(argv, lang='de', **args):
if 'wp' != argv[0]:
return
log.plugin('wp plugin called')
2015-02-09 03:46:17 +01:00
query = ' '.join(argv[1:])
2015-02-09 03:46:17 +01:00
if query == '':
return {
2015-02-09 03:46:17 +01:00
'msg': args['reply_user'] + ': no query given'
}
2015-02-09 03:46:17 +01:00
api = {
'action': 'query',
'prop': 'extracts',
'explaintext': '',
'redirects': '',
'exsentences': 2,
'continue': '',
'format': 'json',
'titles': query
}
apiurl = 'https://%s.wikipedia.org/w/api.php?%s' % (
lang, urllib.parse.urlencode(api)
)
log.plugin('wp: fetching %s' % apiurl)
2015-02-18 00:40:18 +01:00
try:
2015-02-06 13:19:14 +01:00
response = urllib.request.urlopen(apiurl)
buf = response.read(BUFSIZ)
2015-02-18 00:40:18 +01:00
j = json.loads(buf.decode('utf8'))
2015-02-06 15:27:11 +01:00
2015-02-06 14:07:58 +01:00
page = next(iter(j['query']['pages'].values()))
2015-02-09 03:46:17 +01:00
short = page.get('extract', None)
linktitle = page.get('title', query).replace(' ', '_')
2015-02-09 04:01:35 +01:00
link = 'https://%s.wikipedia.org/wiki/%s' % (
lang, urllib.parse.quote(linktitle)
)
except Exception as e:
log.plugin('wp(%s) failed: %s, %s' % (query, e, traceback.format_exc()))
return {
2015-02-09 03:46:17 +01:00
'msg': args['reply_user'] + ': something failed: %s' % e
}
2015-02-06 14:07:58 +01:00
if short is not None:
2015-02-07 02:12:41 +01:00
return {
'msg': args['reply_user'] + ': %s (<%s>)' % (
2015-02-09 03:46:17 +01:00
short if short.strip() else '(nix)', link
)
}
2015-02-09 03:46:17 +01:00
elif 'missing' in page:
return {
'msg': 'Article "%s" not found' % page.get('title', query)
}
else:
2015-02-09 03:46:17 +01:00
return {
'msg': 'json data seem to be broken'
}
2015-02-11 01:44:21 +01:00
@pluginfunction('excuse', 'prints BOFH style excuses', ptypes_COMMAND)
def command_dummy(argv, **args):
if 'excuse' != argv[0]:
return
log.plugin('BOFH plugin called')
2015-02-11 01:44:21 +01:00
excuse = random.sample(excuses, 1)[0]
return {
'msg': args['reply_user'] + ': ' + excuse
}
2015-02-11 01:33:48 +01:00
#@pluginfunction('dummy', 'dummy description', ptypes_COMMAND)
#def command_dummy(argv, **args):
# if 'dummy' != argv[0]:
# return
#
# log.plugin('dummy plugin called')
#
# return {
# 'msg': args['reply_user'] + ': dummy plugin called'
# }
def else_command(args):
log.plugin('sent short info')
return {
'msg': args['reply_user'] + ''': I'm a bot (highlight me with 'info' for more information).'''
}
def data_parse_commands(msg_obj):
2015-06-20 14:18:50 +02:00
global got_hangup
data = msg_obj['body']
words = data.split()
if 2 > len(words): # need at least two words
return None
# don't reply if beginning of the text matches bot_user
if not data.startswith(conf('bot_user')):
return None
if 'hangup' in data:
log.warn('received hangup: ' + data)
2015-06-20 14:18:50 +02:00
got_hangup = True
sys.exit(1)
return None
reply_user = msg_obj['mucnick']
for p in plugins[ptypes_COMMAND]:
if ratelimit_exceeded(p.ratelimit_class):
continue
if not p.is_enabled:
continue
ret = p(
data = data,
cmd_list = [pl.plugin_name for pl in plugins[ptypes_COMMAND]],
parser_list = [pl.plugin_name for pl in plugins[ptypes_PARSE]],
reply_user = reply_user,
msg_obj = msg_obj,
argv = words[1:]
)
if None != ret:
if 'msg' in list(ret.keys()):
2014-12-14 15:54:57 +01:00
ratelimit_touch(RATE_CHAT)
if ratelimit_exceeded(RATE_CHAT):
return False
2014-12-14 15:54:57 +01:00
send_reply(ret['msg'], msg_obj)
return None
ret = else_command({'reply_user': reply_user})
if None != ret:
if ratelimit_exceeded(RATE_GLOBAL):
return False
if 'msg' in list(ret.keys()):
2014-12-14 15:54:57 +01:00
send_reply(ret['msg'], msg_obj)
if debug_enabled():
def _send_reply(a, msg_obj):
log.info('send_reply[%s]' % msg_obj, a)
2014-12-14 16:26:48 +01:00
def _conf(ignored):
return 'bot'
def _ratelimit_exceeded(ignored=None):
return False
def _ratelimit_touch(ignored=None):
return True
try:
send_reply
except NameError:
send_reply = _send_reply
try:
conf
except NameError:
conf = _conf
try:
ratelimit_exceeded
except NameError:
ratelimit_exceeded = _ratelimit_exceeded
try:
ratelimit_touch
except NameError:
ratelimit_touch = _ratelimit_touch
log.info('debugging enabled')
2015-02-05 19:00:38 +01:00
def register(func_type):
2015-02-09 03:46:17 +01:00
'''
2015-02-05 19:00:38 +01:00
Register plugins.
2015-02-05 19:00:38 +01:00
Arguments:
func_type -- plugin functions with this type (ptypes) will be loaded
2015-02-09 03:46:17 +01:00
'''
functions = [
f for ignored, f in globals().items()
if
type(f) == types.FunctionType
and f.__dict__.get('is_plugin', False)
and f.plugin_type == func_type
]
2015-06-21 00:56:33 +02:00
log.info('auto-reg %s: %s' % (func_type, ', '.join(
2015-02-09 03:46:17 +01:00
f.plugin_name for f in functions
)))
2015-02-05 19:00:38 +01:00
for f in functions:
register_plugin(f, func_type)
def register_plugin(function, func_type):
try:
plugins[func_type].append(function)
2015-02-05 19:00:38 +01:00
except Exception as e:
log.warn('registering %s failed: %s, %s' %
(function, e, traceback.format_exc()))
def register_all():
register(ptypes_PARSE)
register(ptypes_COMMAND)
2014-09-29 19:15:00 +02:00
def event_trigger():
2015-06-20 14:18:50 +02:00
if got_hangup:
return False
2014-09-29 19:15:00 +02:00
if 0 == len(joblist):
return
now = time.time()
for (i, (t, callback, args)) in enumerate(joblist):
2014-09-29 19:15:00 +02:00
if t < now:
2014-12-13 22:57:15 +01:00
callback(*args)
2014-09-29 19:15:00 +02:00
del(joblist[i])