1 files changed, 16 insertions, 6 deletions

diff --git a/src/Network/Tox/Onion/Handlers.hs b/src/Network/Tox/Onion/Handlers.hs
index 91dd843e..167df336 100644
--- a/src/Network/Tox/Onion/Handlers.hs
+++ b/src/Network/Tox/Onion/Handlers.hs
@@ -159,9 +159,19 @@ handlers net routing toks keydb AnnounceType
     $ announceH routing toks keydb
 handlers net _ _ keydb _ = Just $ NoReply Right $ dataToRouteH keydb net
 
+
+data Rendezvous = Rendezvous
+    { rendezvousKey :: PublicKey
+    , rendezvousNode :: NodeInfo
+    }
+ deriving Eq
+
+instance Show Rendezvous where
+    show (Rendezvous k ni) = concat [show $ key2id k, ":", show ni]
+
 toxidSearch :: (TransactionId -> OnionDestination r -> STM (OnionDestination r, Int))
                -> Client r
-               -> Search NodeId (IP, PortNumber) (Maybe Nonce32) NodeInfo PublicKey
+               -> Search NodeId (IP, PortNumber) (Maybe Nonce32) NodeInfo Rendezvous
 toxidSearch getTimeout client = Search
     { searchSpace       = toxSpace
     , searchNodeAddress = nodeIP &&& nodePort
@@ -194,23 +204,23 @@ announceSerializer getTimeout = MethodSerializer
         _                              -> Nothing
     }
 
-unwrapAnnounceResponse :: AnnounceResponse -> ([NodeInfo], [PublicKey], Maybe Nonce32)
-unwrapAnnounceResponse (AnnounceResponse is_stored (SendNodes ns))
+unwrapAnnounceResponse :: NodeInfo -> AnnounceResponse -> ([NodeInfo], [Rendezvous], Maybe Nonce32)
+unwrapAnnounceResponse ni (AnnounceResponse is_stored (SendNodes ns))
     = case is_stored of
         NotStored n32    -> (ns, [],  Just n32)
-        SendBackKey k    -> (ns, [k], Nothing)
+        SendBackKey k    -> (ns, [Rendezvous k ni], Nothing)
         Acknowledged n32 -> (ns, [],  Just n32)
 
 announce :: (TransactionId -> OnionDestination r -> STM (OnionDestination r, Int))
             -> Client r
             -> NodeId
             -> NodeInfo
-            -> IO (Maybe ([NodeInfo],[PublicKey],Maybe Nonce32))
+            -> IO (Maybe ([NodeInfo],[Rendezvous],Maybe Nonce32))
 announce getTimeout client nid ni =
     -- Four tries and then we tap out.
     flip fix 4 $ \loop n -> do
     let oaddr = OnionDestination ni Nothing
     mb <- QR.sendQuery client (announceSerializer getTimeout) (AnnounceRequest zeros32 nid zeroID) oaddr
     maybe (if n>0 then loop $! n - 1 else return Nothing)
-          (return . Just . unwrapAnnounceResponse)
+          (return . Just . unwrapAnnounceResponse ni)
           $ join mb