Add externalAddr function

1 files changed, 11 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/Client/Types.hs b/src/Network/BitTorrent/Client/Types.hs
index 603142d5..e80578a3 100644
--- a/src/Network/BitTorrent/Client/Types.hs
+++ b/src/Network/BitTorrent/Client/Types.hs
@@ -3,6 +3,7 @@ module Network.BitTorrent.Client.Types
        ( -- * Core types
          Handle (..)
        , Client (..)
+       , externalAddr
 
          -- * Monad BitTorrent
        , BitTorrent (..)
@@ -52,6 +53,16 @@ instance Eq Client where
 instance Ord Client where
   compare = comparing clientPeerId
 
+-- | External IP address of a host running a bittorrent client
+-- software may be used to acknowledge remote peer the host connected
+-- to. See 'Network.BitTorrent.Exchange.Message.ExtendedHandshake'.
+externalAddr :: Client -> PeerAddr (Maybe IP)
+externalAddr Client {..} = PeerAddr
+  { peerId   = Just clientPeerId
+  , peerHost = Nothing -- TODO return external IP address, if known
+  , peerPort = clientListenerPort
+  }
+
 {-----------------------------------------------------------------------
 --  BitTorrent monad
 -----------------------------------------------------------------------}