From fea26d90875e62f7de03a121acac78bac5dd4bc1 Mon Sep 17 00:00:00 2001
From: zixuanzh <zixuanzh@seas.upenn.edu>
Date: Sun, 17 Mar 2019 19:20:46 -0400
Subject: [PATCH] refactor new_node

---
 libp2p/__init__.py | 60 +++++++++++++++++++++++++++++++++++-----------
 1 file changed, 46 insertions(+), 14 deletions(-)

diff --git a/libp2p/__init__.py b/libp2p/__init__.py
index 0fa2fdb..d83ec3c 100644
--- a/libp2p/__init__.py
+++ b/libp2p/__init__.py
@@ -11,6 +11,9 @@ from .transport.tcp.tcp import TCP
 
 
 async def cleanup_done_tasks():
+    """
+    clean up asyncio done tasks to free up resources
+    """
     while True:
         for task in asyncio.all_tasks():
             if task.done():
@@ -20,30 +23,59 @@ async def cleanup_done_tasks():
         # Some sleep necessary to context switch
         await asyncio.sleep(3)
 
-async def new_node(
+def initialize_default_swarm(
         id_opt=None, transport_opt=None,
-        muxer_opt=None, sec_opt=None, peerstore=None):
-
-    if id_opt is None:
+        muxer_opt=None, sec_opt=None, peerstore_opt=None):
+    """
+    initialize swarm when no swarm is passed in
+    :param id_opt: optional id for host
+    :param transport_opt: optional choice of transport upgrade
+    :param muxer_opt: optional choice of stream muxer
+    :param sec_opt: optional choice of security upgrade
+    :param peerstore_opt: optional peerstore
+    :return: return a default swarm instance
+    """
+    # pylint: disable=too-many-arguments
+    if not id_opt:
         new_key = RSA.generate(2048, e=65537)
         id_opt = id_from_public_key(new_key.publickey())
         # private_key = new_key.exportKey("PEM")
 
     transport_opt = transport_opt or ["/ip4/127.0.0.1/tcp/8001"]
-    transport_opt = [multiaddr.Multiaddr(t) for t in transport_opt]
-    muxer_opt = muxer_opt or ["mplex/6.7.0"]
-    sec_opt = sec_opt or ["secio"]
-    peerstore = peerstore or PeerStore()
-
-    upgrader = TransportUpgrader(sec_opt, transport_opt)
-    swarm = Swarm(id_opt, peerstore, upgrader)
+    transport = [multiaddr.Multiaddr(t) for t in transport_opt]
+    # TODO wire muxer up with swarm
+    # muxer = muxer_opt or ["mplex/6.7.0"]
+    sec = sec_opt or ["secio"]
+    peerstore = peerstore_opt or PeerStore()
+    upgrader = TransportUpgrader(sec, transport)
+    swarm_opt = Swarm(id_opt, peerstore, upgrader)
     tcp = TCP()
-    swarm.add_transport(tcp)
-    await swarm.listen(transport_opt[0])
+    swarm_opt.add_transport(tcp)
+
+    return swarm_opt
+
+async def new_node(
+        swarm_opt=None, id_opt=None, transport_opt=None,
+        muxer_opt=None, sec_opt=None, peerstore_opt=None):
+    """
+    create new libp2p node
+    :param id_opt: optional id for host
+    :param transport_opt: optional choice of transport upgrade
+    :param muxer_opt: optional choice of stream muxer
+    :param sec_opt: optional choice of security upgrade
+    :param peerstore_opt: optional peerstore
+    :return: return a default swarm instance
+    """
+    # pylint: disable=too-many-arguments
+    if not swarm_opt:
+        swarm_opt = initialize_default_swarm(
+            id_opt=id_opt, transport_opt=transport_opt,
+            muxer_opt=muxer_opt, sec_opt=sec_opt,
+            peerstore_opt=peerstore_opt)
 
     # TODO enable support for other host type
     # TODO routing unimplemented
-    host = BasicHost(swarm)
+    host = BasicHost(swarm_opt)
 
     # Kick off cleanup job
     asyncio.ensure_future(cleanup_done_tasks())