oonbotti2/ircbot.py

#!/usr/bin/python
import threading
import time
import socket
import sys

import botcmd

class Channel:
	def __init__(self):
		self.lock = threading.Lock()
		self.msg = []
	
	def send(self, msg):
		self.lock.acquire()
		self.msg.append(msg)
		self.lock.release()
	
	def recv(self, wait = True):
		while True:
			self.lock.acquire()
			
			if len(self.msg) > 0:
				msg = self.msg.pop(0)
				self.lock.release()
				return msg
			
			if not wait:
				self.lock.release()
				return None
			
			self.lock.release()
			time.sleep(0.1)

class Irc:
	def __init__(self, chan, nick, inpc):
		self.chan = chan
		self.nick = nick
		self.inpc = inpc
	
	def send(self, msg):
		self.inpc.send(msg)
	
	def recv(self, wait=True):
		return self.inpc.recv(wait)
	
	def msg(self, chan, msg):
		self.inpc.send('PRIVMSG %s :%s' % (chan, msg))

class Connhandler(threading.Thread):
	def __init__(self, server, port, chan, nick, botname, inpc, logc):
		threading.Thread.__init__(self)
		
		self.server = server
		self.port = port
		self.nick = nick
		self.name = botname
		self.chan = chan
		self.inpc = inpc
		self.logc = logc
	
	def send(self, s):
		s = s.replace('\n', '\\n').replace('\r', '\\r') # Sanitize output
		if len(s) > 512:
			s = s[:512]
		
		self.sock.send(s + '\r\n')
		if s.split(' ')[0] != 'PONG':
			self.logc.send(s + '\n')
	
	def check(self, line):
		args = line.split(' ')
		if args[0] == 'PING':
			self.send('PONG :hjdicks')
		else:
			self.logc.send(line + '\n')
			Threadwrapper(botcmd.parse, (line, Irc(self.chan, self.nick, self.inpc))).start()
	def run(self):
		self.sock = None
		for af, socktype, proto, canonname, sa in socket.getaddrinfo(self.server, self.port, socket.AF_UNSPEC, socket.SOCK_STREAM):
			try:
				self.sock = socket.socket(af, socktype, proto)
			except socket.error:
				self.sock = None
				conntinue
			try:
				self.sock.connect(sa)
			except socket.error:
				self.sock.close()
				self.sock = None
				continue
			break
		
		if self.sock is None:
			self.logc.send('QUIT');
			sys.exit(1);
		
		self.sock.settimeout(0.1)
		
		self.send('NICK %s' % self.nick)
		self.send('USER %s a a :%s' % (self.nick, self.name))
		
		f = open('startcmd.txt', 'r')
		for i in f:
			if i[-1] == '\n':
				i = i[:-1]
			self.send(i)
		f.close()
		
		time.sleep(20) # Wait for nickserv auth to take effect
		for i in self.chan.split(' '):
			self.send('JOIN %s' % i)
		
		buf = ''
		while True:
			while True:
				try:
					data = self.sock.recv(4096)
					break
				except:
					pass
				
				cmd = self.inpc.recv(wait = False)
				if cmd == 'QUIT':
					data = None
					self.logc.send('QUIT')
					break
				elif cmd:
					self.send(cmd)
				
				time.sleep(0.1)
			
			if not data:
				break
			
			buf += data
			buf = buf.split('\n')
			for line in buf[:-1]:
				if line[-1] == '\r':
					line = line[:-1]
				self.check(line)
			buf = buf[-1]
		
		self.sock.close()

class Keyhandler(threading.Thread):
	def __init__(self, outc):
		self.outc = outc
		threading.Thread.__init__(self)
	
	def run(self):
		while True:
			line = raw_input()
			if line == '':
				continue
			
			c = line.split(' ')
			if c[0] in botcmd.concmd:
				botcmd.execcmd(c)
			if c[0] == '/j' and len(c) == 2:
				self.outc.send('JOIN ' + c[1])
			elif c[0] == '/m' and len(c) > 2:
				self.outc.send('PRIVMSG %s :%s' % (c[1], ' '.join(c[2:])))
			elif c[0] == '/q':
				if len(c) > 1:
					self.outc.send('QUIT :%s' % ' '.join(c[1:]))
				self.outc.send('QUIT')
				break
			elif c[0][0] == '/' and c[0] not in botcmd.concmd:
				self.outc.send(c[0][1:].upper() + ' ' + ' '.join(c[1:]))

class Loghandler(threading.Thread):
	def __init__(self, inpc):
		self.inpc = inpc
		threading.Thread.__init__(self)
	
	def run(self):
		while True:
			s = self.inpc.recv()
			if s == 'QUIT':
				break
			sys.stdout.write(''.join([i if ord(i)>=32 or i=='\n' else '^'+chr(ord(i)+64) for i in s]))

class Threadwrapper(threading.Thread):
	def __init__(self, func, arg):
		self.func = func
		self.arg = arg
		threading.Thread.__init__(self)
	
	def run(self):
		self.func(self.arg)

if len(sys.argv)!=5:
	print 'Usage: ' + sys.argv[0] + ' server port channels nick'
else:
	botcmd.init()
	keych = Channel()
	logch = Channel()
	Keyhandler(keych).start()
	Loghandler(logch).start()
	Connhandler(sys.argv[1], int(sys.argv[2]), sys.argv[3], sys.argv[4], sys.argv[4], keych, logch).start()