XMPP: Type-checking on various uses of SockAddr.

1 files changed, 6 insertions, 6 deletions

diff --git a/Presence/PeerResolve.hs b/Presence/PeerResolve.hs
index 900688c3..62becfe1 100644
--- a/Presence/PeerResolve.hs
+++ b/Presence/PeerResolve.hs
@@ -2,6 +2,7 @@ module PeerResolve
     ( peerKeyToResolvedNames
     , resolvePeer
     , parseAddress
+    , unsafeParseAddress
     , strip_brackets
     , withPort
     ) where
@@ -17,11 +18,10 @@ import ConnectionKey
 global_dns_cache :: DNSCache
 global_dns_cache = unsafePerformIO $ newDNSCache
 
-resolvePeer :: Text -> IO [SockAddr]
-resolvePeer addrtext = forwardResolve global_dns_cache addrtext
+resolvePeer :: Text -> IO [PeerAddress]
+resolvePeer addrtext = map PeerAddress <$> forwardResolve global_dns_cache addrtext
 
-peerKeyToResolvedNames :: ConnectionKey -> IO [Text]
-peerKeyToResolvedNames k@(ClientKey { localAddress=addr }) = return []
-peerKeyToResolvedNames k@(PeerKey { callBackAddress=addr }) = do
-    reverseResolve global_dns_cache addr
+peerKeyToResolvedNames :: PeerAddress -> IO [Text]
+peerKeyToResolvedNames (PeerAddress addr)
+    = reverseResolve global_dns_cache addr