102 lines
3.9 KiB
Python
102 lines
3.9 KiB
Python
|
import os, select, sys, threading, time, traceback
|
||
|
|
import EventManager, IRCServer, ModuleManager, Timer
|
||
|
|
|
||
|
|
class Bot(object):
|
||
|
|
def __init__(self):
|
||
|
|
self.lock = threading.Lock()
|
||
|
|
self.database = None
|
||
|
|
self.config = None
|
||
|
|
self.bot_directory = os.path.dirname(os.path.realpath(__file__))
|
||
|
|
self.servers = {}
|
||
|
|
self.running = True
|
||
|
|
self.poll = select.epoll()
|
||
|
|
self.modules = ModuleManager.ModuleManager(self)
|
||
|
|
self.events = EventManager.EventHook(self)
|
||
|
|
self.timers = []
|
||
|
|
self.last_ping = None
|
||
|
|
|
||
|
|
def add_server(self, id, hostname, port, password, ipv4, tls,
|
||
|
|
nickname, username, realname, connect=False):
|
||
|
|
new_server = IRCServer.Server(id, hostname, port, password,
|
||
|
|
ipv4, tls, nickname, username, realname, self)
|
||
|
|
self.servers[new_server.fileno()] = new_server
|
||
|
|
if connect:
|
||
|
|
self.connect(new_server)
|
||
|
|
def connect(self, server):
|
||
|
|
try:
|
||
|
|
server.connect()
|
||
|
|
except:
|
||
|
|
sys.stderr.write("Failed to connect to %s\n" % str(server))
|
||
|
|
traceback.print_exc()
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
def connect_all(self):
|
||
|
|
for server in self.servers.values():
|
||
|
|
if self.connect(server):
|
||
|
|
self.poll.register(server.fileno(), select.EPOLLOUT)
|
||
|
|
|
||
|
|
def add_timer(self, function, delay, *args, **kwargs):
|
||
|
|
timer = Timer.Timer(function, delay, *args, **kwargs)
|
||
|
|
timer.set_started_time()
|
||
|
|
self.timers.append(timer)
|
||
|
|
def next_timer(self):
|
||
|
|
next = None
|
||
|
|
for timer in self.timers:
|
||
|
|
time_left = timer.time_left()
|
||
|
|
if not next or time_left < next:
|
||
|
|
next = time_left
|
||
|
|
return next or 30
|
||
|
|
def call_timers(self):
|
||
|
|
for timer in self.timers[:]:
|
||
|
|
if timer.due():
|
||
|
|
timer.call()
|
||
|
|
if timer.done():
|
||
|
|
self.timers.remove(timer)
|
||
|
|
|
||
|
|
def register_read(self, server):
|
||
|
|
self.poll.modify(server.fileno(), select.EPOLLIN)
|
||
|
|
def register_write(self, server):
|
||
|
|
self.poll.modify(server.fileno(), select.EPOLLOUT)
|
||
|
|
def register_both(self, server):
|
||
|
|
self.poll.modify(server.fileno(),
|
||
|
|
select.EPOLLIN|select.EPOLLOUT)
|
||
|
|
|
||
|
|
def since_last_read(self, server):
|
||
|
|
return time.time()-server.last_read
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
while self.running:
|
||
|
|
self.lock.acquire()
|
||
|
|
events = self.poll.poll(self.next_timer())
|
||
|
|
self.call_timers()
|
||
|
|
for fd, event in events:
|
||
|
|
if fd in self.servers:
|
||
|
|
server = self.servers[fd]
|
||
|
|
if event & select.EPOLLIN:
|
||
|
|
lines = server.read()
|
||
|
|
for line in lines:
|
||
|
|
print(line)
|
||
|
|
server.parse_line(line)
|
||
|
|
elif event & select.EPOLLOUT:
|
||
|
|
server._send()
|
||
|
|
self.register_read(server)
|
||
|
|
elif event & select.EPULLHUP:
|
||
|
|
print("hangup")
|
||
|
|
if not self.last_ping or time.time()-self.last_ping >= 60:
|
||
|
|
for server in self.servers.values():
|
||
|
|
server.send_ping()
|
||
|
|
self.last_ping = time.time()
|
||
|
|
for server in list(self.servers.values()):
|
||
|
|
if server.last_read and self.since_last_read(server
|
||
|
|
) > 160:
|
||
|
|
print("pingout from %s" % str(server))
|
||
|
|
server.disconnect()
|
||
|
|
if not server.connected:
|
||
|
|
self.poll.unregister(server.fileno())
|
||
|
|
del self.servers[server.fileno()]
|
||
|
|
# add reconnect here
|
||
|
|
print("disconnected from %s" % str(server))
|
||
|
|
elif server.waiting_send():
|
||
|
|
self.register_both(server)
|
||
|
|
self.lock.release()
|