Fix typecheck errors and improve WebSocket transport implementation

- Fix INotifee interface compliance in WebSocket demo
- Fix handler function signatures to be async (THandler compatibility)
- Fix is_closed method usage with proper type checking
- Fix pytest.raises multiple exception type issue
- Fix line length violations (E501) across multiple files
- Add debugging logging to Noise security module for troubleshooting
- Update WebSocket transport examples and tests
- Improve transport registry error handling

examples/transport_integration_demo.py

@@ -11,13 +11,14 @@ This script demonstrates:
 
 import asyncio
 import logging
-import sys
 from pathlib import Path
+import sys
 
 # Add the libp2p directory to the path so we can import it
 sys.path.insert(0, str(Path(__file__).parent.parent))
 
 import multiaddr
+
 from libp2p.transport import (
     create_transport,
     create_transport_for_multiaddr,
@@ -25,9 +26,8 @@ from libp2p.transport import (
     get_transport_registry,
     register_transport,
 )
-from libp2p.transport.upgrader import TransportUpgrader
 from libp2p.transport.tcp.tcp import TCP
-from libp2p.transport.websocket.transport import WebsocketTransport
+from libp2p.transport.upgrader import TransportUpgrader
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -38,20 +38,21 @@ def demo_transport_registry():
     """Demonstrate the transport registry functionality."""
     print("🔧 Transport Registry Demo")
     print("=" * 50)
-    
+
     # Get the global registry
     registry = get_transport_registry()
-    
+
     # Show supported protocols
     supported = get_supported_transport_protocols()
     print(f"Supported transport protocols: {supported}")
-    
+
     # Show registered transports
     print("\nRegistered transports:")
     for protocol in supported:
         transport_class = registry.get_transport(protocol)
-        print(f"  {protocol}: {transport_class.__name__}")
-    
+        class_name = transport_class.__name__ if transport_class else "None"
+        print(f"  {protocol}: {class_name}")
+
     print()
 
 
@@ -59,21 +60,21 @@ def demo_transport_factory():
     """Demonstrate the transport factory functions."""
     print("🏭 Transport Factory Demo")
     print("=" * 50)
-    
+
     # Create a dummy upgrader for WebSocket transport
     upgrader = TransportUpgrader({}, {})
-    
+
     # Create transports using the factory function
     try:
         tcp_transport = create_transport("tcp")
         print(f"✅ Created TCP transport: {type(tcp_transport).__name__}")
-        
+
         ws_transport = create_transport("ws", upgrader)
         print(f"✅ Created WebSocket transport: {type(ws_transport).__name__}")
-        
+
     except Exception as e:
         print(f"❌ Error creating transport: {e}")
-    
+
     print()
 
 
@@ -81,10 +82,10 @@ def demo_multiaddr_transport_selection():
     """Demonstrate automatic transport selection based on multiaddrs."""
     print("🎯 Multiaddr Transport Selection Demo")
     print("=" * 50)
-    
+
     # Create a dummy upgrader
     upgrader = TransportUpgrader({}, {})
-    
+
     # Test different multiaddr types
     test_addrs = [
         "/ip4/127.0.0.1/tcp/8080",
@@ -92,20 +93,20 @@ def demo_multiaddr_transport_selection():
         "/ip6/::1/tcp/8080/ws",
         "/dns4/example.com/tcp/443/ws",
     ]
-    
+
     for addr_str in test_addrs:
         try:
             maddr = multiaddr.Multiaddr(addr_str)
             transport = create_transport_for_multiaddr(maddr, upgrader)
-            
+
             if transport:
                 print(f"✅ {addr_str} -> {type(transport).__name__}")
             else:
                 print(f"❌ {addr_str} -> No transport found")
-                
+
         except Exception as e:
             print(f"❌ {addr_str} -> Error: {e}")
-    
+
     print()
 
 
@@ -113,34 +114,37 @@ def demo_custom_transport_registration():
     """Demonstrate how to register custom transports."""
     print("🔧 Custom Transport Registration Demo")
     print("=" * 50)
-    
-    # Create a dummy upgrader
-    upgrader = TransportUpgrader({}, {})
-    
+
     # Show current supported protocols
     print(f"Before registration: {get_supported_transport_protocols()}")
-    
+
     # Register a custom transport (using TCP as an example)
     class CustomTCPTransport(TCP):
         """Custom TCP transport for demonstration."""
+
         def __init__(self):
             super().__init__()
             self.custom_flag = True
-    
+
     # Register the custom transport
     register_transport("custom_tcp", CustomTCPTransport)
-    
+
     # Show updated supported protocols
     print(f"After registration: {get_supported_transport_protocols()}")
-    
+
     # Test creating the custom transport
     try:
         custom_transport = create_transport("custom_tcp")
         print(f"✅ Created custom transport: {type(custom_transport).__name__}")
-        print(f"   Custom flag: {custom_transport.custom_flag}")
+        # Check if it has the custom flag (type-safe way)
+        if hasattr(custom_transport, "custom_flag"):
+            flag_value = getattr(custom_transport, "custom_flag", "Not found")
+            print(f"   Custom flag: {flag_value}")
+        else:
+            print("   Custom flag: Not found")
     except Exception as e:
         print(f"❌ Error creating custom transport: {e}")
-    
+
     print()
 
 
@@ -148,7 +152,7 @@ def demo_integration_with_libp2p():
     """Demonstrate how the new system integrates with libp2p."""
     print("🚀 Libp2p Integration Demo")
     print("=" * 50)
-    
+
     print("The new transport system integrates seamlessly with libp2p:")
     print()
     print("1. ✅ Automatic transport selection based on multiaddr")
@@ -157,7 +161,7 @@ def demo_integration_with_libp2p():
     print("4. ✅ Easy registration of new transport protocols")
     print("5. ✅ No changes needed to existing libp2p code")
     print()
-    
+
     print("Example usage in libp2p:")
     print("  # This will automatically use WebSocket transport")
     print("  host = new_host(listen_addrs=['/ip4/127.0.0.1/tcp/8080/ws'])")
@@ -165,7 +169,7 @@ def demo_integration_with_libp2p():
     print("  # This will automatically use TCP transport")
     print("  host = new_host(listen_addrs=['/ip4/127.0.0.1/tcp/8080'])")
     print()
-    
+
     print()
 
 
@@ -174,14 +178,14 @@ async def main():
     print("🎉 Py-libp2p Transport Integration Demo")
     print("=" * 60)
     print()
-    
+
     # Run all demos
     demo_transport_registry()
     demo_transport_factory()
     demo_multiaddr_transport_selection()
     demo_custom_transport_registration()
     demo_integration_with_libp2p()
-    
+
     print("🎯 Summary of New Features:")
     print("=" * 40)
     print("✅ Transport Registry: Central registry for all transport implementations")
@@ -202,4 +206,5 @@ if __name__ == "__main__":
     except Exception as e:
         print(f"\n❌ Demo failed with error: {e}")
         import traceback
+
         traceback.print_exc()