Add pretty instances for NodeInfos

1 files changed, 17 insertions, 1 deletions

diff --git a/src/Network/BitTorrent/Core/PeerAddr.hs b/src/Network/BitTorrent/Core/PeerAddr.hs
index 9f5c7c5d..4905f910 100644
--- a/src/Network/BitTorrent/Core/PeerAddr.hs
+++ b/src/Network/BitTorrent/Core/PeerAddr.hs
@@ -50,7 +50,7 @@ import Data.String
 import Data.Typeable
 import Data.Word
 import Network.Socket
-import Text.PrettyPrint hiding ((<>))
+import Text.PrettyPrint as PP hiding ((<>))
 import Text.PrettyPrint.Class
 import Text.Read (readMaybe)
 import qualified Text.ParserCombinators.ReadP as RP
@@ -86,6 +86,10 @@ instance Hashable PortNumber where
   hashWithSalt s = hashWithSalt s . fromEnum
   {-# INLINE hashWithSalt #-}
 
+instance Pretty PortNumber where
+  pretty = PP.int . fromEnum
+  {-# INLINE pretty #-}
+
 {-----------------------------------------------------------------------
 --  IP addr
 -----------------------------------------------------------------------}
@@ -160,6 +164,18 @@ instance Serialize IPv6 where
     put ip = put $ toHostAddress6 ip
     get = fromHostAddress6 <$> get
 
+instance Pretty IPv4 where
+  pretty = PP.text . show
+  {-# INLINE pretty #-}
+
+instance Pretty IPv6 where
+  pretty = PP.text . show
+  {-# INLINE pretty #-}
+
+instance Pretty IP where
+  pretty = PP.text . show
+  {-# INLINE pretty #-}
+
 {-----------------------------------------------------------------------
 --  Peer addr
 -----------------------------------------------------------------------}