]> git.rkrishnan.org Git - functorrent.git/commitdiff
Add Strings, instance of the Show typeclass.
authorRamakrishnan Muthukrishnan <ram@rkrishnan.org>
Wed, 4 Feb 2015 11:30:23 +0000 (17:00 +0530)
committerRamakrishnan Muthukrishnan <ram@rkrishnan.org>
Wed, 4 Feb 2015 11:30:23 +0000 (17:00 +0530)
src/Bencode.hs

index 0da372bf07b0cc29a2be5bb754603f9e3d41d5b4..e1f6dd33d802652b56d78d4d3375124d659fd8fe 100644 (file)
@@ -8,8 +8,6 @@ import Text.ParserCombinators.Parsec
 import Control.Applicative ((<*))
 import Data.Functor
 
-
-
 data BVal =
     Bint Integer
   | Bstr BC.ByteString
@@ -17,6 +15,11 @@ data BVal =
   | Bdict (M.Map BVal BVal)
   deriving (Ord, Eq)
 
+instance Show BVal where
+  show (Bint i) = show i
+  show (Bstr s) = "\"" ++ BC.unpack s ++ "\""
+  show (Blist xs) = show xs
+  show (Bdict m) = show (M.toList m)
 -- $setup
 -- >>> import Data.Either
 
@@ -66,9 +69,21 @@ bencInt = do _ <- spaces
                      parseNumber '-' (d'':_) | d'' == '0' = unexpected "numbers cannot be left-padded with zeros"
                      parseNumber d'' ds'' = return (d'':ds'')
 
-bencParser :: ParsecBS.Parser BVal
-bencParser = Bstr <$> bencStr <|>
-             Bint <$> bencInt
+-- | parse lists
+--
+-- >>> parse bencList "Blist" (BC.pack "le")
+-- Right []
+-- >>> parse bencList "Blist" (BC.pack "l4:spam4:eggse")
+-- Right ["spam","eggs"]
+
+bencList :: ParsecBS.Parser [BVal]
+bencList = do _ <- spaces
+              between (char 'l') (char 'e') (many bencVal)
+
+bencVal :: ParsecBS.Parser BVal
+bencVal = Bstr <$> bencStr <|>
+          Bint <$> bencInt <|>
+          Blist <$> bencList
 
 decode :: BC.ByteString -> Either ParseError BVal
-decode = parse bencParser "BVal"
+decode = parse bencVal "BVal"