+ Get torrent info with caching.

1 files changed, 9 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/Sessions.hs b/src/Network/BitTorrent/Sessions.hs
index 5047f06c..4e58be5c 100644
--- a/src/Network/BitTorrent/Sessions.hs
+++ b/src/Network/BitTorrent/Sessions.hs
@@ -33,6 +33,7 @@ module Network.BitTorrent.Sessions
        , getPeerCount
        , getSwarm
        , getStorage
+       , getTorrentInfo
        , openSwarmSession
 
          -- * Swarm
@@ -250,6 +251,14 @@ getSwarm cs @ ClientSession {..} ih = do
 getStorage :: ClientSession -> InfoHash -> IO Storage
 getStorage cs ih = storage <$> getSwarm cs ih
 
+getTorrentInfo :: ClientSession -> InfoHash -> IO (Maybe Torrent)
+getTorrentInfo cs ih = do
+  tstatus <- torrentPresence cs ih
+  case tstatus of
+    Unknown                        -> return   Nothing
+    Active     (SwarmSession {..}) -> return $ Just torrentMeta
+    Registered (TorrentLoc   {..}) -> Just <$> fromFile metafilePath
+
 -- | Get the number of connected peers in the given swarm.
 getSessionCount :: SwarmSession -> IO SessionCount
 getSessionCount SwarmSession {..} = do