dovecot/core-2.2: src/doveadm/doveadm-mail-fetch.c annotate

annotate src/doveadm/doveadm-mail-fetch.c @ 11208:f10ad708d748 HEAD

doveadm: Reduced code duplication.

author	Timo Sirainen <tss@iki.fi>
date	Wed, 28 Apr 2010 23:18:36 +0300
parents	1ec4eb879f0e
children	c7cb58f1d2c6

rev	line source
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	1 /* Copyright (c) 2010 Dovecot authors, see the included COPYING file */
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	2
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	3 #include "lib.h"
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	4 #include "array.h"
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	5 #include "istream.h"
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	6 #include "ostream.h"
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	7 #include "base64.h"
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	8 #include "randgen.h"
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	9 #include "str.h"
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	10 #include "message-size.h"
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	11 #include "imap-util.h"
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	12 #include "mail-namespace.h"
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	13 #include "mail-storage.h"
11208 f10ad708d748 doveadm: Reduced code duplication. Timo Sirainen <tss@iki.fi> parents: 11207 diff changeset	14 #include "mail-search.h"
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	15 #include "doveadm-mail.h"
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	16 #include "doveadm-mail-list-iter.h"
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	17
11207 1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	18 #include <stdio.h>
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	19
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	20 struct fetch_context {
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	21 struct mail_search_args *search_args;
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	22 struct ostream *output;
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	23 struct mail *mail;
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	24
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	25 ARRAY_DEFINE(fields, const struct fetch_field);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	26 enum mail_fetch_field wanted_fields;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	27
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	28 string_t *hdr;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	29 const char *prefix;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	30 };
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	31
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	32 static int fetch_mailbox(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	33 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	34 const char *value;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	35
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	36 if (mail_get_special(ctx->mail, MAIL_FETCH_MAILBOX_NAME, &value) < 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	37 return -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	38 str_append(ctx->hdr, value);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	39 return 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	40 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	41
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	42 static int fetch_seq(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	43 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	44 str_printfa(ctx->hdr, "%u", ctx->mail->seq);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	45 return 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	46 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	47
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	48 static int fetch_uid(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	49 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	50 str_printfa(ctx->hdr, "%u", ctx->mail->seq);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	51 return 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	52 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	53
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	54 static int fetch_guid(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	55 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	56 const char *value;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	57
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	58 if (mail_get_special(ctx->mail, MAIL_FETCH_GUID, &value) < 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	59 return -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	60 str_append(ctx->hdr, value);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	61 return 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	62 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	63
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	64 static int fetch_flags(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	65 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	66 imap_write_flags(ctx->hdr, mail_get_flags(ctx->mail),
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	67 mail_get_keywords(ctx->mail));
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	68 return 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	69 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	70
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	71 static void flush_hdr(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	72 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	73 o_stream_send(ctx->output, str_data(ctx->hdr), str_len(ctx->hdr));
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	74 str_truncate(ctx->hdr, 0);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	75 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	76
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	77 static int fetch_hdr(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	78 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	79 struct istream *input;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	80 struct message_size hdr_size;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	81 int ret = 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	82
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	83 if (mail_get_stream(ctx->mail, &hdr_size, NULL, &input) < 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	84 return -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	85
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	86 str_append_c(ctx->hdr, '\n');
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	87 flush_hdr(ctx);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	88 input = i_stream_create_limit(input, hdr_size.physical_size);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	89 while (!i_stream_is_eof(input)) {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	90 if (o_stream_send_istream(ctx->output, input) <= 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	91 i_fatal("write(stdout) failed: %m");
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	92 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	93 if (input->stream_errno != 0) {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	94 i_error("read() failed: %m");
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	95 ret = -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	96 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	97 i_stream_unref(&input);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	98 o_stream_flush(ctx->output);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	99 return ret;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	100 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	101
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	102 static int fetch_body(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	103 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	104 struct istream *input;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	105 struct message_size hdr_size;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	106 int ret = 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	107
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	108 if (mail_get_stream(ctx->mail, &hdr_size, NULL, &input) < 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	109 return -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	110
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	111 str_append_c(ctx->hdr, '\n');
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	112 flush_hdr(ctx);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	113 i_stream_skip(input, hdr_size.physical_size);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	114 while (!i_stream_is_eof(input)) {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	115 if (o_stream_send_istream(ctx->output, input) <= 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	116 i_fatal("write(stdout) failed: %m");
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	117 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	118 if (input->stream_errno != 0) {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	119 i_error("read() failed: %m");
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	120 ret = -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	121 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	122 o_stream_flush(ctx->output);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	123 return ret;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	124 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	125
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	126 static int fetch_text(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	127 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	128 struct istream *input;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	129 int ret = 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	130
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	131 if (mail_get_stream(ctx->mail, NULL, NULL, &input) < 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	132 return -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	133
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	134 str_append_c(ctx->hdr, '\n');
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	135 flush_hdr(ctx);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	136 while (!i_stream_is_eof(input)) {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	137 if (o_stream_send_istream(ctx->output, input) <= 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	138 i_fatal("write(stdout) failed: %m");
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	139 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	140 if (input->stream_errno != 0) {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	141 i_error("read() failed: %m");
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	142 ret = -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	143 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	144 o_stream_flush(ctx->output);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	145 return ret;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	146 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	147
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	148 static int fetch_size_physical(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	149 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	150 uoff_t size;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	151
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	152 if (mail_get_physical_size(ctx->mail, &size) < 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	153 return -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	154 str_printfa(ctx->hdr, "%"PRIuUOFF_T, size);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	155 return 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	156 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	157
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	158 static int fetch_size_virtual(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	159 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	160 uoff_t size;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	161
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	162 if (mail_get_virtual_size(ctx->mail, &size) < 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	163 return -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	164 str_printfa(ctx->hdr, "%"PRIuUOFF_T, size);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	165 return 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	166 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	167
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	168 static int fetch_date_received(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	169 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	170 time_t t;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	171
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	172 if (mail_get_received_date(ctx->mail, &t) < 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	173 return -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	174 str_printfa(ctx->hdr, "%s", unixdate2str(t));
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	175 return 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	176 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	177
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	178 static int fetch_date_sent(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	179 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	180 time_t t;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	181 int tz;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	182 char chr;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	183
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	184 if (mail_get_date(ctx->mail, &t, &tz) < 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	185 return -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	186
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	187 chr = tz < 0 ? '-' : '+';
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	188 if (tz < 0) tz = -tz;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	189 str_printfa(ctx->hdr, "%s (%c%02u%02u)", unixdate2str(t),
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	190 chr, tz/60, tz%60);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	191 return 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	192 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	193
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	194 static int fetch_date_saved(struct fetch_context *ctx)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	195 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	196 time_t t;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	197
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	198 if (mail_get_save_date(ctx->mail, &t) < 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	199 return -1;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	200 str_printfa(ctx->hdr, "%s", unixdate2str(t));
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	201 return 0;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	202 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	203
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	204 struct fetch_field {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	205 const char *name;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	206 enum mail_fetch_field wanted_fields;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	207 int (print)(struct fetch_context ctx);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	208 };
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	209
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	210 static const struct fetch_field fetch_fields[] = {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	211 { "mailbox", 0, fetch_mailbox },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	212 { "seq", 0, fetch_seq },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	213 { "uid", 0, fetch_uid },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	214 { "guid", 0, fetch_guid },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	215 { "flags", MAIL_FETCH_FLAGS, fetch_flags },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	216 { "hdr", MAIL_FETCH_STREAM_HEADER, fetch_hdr },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	217 { "body", MAIL_FETCH_STREAM_BODY, fetch_body },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	218 { "text", MAIL_FETCH_STREAM_HEADER \|
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	219 MAIL_FETCH_STREAM_BODY, fetch_text },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	220 { "size.physical", MAIL_FETCH_PHYSICAL_SIZE, fetch_size_physical },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	221 { "size.virtual", MAIL_FETCH_VIRTUAL_SIZE, fetch_size_virtual },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	222 { "date.received", MAIL_FETCH_RECEIVED_DATE, fetch_date_received },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	223 { "date.sent", MAIL_FETCH_DATE, fetch_date_sent },
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	224 { "date.saved", MAIL_FETCH_SAVE_DATE, fetch_date_saved }
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	225 };
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	226
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	227 static const struct fetch_field fetch_field_find(const char name)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	228 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	229 unsigned int i;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	230
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	231 for (i = 0; i < N_ELEMENTS(fetch_fields); i++) {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	232 if (strcmp(fetch_fields[i].name, name) == 0)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	233 return &fetch_fields[i];
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	234 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	235 return NULL;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	236 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	237
11207 1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	238 static void print_fetch_fields(void)
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	239 {
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	240 unsigned int i;
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	241
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	242 fprintf(stderr, "Available fetch fields: %s", fetch_fields[0].name);
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	243 for (i = 1; i < N_ELEMENTS(fetch_fields); i++)
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	244 fprintf(stderr, " %s", fetch_fields[i].name);
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	245 fprintf(stderr, "\n");
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	246 }
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	247
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	248 static void parse_fetch_fields(struct fetch_context ctx, const char str)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	249 {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	250 const char const fields, *name;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	251 const struct fetch_field *field;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	252
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	253 t_array_init(&ctx->fields, 32);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	254 fields = t_strsplit_spaces(str, " ");
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	255 for (; *fields != NULL; fields++) {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	256 name = t_str_lcase(*fields);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	257
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	258 field = fetch_field_find(name);
11207 1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	259 if (field == NULL) {
1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	260 print_fetch_fields();
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	261 i_fatal("Unknown fetch field: %s", name);
11207 1ec4eb879f0e doveadm fetch: If invalid fetch field is given, print list of available fields. Timo Sirainen <tss@iki.fi> parents: 11206 diff changeset	262 }
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	263 ctx->wanted_fields \|= field->wanted_fields;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	264
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	265 array_append(&ctx->fields, field, 1);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	266 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	267 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	268
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	269 static void
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	270 cmd_fetch_box(struct fetch_context ctx, struct mailbox box)
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	271 {
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	272 struct mail_storage *storage = mailbox_get_storage(box);
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	273 struct mailbox_transaction_context *t;
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	274 struct mail_search_context *search_ctx;
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	275 struct mail *mail;
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	276 const struct fetch_field *field;
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	277
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	278 if (mailbox_sync(box, MAILBOX_SYNC_FLAG_FULL_READ) < 0) {
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	279 i_error("Syncing mailbox %s failed: %s", mailbox_get_vname(box),
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	280 mail_storage_get_last_error(storage, NULL));
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	281 return;
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	282 }
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	283
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	284 mail_search_args_init(ctx->search_args, box, FALSE, NULL);
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	285 t = mailbox_transaction_begin(box, 0);
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	286 search_ctx = mailbox_search_init(t, ctx->search_args, NULL);
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	287 mail = mail_alloc(t, ctx->wanted_fields, NULL);
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	288 while (mailbox_search_next(search_ctx, mail)) {
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	289 str_truncate(ctx->hdr, 0);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	290 str_append(ctx->hdr, ctx->prefix);
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	291
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	292 ctx->mail = mail;
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	293 array_foreach(&ctx->fields, field) {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	294 str_printfa(ctx->hdr, "%s: ", field->name);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	295 if (field->print(ctx) < 0) {
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	296 i_error("fetch(%s) failed for box=%s uid=%u: %s",
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	297 field->name, mailbox_get_vname(box),
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	298 mail->uid, mail_storage_get_last_error(storage, NULL));
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	299 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	300 str_append_c(ctx->hdr, '\n');
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	301 }
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	302 flush_hdr(ctx);
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	303
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	304 ctx->mail = NULL;
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	305 }
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	306 mail_free(&mail);
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	307 if (mailbox_search_deinit(&search_ctx) < 0) {
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	308 i_error("Search failed: %s",
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	309 mail_storage_get_last_error(storage, NULL));
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	310 }
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	311 mail_search_args_deinit(ctx->search_args);
11147 51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	312 (void)mailbox_transaction_commit(&t);
51eea5bb3336 doveadm: Added initial code for fetch command. Timo Sirainen <tss@iki.fi> parents: diff changeset	313 }
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	314
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	315 void cmd_fetch(struct mail_user user, const char const args[])
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	316 {
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	317 const enum mailbox_list_iter_flags iter_flags =
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	318 MAILBOX_LIST_ITER_VIRTUAL_NAMES \|
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	319 MAILBOX_LIST_ITER_NO_AUTO_INBOX \|
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	320 MAILBOX_LIST_ITER_RETURN_NO_FLAGS;
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	321 const char *fetch_fields = args[0];
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	322 struct fetch_context ctx;
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	323 struct doveadm_mail_list_iter *iter;
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	324 const struct mailbox_info *info;
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	325 struct mailbox *box;
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	326 const char *storage_name;
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	327 unsigned char prefix_buf[9];
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	328
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	329 memset(&ctx, 0, sizeof(ctx));
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	330 if (fetch_fields == NULL \|\| args[1] == NULL)
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	331 doveadm_mail_help_name("fetch");
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	332 parse_fetch_fields(&ctx, fetch_fields);
11208 f10ad708d748 doveadm: Reduced code duplication. Timo Sirainen <tss@iki.fi> parents: 11207 diff changeset	333 ctx.search_args = doveadm_mail_build_search_args(args + 1);
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	334
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	335 ctx.output = o_stream_create_fd(STDOUT_FILENO, 0, FALSE);
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	336
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	337 random_fill_weak(prefix_buf, sizeof(prefix_buf));
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	338 ctx.hdr = str_new(default_pool, 512);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	339 str_append(ctx.hdr, "===");
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	340 base64_encode(prefix_buf, sizeof(prefix_buf), ctx.hdr);
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	341 str_append_c(ctx.hdr, '\n');
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	342 ctx.prefix = t_strdup(str_c(ctx.hdr));
c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	343 str_truncate(ctx.hdr, 0);
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	344
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	345 iter = doveadm_mail_list_iter_init(user, ctx.search_args, iter_flags);
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	346 while ((info = doveadm_mail_list_iter_next(iter)) != NULL) T_BEGIN {
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	347 storage_name = mail_namespace_get_storage_name(info->ns,
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	348 info->name);
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	349 box = mailbox_alloc(info->ns->list, storage_name,
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	350 MAILBOX_FLAG_KEEP_RECENT \|
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	351 MAILBOX_FLAG_IGNORE_ACLS);
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	352 (void)cmd_fetch_box(&ctx, box);
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	353 mailbox_free(&box);
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	354 } T_END;
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	355 doveadm_mail_list_iter_deinit(&iter);
dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	356 o_stream_unref(&ctx.output);
11206 c8981561c5f2 doveadm fetch: Added "fields to fetch" parameter. Timo Sirainen <tss@iki.fi> parents: 11205 diff changeset	357 str_free(&ctx.hdr);
11205 dab4fb9f8140 doveadm fetch: Added support for fetching mails from multiple mailboxes. Timo Sirainen <tss@iki.fi> parents: 11175 diff changeset	358 }

Mercurial > dovecot > core-2.2

annotate src/doveadm/doveadm-mail-fetch.c @ 11208:f10ad708d748 HEAD