38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from Crypto.PublicKey import RSA
|
|
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
|
|
|
|
|
|
class Libp2p():
|
|
|
|
def __init__(self, id_opt=None, transport_opt=["/ip4/127.0.0.1/tcp/8001"], \
|
|
muxer_opt=["mplex/6.7.0"], sec_opt=["secio"], peerstore=PeerStore()):
|
|
|
|
if id_opt:
|
|
self.id_opt = id_opt
|
|
else:
|
|
new_key = RSA.generate(2048, e=65537)
|
|
self.id_opt = new_key.publickey().exportKey("PEM")
|
|
self.private_key = new_key.exportKey("PEM")
|
|
|
|
self.transport_opt = transport_opt
|
|
self.muxer_opt = muxer_opt
|
|
self.sec_opt = sec_opt
|
|
self.peerstore = peerstore
|
|
|
|
async def new_node(self):
|
|
|
|
upgrader = TransportUpgrader(self.sec_opt, self.transport_opt)
|
|
swarm = Swarm(self.id_opt, self.peerstore, upgrader)
|
|
tcp = TCP()
|
|
swarm.add_transport(tcp)
|
|
await swarm.listen(self.transport_opt[0])
|
|
host = BasicHost(swarm)
|
|
|
|
# TODO MuxedConnection currently contains all muxing logic (move to a Muxer)
|
|
# TODO routing unimplemented
|
|
return host
|