1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
-- |
-- Module : Codec.Binary.QuotedPrintable
-- Copyright : (c) 2009 Magnus Therning
-- License : BSD3
--
-- Implementation of Quoted-Printable based on RFC 2045
-- (<http://tools.ietf.org/html/rfc2045>).
--
-- This encoding encodes _everything_ that is passed in, it will not try to
-- guess the native line ending for your architecture. In other words, if you
-- are using this to encode text you need to split it into separate lines
-- before encoding and chopping it up.
--
-- Further documentation and information can be found at
-- <http://www.haskell.org/haskellwiki/Library/Data_encoding>.
module Codec.Binary.QuotedPrintable
( EncIncData(..)
, EncIncRes(..)
, encodeInc
, encode
, DecIncData(..)
, DecIncRes(..)
, decodeInc
, decode
, chop
, unchop
) where
import Codec.Binary.Util
import Data.Char
import Data.Maybe
import Data.Word
-- {{{1 encode
-- | Incremental encoder function.
encodeInc :: EncIncData -> EncIncRes String
encodeInc e = eI e
where
enc [] = []
enc (o:os)
| o < 33 || o == 61 || o > 126 = ('=' : toHex o) ++ enc os
| otherwise = chr (fromIntegral o) : enc os
eI EDone = EFinal []
eI (EChunk bs) = EPart (enc bs) encodeInc
-- | Encode data.
encode :: [Word8] -> String
encode = encoder encodeInc
-- {{{1 decode
-- | Incremental decoder function.
decodeInc :: DecIncData String -> DecIncRes String
decodeInc d = dI [] d
where
dI [] DDone = DFinal [] []
dI lo DDone = DFail [] lo
dI lo (DChunk s) = doDec [] (lo ++ s)
where
doDec acc [] = DPart acc (dI [])
doDec acc s'@('=':c0:c1:cs) = let
o = fromHex [c0, c1]
in if isJust o
then doDec (acc ++ [fromJust o]) cs
else DFail acc s'
doDec acc s'@(c:cs)
| c /= '=' = doDec (acc ++ [fromIntegral $ ord c]) cs
| otherwise = DPart acc (dI s')
-- | Decode data.
decode :: String -> Maybe [Word8]
decode = decoder decodeInc
-- {{{1 chop
-- | Chop up a string in parts.
chop :: Int -- ^ length of individual lines (values @\< 4@ are ignored)
-> String
-> [String]
chop n "" = []
chop n s = let
n' = max 3 $ n - 1
_c i ts "" acc = ts : acc
_c i ts tss@('=' : tss') acc
| i + 2 < n' = _c (i + 1) ('=' : ts) tss' acc
| otherwise = _c 0 "" tss (('=' : ts) : acc)
_c i ts tss@(c : tss') acc
| i < n' = _c (i + 1) (c : ts) tss' acc
| otherwise = _c 0 "" tss (('=' : ts) : acc)
in map reverse . reverse $ _c 0 "" s []
-- {{{1 unchop
-- | Concatenate the list of strings into one long string.
unchop :: [String] -> String
unchop [] = ""
unchop (s : ss) = let
dropLast = last s == '='
len = length s
in if dropLast
then take (len - 1) s ++ unchop ss
else s ++ unchop ss
|