Add decodeToFile

2 files changed, 16 insertions, 3 deletions

diff --git a/archive.h b/archive.h
index a7377ab..d0bc00d 100644
--- a/archive.h
+++ b/archive.h
@@ -154,6 +154,7 @@ char *templateURL(const char *template, const struct Variable vars[]);
 
 char *decodeHeader(const char *header);
 char *decodeToString(const struct BodyPart *part, const char *content);
+int decodeToFile(FILE *file, const struct BodyPart *part, const char *content);
 
 static inline const char *pathUID(uint32_t uid, const char *type) {
 	static char buf[PATH_MAX];
diff --git a/decode.c b/decode.c
index 24de9d3..35e6dfd 100644
--- a/decode.c
+++ b/decode.c
@@ -244,14 +244,26 @@ char *decodeHeader(const char *header) {
 	return bufferString(&buf);
 }
 
-char *decodeToString(const struct BodyPart *part, const char *src) {
-	struct Buffer dst = bufferAlloc(strlen(src) + 1);
+static const char *partCharset(const struct BodyPart *part) {
 	const char *charset = NULL;
 	for (size_t i = 0; i + 1 < part->params.len; i += 2) {
 		const char *key = dataCheck(part->params.ptr[i], String).string;
 		if (strcasecmp(key, "charset")) continue;
 		charset = dataCheck(part->params.ptr[i + 1], String).string;
 	}
-	decode(&dst, part->encoding, charset, src);
+	return charset;
+}
+
+char *decodeToString(const struct BodyPart *part, const char *src) {
+	struct Buffer dst = bufferAlloc(strlen(src) + 1);
+	decode(&dst, part->encoding, partCharset(part), src);
 	return bufferString(&dst);
 }
+
+int decodeToFile(FILE *file, const struct BodyPart *part, const char *src) {
+	struct Buffer dst = bufferAlloc(strlen(src));
+	decode(&dst, part->encoding, partCharset(part), src);
+	size_t n = fwrite(dst.ptr, dst.len, 1, file);
+	free(dst.ptr);
+	return (n ? 0 : -1);
+}