added WebSocket transport support

Signed-off-by: GautamBytes <manchandanigautam@gmail.com>
This commit is contained in:
GautamBytes
2025-07-20 09:23:42 +00:00
parent 3507531344
commit 187418378a
12 changed files with 401 additions and 0 deletions

View File

View File

@ -0,0 +1,18 @@
{
"name": "src",
"version": "1.0.0",
"main": "ping.js",
"scripts": {
"test": "echo \"Error: no test specified\" && exit 1"
},
"keywords": [],
"author": "",
"license": "ISC",
"description": "",
"dependencies": {
"@libp2p/ping": "^2.0.36",
"@libp2p/websockets": "^9.2.18",
"libp2p": "^2.9.0",
"multiaddr": "^10.0.1"
}
}

View File

@ -0,0 +1,35 @@
import { createLibp2p } from 'libp2p'
import { webSockets } from '@libp2p/websockets'
import { ping } from '@libp2p/ping'
import { plaintext } from '@libp2p/insecure'
import { mplex } from '@libp2p/mplex'
async function main() {
const node = await createLibp2p({
transports: [ webSockets() ],
connectionEncryption: [ plaintext() ],
streamMuxers: [ mplex() ],
services: {
// installs /ipfs/ping/1.0.0 handler
ping: ping()
},
addresses: {
listen: ['/ip4/127.0.0.1/tcp/0/ws']
}
})
await node.start()
console.log(node.peerId.toString())
for (const addr of node.getMultiaddrs()) {
console.log(addr.toString())
}
// Keep the process alive
await new Promise(() => {})
}
main().catch(err => {
console.error(err)
process.exit(1)
})

View File

@ -0,0 +1,85 @@
import os
import signal
import subprocess
import pytest
from multiaddr import Multiaddr
import trio
from trio.lowlevel import open_process
from libp2p.crypto.secp256k1 import create_new_key_pair
from libp2p.custom_types import TProtocol
from libp2p.host.basic_host import BasicHost
from libp2p.network.swarm import Swarm
from libp2p.peer.id import ID
from libp2p.peer.peerinfo import PeerInfo
from libp2p.peer.peerstore import PeerStore
from libp2p.security.insecure.transport import InsecureTransport
from libp2p.stream_muxer.mplex.mplex import MPLEX_PROTOCOL_ID, Mplex
from libp2p.transport.upgrader import TransportUpgrader
from libp2p.transport.websocket.transport import WebsocketTransport
PLAINTEXT_PROTOCOL_ID = "/plaintext/1.0.0"
@pytest.mark.trio
async def test_ping_with_js_node():
# 1) Path to the JS node script
js_node_dir = os.path.join(os.path.dirname(__file__), "js_libp2p", "js_node", "src")
script_name = "ws_ping_node.mjs"
# 2) Launch the JS libp2p node (long-running)
proc = await open_process(
["node", script_name],
stdout=subprocess.PIPE,
stderr=subprocess.DEVNULL,
cwd=js_node_dir,
)
try:
# 3) Read first two lines (PeerID and multiaddr)
buffer = b""
with trio.fail_after(10):
while buffer.count(b"\n") < 2:
chunk = await proc.stdout.receive_some(1024) # type: ignore
if not chunk:
break
buffer += chunk
lines = buffer.decode().strip().split("\n")
peer_id_line, addr_line = lines[0], lines[1]
peer_id = ID.from_base58(peer_id_line)
maddr = Multiaddr(addr_line)
# 4) Set up Python host
key_pair = create_new_key_pair()
py_peer_id = ID.from_pubkey(key_pair.public_key)
peer_store = PeerStore()
peer_store.add_key_pair(py_peer_id, key_pair)
upgrader = TransportUpgrader(
secure_transports_by_protocol={
TProtocol(PLAINTEXT_PROTOCOL_ID): InsecureTransport(key_pair)
},
muxer_transports_by_protocol={TProtocol(MPLEX_PROTOCOL_ID): Mplex},
)
transport = WebsocketTransport()
swarm = Swarm(py_peer_id, peer_store, upgrader, transport)
host = BasicHost(swarm)
# 5) Connect to JS node
peer_info = PeerInfo(peer_id, [maddr])
await host.connect(peer_info)
assert host.get_network().connections.get(peer_id) is not None
await trio.sleep(0.1)
# 6) Ping protocol
stream = await host.new_stream(peer_id, [TProtocol("/ipfs/ping/1.0.0")])
await stream.write(b"ping")
data = await stream.read(4)
assert data == b"pong"
# 7) Cleanup
await host.close()
finally:
proc.send_signal(signal.SIGTERM)
await trio.sleep(0)