feat: implement WebSocket transport with transport registry system - Add transport_registry.py for centralized transport management - Integrate WebSocket transport with new registry - Add comprehensive test suite for transport registry - Include WebSocket examples and demos - Update transport initialization and swarm integration

libp2p/transport/__init__.py

@@ -1,7 +1,44 @@
 from .tcp.tcp import TCP
 from .websocket.transport import WebsocketTransport
+from .transport_registry import (
+    TransportRegistry, 
+    create_transport_for_multiaddr,
+    get_transport_registry,
+    register_transport,
+    get_supported_transport_protocols,
+)
+
+def create_transport(protocol: str, upgrader=None):
+    """
+    Convenience function to create a transport instance.
+    
+    :param protocol: The transport protocol ("tcp", "ws", or custom)
+    :param upgrader: Optional transport upgrader (required for WebSocket)
+    :return: Transport instance
+    """
+    # First check if it's a built-in protocol
+    if protocol == "ws":
+        if upgrader is None:
+            raise ValueError(f"WebSocket transport requires an upgrader")
+        return WebsocketTransport(upgrader)
+    elif protocol == "tcp":
+        return TCP()
+    else:
+        # Check if it's a custom registered transport
+        registry = get_transport_registry()
+        transport_class = registry.get_transport(protocol)
+        if transport_class:
+            return registry.create_transport(protocol, upgrader)
+        else:
+            raise ValueError(f"Unsupported transport protocol: {protocol}")
 
 __all__ = [
     "TCP",
     "WebsocketTransport",
+    "TransportRegistry",
+    "create_transport_for_multiaddr",
+    "create_transport",
+    "get_transport_registry",
+    "register_transport",
+    "get_supported_transport_protocols",
 ]