#!/usr/bin/python
import re

class Autoexec(object):
	def __init__(self):
		self.networks={}
	def onModuleAdd(self, IRC, label, onconnect=None, onregister=None, autojoin=None, usermodes=None, wallet=None, opername=None, opermodes=None, snomasks=None, operexec=None, operjoin=None):
		labels=[v[0] for v in self.networks.values()]
		if label in labels:
			raise BaseException, "Label already exists"
		if IRC in self.networks.keys():
			raise BaseException, "Network already exists"
		self.networks[IRC]=(label, onconnect, onregister, autojoin, usermodes, wallet, opername, opermodes, snomasks, operexec, operjoin)
	def onModuleRem(self, IRC):
		del self.networks[IRC]
	def onConnect(self, IRC):
		(label, onconnect, onregister, autojoin, usermodes, wallet, opername, opermodes, snomasks, operexec, operjoin)=self.networks[IRC]
		if onconnect:
			for line in onconnect:
				IRC.raw(line, origin=self)
	def onRegistered(self, IRC):
		(label, onconnect, onregister, autojoin, usermodes, wallet, opername, opermodes, snomasks, operexec, operjoin)=self.networks[IRC]
		if onregister:
			for line in onregister:
				IRC.raw(line, origin=self)
		if usermodes:
			IRC.raw("MODE %s %s"%(IRC.identity.nick, usermodes), origin=self)
		if opername and wallet and "%s/opers/%s"%(label, opername) in wallet.keys():
			IRC.raw("OPER %s %s"%(opername, wallet["%s/opers/%s"%(label, opername)]), origin=self)
		if autojoin:
			IRC.raw("JOIN %s"%(",".join(autojoin)), origin=self)
	def onRecv(self, IRC, line, data):
		if data==None:
			return
		(label, onconnect, onregister, autojoin, usermodes, wallet, opername, opermodes, snomasks, operexec, operjoin)=self.networks[IRC]
		(origin, ident, host, cmd, target, params, extinfo)=data
		if cmd=="381" and opermodes:
			if operexec:
				for line in operexec:
					IRC.raw(line, origin=self)
			if opermodes:
				IRC.raw("MODE %s %s"%(IRC.identity.nick, opermodes), origin=self)
			if snomasks:
				IRC.raw("MODE %s +s %s"%(IRC.identity.nick, snomasks), origin=self)
			if operjoin:
				IRC.raw("JOIN %s"%(",".join(operjoin)), origin=self)

class NickServ(object):
	def __init__(self):
		self.networks={}
	def onModuleAdd(self, IRC, label, wallet=None, autojoin=None):
		labels=[v[0] for v in self.networks.values()]
		#print labels
		if label in labels:
			raise BaseException, "Label already exists"
		if IRC in self.networks.keys():
			raise BaseException, "Network already exists"
		self.networks[IRC]=(label, wallet, autojoin)
	def onModuleRem(self, IRC):
		del self.networks[IRC]
	def onRecv(self, IRC, line, data):
		if data==None: return
		(origin, ident, host, cmd, target, params, extinfo)=data
		label, wallet, autojoin=self.networks[IRC]
		if target==IRC.identity.nick and origin=="NickServ" and re.match("This nickname is registered and protected.", extinfo) and wallet and "%s/NickServ/%s"%(label, target.lower()) in wallet.keys():
			IRC.user("NickServ").msg("identify %s" % wallet["%s/NickServ/%s"%(label, target.lower())])
		if cmd=="900" and autojoin:
			IRC.raw("JOIN %s"%(",".join(autojoin)), origin=self)