Actually grab resume timestamp before it's overwritten (resume.py)

1 files changed, 9 insertions, 1 deletions

diff --git a/modules/resume.py b/modules/resume.py
index 9f10086b..4b3352dc 100644
--- a/modules/resume.py
+++ b/modules/resume.py
@@ -15,6 +15,11 @@ class Module(ModuleManager.BaseModule):
     def _get_timestamp(self, server):
         return server.get_setting("last-read", None)
 
+    @utils.hook("new.server")
+    def new_server(self, event):
+        resume_timestamp = self._get_timestamp(event["server"])
+        event["server"]._resume_timestamp = resume_timestamp
+
     @utils.hook("received.cap.ls")
     def on_cap_ls(self, event):
         if CAP in event["capabilities"]:
@@ -29,18 +34,21 @@ class Module(ModuleManager.BaseModule):
     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":
             token = self._get_token(event["server"])
             self._set_token(event["server"], event["args"][1], new=True)
 
             if token:
-                timestamp = self._get_timestamp(event["server"])
+                timestamp = event["server"]._resume_timestamp
 
                 event["server"].send("RESUME %s%s" %
                     (token, " %s" % timestamp if timestamp else ""))
                 event["server"].cap_started = False
+
         event["server"].capability_done("resume")
 
     @utils.hook("received.numeric.001")