feat: identify-push protocol. initial release

examples/identify_push/identify_push_example.py

@@ -0,0 +1,97 @@
+#!/usr/bin/env python3
+"""
+Example demonstrating the identify/push protocol.
+
+This example shows how to:
+1. Set up a host with the identify/push protocol handler
+2. Connect to another peer
+3. Push identify information to the peer
+4. Receive and process identify/push messages
+"""
+
+import logging
+
+import trio
+
+from libp2p import (
+    new_host,
+)
+from libp2p.crypto.secp256k1 import (
+    create_new_key_pair,
+)
+from libp2p.custom_types import (
+    TProtocol,
+)
+from libp2p.identity.identify import (
+    identify_handler_for,
+)
+from libp2p.identity.identify_push import (
+    ID_PUSH,
+    identify_push_handler_for,
+    push_identify_to_peer,
+)
+from libp2p.peer.peerinfo import (
+    info_from_p2p_addr,
+)
+
+# Configure logging
+logging.basicConfig(
+    level=logging.DEBUG,
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+)
+logger = logging.getLogger(__name__)
+
+
+async def main() -> None:
+    # Create key pairs for the two hosts
+    key_pair_1 = create_new_key_pair()
+    key_pair_2 = create_new_key_pair()
+
+    # Create the first host
+    host_1 = new_host(key_pair=key_pair_1)
+
+    # Set up the identify and identify/push handlers
+    host_1.set_stream_handler(TProtocol("/ipfs/id/1.0.0"), identify_handler_for(host_1))
+    host_1.set_stream_handler(ID_PUSH, identify_push_handler_for(host_1))
+
+    # Create the second host
+    host_2 = new_host(key_pair=key_pair_2)
+
+    # Set up the identify and identify/push handlers
+    host_2.set_stream_handler(TProtocol("/ipfs/id/1.0.0"), identify_handler_for(host_2))
+    host_2.set_stream_handler(ID_PUSH, identify_push_handler_for(host_2))
+
+    # Start listening on random ports using the run context manager
+    import multiaddr
+
+    listen_addr_1 = multiaddr.Multiaddr("/ip4/127.0.0.1/tcp/0")
+    listen_addr_2 = multiaddr.Multiaddr("/ip4/127.0.0.1/tcp/0")
+
+    async with host_1.run([listen_addr_1]), host_2.run([listen_addr_2]):
+        # Get the addresses of both hosts
+        addr_1 = host_1.get_addrs()[0]
+        logger.info("Host 1 listening on %s", addr_1)
+
+        addr_2 = host_2.get_addrs()[0]
+        logger.info("Host 2 listening on %s", addr_2)
+
+        # Connect host_2 to host_1
+        peer_info = info_from_p2p_addr(addr_1)
+        await host_2.connect(peer_info)
+        logger.info("Host 2 connected to Host 1")
+
+        # Wait a bit for the connection to establish
+        await trio.sleep(1)
+
+        # Push identify information from host_1 to host_2
+        logger.info("Host 1 pushing identify information to Host 2")
+        await push_identify_to_peer(host_1, host_2.get_id())
+
+        # Wait a bit for the push to complete
+        await trio.sleep(1)
+
+        logger.info("Example completed successfully")
+
+
+if __name__ == "__main__":
+    trio.run(main)