Add withSession and getStatus functions

1 files changed, 19 insertions, 3 deletions

diff --git a/src/Network/BitTorrent/Tracker/Session.hs b/src/Network/BitTorrent/Tracker/Session.hs
index 9b7654dd..b5c05116 100644
--- a/src/Network/BitTorrent/Tracker/Session.hs
+++ b/src/Network/BitTorrent/Tracker/Session.hs
@@ -12,21 +12,28 @@ module Network.BitTorrent.Tracker.Session
          Session
        , newSession
        , closeSession
+       , withSession
 
-         -- * Trackers
-       , notify
+         -- * Query
+       , Status (..)
+       , getStatus
        , askPeers
 
+         -- * Events
+       , Event (..)
+       , notify
+
          -- * Tracker Exchange
          -- | BEP28: <http://www.bittorrent.org/beps/bep_0028.html>
        , addTracker
        , removeTracker
-       , getTrackers
        , getTrustedTrackers
        ) where
 
 import Control.Applicative
+import Control.Exception
 import Control.Concurrent
+
 import Data.Default
 import Data.Fixed
 import Data.Foldable
@@ -128,6 +135,15 @@ newSession ih origUris = do
 closeSession :: Session -> IO ()
 closeSession _ = return ()
 
+-- | Normally you need to use 'Control.Monad.Trans.Resource.alloc'.
+withSession :: InfoHash -> TrackerList URI -> (Session -> IO ()) -> IO ()
+withSession ih uris = bracket (newSession ih uris) closeSession
+
+-- | Get last announced status. The only action can alter this status
+-- is 'notify'.
+getStatus :: Session -> IO Status
+getStatus Session {..} = takeMVar currentStatus
+
 seconds :: Int -> NominalDiffTime
 seconds n = realToFrac (toEnum n :: Uni)