irc.py

import logging
import socket

import pydle

import config

BotBase = pydle.featurize(pydle.features.RFC1459Support, pydle.features.TLSSupport)

logger = logging.getLogger("IRC")
logger.setLevel(logging.DEBUG)

cfg = config.init()


class IRC(BotBase):
    tracking = None
    RECONNECT_MAX_ATTEMPTS = 100
    announcer = None

    # temp fix until pydle handles connect failures
    def connect(self, *args, **kwargs):
        try:
            super().connect(*args, **kwargs)
        except socket.error:
            self.on_disconnect(expected=False)

    def set_tracker(self, track):
        self.tracking = track

    def on_connect(self):
        logger.info("Connecting to: %s, joining %s", self.tracking['irc_host'], self.tracking['irc_channel'])

        nick_pass = cfg["{}.nick_pass".format(self.tracking['name'].lower())]
        torrent_pass = cfg["{}.torrent_pass".format(self.tracking['name'].lower())]
        announcer = cfg["{}.announcer".format(self.tracking['name'].lower())]


        #only assuming authstring if announcer, this was for PTP need to look into making this more dynamic
        if announcer is not None and len(announcer) > 1:
            announcer = cfg["{}.announcer".format(self.tracking['name'].lower())]
            irc_channel = self.tracking['irc_channel']
            irc_key = cfg["{}.irc_key".format(self.tracking['name'].lower())]
            site_username = cfg["{}.site_username".format(self.tracking['name'].lower())]
            authstring = 'ENTER {} {} {}'.format(site_username, irc_key, irc_channel)
            self.rawmsg('NICKSERV', 'IDENTIFY', nick_pass)
            self.message(announcer, authstring)
        elif nick_pass is not None and len(nick_pass) > 1:
            self.rawmsg('NICKSERV', 'IDENTIFY', nick_pass)
            self.join(self.tracking['irc_channel'])
        else:
            self.join(self.tracking['irc_channel'])

    def on_raw(self, message):
        super().on_raw(message)

        if cfg["{}.nick".format(self.tracking['name'].lower())] in message._raw and '+r' in message._raw:
            logger.debug("Identified with NICKSERV - joining %s", self.tracking['irc_channel'])
            self.join(self.tracking['irc_channel'])

    def on_raw_900(self, message):
        logger.debug("Identified with NICKSERV - joining %s", self.tracking['irc_channel'])
        self.join(self.tracking['irc_channel'])

    def on_message(self, source, target, message):
        if source[0] != '#':
            logger.debug("%s sent us a message: %s", target, message)
        else:
            self.tracking['plugin'].parse(message)

    def on_invite(self, channel, by):
        if channel == self.tracking['irc_channel']:
            self.join(self.tracking['irc_channel'])


pool = pydle.ClientPool()
clients = []


def start(trackers):
    global cfg, pool, clients

    for tracker in trackers.loaded:
        logger.info("Pooling server: %s:%d %s", tracker['irc_host'], tracker['irc_port'], tracker['irc_channel'])

        nick = cfg["{}.nick".format(tracker['name'].lower())]
        client = IRC(nick)

        client.set_tracker(tracker)
        clients.append(client)
        try:
            pool.connect(client, hostname=tracker['irc_host'], port=tracker['irc_port'],
                         tls=tracker['irc_tls'], tls_verify=tracker['irc_tls_verify'])
        except Exception as ex:
            logger.exception("Error while connecting to: %s", tracker['irc_host'])

    try:
        pool.handle_forever()
    except Exception as ex:
        logger.exception("Exception pool.handle_forever:")


def stop():
    global pool

    for tracker in clients:
        logger.debug("Removing tracker: %s", tracker.tracking['name'])
        pool.disconnect(tracker)