matu
/
Heroku
mirror of https://github.com/coddrago/Heroku
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Automatically revoke HikkaAPI token if it expires \ becomes invalid

pull/1/head
hikari.ftg 2022-04-07 09:14:27 +00:00
parent db14892ce2
commit 0139b7bd9c
1 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
hikka/modules/hikka_dl.py
View File

@ -71,26 +71,31 @@ class HikkaDLMod(loader.Module):
while True:
try:
await self._wss()
except websockets.exceptions.ConnectionClosedError:
await self._get_token()
except Exception:
logger.debug("Socket disconnected, retry in 10 sec")
await asyncio.sleep(10)
async def _get_token(self) -> None:
async with self._client.conversation(self._bot) as conv:
m = await conv.send_message("/token")
r = await conv.get_response()
token = r.raw_text
await m.delete()
await r.delete()
if not token.startswith("kirito_") and not token.startswith("asuna_"):
raise LoadError("Can't get token")
self.set("token", token)
async def client_ready(self, client, db) -> None:
self._db = db
self._client = client
self._bot = "@hikka_userbot"
if not self.get("token"):
async with client.conversation(self._bot) as conv:
m = await conv.send_message("/token")
r = await conv.get_response()
token = r.raw_text
await m.delete()
await r.delete()
if not token.startswith("kirito_") and not token.startswith("asuna_"):
raise LoadError("Can't get token")
self.set("token", token)
await self._get_token()
self._task = asyncio.ensure_future(self._connect())