Read instance for NodeId KMessageOf.

1 files changed, 3 insertions, 2 deletions

diff --git a/examples/dhtd.hs b/examples/dhtd.hs
index de1f11ab..68f91446 100644
--- a/examples/dhtd.hs
+++ b/examples/dhtd.hs
@@ -226,7 +226,8 @@ instance Kademlia Tox.Message where
 instance Pretty (NodeId Tox.Message) where
 instance Pretty (NodeInfo Tox.Message IPv4 ()) where
 instance Pretty (NodeInfo Tox.Message IPv4 Bool) where -- TODO
-instance Read (NodeId KMessageOf) where -- TODO
+instance Read (NodeId KMessageOf) where
+    readsPrec d s = map (\(ih,s) -> (toNodeId (ih::InfoHash),s)) $ readsPrec d s
 instance Read (NodeId Tox.Message) where -- TODO
 instance Serialize (FindNode Tox.Message IPv4) where
     get = error "TODO get"
@@ -275,7 +276,7 @@ clientSession bt tox signalQuit isBt sock n h = do
         cmd  (GenericDHT action) = cmd0 $ join $ either (flip runDHT action) (flip runDHT action) dht
         cmd  (BtDHT action)      = cmd0 $ join $ runDHT bt action
         (c,args) = second (dropWhile isSpace) $ break isSpace line
-        switchNetwork dest = clientSession bt tox signalQuit dest sock n h
+        switchNetwork dest = hPutClient h ("Network: "++if dest then "mainline" else "tox") >> clientSession bt tox signalQuit dest sock n h
     case (c,args) of
 
         ("bt", _) -> switchNetwork True