bitbot-3.11-fork/IRCLineHandler.py

import re, threading
import Utils

RE_PREFIXES = re.compile(r"\bPREFIX=\((\w+)\)(\W+)(?:\b|$)")
RE_CHANMODES = re.compile(
    r"\bCHANMODES=(\w*),(\w*),(\w*),(\w*)(?:\b|$)")
RE_CHANTYPES = re.compile(r"\bCHANTYPES=(\W+)(?:\b|$)")
RE_MODES = re.compile(r"[-+]\w+")

CAPABILITIES = {"message-tags", "multi-prefix", "chghost", "invite-notify",
    "account-tag", "account-notify", "extended-join"}

class LineHandler(object):
    def __init__(self, bot, events):
        self.bot = bot
        self.events = events
        events.on("raw").on("PING").hook(self.ping)

        events.on("raw").on("001").hook(self.handle_001, default_event=True)
        events.on("raw").on("005").hook(self.handle_005)
        events.on("raw").on("311").hook(self.handle_311, default_event=True)
        events.on("raw").on("332").hook(self.handle_332)
        events.on("raw").on("333").hook(self.handle_333)
        events.on("raw").on("353").hook(self.handle_353, default_event=True)
        events.on("raw").on("366").hook(self.handle_366, default_event=True)
        events.on("raw").on("421").hook(self.handle_421, default_event=True)
        events.on("raw").on("352").hook(self.handle_352, default_event=True)
        events.on("raw").on("324").hook(self.handle_324, default_event=True)
        events.on("raw").on("329").hook(self.handle_329, default_event=True)
        events.on("raw").on("433").hook(self.handle_433, default_event=True)
        events.on("raw").on("477").hook(self.handle_477, default_event=True)

        events.on("raw").on("JOIN").hook(self.join)
        events.on("raw").on("PART").hook(self.part)
        events.on("raw").on("QUIT").hook(self.quit)
        events.on("raw").on("NICK").hook(self.nick)
        events.on("raw").on("MODE").hook(self.mode)
        events.on("raw").on("KICK").hook(self.kick)
        events.on("raw").on("INVITE").hook(self.invite)
        events.on("raw").on("TOPIC").hook(self.topic)
        events.on("raw").on("PRIVMSG").hook(self.privmsg)
        events.on("raw").on("NOTICE").hook(self.notice)
        events.on("raw").on("CHGHOST").hook(self.chghost)
        events.on("raw").on("ACCOUNT").hook(self.account)

        events.on("raw").on("CAP").hook(self.cap)
        events.on("raw").on("authenticate").hook(self.authenticate)

    def handle(self, server, line):
        original_line = line
        tags = {}
        prefix = None
        command = None

        if line[0] == "@":
            tags_prefix, line = line[1:].split(" ", 1)
            for tag in tags_prefix.split(";"):
                if tag:
                    tag_split = tag.split("=", 1)
                    tags[tag_split[0]] = tag_split[1:]

        if line[0] == ":":
            prefix, command, line = line[1:].split(" ", 2)
        else:
            command, line = line.split(" ", 1)

        arbitrary = None
        if ":" in line:
            line, arbitrary = line.split(":", 1)
            if line.endswith(" "):
                line = line[:-1]
        args = line.split(" ")

        hooks = self.events.on("raw").on(command).get_hooks()
        default_event = False
        for hook in hooks:
            if hook.kwargs.get("default_event", False):
                default_event = True
                break

        #server, prefix, command, args, arbitrary
        self.events.on("raw").on(command).call(server=server,
            prefix=prefix, args=args, arbitrary=arbitrary, tags=tags)
        if default_event or not hooks:
            if command.isdigit():
                self.events.on("received").on("numeric").on(command
                    ).call(line=original_line, server=server, tags=tags,
                    line_split=original_line.split(" "), number=command)
            else:
                self.events.on("received").on(command).call(
                    line=original_line, line_split=original_line.split(" "),
                    command=command, server=server, tags=tags)

    # ping from the server
    def ping(self, event):
        nonce = event["arbitrary"]
        event["server"].send_pong(nonce)

    # first numeric line the server sends
    def handle_001(self, event):
        event["server"].name = Utils.remove_colon(event["prefix"])
        event["server"].set_own_nickname(event["args"][0])
        event["server"].send_whois(event["server"].nickname)

    # server telling us what it supports
    def handle_005(self, event):
        isupport_line = " ".join(event["args"][1:])
        if "NAMESX" in isupport_line:
            event["server"].send("PROTOCTL NAMESX")
        match = re.search(RE_PREFIXES, isupport_line)
        if match:
            event["server"].mode_prefixes.clear()
            modes = match.group(1)
            prefixes = match.group(2)
            for i, prefix in enumerate(prefixes):
                if i < len(modes):
                    event["server"].mode_prefixes[prefix] = modes[i]
        match = re.search(RE_CHANMODES, isupport_line)
        if match:
            event["server"].channel_modes = list(match.group(4))
        match = re.search(RE_CHANTYPES, isupport_line)
        if match:
            event["server"].channel_types = list(match.group(1))
        self.events.on("received").on("numeric").on("005").call(
            isupport=isupport_line, server=event["server"])

    # whois respose (nickname, username, realname, hostname)
    def handle_311(self, event):
        nickname = event["args"][1]
        if event["server"].is_own_nickname(nickname):
            target = event["server"]
        else:
            target = event["server"].get_user(nickname)
        target.username = event["args"][2]
        target.hostname = event["args"][3]
        target.realname = event["arbitrary"]

    # on-join channel topic line
    def handle_332(self, event):
        channel = event["server"].get_channel(event["args"][1])

        channel.set_topic(event["arbitrary"])
        self.events.on("received").on("numeric").on("332"
            ).call(channel=channel, server=event["server"],
            topic=event["arbitrary"])

    # channel topic changed
    def topic(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        user = event["server"].get_user(nickname)
        channel = event["server"].get_channel(event["args"][0])

        channel.set_topic(event["arbitrary"])
        self.events.on("received").on("topic").call(channel=channel,
            server=event["server"], topic=event["arbitrary"], user=user)

    # on-join channel topic set by/at
    def handle_333(self, event):
        channel = event["server"].get_channel(event["args"][1])

        topic_setter_hostmask = event["args"][2]
        nickname, username, hostname = Utils.seperate_hostmask(
            topic_setter_hostmask)
        topic_time = int(event["args"][3]) if event["args"][3].isdigit(
            ) else None

        channel.set_topic_setter(nickname, username, hostname)
        channel.set_topic_time(topic_time)
        self.events.on("received").on("numeric").on("333"
            ).call(channel=channel, setter=nickname, set_at=topic_time,
            server=event["server"])

    # on-join user list with status symbols
    def handle_353(self, event):
        channel = event["server"].get_channel(event["args"][2])
        nicknames = event["arbitrary"].split()
        for nickname in nicknames:
            if nickname.strip():
                modes = set([])

                while nickname[0] in event["server"].mode_prefixes:
                    modes.add(event["server"].mode_prefixes[nickname[0]])
                    nickname = nickname[1:]

                user = event["server"].get_user(nickname)
                user.join_channel(channel)
                channel.add_user(user)
                for mode in modes:
                    channel.add_mode(mode, nickname)

    # on-join user list has finished
    def handle_366(self, event):
        event["server"].send_who(event["args"][1])

    # on user joining channel
    def join(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        account = None
        realname = None
        if len(event["args"]) == 2:
            channel = event["server"].get_channel(event["args"][0])
            if not event["args"] == "*":
                account = event["args"][1]
            realname = event["arbitrary"]
        else:
            channel = event["server"].get_channel(
                event["arbitrary"] or event["args"][0])

        if not event["server"].is_own_nickname(nickname):
            user = event["server"].get_user(nickname)
            if not event["server"].has_user(nickname):
                user.username = username
                user.hostname = hostname
            channel.add_user(user)
            user.join_channel(channel)
            self.events.on("received").on("join").call(channel=channel,
                user=user, server=event["server"], account=account,
                realname=realname)
        else:
            if channel.name in event["server"].attempted_join:
                del event["server"].attempted_join[channel.name]
            self.events.on("self").on("join").call(channel=channel,
                server=event["server"], account=account, realname=realname)
            channel.send_mode()

    # on user parting channel
    def part(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        channel = event["server"].get_channel(event["args"][0])
        reason = event["arbitrary"] or ""

        if not event["server"].is_own_nickname(nickname):
            user = event["server"].get_user(nickname)
            self.events.on("received").on("part").call(channel=channel,
                reason=reason, user=user, server=event["server"])
            channel.remove_user(user)
            user.part_channel(channel)
            if not len(user.channels):
                event["server"].remove_user(user)
        else:
            event["server"].remove_channel(channel)
            self.events.on("self").on("part").call(channel=channel,
                reason=reason, server=event["server"])

    # unknown command sent by us, oops!
    def handle_421(self, event):
        print("warning: unknown command '%s'." % event["args"][1])

    # a user has disconnected!
    def quit(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        reason = event["arbitrary"] or ""

        if not event["server"].is_own_nickname(nickname):
            user = event["server"].get_user(nickname)
            event["server"].remove_user(user)
            self.events.on("received").on("quit").call(reason=reason,
                user=user, server=event["server"])
        else:
            event["server"].disconnect()

    # the server is telling us about its capabilities!
    def cap(self, event):
        capabilities = (event["arbitrary"] or "").split(" ")
        subcommand = event["args"][1].lower()

        if subcommand == "ls":
            matched_capabilities = set(capabilities) & CAPABILITIES
            if matched_capabilities:
                event["server"].queue_capabilities(matched_capabilities)

        self.events.on("received").on("cap").on(subcommand).call(
            capabilities=capabilities, server=event["server"])

        if subcommand == "ls":
            if event["server"].has_capability_queue():
                event["server"].send_capability_queue()
            else:
                event["server"].send_capability_end()
        elif subcommand == "ack" or subcommand == "nack":
            if not event["server"].waiting_for_capabilities():
                event["server"].send_capability_end()

    # the server is asking for authentication
    def authenticate(self, event):
        self.events.on("received").on("authenticate").call(
            message=event["args"][0], server=event["server"])

    # someone has changed their nickname
    def nick(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        new_nickname = event["arbitrary"]
        if not event["server"].is_own_nickname(nickname):
            user = event["server"].get_user(nickname)
            old_nickname = user.nickname
            user.set_nickname(new_nickname)
            event["server"].change_user_nickname(old_nickname, new_nickname)
            self.events.on("received").on("nick").call(
                new_nickname=new_nickname, old_nickname=old_nickname,
                user=user, server=event["server"])
        else:
            old_nickname = event["server"].nickname
            event["server"].set_own_nickname(new_nickname)
            self.events.on("self").on("nick").call(
                server=event["server"], new_nickname=new_nickname,
                old_nickname=old_nickname)

    # something's mode has changed
    def mode(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        user = event["server"].get_user(nickname)
        target = event["args"][0]
        is_channel = target[0] in event["server"].channel_types
        if is_channel:
            channel = event["server"].get_channel(target)
            remove = False
            args  = event["args"][2:]
            _args = args[:]
            modes = RE_MODES.findall(event["args"][1])
            for chunk in modes:
                remove = chunk[0] == "-"
                for mode in chunk[1:]:
                    if mode in event["server"].channel_modes:
                        channel.change_mode(remove, mode)
                    elif mode in event["server"].mode_prefixes.values(
                            ) and len(args):
                        channel.change_mode(remove, mode, args.pop(0))
                    else:
                        args.pop(0)
            self.events.on("received").on("mode").on("channel").call(
                modes=modes, mode_args=_args, channel=channel,
                server=event["server"], user=user)
        elif event["server"].is_own_nickname(target):
            modes = RE_MODES.findall(event["arbitrary"] or args[1])
            for chunk in modes:
                remove = chunk[0] == "-"
                for mode in chunk[1:]:
                    event["server"].change_own_mode(remove, mode)
            self.events.on("self").on("mode").call(modes=modes,
                server=event["server"])

    # someone (maybe me!) has been invited somewhere
    def invite(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        target_channel = event["arbitrary"] or event["args"][1]
        user = event["server"].get_user(nickname)
        target_user = event["server"].get_user(event["args"][0])
        self.events.on("received").on("invite").call(
            user=user, target_channel=target_channel,
            server=event["server"], target_user=target_user)

    # we've received a message
    def privmsg(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        user = event["server"].get_user(nickname)
        message = event["arbitrary"] or ""
        message_split = message.split(" ")
        target = event["args"][0]
        action = message.startswith("\01ACTION ") and message.endswith("\01")
        if action:
            message = message.replace("\01ACTION ", "", 1)[:-1]
        if target[0] in event["server"].channel_types:
            channel = event["server"].get_channel(event["args"][0])
            self.events.on("received").on("message").on("channel").call(
                user=user, message=message, message_split=message_split,
                channel=channel, action=action, server=event["server"])
            channel.buffer.add_line(user.nickname, message, action)
        elif event["server"].is_own_nickname(target):
            self.events.on("received").on("message").on("private").call(
                user=user, message=message, message_split=message_split,
                action=action, server=event["server"])
            user.buffer.add_line(user.nickname, message, action)

    # we've received a notice
    def notice(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        message = event["arbitrary"] or ""
        message_split = message.split(" ")
        target = event["args"][0]
        if nickname == event["server"].name or target == "*":
            self.events.on("received.server-notice").call(
                message=message, message_split=message_split,
                server=event["server"])
        else:
            user = event["server"].get_user(nickname)
            if target[0] in event["server"].channel_types:
                channel = event["server"].get_channel(target)
                self.events.on("received.notice.channel").call(
                    message=message, message_split=message_split,
                    user=user, server=event["server"], channel=channel)
            elif event["server"].is_own_nickname(target):
                self.events.on("received.notice.private").call(
                    message=message, message_split=message_split,
                    user=user, server=event["server"])

    # a user's username and/or hostname has changed
    def chghost(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        user = event["server"].get_user("nickanme")
        username = event["args"][0]
        hostname = event["args"][1]
        user.username = username
        user.hostname = hostname

    def account(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        user = event["server"].get_user("nickname")

        if not event["args"][0] == "*":
            self.events.on("received.account.login").call(user=user,
                server=event["server"], account=account)
        else:
            self.events.on("received.account.logout").call(user=user,
                server=event["server"])

    # response to a WHO command for user information
    def handle_352(self, event):
        user = event["server"].get_user(event["args"][5])
        user.username = event["args"][2]
        user.hostname = event["args"][3]

    # response to an empty mode command
    def handle_324(self, event):
        channel = event["server"].get_channel(event["args"][1])
        modes = event["args"][2]
        if modes[0] == "+" and modes[1:]:
            for mode in modes[1:]:
                if mode in event["server"].channel_modes:
                    channel.add_mode(mode)

    # channel creation unix timestamp
    def handle_329(self, event):
        channel = event["server"].get_channel(event["args"][1])
        channel.creation_timestamp = int(event["args"][2])

    # nickname already in use
    def handle_433(self, event):
        pass

    # we need a registered nickname for this channel
    def handle_477(self, event):
        if event["args"][1].lower() in event["server"].attempted_join:
            self.bot.add_timer("rejoin", 5,
                channel_name=event["args"][1],
                key=event["server"].attempted_join[event["args"][1].lower()],
                server_id=event["server"].id)

    # someone's been kicked from a channel
    def kick(self, event):
        nickname, username, hostname = Utils.seperate_hostmask(
            event["prefix"])
        user = event["server"].get_user(nickname)
        target = event["args"][1]
        channel = event["server"].get_channel(event["args"][0])
        reason = event["arbitrary"] or ""

        if not event["server"].is_own_nickname(target):
            target_user = event["server"].get_user(target)
            self.events.on("received").on("kick").call(channel=channel,
                reason=reason, target_user=target_user, user=user,
                server=event["server"])
        else:
            self.events.on("self").on("kick").call(channel=channel,
                reason=reason, user=user, server=event["server"])