1 files changed, 48 insertions, 0 deletions
diff --git a/src/core_modules/nick_regain.py b/src/core_modules/nick_regain.py
new file mode 100644
index 00000000..cf1dfa48
--- /dev/null
+++ b/src/core_modules/nick_regain.py
@@ -0,0 +1,48 @@
+from src import ModuleManager, utils
+
+class Module(ModuleManager.BaseModule):
+    def _done_connecting(self, server):
+        target_nick = server.connection_params.nickname
+        if not server.irc_equals(server.nickname, 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"])
+
+    @utils.hook("self.nick")
+    def self_nick(self, event):
+        target_nick = event["server"].connection_params.nickname
+        if event["server"].irc_equals(event["new_nickname"], target_nick):
+            if "MONITOR" in event["server"].isupport:
+                event["server"].send_raw("MONITOR - %s " % target_nick)
+
+    @utils.hook("received.731")
+    def mon_offline(self, event):
+        target_nick = event["server"].connection_params.nickname
+        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 = server.connection_params.nickname
+        if not server.irc_equals(server.nickname, target_nick):
+            server.send_raw("ISON %s" % target_nick)
+            timer.redo()
+
+    @utils.hook("received.303")
+    def ison_response(self, event):
+        target_nick = event["server"].connection_params.nickname
+        if not event["line"].args[1] and not event["server"].irc_equals(
+                event["server"].nickname, target_nick):
+            event["server"].send_nick(target_nick)
+