PyLink/protocols/inspircd.py

import threading
import socket
import time
import re
import string
import sys

# TODO: make PyLink a package so I don't have to hack the import system
# like this.
from os import sys, path
sys.path.append(path.dirname(path.dirname(path.abspath(__file__))))
from main import IrcUser

# From http://www.inspircd.org/wiki/Modules/spanningtree/UUIDs.html
chars = string.digits + string.ascii_uppercase
iters = [iter(chars) for _ in range(6)]
a = [next(i) for i in iters]

def next_uid(sid, level=-1):
    try:
        a[level] = next(iters[level])
        return sid + ''.join(a)
    except StopIteration:
        return UID(level-1)

def _sendFromServer(irc, msg):
    irc.send(':%s %s' % (irc.sid, msg))

def _sendFromUser(irc, msg, user=None):
    if user is None:
        user = irc.pseudoclient.uid
    irc.send(':%s %s' % (user, msg))

def _join(irc, channel):
    _sendFromUser(irc, "FJOIN {channel} {ts} +nt :,{uid}".format(sid=irc.sid,
             ts=int(time.time()), uid=irc.pseudoclient.uid, channel=channel))

def _uidToNick(irc, uid):
    for k, v in irc.users.items():
        if v.uid == uid:
            return k

def connect(irc):
    ts = int(time.time())
    host = irc.serverdata["hostname"]
    uid = next_uid(irc.sid)
    irc.pseudoclient = IrcUser('PyLink', ts, uid, 'pylink', host,
                               'PyLink Client')
    irc.users['PyLink'] = irc.pseudoclient
    
    f = irc.send
    f('CAPAB START 1203')
    # This is hard coded atm... We should fix it eventually...
    f('CAPAB CAPABILITIES :NICKMAX=32 HALFOP=0 CHANMAX=65 MAXMODES=20'
      ' IDENTMAX=12 MAXQUIT=255 PROTOCOL=1203')
    f('CAPAB END')
    # TODO: check recvpass here
    f('SERVER {host} {Pass} 0 {sid} :PyLink Service'.format(host=host,
      Pass=irc.serverdata["sendpass"], sid=irc.sid))
    f(':%s BURST %s' % (irc.sid, ts))
    # InspIRCd documentation:
    # :751 UID 751AAAAAA 1220196319 Brain brainwave.brainbox.cc
    # netadmin.chatspike.net brain 192.168.1.10 1220196324 +Siosw
    # +ACKNOQcdfgklnoqtx :Craig Edwards
    f(":{sid} UID {uid} {ts} PyLink {host} {host} pylink 127.0.0.1 {ts} +o +"
      " :PyLink Client".format(sid=irc.sid, ts=ts,
                               host=host,
                               uid=uid))
    f(':%s ENDBURST' % (irc.sid))
    _join(irc, irc.serverdata["channel"])

# :7NU PING 7NU 0AL
def handle_ping(irc, servernumeric, command, args):
    if args[3] == irc.sid:
        _sendFromServer(irc, 'PONG %s' % args[2])

def handle_privmsg(irc, numeric, command, args):
    # _sendFromUser(irc, 'PRIVMSG %s :hello!' % numeric)
    print(irc.users)
    print(irc.channels)

def handle_error(irc, numeric, command, args):
    print('Received an ERROR, killing!')
    irc.connected = False
    sys.exit(1)

def handle_fjoin(irc, servernumeric, command, args):
    # :70M FJOIN #chat 1423790411 +AFPfjnt 6:5 7:5 9:5 :o,1SRAABIT4 v,1IOAAF53R <...>
    channel = args[0]
    # tl;dr InspIRCd sends each user's channel data in the form of 'modeprefix(es),UID'
    # We'll save each user in this format too, at least for now.
    print(args)
    users = args[-1].split()
    users = [x.split(',') for x in users]
    print(users)
    
    '''
    if channel not in irc.channels.keys():
        irc.channels[channel]['users'] = users
    else:
        old_users = irc.channels[channel]['users'].copy()
        old_users.update(users)
    '''

def handle_uid(irc, numeric, command, args):
    # :1SR UID 1SRAAAAAU 1428974823 synnero ow.my.eye.rs ow.my.eye.rs GLolol 2604:180:1::d34d:d87b 1425951245 +Wiosw +ACGKNOQXacfgklnoqvx :move along, nothing to see here!
    uid, ts, nick, realhost, host, ident, ip = args[0:7]
    realname = args[-1]
    irc.users[nick] = IrcUser(nick, ts, uid, ident, host, realname, realhost, ip)

def handle_quit(irc, numeric, command, args):
    # :1SRAAGB4T QUIT :Quit: quit message goes here
    nick = _uidToNick(irc, numeric)
    del irc.users[nick]
    for k, v in irc.channels.items():
        try:
            del irc.channels[k][users][v]
        except KeyError:
            pass

def handle_events(irc, data):
    # Each server message looks something like this:
    # :70M FJOIN #chat 1423790411 +AFPfjnt 6:5 7:5 9:5 :v,1SRAAESWE
    # :<sid> <command> <argument1> <argument2> ... :final multi word argument
    try:
        args = data.split()
        real_args = []
        for arg in args:
            real_args.append(arg)
            # If the argument starts with ':' and ISN'T the first argument.
            # The first argument is used for denoting the source UID/SID.
            if arg.startswith(':') and args.index(arg) != 0:
                # : is used for multi-word arguments that last until the end
                # of the message. We can use list splicing here to turn them all
                # into one argument.
                index = args.index(arg)  # Get array index of arg
                # Replace it with everything left on that line
                arg = ' '.join(args[index:])[1:]
                # Cut the original argument list at before multi-line one, and
                # merge them together.
                real_args = args[:index]
                real_args.append(arg)
                break
        real_args[0] = real_args[0].split(':', 1)[1]
        args = real_args

        numeric = args[0]
        command = args[1]
    except IndexError:
        return

    # We will do wildcard event handling here. Unhandled events are just ignored, yay!
    try:
        func = globals()['handle_'+command.lower()]
        func(irc, numeric, command, args)
    except KeyError:  # unhandled event
        pass