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()
|
