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

66 lines
2.5 KiB
Python

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)