FireBot/handlers.py

import random as r
import config as conf
import re
import commands as cmds
from typing import Union
from overrides import bytes, bbytes
from importlib import reload

def CTCP(bot, msg: str, sender: str = "", isRaw: bool = False) -> bool:
    if isRaw:
        sender = msg.split("!", 1)[0][1:]
        message = msg.split("PRIVMSG", 1)[1].split(":", 1)[1].strip()
    kind = msg.split("\x01")[1].split(" ", 1)[0]
    bot.log(f'Responding to CTCP "{kind}" from {sender}')
    if kind == "VERSION":
        bot.notice(
            f"\x01VERSION FireBot {conf.__version__} (https://git.amcforum.wiki/Firepup650/fire-ircbot)\x01",
            sender,
            True,
        )
        return True
    elif kind == "USERINFO":
        bot.notice("\x01USERINFO FireBot (Firepup's bot)\x01", sender, True)
        return True
    elif kind == "SOURCE":
        bot.notice(
            "\x01SOURCE https://git.amcforum.wiki/Firepup650/fire-ircbot\x01",
            sender,
            True,
        )
        return True
    elif kind == "FINGER":
        bot.notice("\x01FINGER Firepup's bot\x01", sender, True)
        return True
    elif kind == "CLIENTINFO":
        bot.notice(
            "\x01CLIENTINFO ACTION VERSION USERINFO SOURCE FINGER\x01", sender, True
        )
        return True
    bot.log(f'Unknown CTCP "{kind}"', "WARN")
    return False

def PRIVMSG(bot, msg: str) -> Union[None, str]:
    # Format of ":[Nick]![ident]@[host|vhost] PRIVMSG [channel] :[message]”
    name = msg.split("!", 1)[0][1:]
    if (name.startswith("saxjax") and bot.server == "efnet") or (
        name == "ReplIRC" and bot.server == "replirc"
    ):
        if msg.find("<") != -1 and msg.find(">") != -1:
            Nname = msg.split("<", 1)[1].split(">", 1)[0].strip()
            if name == "ReplIRC":
                name = Nname[4:]
            else:
                name = Nname
            message = msg.split(">", 1)[1].strip()
        else:
            message = (
                msg.split("PRIVMSG", 1)[1].split(":", 1)[1].strip()
            )
    elif name == bot.nick:
        return  # type: ignore
    else:
        message = msg.split("PRIVMSG", 1)[1].split(":", 1)[1].strip()
    chan = msg.split("PRIVMSG", 1)[1].split(":", 1)[0].strip()
    bot.log(
        f'Got "{bytes(message).lazy_decode()}" from "{name}" in "{chan}"',
    )
    if len(name) > bot.nicklen:
        bot.log(f"Name too long ({len(name)} > {bot.nicklen})")
        return  # type: ignore
    elif chan not in bot.channels:
        bot.log(
            f"Channel not in channels ({chan} not in {bot.channels})",
            "WARN",
        )
        if not chan.startswith(("#", "+", "&")):
            chan = name
    else:
        bot.channels[chan] += 1
    if "goat" in name.lower() and bot.gmode == True:
        cmds.goat(bot, chan, name, message)
    handled = False
    for cmd in cmds.data:
        triggers = [cmd]
        triggers.extend(cmds.data[cmd]["aliases"])
        triggers = list(
            call.replace("$BOTNICK", bot.nick.lower())
            for call in triggers
        )
        if conf.mfind(
            message.lower(),
            triggers,
            cmds.data[cmd]["prefix"],
        ):
            if (
                "admin" in cmds.data[cmd] and cmds.data[cmd]["admin"]
            ) and name not in bot.adminnames:
                bot.msg(
                    f"Sorry {name}, you don't have permission to use {cmd.strip()}.",
                    chan,
                )
            else:
                cmds.call[cmd](bot, chan, name, message)
            handled = True
            break
    if not handled:
        for check in cmds.checks:
            if re.search(
                check.replace("$MAX", str(bot.nicklen)).replace(
                    "$BOTNICK", bot.nick
                ),
                message,
            ):
                cmds.call[check](bot, chan, name, message)
                handled = True
                break
    if not handled and conf.mfind(message, ["reload"]):
        if name in bot.adminnames:
            return "reload"
        else:
            bot.msg(
                f"Sorry {name}, you don't have permission to use reload.",
                chan,
            )
        handled = True
    if not handled and len(message.split("\x01")) == 3:
        if not bot.CTCP(message, name):
            CTCP = message.split("\x01")[1]
            if CTCP == "ACTION ducks":
                bot.msg("\x01ACTION gets hit by a duck\x01", chan)
            elif CTCP.startswith("ACTION ducks"):
                bot.msg(
                    f"\x01ACTION gets hit by {CTCP.split(' ', 2)[2]}\x01",
                    chan,
                )
    if chan in bot.channels and bot.channels[chan] >= bot.interval:
        r.seed()
        bot.channels[chan] = 0
        mm = open("mastermessages.txt", "r")
        q = mm.readlines()
        sel = conf.decode_escapes(
            str(r.sample(q, 1))
            .strip("[]'")
            .replace("\\n", "")
            .strip('"')
        )
        bot.msg(f"[QUOTE] {sel}", chan)
        mm.close()
    return  # type: ignore