Add base64TextLenient

haskell-works · Jan 4, 2025 · e964b9c · e964b9c
1 parent 54bc615
commit e964b9c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/Data/RdsData/Decode/Row.hs b/src/Data/RdsData/Decode/Row.hs
@@ -25,6 +25,7 @@ module Data.RdsData.Decode.Row
   , bytestring
   , lazyBytestring
   , base64Text
+  , base64TextLenient
   , lazyBase64Text
   , timeOfDay
   , day
@@ -175,6 +176,10 @@ base64Text :: DecodeRow ByteString
 base64Text =
   column DV.base64Text
 
+base64TextLenient :: DecodeRow ByteString
+base64TextLenient =
+  column DV.base64TextLenient
+
 lazyBase64Text :: DecodeRow LBS.ByteString
 lazyBase64Text =
   column DV.lazyBase64Text

diff --git a/src/Data/RdsData/Decode/Value.hs b/src/Data/RdsData/Decode/Value.hs
@@ -33,6 +33,7 @@ module Data.RdsData.Decode.Value
   , lazyText
   , lazyBytestring
   , base64Text
+  , base64TextLenient
   , lazyBase64Text
   , string
   , json
@@ -262,6 +263,12 @@ base64Text = do
     Right a -> pure a
     Left e  -> decodeValueFailed "base64-text" "Text" (Just (T.pack e))
 
+base64TextLenient :: DecodeValue ByteString
+base64TextLenient = do
+  t <- text
+  let b64 = T.encodeUtf8 t
+  pure $ B64.decodeLenient b64
+
 lazyBase64Text :: DecodeValue LBS.ByteString
 lazyBase64Text = do
   t <- lazyText