+ Get torrent info with caching.

3 files changed, 11 insertions, 1 deletions

diff --git a/src/Network/BitTorrent.hs b/src/Network/BitTorrent.hs
index ad96a1b8..8bbd774e 100644
--- a/src/Network/BitTorrent.hs
+++ b/src/Network/BitTorrent.hs
@@ -23,6 +23,7 @@ module Network.BitTorrent
        , getSessionCount
        , getSwarm
        , getStorage
+       , getTorrentInfo
 
          -- * Extensions
        , Extension
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
diff --git a/src/System/Torrent/Storage.hs b/src/System/Torrent/Storage.hs
index c33be89c..99d164f2 100644
--- a/src/System/Torrent/Storage.hs
+++ b/src/System/Torrent/Storage.hs
@@ -299,7 +299,7 @@ readFD fd @ FD {..} offset reqSize =
         readAhead fd offset expSize
         return $ Right bs
 
--- TODO
+-- TODO implement COW; needed for applications which want to change files.
 writeFD :: FD -> ByteString -> Offset -> IO ()
 writeFD FD {..} bs offset = return ()