bot.py

#!/usr/env/bin python
# coding=utf-8
#
# Himawari: Intelligent Mocking Agent for Writing Arbitrary Rants to IRC
# Written by Kang Seonghoon. Dedicated to the Public Domain.
# http://himawari.mearie.org/
#

import sys
import re
import select
import socket
import time
import signal
import traceback

if len(sys.argv) < 5:
    print >>sys.stderr, 'Usage: python %s <host> <port> <nick> <dbpath>' % sys.argv[0]
    raise SystemExit(1)

sys.modules['bot'] = sys.modules['__main__']
DBPATH = sys.argv[4]
import botimpl # requires certain APIs

LINEPARSE = re.compile("^(:(?P<prefix>[^ ]+) +)?(?P<command>[^ ]+)(?P<param>( +[^:][^ ]*)*)(?: +:(?P<message>.*))?$")

s = socket.create_connection((sys.argv[1], sys.argv[2]))
NICK = sys.argv[3]

def send(l):
    s.send('%s\r\n' % l.replace('\r','').replace('\n','').replace('\0',''))
    print '>>', l

def halt(msg='그럼 이만!'):
    send('QUIT :%s' % msg);
    s.close()
    raise SystemExit
signal.signal(signal.SIGINT, lambda sig, frame: halt())

def say(to, msg):
    send('PRIVMSG %s :%s' % (to, msg))

class ExecutionTimedOut(Exception): pass

def sayerr(to):
    if to:
        ty, exc, tb = sys.exc_info()
        if ty != ExecutionTimedOut:
            say(to, '\00304!ERROR! %s (%s)' % (ty, exc))
    traceback.print_exception(ty, exc, tb)

def safeexec(to, f, args=(), kwargs={}):
    def alarm(sig, frame):
        #for i in dir(frame):
        #    if i.startswith('f_'): print i, repr(getattr(frame,i))[:120]
        raise ExecutionTimedOut('execution timed out')
    try:
        try:
            signal.signal(signal.SIGALRM, alarm)
            signal.alarm(botimpl.TIMEOUT)
            f(*args, **kwargs)
        except Exception:
            sayerr(to)
        finally:
            signal.signal(signal.SIGALRM, signal.SIG_DFL)
            signal.alarm(0)
    except ExecutionTimedOut:
        signal.signal(signal.SIGALRM, signal.SIG_DFL)
        signal.alarm(0)

send('USER himawari himawari ruree.net :Furutani Himawari')
send('NICK %s' % NICK)
nexttime = time.time() + botimpl.TICK
while True:
    line = ''
    while not line.endswith('\r\n'):
        ch = s.recv(1)
        if ch == '': break
        line += ch
    if not line:
        print '*** connection failed'
        break
    line = line.rstrip('\r\n')
    print '<<', line
    m = LINEPARSE.match(line)
    if m:
        prefix = m.group('prefix') or ''
        command = m.group('command').lower()
        params = (m.group('param') or '').split()
        if m.group('message') is not None:
            params.append(m.group('message'))

        if command == '001': # welcome
            safeexec(None, getattr(botimpl, 'start', None), ())
        elif command == 'ping':
            if len(params) < 1: continue
            send('PONG :%s' % params[0])
        elif command == 'invite':
            if len(params) < 2: continue
            # params[0] should be NICK
            send('JOIN %s' % params[1])
            safeexec(None, getattr(botimpl, 'welcome', None), (params[1], prefix))
        elif command == 'privmsg':
            if len(params) < 2 or not params[0].startswith('#'): continue
            if ''.join(params[1].split()).lower() in ('%s,reload' % NICK, '%s:reload' % NICK):
                safeexec(params[0], reload, (botimpl,))
                say(params[0], '재기동했습니다.')
                # safeguard
                if not isinstance(getattr(botimpl, 'TICK', None), int):
                    botimpl.TICK = 10
                if not isinstance(getattr(botimpl, 'TIMEOUT', None), int):
                    botimpl.TIMEOUT = 5
            else:
                safeexec(params[0], getattr(botimpl, 'msg', None), (params[0], prefix, params[1]))
        elif command == 'join':
            if len(params) < 1: continue
            safeexec(params[0], getattr(botimpl, 'onenter', None), (params[0], prefix))
        elif command == 'part' or command == 'quit' or command == 'kill':
            if len(params) < 1: continue
            safeexec(params[0], getattr(botimpl, 'onexit', None), (params[0], prefix, command, prefix) + tuple(params[1:2]))
        elif command == 'kick':
            if len(params) < 2: continue
            safeexec(params[0], getattr(botimpl, 'onexit', None), (params[0], prefix, 'kick', params[1]) + tuple(params[2:3]))
        elif command == 'nick':
            if len(params) < 1: continue
            safeexec(None, getattr(botimpl, 'onnickchange', None), (prefix, params[0]))
    while not select.select([s.fileno()], [], [], max(0, nexttime - time.time()))[0]:
        if nexttime < time.time(): nexttime = time.time() + botimpl.TICK
        safeexec(None, getattr(botimpl, 'idle', None))