summaryrefslogtreecommitdiff
path: root/email/quoted-printable.scm
blob: 84c52c1d613bdfd40fe94a2fe3e5018d1017ef68 (plain)
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
;;; guile-email --- Guile email parser
;;; Copyright © 2018 Arun Isaac <arunisaac@systemreboot.net>
;;;
;;; This file is part of guile-email.
;;;
;;; guile-email is free software; you can redistribute it and/or modify
;;; it under the terms of the GNU Affero General Public License as
;;; published by the Free Software Foundation; either version 3 of the
;;; License, or (at your option) any later version.
;;;
;;; guile-email is distributed in the hope that it will be useful, but
;;; WITHOUT ANY WARRANTY; without even the implied warranty of
;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
;;; Affero General Public License for more details.
;;;
;;; You should have received a copy of the GNU Affero General Public
;;; License along with guile-email.  If not, see
;;; <http://www.gnu.org/licenses/>.

(define-module (email quoted-printable)
  #:use-module (rnrs bytevectors)
  #:use-module (rnrs io ports)
  #:use-module (ice-9 format)
  #:use-module (ice-9 match)
  #:use-module (srfi srfi-26)
  #:export (quoted-printable-decode
            quoted-printable-encode
            q-encoding-decode
            q-encoding-encode))

;; TODO: Error out on invalid quoted-printable input
(define quoted-printable-decode
  (match-lambda*
    (((? string? str))
     (call-with-input-string str quoted-printable-decode))
    (((? port? in))
     (call-with-bytevector-output-port
      (cut quoted-printable-decode in <>)))
    (((? port? in) (? port? out))
     (let ((c (read-char in)))
       (cond
        ((eof-object? c) out)
        ((char=? c #\=)
         ;; TODO: Support "\r\n" line ending
         (let ((c1 (read-char in)))
           (unless (char=? c1 #\Newline)
             (let ((c2 (read-char in)))
               (put-u8 out (string->number (string c1 c2) 16)))))
         (quoted-printable-decode in out))
        (else (put-u8 out (char->integer c))
              (quoted-printable-decode in out)))))))

(define* (quoted-printable-style-encode
          in out literal-char-set
          #:optional (number-of-chars-left-on-this-line 76))
  (let ((x (get-u8 in))
        (put-into-output
         (lambda (str)
           (let* ((len (string-length str))
                  (break-line? (<= number-of-chars-left-on-this-line len)))
             (put-string
              out (string-append (if break-line? "=\n" "") str))
             (if break-line?
                 (- 76 len)
                 (- number-of-chars-left-on-this-line len))))))
    (unless (eof-object? x)
      (let ((c (integer->char x)))
        (quoted-printable-style-encode
         in out literal-char-set
         (put-into-output
          (if (char-set-contains? literal-char-set c)
              (string c)
              (format #f "=~:@(~2,'0x~)" x))))))))

;; Character set of characters to be represented by themselves in
;; quoted-printable encoding
(define %quoted-printable-literal-char-set
  (char-set-delete
   (ucs-range->char-set (char->integer #\space)
                        (char->integer #\delete))
   #\=))

;; Character set of characters to be represented by themselves in Q
;; encoding
(define %q-encoding-literal-char-set
  (char-set-delete
   %quoted-printable-literal-char-set
   #\? #\_))

(define quoted-printable-encode
  (match-lambda*
    (((? bytevector? bv))
     (call-with-port (open-bytevector-input-port bv)
                     quoted-printable-encode))
    (((? port? in))
     (call-with-output-string
       (cut quoted-printable-encode in <>)))
    (((? port? in) (? port? out))
     (quoted-printable-style-encode
      in out %quoted-printable-literal-char-set))))

(define (q-encoding-decode str)
  (quoted-printable-decode
   (string-map
    (lambda (c)
      (if (char=? c #\_) #\Space c))
    str)))

(define (q-encoding-encode bv)
  (string-map
   (lambda (c)
     (if (char=? c #\Space) #\_ c))
   (call-with-port
    (open-bytevector-input-port bv)
    (lambda (in)
      (call-with-output-string
        (cut quoted-printable-style-encode in <>
             %q-encoding-literal-char-set))))))