module Data.PEM.Writer
( pemWriteBuilder
, pemWriteLBS
, pemWriteBS
) where
import Data.PEM.Types
import Data.ByteString (ByteString)
import qualified Data.ByteString as B
import qualified Data.ByteString.Char8 as BC
import qualified Data.ByteString.Lazy as L
import qualified Data.ByteString.Base64 as Base64
import Data.Serialize.Builder
import Data.Monoid
import Data.List
pemWriteBuilder :: PEM -> Builder
pemWriteBuilder pem = mconcat $ intersperse eol $ concat ([begin]:header:section:[end]:[[empty]])
where begin = mconcat $ map fromByteString ["-----BEGIN ", sectionName, "-----" ]
end = mconcat $ map fromByteString ["-----END ", sectionName, "-----" ]
section = map fromByteString $ (splitChunks $ Base64.encode $ pemContent pem)
header = if null $ pemHeader pem
then []
else concatMap toHeader (pemHeader pem) ++ [empty]
toHeader (k,v) = [ mconcat $ map fromByteString [ bk, ":", v ] ]
where bk = BC.pack k
sectionName = BC.pack $ pemName pem
splitChunks b
| B.length b > 64 = let (x,y) = B.splitAt 64 b in x : splitChunks y
| otherwise = [b]
eol = fromByteString $ B.singleton 0x0a
pemWriteBS :: PEM -> ByteString
pemWriteBS = toByteString . pemWriteBuilder
pemWriteLBS :: PEM -> L.ByteString
pemWriteLBS = toLazyByteString . pemWriteBuilder