bitbot-3.11-fork/src/core_modules/nick_regain.py

from src import ModuleManager, utils

class Module(ModuleManager.BaseModule):
    def _done_connecting(self, server):
        target_nick = self._target(server)
        if not self._regained(server, target_nick):
            if "MONITOR" in server.isupport:
                server.send_raw("MONITOR + %s" % target_nick)
            else:
                self.timers.add("ison-check", self._ison_check, 30,
                    server=server)

    @utils.hook("received.376")
    def end_of_motd(self, event):
        self._done_connecting(event["server"])
    @utils.hook("received.422")
    def no_motd(self, event):
        self._done_connecting(event["server"])

    def _regained(self, server, target_nickname):
        return server.irc_equals(target_nickname, server.nickname)
    def _target(self, server):
        return server.connection_params.nickname

    @utils.hook("self.nick")
    def self_nick(self, event):
        target_nick = self._target(event["server"])
        if self._regained(event["server"], target_nick):
            if "MONITOR" in event["server"].isupport:
                event["server"].send_raw("MONITOR - %s" % target_nick)

    @utils.hook("received.nick")
    def nick(self, event):
        self._check(event["server"], event["old_nickname"])
    @utils.hook("received.quit")
    def quit(self, event):
        self._check(event["server"], event["user"].nickname)

    def _check(self, server, nickname):
        target_nick = self._target(server)
        if (not self._regained(server, target_nick)
                and server.irc_equals(nickname, target_nick)):
            server.send_nick(target_nick)

    @utils.hook("received.731")
    def mon_offline(self, event):
        target_nick = self._target(event["server"])
        nicks = event["line"].args[1].split(",")
        nicks = [event["server"].irc_lower(n) for n in nicks]
        if event["server"].irc_lower(target_nick) in nicks:
            event["server"].send_nick(target_nick)

    def _ison_check(self, timer):
        server = timer.kwargs["server"]
        target_nick = self._target(server)
        if not self._regained(server, target_nick):
            server.send_raw("ISON %s" % target_nick)
            timer.redo()

    @utils.hook("received.303")
    def ison_response(self, event):
        target_nick = self._target(event["server"])
        if (not event["line"].args[1] and
                not self._regained(event["server"], target_nick)):
            event["server"].send_nick(target_nick)