From 84c9ddc2ddf6168d04604488b9676be5d89f6be0 Mon Sep 17 00:00:00 2001
From: Akash Mondal <akmo3901@gmail.com>
Date: Mon, 14 Jul 2025 03:32:44 +0000
Subject: [PATCH] chore: cleanup and doc gen fixes

---
 libp2p/transport/quic/exceptions.py | 10 ++++------
 libp2p/transport/quic/listener.py   |  8 +-------
 libp2p/transport/quic/security.py   | 21 +++------------------
 libp2p/transport/quic/transport.py  | 13 ++-----------
 4 files changed, 10 insertions(+), 42 deletions(-)

diff --git a/libp2p/transport/quic/exceptions.py b/libp2p/transport/quic/exceptions.py
index 643b2edf..2df3dda5 100644
--- a/libp2p/transport/quic/exceptions.py
+++ b/libp2p/transport/quic/exceptions.py
@@ -1,10 +1,8 @@
-from typing import Any, Literal
+"""
+QUIC Transport exceptions
+"""
 
-"""
-QUIC Transport exceptions for py-libp2p.
-Comprehensive error handling for QUIC transport, connection, and stream operations.
-Based on patterns from go-libp2p and js-libp2p implementations.
-"""
+from typing import Any, Literal
 
 
 class QUICError(Exception):
diff --git a/libp2p/transport/quic/listener.py b/libp2p/transport/quic/listener.py
index b1c13562..466f4b6d 100644
--- a/libp2p/transport/quic/listener.py
+++ b/libp2p/transport/quic/listener.py
@@ -70,13 +70,7 @@ class QUICPacketInfo:
 
 class QUICListener(IListener):
     """
-    Enhanced QUIC Listener with proper connection ID handling and protocol negotiation.
-
-    Key improvements:
-    - Proper QUIC packet parsing to extract connection IDs
-    - Version negotiation following RFC 9000
-    - Connection routing based on destination connection ID
-    - Support for connection migration
+    QUIC Listener with connection ID handling and protocol negotiation.
     """
 
     def __init__(
diff --git a/libp2p/transport/quic/security.py b/libp2p/transport/quic/security.py
index 568514d5..08719863 100644
--- a/libp2p/transport/quic/security.py
+++ b/libp2p/transport/quic/security.py
@@ -1,7 +1,5 @@
 """
-QUIC Security implementation for py-libp2p Module 5.
-Implements libp2p TLS specification for QUIC transport with peer identity integration.
-Based on go-libp2p and js-libp2p security patterns.
+QUIC Security helpers implementation
 """
 
 from dataclasses import dataclass, field
@@ -854,7 +852,7 @@ def create_server_tls_config(
         certificate: X.509 certificate
         private_key: Private key corresponding to certificate
         peer_id: Optional peer ID for validation
-        **kwargs: Additional configuration parameters
+        kwargs: Additional configuration parameters
 
     Returns:
         Server TLS configuration
@@ -886,7 +884,7 @@ def create_client_tls_config(
         certificate: X.509 certificate
         private_key: Private key corresponding to certificate
         peer_id: Optional peer ID for validation
-        **kwargs: Additional configuration parameters
+        kwargs: Additional configuration parameters
 
     Returns:
         Client TLS configuration
@@ -935,7 +933,6 @@ class QUICTLSConfigManager:
             peer_id=self.peer_id,
         )
 
-        print("🔧 SECURITY: Created server config")
         return config
 
     def create_client_config(self) -> QUICTLSSecurityConfig:
@@ -952,7 +949,6 @@ class QUICTLSConfigManager:
             peer_id=self.peer_id,
         )
 
-        print("🔧 SECURITY: Created client config")
         return config
 
     def verify_peer_identity(
@@ -1011,14 +1007,3 @@ def generate_libp2p_tls_config(private_key: PrivateKey, peer_id: ID) -> TLSConfi
     """
     generator = CertificateGenerator()
     return generator.generate_certificate(private_key, peer_id)
-
-
-def cleanup_tls_config(config: TLSConfig) -> None:
-    """
-    Clean up TLS configuration.
-
-    For the new implementation, this is mostly a no-op since we don't use
-    temporary files, but kept for compatibility.
-    """
-    # New implementation doesn't use temporary files
-    print("TLS config cleanup completed")
diff --git a/libp2p/transport/quic/transport.py b/libp2p/transport/quic/transport.py
index 65146eca..f577b574 100644
--- a/libp2p/transport/quic/transport.py
+++ b/libp2p/transport/quic/transport.py
@@ -1,8 +1,5 @@
 """
-QUIC Transport implementation for py-libp2p with integrated security.
-Uses aioquic's sans-IO core with trio for native async support.
-Based on aioquic library with interface consistency to go-libp2p and js-libp2p.
-Updated to include Module 5 security integration.
+QUIC Transport implementation
 """
 
 import copy
@@ -79,13 +76,7 @@ logger = logging.getLogger(__name__)
 
 class QUICTransport(ITransport):
     """
-    QUIC Transport implementation following libp2p transport interface.
-
-    Uses aioquic's sans-IO core with trio for native async support.
-    Supports both QUIC v1 (RFC 9000) and draft-29 for compatibility with
-    go-libp2p and js-libp2p implementations.
-
-    Includes integrated libp2p TLS security with peer identity verification.
+    QUIC Stream implementation following libp2p IMuxedStream interface.
     """
 
     def __init__(