Implement AnnounceQuery parsing

1 files changed, 20 insertions, 0 deletions

diff --git a/src/Data/Torrent/InfoHash.hs b/src/Data/Torrent/InfoHash.hs
index 3d2bbe25..a4d6a02e 100644
--- a/src/Data/Torrent/InfoHash.hs
+++ b/src/Data/Torrent/InfoHash.hs
@@ -14,6 +14,7 @@ module Data.Torrent.InfoHash
          InfoHash(..)
 
          -- * Parsing
+       , byteStringToInfoHash
        , textToInfoHash
 
          -- * Rendering
@@ -54,6 +55,16 @@ import Text.PrettyPrint
 import Text.PrettyPrint.Class
 
 
+-- TODO
+--
+-- data Word160 = Word160 {-# UNPACK #-} !Word64
+--                        {-# UNPACK #-} !Word64
+--                        {-# UNPACK #-} !Word32
+--
+-- newtype InfoHash = InfoHash Word160
+--
+-- reason: bytestring have overhead = 8 words, while infohash have length 20 bytes
+
 -- | Exactly 20 bytes long SHA1 hash of the info part of torrent file.
 newtype InfoHash = InfoHash { getInfoHash :: BS.ByteString }
                    deriving (Eq, Ord)
@@ -110,6 +121,15 @@ instance URLShow InfoHash where
 instance Pretty InfoHash where
   pretty = text . BC.unpack . ppHex . getInfoHash
 
+infoHashLen :: Int
+infoHashLen = 20
+
+-- | Convert raw bytes to info hash.
+byteStringToInfoHash :: BS.ByteString -> Maybe InfoHash
+byteStringToInfoHash bs
+  | BS.length bs == infoHashLen = Just (InfoHash bs)
+  |          otherwise          = Nothing
+
 -- | Tries both base16 and base32 while decoding info hash.
 textToInfoHash :: Text -> Maybe InfoHash
 textToInfoHash t