Remove trailing whitespace in the to-hash content.patched

1 files changed, 14 insertions, 2 deletions

diff --git a/Codec/Encryption/OpenPGP/ASCIIArmor/Utils.hs b/Codec/Encryption/OpenPGP/ASCIIArmor/Utils.hs
index f1068fe..e36b8db 100644
--- a/Codec/Encryption/OpenPGP/ASCIIArmor/Utils.hs
+++ b/Codec/Encryption/OpenPGP/ASCIIArmor/Utils.hs
@@ -14,11 +14,23 @@ import qualified Data.ByteString.Char8 as BC8
 import qualified Data.ByteString.Lazy as BL
 import qualified Data.ByteString.Lazy.Char8 as BLC8
 import Data.List (intersperse)
+import Data.Char (isSpace)
+
+removeTrailingSpaces :: ByteString -> ByteString
+removeTrailingSpaces bs = fst $ BC8.spanEnd isSpace bs
+
+toLast :: (a -> a) -> [a] -> [a]
+toLast _ []     = []
+toLast f [x]    = [f x]
+toLast f (x:xs) = x : toLast f xs
+
+removeTrailingSpacesLazy :: BL.ByteString -> BL.ByteString
+removeTrailingSpacesLazy bs = BL.fromChunks $ toLast removeTrailingSpaces $ BL.toChunks bs
 
 crlfUnlines :: [ByteString] -> ByteString
 crlfUnlines [] = B.empty
-crlfUnlines ss = B.concat $ intersperse (BC8.pack "\r\n") ss
+crlfUnlines ss = B.concat $ intersperse (BC8.pack "\r\n") $ map removeTrailingSpaces ss
 
 crlfUnlinesLazy :: [BL.ByteString] -> BL.ByteString
 crlfUnlinesLazy [] = BL.empty
-crlfUnlinesLazy ss = BL.concat $ intersperse (BLC8.pack "\r\n") ss
+crlfUnlinesLazy ss = BL.concat $ intersperse (BLC8.pack "\r\n") $ map removeTrailingSpacesLazy ss