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.events.on("timer").on("reconnect").hook(self.reconnect) self.events.on("boot").on("done").hook(self.setup_timers) 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) if not new_server.get_setting("connect", True): return self.events.on("new").on("server").call(server=new_server) self.servers[new_server.fileno()] = new_server if connect and new_server.get_setting("connect", True): self.connect(new_server) return 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 self.poll.register(server.fileno(), select.EPOLLOUT) return True def connect_all(self): for server in self.servers.values(): if server.get_setting("connect", True): self.connect(server) def setup_timers(self, event): for setting, value in self.find_settings("timer-%"): id = setting.split("timer-", 1)[1] self.add_timer(value["event-name"], value["delay"], value[ "next-due"], id, **value["kwargs"]) def timer_setting(self, timer): self.set_setting("timer-%s" % timer.id, { "event-name": timer.event_name, "delay": timer.delay, "next-due": timer.next_due, "kwargs": timer.kwargs}) def timer_setting_remove(self, timer): self.timers.remove(timer) self.del_setting("timer-%s" % timer.id) def add_timer(self, event_name, delay, next_due=None, id=None, persist=True, **kwargs): timer = Timer.Timer(self, event_name, delay, next_due, **kwargs) if id: timer.id = id else: self.timer_setting(timer) 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.timer_setting_remove(timer) else: self.timer_setting(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 None if not server.last_read else time.time( )-server.last_read def disconnect(self, server): self.poll.unregister(server.fileno()) del self.servers[server.fileno()] def reconnect(self, event): server_details = self.database.get_server(event["server_id"]) server = self.add_server(*server_details, False) if self.connect(server): self.servers[server.fileno()] = server else: event["timer"].redo() def set_setting(self, setting, value): self.database.set_bot_setting(setting, value) def get_setting(self, setting, default=None): return self.database.get_bot_setting(setting, default) def find_settings(self, pattern, default=[]): return self.database.find_bot_settings(pattern, default) def del_setting(self, setting): self.database.del_bot_setting(setting) 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") server.disconnect() for server in list(self.servers.values()): since_last_read = self.since_last_read(server) if since_last_read: if since_last_read > 120: print("pingout from %s" % str(server)) server.disconnect() elif since_last_read > 30 and not server.ping_sent: server.send_ping() server.ping_sent = True if not server.connected: self.disconnect(server) reconnect_delay = self.config.get("reconnect-delay", 10) self.add_timer("reconnect", reconnect_delay, None, None, False, server_id=server.id) print("disconnected from %s, reconnecting in %d seconds" % ( str(server), reconnect_delay)) elif server.waiting_send(): self.register_both(server) self.lock.release()