Removing transport_options parameter

Author: blast-hardcheese

src/replit_river/client_session.py

@@ -125,7 +125,7 @@ async def _handle_messages_from_ws(self) -> None:
                     if not await ws_wrapper.is_open():
                         # We should not process messages if the websocket is closed.
                         break
-                    msg = parse_transport_msg(message, self._transport_options)
+                    msg = parse_transport_msg(message)
 
                     logger.debug(f"{self._transport_id} got a message %r", msg)
 

src/replit_river/client_transport.py

@@ -294,7 +294,7 @@ async def _get_handshake_response_msg(
                     "Handshake failed, conn closed while waiting for response",
                 ) from e
             try:
-                return parse_transport_msg(data, self._transport_options)
+                return parse_transport_msg(data)
             except IgnoreMessageException:
                 logger.debug("Ignoring transport message", exc_info=True)
                 continue

src/replit_river/messages.py

@@ -60,9 +60,7 @@ def formatted_bytes(message: bytes) -> str:
     return " ".join(f"{b:02x}" for b in message)
 
 
-def parse_transport_msg(
-    message: str | bytes, transport_options: TransportOptions
-) -> TransportMessage:
+def parse_transport_msg(message: str | bytes) -> TransportMessage:
     if isinstance(message, str):
         raise IgnoreMessageException(
             f"ignored a message beacuse it was a text frame: {message}"

src/replit_river/server_session.py

@@ -122,7 +122,7 @@ async def _handle_messages_from_ws(self, tg: asyncio.TaskGroup) -> None:
                     if not await ws_wrapper.is_open():
                         # We should not process messages if the websocket is closed.
                         break
-                    msg = parse_transport_msg(message, self._transport_options)
+                    msg = parse_transport_msg(message)
 
                     logger.debug(f"{self._transport_id} got a message %r", msg)
 

src/replit_river/server_transport.py

@@ -73,7 +73,7 @@ async def handshake_to_get_session(
     ) -> ServerSession:
         async for message in websocket:
             try:
-                msg = parse_transport_msg(message, self._transport_options)
+                msg = parse_transport_msg(message)
                 (
                     handshake_request,
                     handshake_response,

src/replit_river/v2/client_session.py

@@ -158,7 +158,7 @@ async def _handle_messages_from_ws(self) -> None:
                     if not self._ws_unwrapped:
                         # We should not process messages if the websocket is closed.
                         break
-                    msg = parse_transport_msg(message, self._transport_options)
+                    msg = parse_transport_msg(message)
 
                     logger.debug(f"{self._transport_id} got a message %r", msg)
 

src/replit_river/v2/client_transport.py

@@ -281,7 +281,7 @@ async def _get_handshake_response_msg(
                     "Handshake failed, conn closed while waiting for response",
                 ) from e
             try:
-                return parse_transport_msg(data, self._transport_options)
+                return parse_transport_msg(data)
             except IgnoreMessageException:
                 logger.debug("Ignoring transport message", exc_info=True)
                 continue