From 26931425c4bfca03f22c6e06bb6dfad84a1a97c7 Mon Sep 17 00:00:00 2001
From: Luna Mendes <lkmnds@gmail.com>
Date: Sat, 22 Sep 2018 19:01:42 -0300
Subject: [PATCH] gateway.websocket: fix resuming woes

---
 README.md                     | 2 +-
 litecord/gateway/websocket.py | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index a82a055..697763e 100644
--- a/README.md
+++ b/README.md
@@ -16,7 +16,7 @@ This project is a rewrite of [litecord-reference].
 [pipenv]: https://github.com/pypa/pipenv
 
 ```sh
-$ git clone https://gitlab.com/lnmds/litecord.git && cd litecord
+$ git clone https://gitlab.com/luna/litecord.git && cd litecord
 
 # Setup the database:
 $ createdb litecord
diff --git a/litecord/gateway/websocket.py b/litecord/gateway/websocket.py
index 7d27441..e4cc4af 100644
--- a/litecord/gateway/websocket.py
+++ b/litecord/gateway/websocket.py
@@ -392,7 +392,7 @@ class GatewayWebsocket:
         })
 
         if not resumable and self.state:
-            self.state_manager.remove(self.state)
+            self.ext.state_manager.remove(self.state)
 
     async def _resume(self, replay_seqs: iter):
         presences = []
@@ -416,7 +416,7 @@ class GatewayWebsocket:
                 await self.send(payload)
         except Exception:
             log.exception('error while resuming')
-            await self.invalidate()
+            await self.invalidate_session()
             return
 
         if presences:
@@ -440,7 +440,7 @@ class GatewayWebsocket:
         try:
             state = self.ext.state_manager.fetch(user_id, sess_id)
         except KeyError:
-            return await self.invalidate(False)
+            return await self.invalidate_session(False)
 
         if seq > state.seq:
             raise WebsocketClose(4007, 'Invalid seq')
@@ -448,7 +448,7 @@ class GatewayWebsocket:
         # check if a websocket isnt on that state already
         if state.ws is not None:
             log.info('Resuming failed, websocket already connected')
-            return await self.invalidate(False)
+            return await self.invalidate_session(False)
 
         # relink this connection
         self.state = state