msgpack-binary-0.0.14: tools/msgpack-parser.hs

{-# LANGUAGE Trustworthy #-}
-- | A MessagePack parser.
--
-- Example usage:
--   $ echo -ne "\x94\x01\xa1\x32\xa1\x33\xa4\x50\x6f\x6f\x66" | ./msgpack-parser
-- or
--   $ echo 'ObjectArray [ObjectInt 97, ObjectStr "test",  ObjectBool True]' | ./msgpack-parser
--
-- This tool performs two symmetrical functions:
--   1. It can decode binary data representing a
--      Data.MessagePack.Object into a human-readable string.
--   2. It can do the reverse: encode a human-readable string into
--      a binary representation of Data.MessagePack.Object.
--
-- No flags are required as it automatically detects which of these
-- two functions it should perform.  This is done by first assuming
-- the input is human readable.  If it fails to parse it, it then
-- considers it as binary data.
--
-- Therefore, given a valid input, the tool has the following property
--   $ ./msgpack-parser < input.bin | ./msgpack-parser
-- will output back the contents of input.bin.
--
-- In case the input is impossible to parse, nothing is output.
--
-- Known bugs:
--   - If no input is given, the tool exits with
--     "Data.Binary.Get.runGet at position 0: not enough bytes"
--   - The tool does not check that all the input is parsed.
--     Therefore, "abc" is interpreted as just "ObjectInt 97".
--
module Main where

import           Control.Applicative        ((<$>), (<|>))
import qualified Data.ByteString.Lazy       as L
import qualified Data.ByteString.Lazy.Char8 as L8
import           Data.Maybe                 (fromMaybe)
import           Data.MessagePack           (Object, pack, unpack)
import           Text.Groom                 (groom)
import           Text.Read                  (readMaybe)


parse :: L.ByteString -> L.ByteString
parse str = fromMaybe L.empty $
  pack <$> (readMaybe $ L8.unpack str :: Maybe Object)
  <|>
  L8.pack . flip (++) "\n" . groom <$> (unpack str :: Maybe Object)


main :: IO ()
main = parse <$> L.getContents >>= L.putStr