bitbot-3.11-fork/modules/channel_op.py

from src import Utils

class Module(object):
    _name = "Channel Op"
    def __init__(self, bot, events, exports):
        self.bot = bot

        events.on("received.command").on("kick", "k").hook(self.kick,
            channel_only=True, require_mode="o", usage="<nickname> [reason]",
            min_args=1, help="Kick a user from the channel")
        events.on("received.command.ban").hook(self.ban, channel_only=True,
            require_mode="o", min_args=1, usage="<nickname/hostmask>",
            help="Ban a user/hostmask from the channel")
        events.on("received.command.unban").hook(self.unban,
            channel_only=True, require_mode="o", usage="<nickname/hostmask>",
            min_args=1, help="Unban a user/hostmask from the channel")

        events.on("received.command").on("kickban", "kb"
            ).hook(self.kickban, channel_only=True, require_mode="o",
            min_args=1, help="Kickban a user from the channel",
            usage="<nickanme> [reason]")

        events.on("received.command.op"
            ).hook(self.op, channel_only=True, require_mode="o",
            help="Give +o to a user", usage="[nickname]")
        events.on("received.command.deop"
            ).hook(self.deop, channel_only=True, require_mode="o",
            help="Take +o from a user", usage="[nickname]")

        events.on("received.command.voice").hook(self.voice,
            channel_only=True, require_mode="o", usage="[nickname]",
             help="Give +v to a user")
        events.on("received.command.devoice").hook(self.devoice,
            channel_only=True, require_mode="o", usage="[nickname]",
            help="Take +v from a user")

        events.on("received.command.topic").hook(self.topic, min_args=1,
            require_mode="o", channel_only=True, usage="<topic>",
            help="Set the topic of the current channel")
        events.on("received.command").on("topicappend", "tappend").hook(
            self.tappend, min_args=1, require_mode="o", channel_only=True,
            usage="<topic>", help="Set the topic of the current channel")

        events.on("received.message.channel").hook(self.highlight_spam)

        exports.add("channelset", {"setting": "highlight-spam-threshold",
            "help": "Set the number of nicknames in a message that "
            "qualifies as spam", "validate": Utils.int_or_none})
        exports.add("channelset", {"setting": "highlight-spam-protection",
            "help": "Enable/Disable highlight spam protection",
            "validate": Utils.bool_or_none})
        exports.add("channelset", {"setting": "highlight-spam-ban",
            "help": "Enable/Disable banning highlight spammers "
            "instead of just kicking", "validate": Utils.bool_or_none})
        exports.add("channelset", {"setting": "ban-format",
            "help": "Set ban format ($n = nick, $u = username, "
            "$h = hostname)"})

    def kick(self, event):
        target = event["args_split"][0]
        target_user = event["server"].get_user(target)
        if event["args_split"][1:]:
            reason = " ".join(event["args_split"][1:])
        else:
            reason = None
        event["stderr"].set_prefix("Kick")
        if event["target"].has_user(target_user):
            if not event["server"].is_own_nickname(target):
                event["target"].send_kick(target, reason)
            else:
                event["stderr"].write("Nope.")
        else:
            event["stderr"].write("That user is not in this channel")

    def _ban_format(self, user, s):
        return s.replace("$n", user.nickname).replace("$u", user.username
            ).replace("$h", user.hostname)
    def _ban(self, channel, ban, user):
        format = channel.get_setting("ban-format", "*!$u@$h")
        hostmask_split = format.split("$$")
        hostmask_split = [self._ban_format(user, s) for s in hostmask_split]
        hostmask = "".join(hostmask_split)
        if ban:
            channel.send_ban(hostmask)
        else:
            channel.send_unban(hostmask)
    def ban(self, event):
        target_user = event["server"].get_user(event["args_split"][0])
        if event["target"].has_user(target_user):
            self._ban(event["target"], True, target_user)
        else:
            event["target"].send_ban(event["args_split"][0])
    def unban(self, event):
        target_user = event["server"].get_user(event["args_split"][0])
        if event["target"].has_user(target_user):
            self._ban(event["target"], False, target_user)
        else:
            event["target"].send_unban(event["args_split"][0])

    def kickban(self, event):
        if event["server"].has_user(event["args_split"][0]):
            self.ban(event)
            self.kick(event)
        else:
            event["stderr"].write("That user is not in this channel")

    def op(self, event):
        target = event["user"].nickname if not event["args_split"] else event[
            "args_split"][0]
        event["target"].send_mode("+o", target)
    def deop(self, event):
        target = event["user"].nickname if not event["args_split"] else event[
            "args_split"][0]
        event["target"].send_mode("-o", target)
    def voice(self, event):
        target = event["user"].nickname if not event["args_split"] else event[
            "args_split"][0]
        event["target"].send_mode("+v", target)
    def devoice(self, event):
        target = event["user"].nickname if not event["args_split"] else event[
            "args_split"][0]
        event["target"].send_mode("-v", target)

    def topic(self, event):
        event["target"].send_topic(event["args"])
    def tappend(self, event):
        event["target"].send_topic(event["target"].topic + event["args"])

    def highlight_spam(self, event):
        if event["channel"].get_setting("highlight-spam-protection", False):
            nicknames = list(map(lambda user: user.nickname,
                event["channel"].users)) + [event["server"].nickname]

            highlights = set(nicknames) & set(event["message_split"])
            if len(highlights) > 1 and len(highlights) >= event["channel"
                    ].get_setting("highlight-spam-threshold", 10):
                has_mode = event["channel"].mode_or_above(event["user"], "v")
                should_ban = event["channel"].get_setting("highlight-spam-ban",
                    False)
                if not has_mode:
                    if should_ban:
                        event["channel"].send_ban("*!%s@%s" % (
                            event["user"].username, event["user"].hostname))
                    event["channel"].send_kick(event["user"].nickname,
                        "highlight spam detected")