summaryrefslogtreecommitdiff
path: root/fot.py
blob: ae6d56439cd034f9895d9f70b84e744b51a4d210 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#!/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 by %s' % (mod.info['title'], mod.info['author'])
				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):
		print 'Connection failed: ', reason
		protocol.ReconnectingClientFactory.clientConnectionFailed(self, connector, reason)

	def clientConnectionLost(self, connector, reason):
		print 'Connection lost: ', 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)

# TODO: Fix user/pass
loginfactory = login.getManholeFactory(globals(), user = 'pass')
reactor.listenTCP(3333, loginfactory)

reactor.run()