mirror of https://github.com/nealey/firebot
115 lines
3.5 KiB
Python
115 lines
3.5 KiB
Python
|
#! /usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import firebot
|
||
|
|
import irc
|
||
|
|
import re
|
||
|
|
import os
|
||
|
|
import random
|
||
|
|
from procbot import ProcBot, Runner
|
||
|
|
|
||
|
|
def esc(arg):
|
||
|
|
return "'" + arg.replace("'", r"'\''") + "'"
|
||
|
|
|
||
|
|
def lesc(args):
|
||
|
|
return [esc(arg) for arg in args]
|
||
|
|
|
||
|
|
class Gallium(firebot.FireBot, ProcBot):
|
||
|
|
opall = False
|
||
|
|
bindings = []
|
||
|
|
|
||
|
|
def cmd_invite(self, sender, forum, addl):
|
||
|
|
# Join any channel to which we're invited
|
||
|
|
self.write('JOIN', forum.name())
|
||
|
|
|
||
|
|
def cmd_join(self, sender, forum, addl):
|
||
|
|
#firebot.FireBot.cmd_join(self, sender, forum, addl)
|
||
|
|
if self.opall:
|
||
|
|
if sender.name() == self.nick:
|
||
|
|
# If it was me, get a channel listing and beg for ops
|
||
|
|
self.write('WHO %s' % (forum.name()))
|
||
|
|
forum.notice('If you op me, I will op everyone who joins this channel.')
|
||
|
|
else:
|
||
|
|
# Otherwise, op the user
|
||
|
|
forum.write(['MODE', forum.name(), '+o'], sender.name())
|
||
|
|
|
||
|
|
def cmd_352(self, sender, forum, addl):
|
||
|
|
# Response to WHO
|
||
|
|
forum = irc.Channel(self, addl[0])
|
||
|
|
who = irc.User(self, addl[4], addl[1], addl[2])
|
||
|
|
self.add_luser(who, forum)
|
||
|
|
|
||
|
|
def server_status(self, sender, forum, addl, match):
|
||
|
|
loadavg = file('/proc/loadavg').read().strip()
|
||
|
|
io_status = file('/proc/io_status').read().strip()
|
||
|
|
forum.msg('%s; load %s' % (io_status, loadavg))
|
||
|
|
bindings.append((re.compile(r"^\008[:, ]+server status"),
|
||
|
|
server_status))
|
||
|
|
|
||
|
|
def unsafe_eval(self, sender, forum, addl, match):
|
||
|
|
if self.debug:
|
||
|
|
txt = match.group(1)
|
||
|
|
r = eval(txt)
|
||
|
|
forum.msg('%s: %r' % (sender.name(), r))
|
||
|
|
bindings.append((re.compile(r"^\008[:, ]+eval (.*)$"),
|
||
|
|
unsafe_eval))
|
||
|
|
|
||
|
|
def randglyph(self, sender, forum, addl, match):
|
||
|
|
count = 0
|
||
|
|
tries = []
|
||
|
|
while count < 6:
|
||
|
|
i = random.randint(0, 0xffff)
|
||
|
|
k = 'U+%04x' % i
|
||
|
|
tries.append(k)
|
||
|
|
r = self.get(k)
|
||
|
|
if r:
|
||
|
|
forum.msg('%s %s' % (k, r))
|
||
|
|
return
|
||
|
|
count += 1
|
||
|
|
forum.msg("Nothing found (tried %s)" % tries)
|
||
|
|
bindings.append((re.compile(r"^u\+rand$"),
|
||
|
|
randglyph))
|
||
|
|
|
||
|
|
def runcmd(self, sender, forum, addl, match):
|
||
|
|
command = match.group('command')
|
||
|
|
args = match.group('args').split(' ')
|
||
|
|
args = [x.replace("'", "'\\''") for x in args]
|
||
|
|
argstr = ' '.join(args)
|
||
|
|
Runner('%s %s' % (command, argstr),
|
||
|
|
lambda l,r: self.proc_cb('%s: ' % command, sender, forum, l, r))
|
||
|
|
bindings.append((re.compile(r"^(?P<command>whois) +(?P<args>.*)$"),
|
||
|
|
runcmd))
|
||
|
|
bindings.append((re.compile(r"^(?P<command>host) +(?P<args>.*)$"),
|
||
|
|
runcmd))
|
||
|
|
|
||
|
|
bindings.extend(firebot.FireBot.bindings)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
import shorturl
|
||
|
|
import socket
|
||
|
|
import daemon
|
||
|
|
import sys
|
||
|
|
|
||
|
|
debug = False
|
||
|
|
if "-d" in sys.argv:
|
||
|
|
debug = True
|
||
|
|
|
||
|
|
if not debug:
|
||
|
|
# Become a daemon
|
||
|
|
log = file('gallium.log', 'a')
|
||
|
|
daemon.daemon('gallium.pid', log, log)
|
||
|
|
|
||
|
|
# Short URL server
|
||
|
|
us = shorturl.start(('', 0))
|
||
|
|
firebot.URLSERVER = (socket.gethostbyaddr(socket.gethostname())[0],
|
||
|
|
us.getsockname()[1])
|
||
|
|
|
||
|
|
gallium = Gallium(('fozzie.woozle.org', 6667),
|
||
|
|
['gallium'],
|
||
|
|
"I'm a little printf, short and stdout",
|
||
|
|
["#woozle", "#gallium"])
|
||
|
|
gallium.debug = debug
|
||
|
|
|
||
|
|
irc.run_forever(0.5)
|