+ Add message and block pprint for logging.

1 files changed, 11 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/PeerWire/Message.hs b/src/Network/BitTorrent/PeerWire/Message.hs
index 650e0082..c98c9808 100644
--- a/src/Network/BitTorrent/PeerWire/Message.hs
+++ b/src/Network/BitTorrent/PeerWire/Message.hs
@@ -1,5 +1,6 @@
 module Network.BitTorrent.PeerWire.Message
        ( Message(..)
+       , ppMessage
        ) where
 
 import Control.Applicative
@@ -129,3 +130,13 @@ instance Serialize Message where
   put (SuggestPiece pix) = putInt 5  >> putWord8 0x0D >> putInt pix
   put (RejectRequest ix) = putInt 13 >> putWord8 0x10 >> put ix
   put (AllowedFast   ix) = putInt 5  >> putWord8 0x11 >> putInt ix
+
+
+-- | Compact output for logging: only useful information but not payload bytes.
+ppMessage :: Message -> String
+ppMessage (Bitfield _)       = "Bitfield "
+ppMessage (Piece blk)        = "Piece "    ++ ppBlock blk
+ppMessage (Cancel ix)        = "Cancel "   ++ ppBlockIx ix
+ppMessage (SuggestPiece pix) = "Suggest"   ++ show pix
+ppMessage (RejectRequest ix) = "Reject"    ++ ppBlockIx ix
+ppMessage msg = show msg
\ No newline at end of file