Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Send request to RESUME a session AFTER we have a new token (resume.py)

Browse source
This commit is contained in:
jesopo 2019-02-10 00:03:14 +00:00
parent e2b019a9b8
commit 20d75e9716
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
modules/resume.py
View file

@ -11,15 +11,12 @@ class Module(ModuleManager.BaseModule):
@utils.hook("received.cap.ls") @utils.hook("received.cap.ls")
def on_cap_ls(self, event): def on_cap_ls(self, event):
nickname, token = self._get_token(event["server"]) nickname, token = self._get_token(event["server"])
if CAP in event["capabilities"] and (not nickname or not token): if CAP in event["capabilities"]:
event["server"].queue_capability(CAP) event["server"].queue_capability(CAP)
@utils.hook("received.cap.ack") @utils.hook("received.cap.ack")
def on_cap_ack(self, event): def on_cap_ack(self, event):
nickname, token = self._get_token(event["server"]) event["server"].wait_for_capability("resume")
if CAP in event["capabilities"] and nickname and token:
event["server"].send("RESUME %s %s" % (nickname, token))
event["server"].cap_started = False
@utils.hook("received.resume") @utils.hook("received.resume")
def on_resume(self, event): def on_resume(self, event):
@ -30,6 +27,11 @@ class Module(ModuleManager.BaseModule):
elif event["args"][0] == "TOKEN": elif event["args"][0] == "TOKEN":
event["server"].connection_params.args["new-resume-token" event["server"].connection_params.args["new-resume-token"
] = event["args"][1] ] = event["args"][1]
nickname, token = self._get_token(event["server"])
if nickname and token:
event["server"].send("RESUME %s %s" % (nickname, token))
event["server"].capability_done("resume")
@utils.hook("received.numeric.001") @utils.hook("received.numeric.001")
def on_connect(self, event): def on_connect(self, event):