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):
		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(' ')
		nick=args[0].split('!')[0][1:]
		chan=args[2] if len(args) >= 3 and args[2][0]=='#' else nick
		if args[0]=='PING':
			self.send('PONG :hjdicks')
		elif args[1]=='INVITE' and args[2]==self.nick and args[3][1:] in self.chan.split(' '):
			botcmd.addauthcmd(nick, 'JOIN '+args[3])
			self.send('PRIVMSG NickServ :ACC '+nick)
		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()
		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()
			c=line.split(' ')
			if c == '':
				continue
			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' and len(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 channel nick'
else:
	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()