mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2025-12-31 20:36:24 +00:00
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from peer.peerstore import PeerStore
|
|
from network.swarm import Swarm
|
|
from host.basic_host import BasicHost
|
|
from transport.upgrader import TransportUpgrader
|
|
from transport.tcp.tcp import TCP
|
|
from Crypto.PublicKey import RSA
|
|
|
|
class Libp2p(object):
|
|
|
|
def __init__(self, idOpt = None, \
|
|
transportOpt = ["/ip4/127.0.0.1/tcp/8001"], \
|
|
muxerOpt = ["mplex/6.7.0"], \
|
|
secOpt = ["secio"], \
|
|
peerstore = PeerStore()):
|
|
|
|
if idOpt:
|
|
self.idOpt = idOpt
|
|
else:
|
|
new_key = RSA.generate(2048, e=65537)
|
|
self.idOpt = new_key.publickey().exportKey("PEM")
|
|
self.private_key = new_key.exportKey("PEM")
|
|
|
|
self.transportOpt = transportOpt
|
|
self.muxerOpt = muxerOpt
|
|
self.secOpt = secOpt
|
|
self.peerstore = peerstore
|
|
|
|
async def new_node(self):
|
|
|
|
upgrader = TransportUpgrader(self.secOpt, self.transportOpt)
|
|
swarm = Swarm(self.idOpt, self.peerstore, upgrader)
|
|
tcp = TCP()
|
|
swarm.add_transport(tcp)
|
|
await swarm.listen(self.transportOpt[0])
|
|
host = BasicHost(swarm)
|
|
|
|
# TODO MuxedConnection currently contains all muxing logic (move to a Muxer)
|
|
# TODO routing unimplemented
|
|
return host
|