1 files changed, 25 insertions, 7 deletions

diff --git a/examples/dhtd.hs b/examples/dhtd.hs
index ef225f0c..a1b8d665 100644
--- a/examples/dhtd.hs
+++ b/examples/dhtd.hs
@@ -49,6 +49,7 @@ import Network.QueryResponse
 import Network.StreamServer
 import Kademlia
 import qualified Mainline
+import qualified Tox
 import Network.DHT.Routing as R
 import Data.Aeson as J (ToJSON, FromJSON)
 import qualified Data.Aeson as J
@@ -280,7 +281,7 @@ clientSession s@Session{..} sock cnum h = do
                     pid <- getProcessID
                     hPutClient h (show pid)
         ("external-ip", _) -> cmd0 $ do
-                    unlines . map (either show show . Mainline.either4or6) <$> externalAddresses
+                    unlines . map (either show show . either4or6) <$> externalAddresses
                         >>= hPutClient h
 #ifdef THREAD_DEBUG
         ("threads", _) -> cmd0 $ do
@@ -470,10 +471,16 @@ main = do
 
     quitBt <- forkListener bt
 
-    tox <- return $ error "TODO: Tox.newClient"
-    quitTox <- return $ return () -- TODO: forkListener tox
+    let toxport = succ $ fromMaybe 33445 (fromIntegral <$> sockAddrPort addr)
+    addrTox <- getBindAddress (show toxport) True
+    (tox,toxR) <- Tox.newClient addrTox
+
+    -- TODO: load saved tox nodes.
+
+    quitTox <- forkListener tox
 
     mainlineSearches <- atomically $ newTVar Map.empty
+    toxSearches <- atomically $ newTVar Map.empty
 
     let mainlineDHT bkts = DHT
             { dhtBuckets = bkts btR
@@ -482,13 +489,13 @@ main = do
                             [ ("node", DHTQuery (Mainline.nodeSearch bt)
                                                 (\ni -> fmap Mainline.unwrapNodes
                                                         . Mainline.findNodeH btR ni
-                                                        . flip Mainline.FindNode (Just Mainline.Want_Both))
+                                                        . flip Mainline.FindNode (Just Want_Both))
                                                 show
                                                 (const Nothing))
                             , ("peer", DHTQuery (Mainline.peerSearch bt)
                                                 (\ni -> fmap Mainline.unwrapPeers
                                                         . Mainline.getPeersH btR swarms ni
-                                                        . flip Mainline.GetPeers (Just Mainline.Want_Both)
+                                                        . flip Mainline.GetPeers (Just Want_Both)
                                                         . (read . show)) -- TODO: InfoHash -> NodeId
                                                 (show . pPrint)
                                                 (Just . show))
@@ -496,9 +503,20 @@ main = do
             , dhtParseId  = readEither :: String -> Either String Mainline.NodeId
             , dhtSearches = mainlineSearches
             }
+        toxDHT bkts = DHT
+            { dhtBuckets = bkts toxR
+            , dhtPing    = Tox.ping tox
+            , dhtQuery   = Map.fromList
+                            [ -- "node"
+                            ]
+            , dhtParseId  = readEither :: String -> Either String Tox.NodeId
+            , dhtSearches = toxSearches
+            }
         dhts = Map.fromList
                 [ ("bt4", mainlineDHT Mainline.routing4)
                 , ("bt6", mainlineDHT Mainline.routing6)
+                , ("tox4", toxDHT Tox.routing4)
+                , ("tox6", toxDHT Tox.routing6)
                 ]
 
     waitForSignal <- do
@@ -521,7 +539,7 @@ main = do
     let bkts4 = Mainline.routing4 btR
     btSaved4 <- loadNodes "bt4" :: IO [Mainline.NodeInfo]
     putStrLn $ "Loaded "++show (length btSaved4)++" nodes for bt4."
-    fallbackNodes4 <- Mainline.bootstrapNodes Mainline.Want_IP4
+    fallbackNodes4 <- Mainline.bootstrapNodes Want_IP4
     fork $ do
         myThreadId >>= flip labelThread "bootstrap.Mainline4"
         bootstrap (Mainline.nodeSearch bt) bkts4 (Mainline.ping bt) btSaved4 fallbackNodes4
@@ -530,7 +548,7 @@ main = do
     btSaved6 <- loadNodes "bt6"
     putStrLn $ "Loaded "++show (length btSaved6)++" nodes for bt6."
     let bkts6 = Mainline.routing6 btR
-    fallbackNodes6 <- Mainline.bootstrapNodes Mainline.Want_IP6
+    fallbackNodes6 <- Mainline.bootstrapNodes Want_IP6
     fork $ do
         myThreadId >>= flip labelThread "bootstrap.Mainline6"
         bootstrap (Mainline.nodeSearch bt) bkts6 (Mainline.ping bt) btSaved6 fallbackNodes6