A much needed inital import.

1 files changed, 122 insertions, 0 deletions

diff --git a/fot.py b/fot.py
new file mode 100755
index 0000000..8975979
--- /dev/null
+++ b/fot.py
@@ -0,0 +1,122 @@
+#!/usr/bin/env python
+
+from twisted.words.protocols import irc
+from twisted.internet import reactor, protocol
+from twisted.python import log
+from twisted.protocols.basic import LineReceiver
+from twisted.manhole.telnet import Shell
+
+import sys, os, traceback, StringIO, md5
+
+from ConfigParser import ConfigParser
+
+import login
+
+config = ConfigParser()
+
+def load_config():
+	print 'Loading config...'
+	config.read([os.path.expanduser('~/.fot')])
+	print 'Done!'
+
+load_config()
+
+modules = {}
+
+def load_modules():
+	print 'Loading modules...'
+	for m in config.options('modules'):
+		if config.getboolean('modules', m):
+			try:
+				if modules.has_key(m):
+					reload(modules[m])
+				else:
+					modules[m] = __import__('modules.' + m, globals(), locals(), ['Module'])
+				mod = modules[m]
+				print '%s version %s by %s (%s)' % (mod.info['title'], mod.info['version'], mod.info['author'], mod.info['license'])
+				del mod
+			except:
+				print 'Failed to load', m
+				raise
+		else:
+			print 'Not loading', m
+	print 'Done!'
+
+load_modules()
+
+bots = []
+
+def refresh_modules():
+	print 'Re-applying modules to bots...'
+	for bot in bots:
+		bot.apply_modules()
+	print 'Done!'
+
+class Bot(irc.IRCClient):
+	def __init__(self):
+		bots.append(self)
+
+	def __del__(self):
+		bots.remove(self)
+
+	def __repr__(self):
+		return '<Bot %s@%s>' % (self.nickname, self.factory.server)
+
+	def apply_modules(self):
+		self.modules = {}
+		for m in modules.keys():
+			if config.has_option(self.factory.server, m) and config.get(self.factory.server, m).strip():
+				self.modules[m] = modules[m].Module(self)
+
+	def connectionMade(self):
+		self.apply_modules()
+		self.nickname = config.get(self.factory.server, 'nickname')
+		irc.IRCClient.connectionMade(self)
+
+	def signedOn(self):
+		for chan in config.get(self.factory.server, 'channels').split(' '):
+			self.join(chan)
+
+	def privmsg(self, nick, channel, msg):
+		for m in self.modules.keys():
+			if (config.has_option(self.factory.server, m) and channel in config.get(self.factory.server, m)) or (channel == self.nickname and nick.split('!')[0] != self.nickname):
+				self.modules[m](nick, channel, msg)
+
+	def kickedFrom(self, channel, kicker, message):
+		self.join(channel)
+
+class BotFactory(protocol.ReconnectingClientFactory):
+	protocol = Bot
+
+	def __init__(self, server, nickname):
+		self.server = server
+		self.nickname = nickname
+
+	def buildProtocol(self, addr):
+		self.resetDelay()
+		return protocol.ReconnectingClientFactory.buildProtocol(self, addr)
+
+	def clientConnectionFailed(self, connector, reason):
+		protocol.ReconnectingClientFactory.clientConnectionFailed(self, connector, reason)
+
+	def clientConnectionLost(self, connector, reason):
+		protocol.ReconnectingClientFactory.clientConnectionLost(self, connector, reason)
+
+print 'Starting per-network instances...'
+for server in config.sections():
+	if not config.has_option(server,'host') or not config.has_option(server, 'port') or not config.has_option(server, 'channels') or config.has_option(server, 'disabled'):
+		continue
+	channels = []
+	ms = [(m, config.get(server, m)) for m in modules.keys() if config.has_option(server, m)]
+	for c in config.get(server, 'channels').split(' '):
+		ch = [x[0] for x in ms if c in x[1]]
+		channels.append('%s (%s)' % (c, ' '.join(ch)))
+	print '%s: %s' % (server, ' '.join(channels))
+	del channels, ms, c, ch, x
+	factory = BotFactory(server, config.get(server, 'nickname'))
+	reactor.connectTCP(config.get(server, 'host'), config.getint(server, 'port'), factory)
+
+loginfactory = login.getManholeFactory(globals(), user = 'pass')
+reactor.listenTCP(3333, loginfactory)
+
+reactor.run()