pluto/handler/irc.py

# Twisted/Klein imports
from twisted.logger import Logger
from twisted.words.protocols import irc
from twisted.internet import protocol, reactor, ssl

# Other library imports
from json import dumps

# Project imports
from settings import settings


class IRCBot(irc.IRCClient):
    def __init__(self, log):
        """
        Initialise IRC bot.
        :param log: logger instance
        :type log: Logger
        """
        self.log = log
        self.nickname = settings.IRC.Nick
        self.password = settings.IRC.Pass
        self.realname = self.nickname
        self.username = self.nickname

        # Don't give away information about our client
        self.userinfo = None
        self.fingerReply = None
        self.versionName = None
        self.sourceURL = None
        self.lineRate = None  # Don't throttle messages, we may need to send a lot

        self.prefix = settings.IRC.Prefix
        self.admins = (settings.IRC.Admins).split("\n")
        self.highlight = (settings.IRC.Highlight).split("\n")

        self.channel = settings.IRC.Channel

    def set_agora(self, agora):
        self.agora = agora

    def set_revolut(self, revolut):
        self.revolut = revolut

    def parse(self, user, host, channel, msg):
        """
        Simple handler for IRC commands.
        :param user: full user string with host
        :param host: user's hostname
        :param channel: channel the message was received on
        :param msg: the message
        :type user: string
        :type host: string
        :type channel: string
        :type msg: string
        """
        spl = msg.split()
        # nick = user.split("!")[0]

        cmd = spl[0]

        if cmd == "trades" and host in self.admins:
            # Get details of open trades and post on IRC
            trades = self.agora.dashboard(send_irc=False)
            if not trades:
                self.msg(channel, "No open trades.")
            for trade_id in trades:
                self.msg(channel, trade_id)

        elif cmd == "create" and host in self.admins and len(spl) == 3:
            # Post an ad on AgoraDesk with the given country and currency code
            posted = self.agora.create_ad(spl[1], spl[2])
            if posted["success"]:
                self.msg(channel, f"{posted['response']['data']['message']}: {posted['response']['data']['ad_id']}")
            else:
                self.msg(channel, posted["response"]["data"]["message"])

        elif cmd == "messages" and host in self.admins and len(spl) == 1:
            # Get all messages for all open trades
            messages = self.agora.get_all_messages()
            if not messages:
                self.msg(channel, "No messages.")
            for message_id in messages:
                for message in messages[message_id]:
                    self.msg(channel, f"{message_id}: {message}")
                self.msg(channel, "---")
            # self.msg(channel, dumps(messages))

        elif cmd == "messages" and host in self.admins and len(spl) == 2:
            # Get all messages for a given trade
            messages = self.agora.get_messages(spl[1], send_irc=False)
            if not messages:
                self.msg(channel, "No messages.")
            for message in messages:
                self.msg(channel, f"{spl[1]}: {message}")

        elif cmd == "dist" and host in self.admins:
            # Distribute out our ad to all countries in the config
            rtrn = self.agora.dist_countries()
            self.msg(channel, dumps(rtrn))

        elif cmd == "find" and host in self.admins and len(spl) == 3:
            # Find a transaction received by Revolut with the given reference and amount
            try:
                int(spl[2])
            except ValueError:
                self.msg(channel, "Amount is not an integer.")
            rtrn = self.tx.find_tx(spl[1], spl[2])
            if rtrn == "AMOUNT_INVALID":
                self.msg(channel, "Reference found but amount invalid.")
            elif not rtrn:
                self.msg(channel, "Reference not found.")
            else:
                return dumps(rtrn)

        elif cmd == "accounts" and host in self.admins:
            accounts = self.revolut.accounts()
            for account in accounts:
                if account["balance"] > 0:
                    self.msg(channel, f"{account['name']} {account['currency']}: {account['balance']}")

        elif cmd == "total" and host in self.admins:
            total_usd = self.revolut.get_total_usd()
            if not total_usd:
                self.msg(channel, "Error getting total balance.")
            self.msg(channel, f"Total: {round(total_usd, 2)}USD")

    def signedOn(self):
        """
        Called when we have signed on to IRC.
        Join our channel.
        """
        self.log.info("Signed on as %s" % (self.nickname))
        self.join(self.channel)

    def joined(self, channel):
        """
        Called when we have joined a channel.
        Setup the Agora LoopingCall to get trades.
        This is here to ensure the IRC client is initialised enough to send the trades.
        :param channel: channel we joined
        :type channel: string
        """
        self.agora.setup_loop()
        self.log.info("Joined channel %s" % (channel))

    def privmsg(self, user, channel, msg):
        """
        Called on received PRIVMSGs.
        Pass through identified commands to the parse function.
        :param user: full user string with host
        :param channel: channel the message was received on
        :param msg: the message
        :type user: string
        :type channel: string
        :type msg: string
        """
        nick = user.split("!")[0]

        if channel == self.nickname:
            channel = nick

        host = user.split("!")[1]
        host = host.split("@")[1]

        ident = user.split("!")[1]
        ident = ident.split("@")[0]
        self.log.info("(%s) %s: %s" % (channel, user, msg))
        if msg[0] == self.prefix:
            if len(msg) > 1:
                if msg.split()[0] != "!":
                    self.parse(user, host, channel, msg[1:])

    def noticed(self, user, channel, msg):
        """
        Called on received NOTICEs.
        :param user: full user string with host
        :param channel: channel the notice was received on
        :param msg: the message
        :type user: string
        :type channel: string
        :type msg: string
        """
        nick = user.split("!")[0]
        if channel == self.nickname:
            channel = nick
        # self.log.info("[%s] %s: %s" % (channel, user, msg))


class IRCBotFactory(protocol.ClientFactory):
    def __init__(self):
        self.log = Logger("irc")

    def set_agora(self, agora):
        self.agora = agora

    def set_revolut(self, revolut):
        self.revolut = revolut

    def buildProtocol(self, addr):
        """
        Custom override for the Twisted buildProtocol so we can access the Protocol instance.
        Passes through the Agora instance to IRC.
        :return: IRCBot Protocol instance
        """
        prcol = IRCBot(self.log)
        self.client = prcol
        self.client.set_agora(self.agora)
        self.client.set_revolut(self.revolut)
        return prcol

    def clientConnectionLost(self, connector, reason):
        """
        Called when connection to IRC server lost. Reconnect.
        :param connector: connector object
        :param reason: reason connection lost
        :type connector: object
        :type reason: string
        """
        self.log.error("Lost connection: {reason}, reconnecting", reason=reason)
        connector.connect()

    def clientConnectionFailed(self, connector, reason):
        """
        Called when connection to IRC server failed. Reconnect.
        :param connector: connector object
        :param reason: reason connection failed
        :type connector: object
        :type reason: string
        """
        self.log.error("Could not connect: {reason}", reason=reason)
        connector.connect()


def bot():
    """
    Load the certificates, start the Bot Factory and connect it to the IRC server.
    :return: Factory instance
    :rtype: Factory
    """
    # Load the certificates
    context = ssl.DefaultOpenSSLContextFactory(settings.IRC.Cert, settings.IRC.Cert)
    # Define the factory instance
    factory = IRCBotFactory()
    reactor.connectSSL(settings.IRC.Host, int(settings.IRC.Port), factory, context)
    return factory