h/core/irc.py

251 lines
8.1 KiB
Python
Raw Normal View History

2009-11-13 01:59:41 +00:00
import re
import socket
import time
2009-11-13 01:59:41 +00:00
import thread
import Queue
from ssl import wrap_socket, CERT_NONE, CERT_REQUIRED, SSLError
2009-11-13 01:59:41 +00:00
2010-03-01 02:32:41 +00:00
2009-11-13 01:59:41 +00:00
def decode(txt):
for codec in ('utf-8', 'iso-8859-1', 'shift_jis', 'cp1252'):
try:
return txt.decode(codec)
except UnicodeDecodeError:
continue
return txt.decode('utf-8', 'ignore')
2010-03-01 02:32:41 +00:00
def censor(text):
replacement = '[censored]'
if 'censored_strings' in bot.config:
words = map(re.escape, bot.config['censored_strings'])
regex = re.compile('(%s)' % "|".join(words))
text = regex.sub(replacement, text)
return text
class crlf_tcp(object):
2009-11-13 01:59:41 +00:00
"Handles tcp connections that consist of utf-8 lines ending with crlf"
def __init__(self, host, port, timeout=300):
self.ibuffer = ""
2009-11-13 01:59:41 +00:00
self.obuffer = ""
2010-03-01 02:32:41 +00:00
self.oqueue = Queue.Queue() # lines to be sent out
self.iqueue = Queue.Queue() # lines that were received
self.socket = self.create_socket()
2009-11-13 01:59:41 +00:00
self.host = host
self.port = port
self.timeout = timeout
2009-11-13 01:59:41 +00:00
def create_socket(self):
return socket.socket(socket.AF_INET, socket.TCP_NODELAY)
2010-03-01 02:32:41 +00:00
2009-11-13 01:59:41 +00:00
def run(self):
self.socket.connect((self.host, self.port))
thread.start_new_thread(self.recv_loop, ())
thread.start_new_thread(self.send_loop, ())
2009-11-13 01:59:41 +00:00
def recv_from_socket(self, nbytes):
return self.socket.recv(nbytes)
2010-03-01 02:32:41 +00:00
def get_timeout_exception_type(self):
return socket.timeout
2010-03-01 02:32:41 +00:00
def handle_receive_exception(self, error, last_timestamp):
if time.time() - last_timestamp > self.timeout:
self.iqueue.put(StopIteration)
self.socket.close()
return True
return False
def recv_loop(self):
last_timestamp = time.time()
while True:
2009-11-21 22:55:52 +00:00
try:
data = self.recv_from_socket(4096)
self.ibuffer += data
if data:
last_timestamp = time.time()
else:
if time.time() - last_timestamp > self.timeout:
self.iqueue.put(StopIteration)
self.socket.close()
return
time.sleep(1)
2011-05-11 20:40:04 +00:00
except (self.get_timeout_exception_type(), socket.error) as e:
if self.handle_receive_exception(e, last_timestamp):
return
2009-11-21 22:55:52 +00:00
continue
while '\r\n' in self.ibuffer:
line, self.ibuffer = self.ibuffer.split('\r\n', 1)
self.iqueue.put(decode(line))
2009-11-13 01:59:41 +00:00
def send_loop(self):
2009-11-13 01:59:41 +00:00
while True:
line = self.oqueue.get().splitlines()[0][:500]
print ">>> %r" % line
self.obuffer += line.encode('utf-8', 'replace') + '\r\n'
while self.obuffer:
sent = self.socket.send(self.obuffer)
self.obuffer = self.obuffer[sent:]
2009-11-13 01:59:41 +00:00
2010-03-01 02:32:41 +00:00
class crlf_ssl_tcp(crlf_tcp):
"Handles ssl tcp connetions that consist of utf-8 lines ending with crlf"
def __init__(self, host, port, ignore_cert_errors, timeout=300):
self.ignore_cert_errors = ignore_cert_errors
crlf_tcp.__init__(self, host, port, timeout)
2010-03-01 02:32:41 +00:00
def create_socket(self):
2010-03-01 02:32:41 +00:00
return wrap_socket(crlf_tcp.create_socket(self), server_side=False,
cert_reqs=CERT_NONE if self.ignore_cert_errors else
CERT_REQUIRED)
def recv_from_socket(self, nbytes):
return self.socket.read(nbytes)
def get_timeout_exception_type(self):
return SSLError
2010-03-01 02:32:41 +00:00
def handle_receive_exception(self, error, last_timestamp):
# this is terrible
if not "timed out" in error.args[0]:
raise
return crlf_tcp.handle_receive_exception(self, error, last_timestamp)
2010-03-01 02:32:41 +00:00
2009-11-13 01:59:41 +00:00
irc_prefix_rem = re.compile(r'(.*?) (.*?) (.*)').match
irc_noprefix_rem = re.compile(r'()(.*?) (.*)').match
irc_netmask_rem = re.compile(r':?([^!@]*)!?([^@]*)@?(.*)').match
irc_param_ref = re.compile(r'(?:^|(?<= ))(:.*|[^ ]+)').findall
class IRC(object):
2009-11-13 01:59:41 +00:00
"handles the IRC protocol"
#see the docs/ folder for more information on the protocol
def __init__(self, server, nick, port=6667, channels=[], conf={}):
self.channels = channels
self.conf = conf
self.server = server
self.port = port
self.nick = nick
2010-03-01 02:32:41 +00:00
self.out = Queue.Queue() # responses from the server are placed here
2009-11-13 01:59:41 +00:00
# format: [rawline, prefix, command, params,
# nick, user, host, paramlist, msg]
self.connect()
thread.start_new_thread(self.parse_loop, ())
def create_connection(self):
return crlf_tcp(self.server, self.port)
2010-03-01 02:32:41 +00:00
def connect(self):
self.conn = self.create_connection()
thread.start_new_thread(self.conn.run, ())
self.set_pass(self.conf.get('server_password'))
self.set_nick(self.nick)
self.cmd("USER",
[conf.get('user', 'skybot'), "3", "*", conf.get('realname',
2010-08-24 21:59:47 +00:00
'Python bot - http://github.com/rmmh/skybot')])
2009-11-13 01:59:41 +00:00
def parse_loop(self):
while True:
msg = self.conn.iqueue.get()
if msg == StopIteration:
self.connect()
continue
2010-03-01 02:32:41 +00:00
if msg.startswith(":"): # has a prefix
2009-11-13 01:59:41 +00:00
prefix, command, params = irc_prefix_rem(msg).groups()
else:
prefix, command, params = irc_noprefix_rem(msg).groups()
nick, user, host = irc_netmask_rem(prefix).groups()
paramlist = irc_param_ref(params)
lastparam = ""
if paramlist:
if paramlist[-1].startswith(':'):
paramlist[-1] = paramlist[-1][1:]
lastparam = paramlist[-1]
2009-11-13 01:59:41 +00:00
self.out.put([msg, prefix, command, params, nick, user, host,
paramlist, lastparam])
if command == "PING":
self.cmd("PONG", paramlist)
2009-11-13 01:59:41 +00:00
def set_pass(self, password):
if password:
self.cmd("PASS", [password])
def set_nick(self, nick):
2009-11-13 01:59:41 +00:00
self.cmd("NICK", [nick])
def join(self, channel):
self.cmd("JOIN", [channel])
2009-11-13 01:59:41 +00:00
def msg(self, target, text):
self.cmd("PRIVMSG", [target, text])
2009-11-13 01:59:41 +00:00
def cmd(self, command, params=None):
if params:
params[-1] = ':' + params[-1]
self.send(command + ' ' + ' '.join(map(censor, params)))
2009-11-13 01:59:41 +00:00
else:
self.send(command)
def send(self, str):
self.conn.oqueue.put(str)
2010-03-01 02:32:41 +00:00
class FakeIRC(IRC):
def __init__(self, server, nick, port=6667, channels=[], conf={}, fn=""):
self.channels = channels
self.conf = conf
self.server = server
self.port = port
self.nick = nick
2010-03-01 02:32:41 +00:00
self.out = Queue.Queue() # responses from the server are placed here
self.f = open(fn, 'rb')
thread.start_new_thread(self.parse_loop, ())
def parse_loop(self):
while True:
msg = decode(self.f.readline()[9:])
if msg == '':
print "!!!!DONE READING FILE!!!!"
return
2010-03-01 02:32:41 +00:00
if msg.startswith(":"): # has a prefix
prefix, command, params = irc_prefix_rem(msg).groups()
else:
prefix, command, params = irc_noprefix_rem(msg).groups()
nick, user, host = irc_netmask_rem(prefix).groups()
paramlist = irc_param_ref(params)
lastparam = ""
if paramlist:
if paramlist[-1].startswith(':'):
paramlist[-1] = paramlist[-1][1:]
lastparam = paramlist[-1]
self.out.put([msg, prefix, command, params, nick, user, host,
paramlist, lastparam])
if command == "PING":
self.cmd("PONG", [params])
def cmd(self, command, params=None):
pass
2010-03-01 02:32:41 +00:00
class SSLIRC(IRC):
2010-03-01 02:32:41 +00:00
def __init__(self, server, nick, port=6667, channels=[], conf={},
ignore_certificate_errors=True):
2010-02-22 23:49:30 +00:00
self.ignore_cert_errors = ignore_certificate_errors
IRC.__init__(self, server, nick, port, channels, conf)
2010-03-01 02:32:41 +00:00
def create_connection(self):
return crlf_ssl_tcp(self.server, self.port, self.ignore_cert_errors)