mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-16 01:44:03 +03:00
40 lines
1016 B
Haskell
40 lines
1016 B
Haskell
|
module Data.TByteString
|
||
|
|
( TByteString
|
||
|
|
, fromText
|
||
|
|
, fromBS
|
||
|
|
, fromLBS
|
||
|
|
) where
|
||
|
|
|
||
|
|
import qualified Data.Aeson as J
|
||
|
|
import qualified Data.ByteString as B
|
||
|
|
import qualified Data.ByteString.Base64 as B64
|
||
|
|
import qualified Data.ByteString.Lazy as BL
|
||
|
|
import qualified Data.Text as T
|
||
|
|
import qualified Data.Text.Encoding as TE
|
||
|
|
|
||
|
|
import Data.Bool (bool)
|
||
|
|
import Prelude
|
||
|
|
|
||
|
|
newtype TByteString
|
||
|
|
= TByteString (Bool, T.Text)
|
||
|
|
deriving (Show, Eq)
|
||
|
|
|
||
|
|
instance J.ToJSON TByteString where
|
||
|
|
toJSON (TByteString (isBase64, t)) =
|
||
|
|
bool (J.toJSON t) (J.toJSON ["Base64", t]) isBase64
|
||
|
|
|
||
|
|
fromText :: T.Text -> TByteString
|
||
|
|
fromText t = TByteString (False, t)
|
||
|
|
|
||
|
|
fromBS :: B.ByteString -> TByteString
|
||
|
|
fromBS bs =
|
||
|
|
TByteString $
|
||
|
|
-- if the bs in not utf-8 encoded, encode it to Base64
|
||
|
|
case TE.decodeUtf8' bs of
|
||
|
|
Left _ -> (True, TE.decodeUtf8 $ B64.encode bs)
|
||
|
|
Right t -> (False, t)
|
||
|
|
|
||
|
|
fromLBS :: BL.ByteString -> TByteString
|
||
|
|
fromLBS =
|
||
|
|
fromBS . BL.toStrict
|