Add message.proto and fix a few mypy-ci errors

Author: sukhman-sukh

libp2p/transport/webrtc/async_bridge.py

@@ -38,7 +38,8 @@ async def __aenter__(self) -> "WebRTCAsyncBridge":
         """Enter async context manager"""
         if not self._in_context:
             self._loop_context = open_loop()
-            await self._loop_context.__aenter__()
+            if self._loop_context:
+                await self._loop_context.__aenter__()
             self._in_context = True
         return self
 

libp2p/transport/webrtc/connection.py

@@ -56,10 +56,10 @@ def get_muxed_conn(self) -> IMuxedConn:
         """Get the underlying muxed connection."""
         return cast(IMuxedConn, self.connection)
 
-    def set_protocol(self, protocol: TProtocol) -> None:
+    def set_protocol(self, protocol_id: TProtocol) -> None:
         """Set the protocol for this stream."""
-        self.protocol = protocol
-        logger.debug(f"Stream {self.stream_id} set protocol: {protocol}")
+        self.protocol = protocol_id
+        logger.debug(f"Stream {self.stream_id} set protocol: {protocol_id}")
 
     def get_protocol(self) -> TProtocol | None:
         """Get the protocol for this stream."""

libp2p/transport/webrtc/pb/message.proto

@@ -0,0 +1,16 @@
+syntax = "proto3";
+
+message Message {
+    // Specifies type in `data` field.
+    enum Type {
+        // String of `RTCSessionDescription.sdp`
+        SDP_OFFER = 0;
+        // String of `RTCSessionDescription.sdp`
+        SDP_ANSWER = 1;
+        // String of `RTCIceCandidate.toJSON()`
+        ICE_CANDIDATE = 2;
+    }
+
+    optional Type type = 1;
+    optional string data = 2;
+}

libp2p/transport/webrtc/private_to_public/transport.py

@@ -186,7 +186,7 @@ async def dial(self, maddr: Multiaddr) -> IRawConnection:
             peer_id = (
                 peer_id_str
                 if isinstance(peer_id_str, ID)
-                else ID.from_base58(peer_id_str)
+                else ID.from_base58(str(peer_id_str))
             )
 
             # Extract port from multiaddr