websocket.py: add ping callback support

ThomasFarstrike · ThomasFarstrike · commit db925ce9fbb8 · 2025-05-30T21:39:19.000+02:00
diff --git a/internal_filesystem/lib/websocket.py b/internal_filesystem/lib/websocket.py
@@ -328,6 +328,9 @@ async def _connect_and_run(self):
                 elif msg.type == WSMsgType.ERROR or ws.ws.closed:
                     _log_error("WebSocket error or closed")
                     raise WebSocketConnectionClosedException("WebSocket closed")
+                elif msg.type == ABNF.OPCODE_PING:
+                    data = msg.data
+                    _run_callback(self.on_ping, self, data, ABNF.OPCODE_PING, True)
 
     async def _send_async(self, data, opcode):
         """Async send implementation."""
