From b3625dbe130f9e274ad8720e8de33fe7f20a6a66 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Thu, 8 Nov 2018 17:09:53 +0000
Subject: Support IRCv3's `draft/resume-0.2`

---
 modules/resume.py | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 modules/resume.py

(limited to 'modules')

diff --git a/modules/resume.py b/modules/resume.py
new file mode 100644
index 00000000..95dadbc5
--- /dev/null
+++ b/modules/resume.py
@@ -0,0 +1,38 @@
+from src import ModuleManager, utils
+
+CAP = "draft/resume-0.2"
+
+class Module(ModuleManager.BaseModule):
+    def _get_token(self, server):
+        return server.connection_params.args.get("resume", [None, None])
+
+    @utils.hook("received.cap.ls")
+    def on_cap_new(self, event):
+        username, token = self._get_token(event["server"])
+        if CAP in event["capabilities"] and (not username or not token):
+            event["server"].queue_capability("draft/resume-0.2")
+
+    @utils.hook("received.cap.ack")
+    def on_cap_ack(self, event):
+        username token = self._get_token(event["server"])
+        if CAP in event["capabilities"] and username and token:
+            event["server"].send("RESUME %s %s" % (username, token))
+            return False
+
+    @utils.hook("received.resume")
+    def on_resume(self, event):
+        if event["args"][0] == "SUCCESS":
+            self.log.info("Successfully resumed session")
+        elif event["args"][0] == "ERR":
+            self.log.info("Failed to resume session: %s" % event["args"][1])
+        elif event["args"][0] == "TOKEN":
+            event["server"].connection_params.args["new-token"
+                ] = event["args"][1]
+
+    @utils.hook("received.numeric.001")
+    def on_connect(self, event):
+        new_token = event["server"].connection_params.get("new-token")
+        if new_token:
+            event["server"].connection_params.args["resume"] = [
+                event["server"].nickname, new_token]
+            del event["server"].connection_params.args["resume"]
-- 
cgit v1.3.1-10-gc9f91