{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE OverloadedStrings #-}
module Data.ByteString.Base64
(
encodeBase64
, encodeBase64'
, decodeBase64
, decodeBase64Untyped
, decodeBase64Lenient
, isBase64
, isValidBase64
) where
import Data.Base64.Types
import Data.ByteString.Internal (ByteString(..))
import Data.ByteString.Base64.Internal
import Data.ByteString.Base64.Internal.Head
import Data.ByteString.Base64.Internal.Tables
import Data.Either (isRight)
import Data.Text (Text)
import qualified Data.Text.Encoding as T
import System.IO.Unsafe
encodeBase64 :: ByteString -> Base64 'StdPadded Text
encodeBase64 :: ByteString -> Base64 'StdPadded Text
encodeBase64 = (ByteString -> Text)
-> Base64 'StdPadded ByteString -> Base64 'StdPadded Text
forall a b. (a -> b) -> Base64 'StdPadded a -> Base64 'StdPadded b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap ByteString -> Text
T.decodeUtf8 (Base64 'StdPadded ByteString -> Base64 'StdPadded Text)
-> (ByteString -> Base64 'StdPadded ByteString)
-> ByteString
-> Base64 'StdPadded Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ByteString -> Base64 'StdPadded ByteString
encodeBase64'
{-# inline encodeBase64 #-}
encodeBase64' :: ByteString -> Base64 'StdPadded ByteString
encodeBase64' :: ByteString -> Base64 'StdPadded ByteString
encodeBase64' = ByteString -> Base64 'StdPadded ByteString
forall (k :: Alphabet) a. a -> Base64 k a
assertBase64 (ByteString -> Base64 'StdPadded ByteString)
-> (ByteString -> ByteString)
-> ByteString
-> Base64 'StdPadded ByteString
forall b c a. (b -> c) -> (a -> b) -> a -> c
. EncodingTable -> ByteString -> ByteString
encodeBase64_ EncodingTable
base64Table
{-# inline encodeBase64' #-}
decodeBase64 :: StdAlphabet k => Base64 k ByteString -> ByteString
decodeBase64 :: forall (k :: Alphabet).
StdAlphabet k =>
Base64 k ByteString -> ByteString
decodeBase64 = ForeignPtr Word8 -> Base64 k ByteString -> ByteString
forall (k :: Alphabet).
ForeignPtr Word8 -> Base64 k ByteString -> ByteString
decodeBase64Typed_ ForeignPtr Word8
decodeB64Table
{-# inline decodeBase64 #-}
decodeBase64Untyped :: ByteString -> Either Text ByteString
decodeBase64Untyped :: ByteString -> Either Text ByteString
decodeBase64Untyped bs :: ByteString
bs@(PS ForeignPtr Word8
_ Int
_ !Int
l)
| Int
l Int -> Int -> Bool
forall a. Eq a => a -> a -> Bool
== Int
0 = ByteString -> Either Text ByteString
forall a b. b -> Either a b
Right ByteString
forall a. Monoid a => a
mempty
| Int
r Int -> Int -> Bool
forall a. Eq a => a -> a -> Bool
== Int
1 = Text -> Either Text ByteString
forall a b. a -> Either a b
Left Text
"Base64-encoded bytestring has invalid size"
| Int
r Int -> Int -> Bool
forall a. Eq a => a -> a -> Bool
/= Int
0 = Text -> Either Text ByteString
forall a b. a -> Either a b
Left Text
"Base64-encoded bytestring requires padding"
| Bool
otherwise = IO (Either Text ByteString) -> Either Text ByteString
forall a. IO a -> a
unsafeDupablePerformIO (IO (Either Text ByteString) -> Either Text ByteString)
-> IO (Either Text ByteString) -> Either Text ByteString
forall a b. (a -> b) -> a -> b
$ ForeignPtr Word8 -> ByteString -> IO (Either Text ByteString)
decodeBase64_ ForeignPtr Word8
decodeB64Table ByteString
bs
where
!r :: Int
r = Int
l Int -> Int -> Int
forall a. Integral a => a -> a -> a
`rem` Int
4
{-# inline decodeBase64Untyped #-}
decodeBase64Lenient :: ByteString -> ByteString
decodeBase64Lenient :: ByteString -> ByteString
decodeBase64Lenient = ForeignPtr Word8 -> ByteString -> ByteString
decodeBase64Lenient_ ForeignPtr Word8
decodeB64Table
{-# inline decodeBase64Lenient #-}
isBase64 :: ByteString -> Bool
isBase64 :: ByteString -> Bool
isBase64 ByteString
bs
= ByteString -> Bool
isValidBase64 ByteString
bs
Bool -> Bool -> Bool
&& Either Text ByteString -> Bool
forall a b. Either a b -> Bool
isRight (ByteString -> Either Text ByteString
decodeBase64Untyped ByteString
bs)
{-# inline isBase64 #-}
isValidBase64 :: ByteString -> Bool
isValidBase64 :: ByteString -> Bool
isValidBase64 = ByteString -> ByteString -> Bool
validateBase64 ByteString
"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
{-# inline isValidBase64 #-}