mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2025-12-31 20:36:24 +00:00
396812e84a
- Implemented full WSS support with TLS configuration - Added handshake timeout and connection state tracking - Created comprehensive test suite with 13+ WSS unit tests - Added Python-to-Python WebSocket peer-to-peer tests - Implemented multiaddr parsing for /ws, /wss, /tls/ws formats - Added connection state tracking and concurrent close handling - Created standalone WebSocket client for testing - Fixed circular import issues with multiaddr utilities - Added debug tools for WebSocket URL testing All WebSocket transport functionality is complete and working. Tests demonstrate WebSocket transport works correctly at the transport layer. Higher-level libp2p protocol compatibility issues remain (same as JS interop).
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
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,
|
|
)
|
|
from .upgrader import TransportUpgrader
|
|
from libp2p.abc import ITransport
|
|
|
|
def create_transport(protocol: str, upgrader: TransportUpgrader | None = None, **kwargs) -> ITransport:
|
|
"""
|
|
Convenience function to create a transport instance.
|
|
|
|
:param protocol: The transport protocol ("tcp", "ws", "wss", or custom)
|
|
:param upgrader: Optional transport upgrader (required for WebSocket)
|
|
:param kwargs: Additional arguments for transport construction (e.g., tls_client_config, tls_server_config)
|
|
:return: Transport instance
|
|
"""
|
|
# First check if it's a built-in protocol
|
|
if protocol in ["ws", "wss"]:
|
|
if upgrader is None:
|
|
raise ValueError(f"WebSocket transport requires an upgrader")
|
|
return WebsocketTransport(
|
|
upgrader,
|
|
tls_client_config=kwargs.get("tls_client_config"),
|
|
tls_server_config=kwargs.get("tls_server_config"),
|
|
handshake_timeout=kwargs.get("handshake_timeout", 15.0)
|
|
)
|
|
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:
|
|
transport = registry.create_transport(protocol, upgrader, **kwargs)
|
|
if transport is None:
|
|
raise ValueError(f"Failed to create transport for protocol: {protocol}")
|
|
return transport
|
|
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",
|
|
]
|