Attempt to improve the ws connection

rustplus/__init__.py

@@ -10,5 +10,5 @@
 
 __name__ = "rustplus"
 __author__ = "olijefferson"
-__version__ = "5.1.9"
+__version__ = "5.1.10"
 __support__ = "Discord: https://discord.gg/nQqJe8qvP8"

rustplus/api/remote/rust_remote_interface.py

@@ -39,6 +39,7 @@ def close(self) -> None:
 
         if self.ws is not None:
             self.ws.close()
+            del self.ws
             self.ws = None
             self.open = False
 

rustplus/api/remote/rustws.py

@@ -1,8 +1,12 @@
 import asyncio
 import logging
 import time
+import socket
+import errno
 from typing import Optional
+from ws4py import exc
 from ws4py.client.threadedclient import WebSocketClient
+from ws4py.websocket import pyOpenSSLError
 
 from .rustplus_pb2 import AppMessage, AppRequest
 from ..structures import RustChatMessage
@@ -184,3 +188,39 @@ async def get_response(self, seq : int, app_request : AppRequest, retry_depth :
             raise RequestError(response.response.error.error)
 
         return response
+
+    def once(self):
+
+        if self.terminated:
+            logging.getLogger("ws4py").debug("WebSocket is already terminated")
+            return False
+        try:
+            b = b''
+            if self._is_secure:
+                b = self._get_from_pending()
+            if not b and not self.buf:
+                try:
+                    b = self.sock.recv(self.reading_buffer_size)
+                except OSError:
+                    self.connect()
+                    return True
+            if not b and not self.buf:
+                return False
+            self.buf += b
+        except (socket.error, OSError, pyOpenSSLError) as e:
+            if hasattr(e, "errno") and e.errno == errno.EINTR:
+                pass
+            else:
+                self.unhandled_error(e)
+                return False
+        else:
+            # process as much as we can
+            # the process will stop either if there is no buffer left
+            # or if the stream is closed
+            # only pass the requested number of bytes, leave the rest in the buffer
+            requested = self.reading_buffer_size
+            if not self.process(self.buf[:requested]):
+                return False
+            self.buf = self.buf[requested:]
+
+        return True