feat: allow bot to recover from more error states

Author: LordOfPolls

interactions/api/gateway/websocket.py

@@ -9,6 +9,7 @@
 
 from aiohttp import WSMsgType
 
+from interactions.client import const
 from interactions.client.errors import WebSocketClosed
 from interactions.client.utils.input_utils import FastJson
 from interactions.models.internal.cooldowns import CooldownSystem
@@ -185,15 +186,16 @@ async def receive(self, force: bool = False) -> str:
 
             if resp.type == WSMsgType.CLOSE:
                 self.logger.debug(f"Disconnecting from gateway! Reason: {resp.data}::{resp.extra}")
-                if resp.data >= 4000:
+                code = int(resp.data)
+                if code not in const.RECOVERABLE_WEBSOCKET_CLOSE_CODES:
                     # This should propagate to __aexit__() which will forcefully shut down everything
                     # and cleanup correctly.
-                    raise WebSocketClosed(resp.data)
+                    raise WebSocketClosed(code)
 
                 if force:
                     raise RuntimeError("Discord unexpectedly wants to close the WebSocket during force receive!")
 
-                await self.reconnect(code=resp.data, resume=resp.data != 1000)
+                await self.reconnect(code=code, resume=code not in const.NON_RESUMABLE_WEBSOCKET_CLOSE_CODES)
                 continue
 
             if resp.type is WSMsgType.CLOSED:

interactions/client/const.py

@@ -76,6 +76,8 @@
     "T",
     "T_co",
     "LIB_PATH",
+    "RECOVERABLE_WEBSOCKET_CLOSE_CODES",
+    "NON_RESUMABLE_WEBSOCKET_CLOSE_CODES",
 )
 
 _ver_info = sys.version_info
@@ -212,3 +214,6 @@ class MentionPrefix(Sentinel):
 
 LIB_PATH = os.sep.join(__file__.split(os.sep)[:-2])
 """The path to the library folder."""
+
+RECOVERABLE_WEBSOCKET_CLOSE_CODES = (4000, 4001, 4002, 4003, 4005, 4007, 4008, 4009)
+NON_RESUMABLE_WEBSOCKET_CLOSE_CODES = (1000, 4007)