Add getSession function

1 files changed, 4 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/Exchange/Wire.hs b/src/Network/BitTorrent/Exchange/Wire.hs
index 4bd342ca..0414ebe7 100644
--- a/src/Network/BitTorrent/Exchange/Wire.hs
+++ b/src/Network/BitTorrent/Exchange/Wire.hs
@@ -56,6 +56,7 @@ module Network.BitTorrent.Exchange.Wire
 
          -- ** Query
        , getConnection
+       , getSession
        , getExtCaps
        , getStats
        , getMetadata
@@ -571,6 +572,9 @@ getStats = lift $ use connStats
 getConnection :: Wire s (Connection s)
 getConnection = lift ask
 
+getSession :: Wire s s
+getSession = lift (asks connSession)
+
 {-----------------------------------------------------------------------
 --  Wrapper
 -----------------------------------------------------------------------}