1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
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-token", (None, None))
def _set_token(Self, server, nickname, token):
server.connection_params.args["resume-token"] = (nickname, token)
@utils.hook("received.cap.ls")
def on_cap_ls(self, event):
nickname, token = self._get_token(event["server"])
if CAP in event["capabilities"]:
event["server"].queue_capability(CAP)
@utils.hook("received.cap.ack")
def on_cap_ack(self, event):
if CAP in event["capabilities"]:
event["server"].wait_for_capability("resume")
@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-resume-token"
] = event["args"][1]
nickname, token = self._get_token(event["server"])
if nickname and token:
event["server"].send("RESUME %s %s" % (nickname, token))
event["server"].cap_started = False
event["server"].capability_done("resume")
@utils.hook("received.numeric.001")
def on_connect(self, event):
new_token = event["server"].connection_params.args.get(
"new-resume-token", None)
if new_token:
self._set_token(event["server"], event["server"].nickname,
new_token)
del event["server"].connection_params.args["new-resume-token"]
@utils.hook("self.nick")
def nick_change(self, event):
nickname, token = self._get_token(event["server"])
if nickname and token:
self._set_token(event["server"], event["new_nickname"], token)
|
