From d8160f03094e6d18509956efe9bb6aeda31d79c5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Roman=20Smr=C5=BE?= <roman.smrz@seznam.cz>
Date: Mon, 28 Jul 2025 19:16:18 +0200
Subject: Return peer address in IO monad

Changelog: API: Replaced `Network.peerAddress` with `getPeerAddress` and added `getPeerAddresses`
---
 src/Erebos/Discovery.hs | 17 +++++++----------
 src/Erebos/Network.hs   | 10 +++++++++-
 2 files changed, 16 insertions(+), 11 deletions(-)

(limited to 'src')

diff --git a/src/Erebos/Discovery.hs b/src/Erebos/Discovery.hs
index ff07964..a83e589 100644
--- a/src/Erebos/Discovery.hs
+++ b/src/Erebos/Discovery.hs
@@ -246,19 +246,16 @@ instance Service DiscoveryService where
         DiscoverySelf addrs priority -> do
             pid <- asks svcPeerIdentity
             peer <- asks svcPeer
+            paddrs <- getPeerAddresses peer
+
             let insertHelper new old | dpPriority new > dpPriority old = new
                                      | otherwise                       = old
-            matchedAddrs <- flip filterM addrs $ \case
-                DiscoveryICE -> do
-                    return True
-
-                DiscoveryIP ipaddr port
-                  | DatagramAddress saddr <- peerAddress peer
-                  , Just paddr <- inetFromSockAddr saddr
-                  -> do
-                    return $ ( ipaddr, port ) == paddr
 
-                _ -> return False
+            let matchedAddrs = flip filter addrs $ \case
+                    DiscoveryICE -> True
+                    DiscoveryIP ipaddr port ->
+                        DatagramAddress (inetToSockAddr ( ipaddr, port )) `elem` paddrs
+                    _ -> False
 
             forM_ (idDataF =<< unfoldOwners pid) $ \sdata -> do
                 let dp = DiscoveryPeer
diff --git a/src/Erebos/Network.hs b/src/Erebos/Network.hs
index 76ecd82..63ce7b8 100644
--- a/src/Erebos/Network.hs
+++ b/src/Erebos/Network.hs
@@ -8,7 +8,7 @@ module Erebos.Network (
     ServerOptions(..), serverIdentity, defaultServerOptions,
 
     Peer, peerServer, peerStorage,
-    PeerAddress(..), peerAddress,
+    PeerAddress(..), getPeerAddress, getPeerAddresses,
     PeerIdentity(..), peerIdentity,
     WaitingRef, wrDigest,
     Service(..),
@@ -139,6 +139,14 @@ data Peer = Peer
     , peerWaitingRefs :: TMVar [WaitingRef]
     }
 
+-- | Get current main address of the peer (used to send new packets).
+getPeerAddress :: MonadIO m => Peer -> m PeerAddress
+getPeerAddress = liftIO . return . peerAddress
+
+-- | Get all known addresses of given peer.
+getPeerAddresses :: MonadIO m => Peer -> m [ PeerAddress ]
+getPeerAddresses = fmap (: []) . getPeerAddress
+
 peerServer :: Peer -> Server
 peerServer = peerServer_
 
-- 
cgit v1.2.3