Acknowledge client/server protocol split

Author: blast-hardcheese

replit_river/client_transport.py

@@ -18,7 +18,7 @@
     RiverException,
 )
 from replit_river.messages import (
-    PROTOCOL_VERSION,
+    CLIENT_PROTOCOL_VERSION,
     FailedSendingMessageException,
     WebsocketClosedException,
     parse_transport_msg,
@@ -227,7 +227,7 @@ async def _send_handshake_request(
     ) -> ControlMessageHandshakeRequest[HandshakeMetadataType]:
         handshake_request = ControlMessageHandshakeRequest[HandshakeMetadataType](
             type="HANDSHAKE_REQ",
-            protocolVersion=PROTOCOL_VERSION,
+            protocolVersion=CLIENT_PROTOCOL_VERSION,
             sessionId=session_id,
             metadata=handshake_metadata,
             expectedSessionState=expected_session_state,

replit_river/messages.py

@@ -29,7 +29,8 @@ class FailedSendingMessageException(Exception):
     pass
 
 
-PROTOCOL_VERSION = "v1.1"
+CLIENT_PROTOCOL_VERSION = "v2.0"
+SERVER_PROTOCOL_VERSION = "v1.1"
 
 
 async def send_transport_message(

replit_river/server_transport.py

@@ -10,7 +10,7 @@
 from websockets.exceptions import ConnectionClosed
 
 from replit_river.messages import (
-    PROTOCOL_VERSION,
+    SERVER_PROTOCOL_VERSION,
     FailedSendingMessageException,
     WebsocketClosedException,
     parse_transport_msg,
@@ -209,15 +209,15 @@ async def _establish_handshake(
             )
             raise InvalidMessageException("failed validate handshake request") from e
 
-        if handshake_request.protocolVersion != PROTOCOL_VERSION:
+        if handshake_request.protocolVersion != SERVER_PROTOCOL_VERSION:
             await self._send_handshake_response(
                 request_message,
                 HandShakeStatus(ok=False, reason="protocol version mismatch"),
                 websocket,
             )
             error_str = (
                 "protocol version mismatch: "
-                + f"{handshake_request.protocolVersion} != {PROTOCOL_VERSION}"
+                + f"{handshake_request.protocolVersion} != {SERVER_PROTOCOL_VERSION}"
             )
             raise InvalidMessageException(error_str)
         if request_message.to != self._transport_id: