1 module FuncTorrent.Metainfo
12 import Prelude hiding (lookup)
13 import Data.ByteString.Char8 (ByteString, unpack)
14 import Data.Map as M ((!), lookup)
15 import Data.Maybe (maybeToList)
17 import FuncTorrent.Bencode (BVal(..), bstrToString)
19 -- only single file mode supported for the time being.
20 data Info = Info { pieceLength :: !Integer
21 , pieces :: !ByteString
22 , private :: !(Maybe Integer)
24 , lengthInBytes :: !Integer
25 , md5sum :: !(Maybe String)
28 data Metainfo = Metainfo { info :: !Info
29 , announceList :: ![String]
30 , creationDate :: !(Maybe String)
31 , comment :: !(Maybe String)
32 , createdBy :: !(Maybe String)
33 , encoding :: !(Maybe String)
36 mkInfo :: BVal -> Maybe Info
37 mkInfo (Bdict m) = let (Bint pieceLength') = m ! "piece length"
38 (Bstr pieces') = m ! "pieces"
40 (Bstr name') = m ! "name"
41 (Bint length') = m ! "length"
43 in Just Info { pieceLength = pieceLength'
47 , lengthInBytes = length'
51 maybeBstrToString :: Maybe BVal -> Maybe String
52 maybeBstrToString (Just (Bstr bs)) = Just $ unpack bs
53 maybeBstrToString _ = Nothing
55 mkMetaInfo :: BVal -> Maybe Metainfo
56 mkMetaInfo (Bdict m) = let (Just info') = mkInfo $ m ! "info"
57 announce' = lookup "announce" m
58 announceList' = lookup "announce-list" m
59 -- creationDate = lookup (Bstr (pack "creation date")) m
60 creationDate' = Nothing
61 comment' = lookup "comment" m
62 createdBy' = lookup "created by" m
63 encoding' = lookup "encoding" m
64 in Just Metainfo { info = info'
65 , announceList = maybeToList (announce' >>= bstrToString)
66 ++ getAnnounceList announceList'
67 , creationDate = creationDate'
68 , comment = maybeBstrToString comment'
69 , createdBy = maybeBstrToString createdBy'
70 , encoding = maybeBstrToString encoding'
72 mkMetaInfo _ = Nothing
74 getAnnounceList :: Maybe BVal -> [String]
75 getAnnounceList Nothing = []
76 getAnnounceList (Just (Bint _)) = []
77 getAnnounceList (Just (Bstr _)) = []
78 getAnnounceList (Just (Blist l)) = map (\s -> case s of
79 (Bstr s') -> unpack s'
80 (Blist s') -> case s' of
81 [Bstr s''] -> unpack s''
85 getAnnounceList (Just (Bdict _)) = []