bitbot-3.11-fork/modules/autojoin.py

from src import ModuleManager, utils

BATCH_MAX = 10

class Module(ModuleManager.BaseModule):
    @utils.hook("received.001")
    def on_connect(self, event):
        channels = event["server"].get_setting("autojoin", [])
        if channels:
            channel_batches = [channels[i:i+BATCH_MAX] for i in range(
                0, len(channels), BATCH_MAX)]
            for channel_batch in channel_batches:
                event["server"].send_joins(channel_batch)

    @utils.hook("self.join")
    def on_join(self, event):
        channels = event["server"].get_setting("autojoin", [])
        channel_name = event["server"].irc_lower(event["channel"].name)
        if not channel_name in channels:
            channels.append(channel_name)
            event["server"].set_setting("autojoin", channels)

    def _remove_channel(self, server, channel_name):
        channels = server.get_setting("autojoin", [])
        if channel_name in channels:
            channels.remove(channel_name)
            server.set_setting("autojoin", channels)
            return True
        else:
            return False

    @utils.hook("self.part")
    def on_part(self, event):
        self._remove_channel(event["server"], event["channel"].name)

    @utils.hook("self.kick")
    def on_kick(self, event):
        self._remove_channel(event["server"], event["channel"].name)

    @utils.hook("raw.received.470")
    def on_linkchannel(self, event):
        initial = event["line"].args[1]
        initial_lower = event["server"].irc_lower(initial)
        linked = event["line"].args[2]

        if self._remove_channel(event["server"], initial_lower):
            self.log.warn(f"{str(event['server'])} "
                f"channel {initial} linked to {linked} "
                "- removed from autojoin")