pyrot/irc/bot.py

#! /usr/bin/env python

import sys, random, string, ssl
import irc.bot#, irc.strings
from irc.client import ip_numstr_to_quad#, ip_quad_to_numstr
from jaraco.stream import buffer
from postgres import Postgres
import commands.public, commands.admin, commands.games, commands.statistics
import events.on_welcome, events.on_join, events.on_kick, events.on_mode, events.on_pubmsg, events.on_action, events.on_whoreply, events.on_nick
from common import log
from common.networkservices import NickServ
from events.common import Lastact, MessageStatistics, Inform

bold = "\x02"
italic = "\x1D"
underline = "\x1F"
reverse = "\x16" 	# swap background and foreground colors ("reverse video")
reset = "\x0F"
blue = "\x0302"
green = "\x0303"
red = "\x0304"
grey = "\x0314"

class PyRot(irc.bot.SingleServerIRCBot):
    def __init__(self, network, db, homechannel, nickname, username, password, host, port=6667, usessl=False, cmdchar="!", helpchar="@"):
        self.network = network
        self.db = db
        self.homechannel = homechannel
        self.password = password
        self.cmdchar = cmdchar
        self.helpchar = helpchar
        self.protectees = {}
        self.channelkeys = {}
        
        log.info("Starting pyRot, the third RotBot by tBkwtWS.")
        log.info("Connecting to " + host + ":" + str(port) + "/" + self.homechannel)
        
        if usessl:
            factory = irc.connection.Factory(wrapper=ssl.wrap_socket)
        else:
            factory = irc.connection.Factory()
        
        irc.client.ServerConnection.buffer_class = buffer.LenientDecodingLineBuffer
        try:
            irc.bot.SingleServerIRCBot.__init__(self, [(host, port)], nickname, username, connect_factory=factory)
        except irc.client.ServerConnectionError:
            sys.stderr.write(sys.exc_info()[1])
    
    # Events.
    def on_nicknameinuse(self, connection, event):
        log.info("Nickname in use, attempting to recover: " + connection.nickname)
        connection.nick(connection.nickname + ''.join(random.choice(string.digits) for _ in range(3)))    # Take temporary nick. Without this recovering via NickServ won't work.
        NickServ.recover_nick(connection, self.password)
    
    def on_welcome(self, connection, event):
        events.on_welcome.process_event(self, connection, event)
    
    def on_error(self, connection, event):
        log.notice(str(event))
        connection.privmsg(self.homechannel, "ERROR: " + str(event))
        Inform.owners(self, connection, "Received error from server: " + str(event))
    
    def on_nick(self, connection, event):
        events.on_nick.process_event(self, connection, event)
    
    def on_join(self, connection, event):
        events.on_join.process_event(self, connection, event)
    
    def on_kick(self, connection, event):
        events.on_kick.process_event(self, connection, event)
    
    def on_mode(self, connection, event):
        events.on_mode.process_event(self, connection, event)
    
    def on_part(self, connection, event):
        log.info(event)
        
        # Update protectees.
        if event.target == self.homechannel and event.source.nick in self.protectees:    # Protectee parted home channel.
            del self.protectees[event.source.nick] # Delete from protectees.
        
        # Update last act.
        if event.arguments:
            Lastact.update(self, event.source.nick, "part", channel=event.target, lastact=event.arguments[0])
        else:
            Lastact.update(self, event.source.nick, "part", channel=event.target)
    
    def on_quit(self, connection, event):
        log.info(event)
        
        # Update protectees.
        if event.source.nick in self.protectees:    # Protectee parted home channel.
            del self.protectees[event.source.nick] # Delete from protectees.
        
        # Update last act.
        if event.arguments:
            Lastact.update(self, event.source.nick, "quit", lastact=event.arguments[0])
        else:
            Lastact.update(self, event.source.nick, "quit")
    
    def on_invite(self, connection, event):
        log.notice(event)
        if event.target == connection.get_nickname(): # Bot invited.
            Inform.operators(self, connection, "Received invitation to " + red + event.arguments[0] + reset + " from " + red + event.source.nick + reset + ".")
    
    def on_topic(self, connection, event):
        log.info(event)
        
        # Update last act.
        Lastact.update(self, event.source.nick, "topic", channel=event.target, lastact=event.arguments[0])

    def on_pubmsg(self, connection,  event):
        events.on_pubmsg.process_event(self, connection, event)
        commands.public.do_command(self, connection, event)
        commands.admin.do_command(self, connection, event)
        commands.statistics.do_command(self, connection, event)
        commands.games.do_command(self, connection, event)
    
    def on_privmsg(self, connection, event):
        log.info(event)
        commands.public.do_command(self, connection, event)
        commands.admin.do_command(self, connection, event)
        commands.statistics.do_command(self, connection, event)
        commands.games.do_command(self, connection, event)
        Inform.owners(self, connection, "PM from " + red + red + event.source.nick + grey + ": " + reset + event.arguments[0] + ".")
    
    def on_pubnotice(self, connection, event):
        log.info(event)
        
        # Update last act.
        Lastact.update(self, event.source.nick, "notice", channel=event.target, lastact=event.arguments[0])
        
        # Save statistic to database.
        MessageStatistics.update(self, event, "notice")
    
    def on_privnotice(self,  connection,  event):
        log.info(event)
        commands.public.do_command(self, connection, event)
        commands.admin.do_command(self, connection, event)
        commands.statistics.do_command(self, connection, event)
        commands.games.do_command(self, connection, event)
        try:
            if event.source.nick == "NickServ" and event.arguments[0].startswith("This nickname is registered"):
                connection.privmsg("NickServ", "identify " + connection.nickname + " " + self.password) # Identify with NickServ.
            if event.source.nick == "ChanServ" and event.arguments[0].startswith("Key for channel ") and len(event.arguments[0]) > 5:   # Received channel key.
                connection.join(event.arguments[0].split(' ')[3], event.arguments[0].split(' ')[5][:-1])
        except:
            pass
        Inform.owners(self, connection, "PM from " + red + red + event.source.nick + grey + ": " + reset + event.arguments[0] + ".")
    
    def on_action(self, connection, event):
        events.on_action.process_event(self, connection, event)
    
    
    def on_whoreply(self, connection, event):
        events.on_whoreply.process_event(self, connection, event)
    
    def on_userhost(self, connection, event):
        log.info(event)
        
    
    def on_yourhost(self, connection, event):
        log.info(event)
    
    def on_yourebannedcreep(self, connection, event):
        log.warning(event)
    
    def on_youwillbebanned(self, connection, event):
        log.warning(event)
        Inform.operators(self, connection, "I will be banned " + red + event.arguments[0] + reset + " from " + red + event.source)
    
    
    # DCC stuff from originalexample file.
    def on_dccmsg(self, c, e):
        log.info(e)
        # non-chat DCC messages are raw bytes; decode as text
        text = e.arguments[0].decode('utf-8')
        c.privmsg("You said: " + text)

    def on_dccchat(self, c, e):
        log.info(e)
        if len(e.arguments) != 2:
            return
        args = e.arguments[1].split()
        if len(args) == 4:
            try:
                address = ip_numstr_to_quad(args[2])
                port = int(args[3])
            except ValueError:
                return
            self.dcc_connect(address, port)

def main():
    
    # Check system arguments.
    if len(sys.argv) != 2:
        print(sys.argv)
        print("Usage: rotbot <server ID from database>")
        sys.exit(1)
    instance = sys.argv[1]  # Instance is the database network id.
    
    # Database.
    db = Postgres("postgres://pyRot:4h8q(.@localhost/pyRot")
    
    # Get network from database.
    try:
        network = db.one("SELECT * FROM networks WHERE id=" + str(instance))
    except:
        print("Invalid network ID.")
        sys.exit(1)
    if network == None:
        print("Invalid network ID.")
        sys.exit(1)
    
    bot = PyRot(network.name, db, network.home_channel, network.nickname, network.username, network.password, network.host, network.port, network.use_ssl,  network.command_character, network.help_character)
    bot.start()

if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        log.info('Interrupted by keyboard.')
        sys.exit(0)