diff options
author | Arun Isaac | 2020-05-25 05:22:17 +0530 |
---|---|---|
committer | Arun Isaac | 2020-05-25 05:44:15 +0530 |
commit | 8a1b7f6b5534bc02692f7a02d0dc287dc46ca4c1 (patch) | |
tree | ce3e9a33b73dd7e7faa66242322b32d73534e134 /email | |
parent | 3b662f55237e037184f92335ea0f4502adc91fe8 (diff) | |
download | guile-email-8a1b7f6b5534bc02692f7a02d0dc287dc46ca4c1.tar.gz guile-email-8a1b7f6b5534bc02692f7a02d0dc287dc46ca4c1.tar.lz guile-email-8a1b7f6b5534bc02692f7a02d0dc287dc46ca4c1.zip |
utils: Do not return eof if matched at beginning.
* email/utils.scm (read-while, read-bytes-till): Do not return eof if
matched at beginning. Return empty string or bytevector respectively.
* tests/utils.scm ("read-bytes-till returns empty bytevector on match
at beginning", "read-while returns empty string on match at
beginning"): New tests.
Diffstat (limited to 'email')
-rw-r--r-- | email/utils.scm | 25 |
1 files changed, 18 insertions, 7 deletions
diff --git a/email/utils.scm b/email/utils.scm index 70153be..984f07e 100644 --- a/email/utils.scm +++ b/email/utils.scm @@ -1,5 +1,5 @@ ;;; guile-email --- Guile email parser -;;; Copyright © 2018, 2019 Arun Isaac <arunisaac@systemreboot.net> +;;; Copyright © 2018, 2019, 2020 Arun Isaac <arunisaac@systemreboot.net> ;;; ;;; This file is part of guile-email. ;;; @@ -24,7 +24,7 @@ #:use-module (ice-9 textual-ports) #:use-module (rnrs bytevectors) #:use-module ((rnrs io ports) - #:select (call-with-bytevector-output-port)) + #:select (call-with-port)) #:use-module (rnrs io simple) #:use-module (srfi srfi-1) #:use-module (srfi srfi-26) @@ -58,8 +58,14 @@ back into PORT." (read-while-loop output)) (else (unget-string port x))))) - (let ((str (call-with-output-string read-while-loop))) - (if (string-null? str) (eof-object) str))) + (call-with-port (open-output-string) + (lambda (out) + (let ((read-result (read-while-loop out)) + (str (get-output-string out))) + (if (and (eof-object? read-result) + (string-null? str)) + read-result + str))))) (define (read-bytes-till port sequence) "Read bytes from PORT until byte SEQUENCE is seen or end-of-file is @@ -82,9 +88,14 @@ reached. If SEQUENCE is seen, unget it to PORT and return." (else (put-u8 out octet) (read-bytes-and-write-till in out sequence))))) - (let ((bv (call-with-bytevector-output-port - (cut read-bytes-and-write-till port <> sequence)))) - (if (bytevector=? bv (make-bytevector 0)) (eof-object) bv))) + (call-with-values open-bytevector-output-port + (lambda (out get-bytevector) + (let ((read-result (read-bytes-and-write-till port out sequence)) + (bv (get-bytevector))) + (if (and (eof-object? read-result) + (bytevector=? bv #vu8())) + read-result + bv))))) (define (get-line-with-delimiter port) "Read a line from PORT and return it as a string including the |