Add openInfoDict function

1 files changed, 7 insertions, 0 deletions

diff --git a/src/System/Torrent/Storage.hs b/src/System/Torrent/Storage.hs
index 5315d9dc..003a4e98 100644
--- a/src/System/Torrent/Storage.hs
+++ b/src/System/Torrent/Storage.hs
@@ -24,6 +24,7 @@ module System.Torrent.Storage
        , Mode (..)
        , def
        , open
+       , openInfoDict
        , close
        , withStorage
 
@@ -54,6 +55,7 @@ import Data.Conduit.Binary as C
 import Data.Conduit.List as C
 import Data.Typeable
 
+import Data.Torrent
 import Data.Torrent.Bitfield as BF
 import Data.Torrent.Layout
 import Data.Torrent.Piece
@@ -94,6 +96,11 @@ open mode s l
   |   s <= 0  = throwIO (InvalidSize s)
   | otherwise = Storage mode s <$> mmapFiles mode l
 
+-- | Like 'open', but use 'InfoDict' file layout.
+openInfoDict :: Mode -> FilePath -> InfoDict -> IO Storage
+openInfoDict mode rootPath InfoDict {..} =
+  open mode (piPieceLength idPieceInfo) (flatLayout rootPath idLayoutInfo)
+
 -- | Unmaps all files forcefully. It is recommended but not required.
 close :: Storage -> IO ()
 close Storage {..} = unmapFiles fileMap