Used partitionTransform to simplify the onion client.

1 files changed, 18 insertions, 1 deletions

diff --git a/dht/src/Data/Tox/Onion.hs b/dht/src/Data/Tox/Onion.hs
index a9bc4e1d..55e81069 100644
--- a/dht/src/Data/Tox/Onion.hs
+++ b/dht/src/Data/Tox/Onion.hs
@@ -19,7 +19,7 @@
 module Data.Tox.Onion where
 
 
-import Network.Address (fromSockAddr,toSockAddr,setPort,either4or6,sockAddrPort,localhost4,localhost6)
+import Network.Address (fromSockAddr,toSockAddr,setPort,either4or6,sockAddrPort,localhost4,localhost6,nullAddress4)
 import Network.QueryResponse
 import Crypto.Tox hiding (encrypt,decrypt)
 import Network.Tox.NodeId
@@ -873,6 +873,19 @@ data OnionRoute = OnionRoute
     }
  deriving Show
 
+dummySecret :: SecretKey
+dummySecret = fromJust $ decodeSecret "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
+
+dummyNodeId :: NodeId
+dummyNodeId = read "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
+
+dummyNode :: NodeInfo
+dummyNode = k where Right k = nodeInfo dummyNodeId nullAddress4
+
+dummyRoute :: OnionRoute
+dummyRoute = OnionRoute dummySecret dummySecret dummySecret
+                        dummyNode dummyNode dummyNode
+                        Nothing
 
 wrapOnion :: Serialize (Forwarding n msg) =>
            TransportCrypto
@@ -956,6 +969,10 @@ instance Read AnnouncedRendezvous where
             }
 
 
+-- | Lookup the secret key for the given toxid public key.  If it is not found,
+-- then the SearchingAlias symbol will be used to indicate that a new temporary
+-- key pair should be generated or that all known keys should be tried until one
+-- succeeds to decrypt the message.
 selectAlias :: TransportCrypto -> NodeId -> STM AliasSelector
 selectAlias crypto pkey = do
         ks <- filter (\(sk,pk) -> pk == id2key pkey)