Bootstrapping rework in progress.

1 files changed, 7 insertions, 2 deletions

diff --git a/examples/dhtd.hs b/examples/dhtd.hs
index d1db1938..7d3661e6 100644
--- a/examples/dhtd.hs
+++ b/examples/dhtd.hs
@@ -61,7 +61,7 @@ import Network.Address hiding (NodeId, NodeInfo(..))
 import Network.QueryResponse
 import Network.StreamServer
 import Network.Kademlia
-import Network.Kademlia.Bootstrap (bootstrap)
+import Network.Kademlia.Bootstrap
 import Network.Kademlia.Search
 import qualified Network.BitTorrent.MainlineDHT as Mainline
 import qualified Network.Tox as Tox
@@ -1323,7 +1323,12 @@ main = do
             case Map.lookup "node" qrys of
                 Just DHTQuery { qsearch = srch } -> do
                     case eqT of
-                        Just witness -> bootstrap (isNodesSearch witness srch) bkts ping btSaved fallbackNodes
+                        Just witness -> let strapper = BucketRefresher
+                                                { refreshSearch = isNodesSearch witness srch
+                                                , refreshBuckets = bkts
+                                                , refreshPing = ping
+                                                }
+                                         in bootstrap strapper btSaved fallbackNodes
                         _            -> error $ "Missing node-search for "++netname++"."
                     saveNodes netname dht
                 Nothing -> return ()