some new plugins (thx to _20h_)
This commit is contained in:
ChChBot admin
parent
63fc042027
commit
0ba2001b62
|
|
@ -0,0 +1,13 @@
|
|||
from util import hook, http, web
|
||||
from subprocess import check_output, CalledProcessError
|
||||
|
||||
@hook.command
|
||||
def freddycode(inp):
|
||||
"""freddycode <code> - Check if the Freddy Fresh code is correct."""
|
||||
|
||||
try:
|
||||
return "Freddy: '%s' ist %s" % (inp, \
|
||||
check_output(["/bin/freddycheck", inp]))
|
||||
except CalledProcessError as err:
|
||||
return "Freddy: Skript returned %s" % (str(err))
|
||||
|
||||
|
|
@ -0,0 +1,53 @@
|
|||
from util import hook
|
||||
import re
|
||||
import time
|
||||
from subprocess import check_output
|
||||
|
||||
def getstatus():
|
||||
try:
|
||||
return check_output("sudo /bin/chch-status", shell=True).strip("\n").decode("utf-8")
|
||||
except:
|
||||
return "unbekannt"
|
||||
|
||||
@hook.command("status", autohelp=False)
|
||||
def cmd_status(inp, reply=None):
|
||||
"""status - Return the door status"""
|
||||
reply("Chaostreff Status: %s" % (getstatus()))
|
||||
|
||||
@hook.event("TOPIC")
|
||||
def topic_update(info, conn=None, chan=None):
|
||||
"""topic_update -- Update the topic on TOPIC command"""
|
||||
status = getstatus()
|
||||
|
||||
topic = info[-1]
|
||||
|
||||
sstr = "Status: %s" % (status)
|
||||
if sstr in topic:
|
||||
return
|
||||
|
||||
if 'Status: ' in topic:
|
||||
new_topic = re.sub("Status: [^ ]*", sstr, topic)
|
||||
else:
|
||||
new_topic = "%s | %s" % (topic.rstrip(' |'), sstr)
|
||||
|
||||
if new_topic != topic:
|
||||
conn.send("TOPIC %s :%s" % (chan, new_topic))
|
||||
|
||||
@hook.event("332")
|
||||
def e332_update(info, conn=None, chan=None):
|
||||
"""e332_update -- run after current topic was requested"""
|
||||
chan = info[1]
|
||||
topic_update(info, conn=conn, chan=chan)
|
||||
|
||||
@hook.singlethread
|
||||
@hook.event("353")
|
||||
def e353_update(info, conn=None, chan=None):
|
||||
"""e353_update -- runs after a channel was joined"""
|
||||
chan = info[2]
|
||||
if chan.lower() == "#chaoschemnitz":
|
||||
conn.send("PRIVMSG Chanserv :op #chaoschemnitz")
|
||||
|
||||
while True:
|
||||
conn.send("TOPIC %s" % (chan))
|
||||
time.sleep(60)
|
||||
|
||||
|
|
@ -0,0 +1,21 @@
|
|||
from util import hook, http, web
|
||||
from subprocess import check_output, CalledProcessError
|
||||
from datetime import datetime
|
||||
|
||||
@hook.command("bw", autohelp=False)
|
||||
def bw(inp):
|
||||
"""bw - list last bandwidth measurement to the outside."""
|
||||
|
||||
try:
|
||||
o = check_output("/bin/chch-bandwidth")
|
||||
except CalledProcessError as err:
|
||||
return "chch-bandwidth: returned %s" % (str(err))
|
||||
|
||||
os = o.split(",")
|
||||
upl = int(os[-1])/1024.0/1024.0
|
||||
dl = int(os[-2])/1024.0/1024.0
|
||||
ts = os[0]
|
||||
tsd = datetime.strptime(ts, "%Y%m%d%H%M%S")
|
||||
|
||||
return "%s: upl = %f Mbit/s; dl = %f Mbit/s;" % (tsd, upl, dl)
|
||||
|
||||
|
|
@ -0,0 +1,183 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
from util import hook
|
||||
import re
|
||||
import time
|
||||
import requests
|
||||
import urllib
|
||||
from subprocess import check_output
|
||||
import json
|
||||
|
||||
def run_ecmd(cmd):
|
||||
# baseuri = "http://netio.chch.lan.ffc/ecmd?"
|
||||
baseuri = "http://10.8.128.35/ecmd?"
|
||||
cmds = "%20".join(cmd)
|
||||
req = requests.get("%s%s" % (baseuri, cmds))
|
||||
return req.text.strip()
|
||||
|
||||
# Lamp handling
|
||||
@hook.command("lamp", autohelp=True)
|
||||
def cmd_lamp(inp, reply=None):
|
||||
"""lamp color [mode] - set the lamp color"""
|
||||
args = inp.split(" ")
|
||||
if len(args) < 1:
|
||||
reply("""lamp color [mode] - set the lamp color""")
|
||||
return
|
||||
|
||||
if len(args[0]) != 6:
|
||||
reply("""lamp color [mode] - set the lamp color""")
|
||||
return
|
||||
|
||||
cmode = "s"
|
||||
if len(args) > 1:
|
||||
if args[1] == "s" or args[1] == "y" or args[1] == "f":
|
||||
cmode = args[1]
|
||||
|
||||
c = []
|
||||
c.append([5, int(args[0][0:2], 16)])
|
||||
c.append([4, int(args[0][2:4], 16)])
|
||||
c.append([3, int(args[0][4:6], 16)])
|
||||
|
||||
for ce in c:
|
||||
res = run_ecmd(["channel", str(ce[0]), str(ce[1]), cmode])
|
||||
if res != "OK":
|
||||
return
|
||||
reply("OK")
|
||||
|
||||
@hook.command("lamp_fadestep", autohelp=True)
|
||||
def cmd_lamp_fadestep(inp, reply=None):
|
||||
"""lamp_fadestep step - set the lamp fadestep"""
|
||||
args = inp.split(" ")
|
||||
|
||||
if len(args) < 1:
|
||||
reply("""lamp_fadestep step - set the lamp fadestep""")
|
||||
return
|
||||
|
||||
reply(run_ecmd(["fadestep", args[0]]))
|
||||
|
||||
@hook.command("lamp_fadestep_get", autohelp=False)
|
||||
def cmd_lamp_fadestep_get(inp, reply=None):
|
||||
"""lamp_fadestep_get - get the lamp fadestep"""
|
||||
reply(run_ecmd(["fadestep"]))
|
||||
|
||||
@hook.command("lamp_channels", autohelp=False)
|
||||
def cmd_lamp_channels(inp, reply=None):
|
||||
"""lamp_chanels - get the lamp channel count"""
|
||||
reply(run_ecmd(["channels"]))
|
||||
|
||||
# Wiki handling
|
||||
def wiki_changes(cmd=False):
|
||||
tmpfile = "/tmp/wikichanges.timestamp.txt"
|
||||
basewikiuri = "https://www.chaoschemnitz.de/index.php?title=%s"
|
||||
wikiapiuri = "https://www.chaoschemnitz.de/api.php?"\
|
||||
"action=query&list=recentchanges&format=json&"\
|
||||
"rcprop=user|userid|comment|parsedcomment|timestamp|"\
|
||||
"title|sha1|sizes|redirect|loginfo|tags|flags"\
|
||||
"&rclist=edit|external|new|log"
|
||||
|
||||
try:
|
||||
fdch = open(tmpfile, "rw")
|
||||
timestamp = fdch.read()
|
||||
fdch.close()
|
||||
except IOError:
|
||||
timestamp = None
|
||||
|
||||
try:
|
||||
r = requests.get(wikiapiuri, verify=False)
|
||||
except:
|
||||
return []
|
||||
|
||||
rarr = []
|
||||
changes = r.json["query"]["recentchanges"]
|
||||
ntimestamp = changes[0]["timestamp"]
|
||||
for change in changes:
|
||||
if change["timestamp"] == timestamp:
|
||||
break
|
||||
uri = basewikiuri % (urllib.quote(change["title"].encode("utf-8"), safe=""))
|
||||
rarr.append("wiki: %s changed '%s' ( %s ) comment: %s" %\
|
||||
(change["user"], change["title"], uri,\
|
||||
change["comment"].strip("\r\n\t")))
|
||||
|
||||
if cmd == False:
|
||||
fdch = open(tmpfile, "w+")
|
||||
fdch.write("%s" % (ntimestamp))
|
||||
fdch.close()
|
||||
|
||||
return rarr
|
||||
|
||||
def print_wiki_changes(info, conn=None, chan=None):
|
||||
"""print_wiki_changes - print wiki changes, when the worker calls"""
|
||||
ch = wiki_changes(cmd=False)
|
||||
if len(ch) == 0:
|
||||
return
|
||||
for c in ch[::-1]:
|
||||
conn.msg("#chaoschemnitz", c)
|
||||
time.sleep(0.5)
|
||||
|
||||
@hook.command("wikichanges", autohelp=False)
|
||||
def cmd_wikichanges(inp, reply=None):
|
||||
"""wikichanges - Return new recent wiki changes"""
|
||||
ch = wiki_changes(cmd=True)
|
||||
if len(ch) == 0:
|
||||
reply("No changes since the last call were made to the wiki.")
|
||||
else:
|
||||
for c in ch[::-1][-4:]:
|
||||
reply(c)
|
||||
time.sleep(0.5)
|
||||
|
||||
# Status handling
|
||||
def getstatus():
|
||||
try:
|
||||
response = urllib.urlopen('https://www.chaoschemnitz.de/chch.json')
|
||||
chch_json = response.read()
|
||||
chch_info = json.loads(chch_json)
|
||||
if chch_info['state']['open']:
|
||||
return "geöffnet".decode("utf-8")
|
||||
else:
|
||||
return "geschlossen"
|
||||
# return check_output("sudo /bin/chch-status", shell=True).strip("\n").decode("utf-8")
|
||||
except:
|
||||
return "unbekannt"
|
||||
|
||||
@hook.command("status", autohelp=False)
|
||||
def cmd_status(inp, reply=None):
|
||||
"""status - Return the door status"""
|
||||
reply("Chaostreff Status: %s" % (getstatus()))
|
||||
|
||||
@hook.event("TOPIC")
|
||||
def topic_update(info, conn=None, chan=None):
|
||||
"""topic_update -- Update the topic on TOPIC command"""
|
||||
status = getstatus()
|
||||
|
||||
topic = info[-1]
|
||||
|
||||
sstr = "Status: %s" % (status)
|
||||
if sstr in topic:
|
||||
return
|
||||
|
||||
if 'Status: ' in topic:
|
||||
new_topic = re.sub("Status: [^ ]*", sstr, topic)
|
||||
else:
|
||||
new_topic = "%s | %s" % (topic.rstrip(' |'), sstr)
|
||||
|
||||
if new_topic != topic:
|
||||
conn.send("TOPIC %s :%s" % (chan, new_topic))
|
||||
|
||||
@hook.event("332")
|
||||
def e332_update(info, conn=None, chan=None):
|
||||
"""e332_update -- run after current topic was requested, runs worker tasks too"""
|
||||
chan = info[1]
|
||||
topic_update(info, conn=conn, chan=chan)
|
||||
print_wiki_changes(info, conn=conn, chan=chan)
|
||||
|
||||
@hook.singlethread
|
||||
@hook.event("353")
|
||||
def e353_update(info, conn=None, chan=None):
|
||||
"""e353_update -- runs after a channel (#chaoschemnitz) was joined"""
|
||||
chan = info[2]
|
||||
if chan.lower() == "#chaoschemnitz":
|
||||
conn.send("PRIVMSG Chanserv :op #chaoschemnitz")
|
||||
|
||||
while True:
|
||||
time.sleep(60)
|
||||
conn.send("TOPIC %s" % (chan))
|
||||
|
||||
|
|
@ -0,0 +1,76 @@
|
|||
import os
|
||||
import codecs
|
||||
import time
|
||||
import re
|
||||
|
||||
from util import hook
|
||||
|
||||
timestamp_format = '%H:%M:%S'
|
||||
|
||||
formats = {
|
||||
'PRIVMSG': '<%(nick)s> %(msg)s',
|
||||
'PART': '-!- %(nick)s [%(user)s@%(host)s] has left %(chan)s',
|
||||
'JOIN': '-!- %(nick)s [%(user)s@%(host)s] has joined %(param0)s',
|
||||
'MODE': '-!- mode/%(chan)s [%(param_tail)s] by %(nick)s',
|
||||
'KICK': '-!- %(param1)s was kicked from %(chan)s by %(nick)s [%(msg)s]',
|
||||
'TOPIC': '-!- %(nick)s changed the topic of %(chan)s to: %(msg)s',
|
||||
'QUIT': '-!- %(nick)s has quit [%(msg)s]',
|
||||
'PING': '',
|
||||
'NOTICE': '-%(nick)s- %(msg)s'
|
||||
}
|
||||
|
||||
ctcp_formats = {
|
||||
'ACTION': '* %(nick)s %(ctcpmsg)s',
|
||||
'VERSION': '%(nick)s has requested CTCP %(ctcpcmd)s from %(chan)s: %(ctcpmsg)s',
|
||||
'PING': '%(nick)s has requested CTCP %(ctcpcmd)s from %(chan)s: %(ctcpmsg)s',
|
||||
'TIME': '%(nick)s has requested CTCP %(ctcpcmd)s from %(chan)s: %(ctcpmsg)s',
|
||||
'FINGER': '%(nick)s has requested CTCP %(ctcpcmd)s from %(chan)s: %(ctcpmsg)s'
|
||||
}
|
||||
|
||||
irc_color_re = re.compile(r'(\x03(\d+,\d+|\d)|[\x0f\x02\x16\x1f])')
|
||||
|
||||
|
||||
def gmtime(format):
|
||||
return time.strftime(format, time.gmtime())
|
||||
|
||||
|
||||
def beautify(input):
|
||||
format = formats.get(input.command, '%(raw)s')
|
||||
args = dict(input)
|
||||
|
||||
leng = len(args['paraml'])
|
||||
for n, p in enumerate(args['paraml']):
|
||||
args['param' + str(n)] = p
|
||||
args['param_' + str(abs(n - leng))] = p
|
||||
|
||||
args['param_tail'] = ' '.join(args['paraml'][1:])
|
||||
args['msg'] = irc_color_re.sub('', args['msg'])
|
||||
|
||||
if input.command == 'PRIVMSG' and input.msg.count('\x01') >= 2:
|
||||
ctcp = input.msg.split('\x01', 2)[1].split(' ', 1)
|
||||
if len(ctcp) == 1:
|
||||
ctcp += ['']
|
||||
args['ctcpcmd'], args['ctcpmsg'] = ctcp
|
||||
format = ctcp_formats.get(args['ctcpcmd'],
|
||||
'%(nick)s [%(user)s@%(host)s] requested unknown CTCP '
|
||||
'%(ctcpcmd)s from %(chan)s: %(ctcpmsg)s')
|
||||
|
||||
return format % args
|
||||
|
||||
@hook.singlethread
|
||||
@hook.event('*')
|
||||
def log(paraml, input=None, bot=None):
|
||||
timestamp = gmtime(timestamp_format)
|
||||
|
||||
if input.command == 'QUIT': # these are temporary fixes until proper
|
||||
input.chan = 'quit' # presence tracking is implemented
|
||||
if input.command == 'NICK':
|
||||
input.chan = 'nick'
|
||||
|
||||
beau = beautify(input)
|
||||
|
||||
if beau == '': # don't log this
|
||||
return
|
||||
|
||||
print timestamp, input.chan, beau.encode('utf8', 'ignore')
|
||||
|
||||
Reference in New Issue