#! /usr/bin/env python # # //Reaktor/IRC/asybot.py # import logging import logging.handlers from asynchat import async_chat as asychat from datetime import datetime as date from datetime import timedelta from re import search, split from signal import SIGALRM, alarm, signal from socket import AF_INET, SOCK_STREAM from textwrap import TextWrapper from time import sleep # s/\x1B\[([0-9]{1,2}(;[0-9]{1,2})?)?[m|K]//g -- removes color codes class asybot(asychat): def __init__( self, server, port, nickname, channels, realname=False, username=False, hostname=False, hammer_interval=10, alarm_timeout=300, kill_timeout=360, loglevel=logging.ERROR, nickserv_password=None ): asychat.__init__(self) # logger magic self.log = logging.getLogger('asybot_' + nickname) self.nickname = nickname if realname: self.realname = realname else: self.realname = nickname if username: self.username = username else: self.username = nickname if hostname: self.hostname = hostname else: self.hostname = nickname self.retry = True self.server = server self.port = port self.channels = channels self.data = '' self.myterminator = '\r\n' self.set_terminator(self.myterminator.encode()) self.create_socket(AF_INET, SOCK_STREAM) self.connect((self.server, self.port)) self.wrapper = TextWrapper(subsequent_indent=" ", width=400) self.nickserv_password = nickserv_password self.log.info('=> irc://%s@%s:%s/%s' % (self.nickname, self.server, self.port, self.channels)) # When we don't receive data for alarm_timeout seconds then issue a # PING every hammer_interval seconds until kill_timeout seconds have # passed without a message. Any incoming message will reset alarm. self.alarm_timeout = alarm_timeout self.hammer_interval = hammer_interval self.kill_timeout = kill_timeout try: signal(SIGALRM, lambda signum, frame: self.alarm_handler()) except Exception as e: print('asybot: ' + str(e)) self.reset_alarm() def reset_alarm(self): self.last_activity = date.now() alarm(self.alarm_timeout) def alarm_handler(self): delta = date.now() - self.last_activity if delta > timedelta(seconds=self.kill_timeout): self.log.error('No data for %s. Giving up...' % delta) self.handle_disconnect() else: self.log.error('No data for %s. PINGing server...' % delta) self.push('PING :%s' % self.nickname) alarm(self.hammer_interval) def collect_incoming_data(self, data): try: self.data += data.decode() except Exception as e: print('error decoding message: ' + str(e)) print('current data: %s' % self.data) print('received data: %s' % data) print('trying to decode as latin1') self.data += data.decode('latin1') def found_terminator(self): self.log.debug('<< %s' % self.data) message = self.data self.data = '' try: _, prefix, command, params, rest, _ = \ split( '^(?::(\S+)\s)?(\S+)((?:\s[^:]\S*)*)(?:\s:(.*))?$', message, ) except Exception as e: print("cannot split message :(\nmsg: %s" % message) return params = params.split(' ')[1:] if command == 'PING': self.push('PONG :%s' % rest) self.log.debug("Replying to servers PING with PONG :%s" % rest) self.on_ping(prefix, command, params, rest) elif command == 'PRIVMSG': self.on_privmsg(prefix, command, params, rest) elif command == 'INVITE': self.on_invite(prefix, command, params, rest) elif command == 'KICK': self.on_kick(prefix, command, params, rest) elif command == 'JOIN': self.on_join(prefix, command, params, rest) elif command == '433': # ERR_NICKNAMEINUSE, retry with another name self.on_nickinuse(prefix, command, params, rest) elif command == '376' or command == '422': self.on_welcome(prefix, command, params, rest) self.reset_alarm() def push(self, message): try: self.log.debug('>> %s' % message) msg = (message + self.myterminator).encode() self.log.debug('>> %s' % msg) asychat.push(self, msg) except Exception as e: self.log.error("Exception durin push {}".format(e)) pass def disconnect(self): self.push('QUIT') self.close() def reconnect(self): if self.connected: self.push('QUIT') self.close() self.create_socket(AF_INET, SOCK_STREAM) self.connect((self.server, self.port)) def handle_connect(self): if self.nickserv_password: self.push(f'PASS {self.nickserv_password}') self.push('NICK %s' % self.nickname) self.push('USER %s %s %s :%s' % (self.username, self.hostname, self.server, self.realname)) def handle_disconnect(self): if self.retry: self.reconnect() else: self.log.error('No retry set, giving up') def PRIVMSG(self, target, text): for line in self.wrapper.wrap(text): msg = 'PRIVMSG %s :%s' % (','.join(target), line) self.log.info(msg) self.push(msg) sleep(1) def ME(self, target, text): self.PRIVMSG(target, ('ACTION ' + text + '')) def on_welcome(self, prefix, command, params, rest): self.push('JOIN %s' % ','.join(self.channels)) def on_kick(self, prefix, command, params, rest): self.log.debug(params) if params[-1] == self.nickname: for chan in params[:-1]: self.channels.remove(chan) def on_join(self, prefix, command, params, rest): pass def on_ping(self, prefix, command, params, rest): pass def on_privmsg(self, prefix, command, params, rest): pass def on_invite(self, prefix, command, params, rest): pass def on_nickinuse(self, prefix, command, params, rest): regex = search('(\d+)$', self.nickname) if regex: theint = int(regex.group(0)) self.nickname = self.nickname.strip(str(theint)) + str(theint + 1) else: self.nickname = self.nickname + '0' self.handle_connect()