Fix test

2 files changed, 15 insertions, 2 deletions

diff --git a/src/Data/Torrent.hs b/src/Data/Torrent.hs
index fc0cb10b..4d6461af 100644
--- a/src/Data/Torrent.hs
+++ b/src/Data/Torrent.hs
@@ -26,6 +26,7 @@
 module Data.Torrent
        ( -- * Info dictionary
          InfoDict (..)
+       , infoDictionary
 
          -- ** Lenses
        , infohash
@@ -131,6 +132,13 @@ instance Hashable InfoDict where
   hash = Hashable.hash . idInfoHash
   {-# INLINE hash #-}
 
+-- | Smart constructor: add a info hash to info dictionary.
+infoDictionary :: LayoutInfo -> PieceInfo -> Bool -> InfoDict
+infoDictionary li pinfo private = InfoDict ih li pinfo private
+  where
+    ih = IH.hashlazy $ encode $ InfoDict fake_ih li pinfo private
+    fake_ih = InfoHash ""
+
 getPrivate :: Get Bool
 getPrivate = (Just True ==) <$>? "private"
 
@@ -155,8 +163,8 @@ instance BEncode InfoDict where
 ppPrivacy :: Bool -> Doc
 ppPrivacy privacy = "Privacy: " <> if privacy then "private" else "public"
 
-ppAdditionalInfo :: InfoDict -> Doc
-ppAdditionalInfo layout = PP.empty
+--ppAdditionalInfo :: InfoDict -> Doc
+--ppAdditionalInfo layout = PP.empty
 
 instance Pretty InfoDict where
   pretty InfoDict {..} =
@@ -246,6 +254,9 @@ instance BEncode URI where
   fromBEncode b           = decodingError $ "url <" ++ show b ++ ">"
   {-# INLINE fromBEncode #-}
 
+--pico2uni :: Pico -> Uni
+--pico2uni = undefined
+
 -- TODO move to bencoding
 instance BEncode POSIXTime where
   toBEncode pt = toBEncode (floor pt :: Integer)
diff --git a/src/Data/Torrent/Piece.hs b/src/Data/Torrent/Piece.hs
index 6cff53d6..7eb4e3d5 100644
--- a/src/Data/Torrent/Piece.hs
+++ b/src/Data/Torrent/Piece.hs
@@ -25,6 +25,7 @@ module Data.Torrent.Piece
        , isPiece
 
          -- * Piece control
+       , HashArray (..)
        , PieceInfo (..)
        , pieceCount
 
@@ -138,6 +139,7 @@ isPiece pieceLen blk @ (Block i offset _) =
 -- Piece control
 -----------------------------------------------------------------------}
 
+-- | A flat array of SHA1 sums of each piece.
 newtype HashArray = HashArray { unHashArray :: ByteString }
                     deriving (Show, Read, Eq, BEncode)