1 files changed, 5 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/Exchange.hs b/src/Network/BitTorrent/Exchange.hs
index 2eedc6bd..3d05f7fc 100644
--- a/src/Network/BitTorrent/Exchange.hs
+++ b/src/Network/BitTorrent/Exchange.hs
@@ -58,6 +58,7 @@ module Network.BitTorrent.Exchange
        , Event(..)
        , awaitEvent
        , yieldEvent
+       , handleEvent
 
          -- * Exceptions
        , disconnect
@@ -444,6 +445,10 @@ yieldEvent (Fragment  blk) = do
     then yieldMessage (Piece blk)
     else return ()
 
+
+handleEvent :: (Event -> P2P Event) -> P2P ()
+handleEvent action = awaitEvent >>= action >>= yieldEvent
+
 --flushBroadcast :: P2P ()
 --flushBroadcast = nextBroadcast >>= maybe (return ()) go
 --  where