From a9f3ad4fcbaf1415ec40a82125dbf4064dbdc819 Mon Sep 17 00:00:00 2001 From: Carl Worth Date: Wed, 11 Nov 2009 10:54:12 -0800 Subject: [PATCH] notmuch reply: Use GMime to construct the header for the reply. The advantage here is that we actually get the necessary folding of long headers, (particularly the References header, but also things like Subject). This also gives us parsed recipient addresses so that we can easily elide the sender's address(es) from the recipient list (just as soon as we have a configured value for the recipient's address(es)). --- notmuch-reply.c | 163 +++++++++++++++++++++++++++++++++--------------- notmuch.c | 2 + show-message.c | 6 -- 3 files changed, 116 insertions(+), 55 deletions(-) diff --git a/notmuch-reply.c b/notmuch-reply.c index 40ef4c64..35a735f1 100644 --- a/notmuch-reply.c +++ b/notmuch-reply.c @@ -71,18 +71,79 @@ reply_part(GMimeObject *part, int *part_count) } } +static void +add_recipients_for_address_list (GMimeMessage *message, + GMimeRecipientType type, + InternetAddressList *list) +{ + InternetAddress *address; + int i; + + for (i = 0; i < internet_address_list_length (list); i++) { + address = internet_address_list_get_address (list, i); + if (INTERNET_ADDRESS_IS_GROUP (address)) { + InternetAddressGroup *group; + InternetAddressList *group_list; + + group = INTERNET_ADDRESS_GROUP (address); + group_list = internet_address_group_get_members (group); + if (group_list == NULL) + continue; + + add_recipients_for_address_list (message, type, group_list); + } else { + InternetAddressMailbox *mailbox; + const char *name; + const char *addr; + + mailbox = INTERNET_ADDRESS_MAILBOX (address); + + name = internet_address_get_name (address); + addr = internet_address_mailbox_get_addr (mailbox); + + g_mime_message_add_recipient (message, type, name, addr); + } + } +} + +static void +add_recipients_for_string (GMimeMessage *message, + GMimeRecipientType type, + const char *recipients) +{ + InternetAddressList *list; + + list = internet_address_list_parse_string (recipients); + if (list == NULL) + return; + + add_recipients_for_address_list (message, type, list); +} + int notmuch_reply_command (void *ctx, int argc, char *argv[]) { void *local = talloc_new (ctx); - char *query_string; - notmuch_database_t *notmuch = NULL; notmuch_query_t *query = NULL; + notmuch_database_t *notmuch = NULL; + GMimeMessage *reply = NULL; + char *query_string; notmuch_messages_t *messages; notmuch_message_t *message; int ret = 0; - int has_recipient; - const char *subject, *to, *references; + const char *subject, *recipients; + const char *in_reply_to, *orig_references, *references; + char *reply_headers; + struct { + const char *header; + GMimeRecipientType recipient_type; + } reply_to_map[] = { + { "from", GMIME_RECIPIENT_TYPE_TO }, + { "to", GMIME_RECIPIENT_TYPE_TO }, + { "cc", GMIME_RECIPIENT_TYPE_CC }, + { "bcc", GMIME_RECIPIENT_TYPE_BCC } + }; + unsigned int i; notmuch = notmuch_database_open (NULL); if (notmuch == NULL) { @@ -110,58 +171,59 @@ notmuch_reply_command (void *ctx, int argc, char *argv[]) { message = notmuch_messages_get (messages); + /* The 1 means we want headers in a "pretty" order. */ + reply = g_mime_message_new (1); + if (reply == NULL) { + fprintf (stderr, "Out of memory\n"); + ret = 1; + goto DONE; + } + + /* XXX: We need a configured email address (or addresses) for + * the user here, so that we can prevent replying to the user, + * and also call _mime_message_set_sender to set From: (either + * from the first "owned" address mentioned as a recipient in + * the original message, or else some default address). + */ + subject = notmuch_message_get_header (message, "subject"); - /* XXX: Should have the user's email address(es) configured - * somewhere, and should fish it out of any recipient headers - * to reply by default from the same address that the original - * email was sent to */ - printf ("From: \n"); - - /* XXX: Should fold long recipient lists. */ - printf ("To:"); - has_recipient = 0; - - to = notmuch_message_get_header (message, "from"); - if (to) { - printf (" %s", to); - has_recipient = 1; - } - to = notmuch_message_get_header (message, "to"); - if (to) { - printf ("%s%s", - has_recipient ? ", " : " ", to); - has_recipient = 1; - } - to = notmuch_message_get_header (message, "cc"); - if (to) { - printf ("%s%s", - has_recipient ? ", " : " ", to); - has_recipient = 1; - } - to = notmuch_message_get_header (message, "bcc"); - if (to) { - printf ("%s%s", - has_recipient ? ", " : " ", to); - has_recipient = 1; - } - - printf ("\n"); - if (strncasecmp (subject, "Re:", 3)) subject = talloc_asprintf (ctx, "Re: %s", subject); - printf ("Subject: %s\n", subject); + g_mime_message_set_subject (reply, subject); - printf ("In-Reply-To: <%s>\n", - notmuch_message_get_message_id (message)); + for (i = 0; i < ARRAY_SIZE (reply_to_map); i++) { + recipients = notmuch_message_get_header (message, + reply_to_map[i].header); + add_recipients_for_string (reply, + reply_to_map[i].recipient_type, + recipients); + } - /* XXX: Should fold long references lists. */ - references = notmuch_message_get_header (message, "references"); - printf ("References: %s <%s>\n", - references ? references : "", - notmuch_message_get_message_id (message)); + in_reply_to = talloc_asprintf (ctx, "<%s>", + notmuch_message_get_message_id (message)); - printf ("--text follows this line--\n"); + g_mime_object_set_header (GMIME_OBJECT (reply), + "In-Reply-To", in_reply_to); + + orig_references = notmuch_message_get_header (message, "references"); + references = talloc_asprintf (ctx, "%s%s%s", + orig_references ? orig_references : "", + orig_references ? " " : "", + in_reply_to); + g_mime_object_set_header (GMIME_OBJECT (reply), + "References", references); + + reply_headers = g_mime_object_to_string (GMIME_OBJECT (reply)); + printf ("%s", reply_headers); + free (reply_headers); + + g_object_unref (G_OBJECT (reply)); + reply = NULL; + + printf ("On %s, %s wrote:\n", + notmuch_message_get_header (message, "date"), + notmuch_message_get_header (message, "from")); show_message_body (notmuch_message_get_filename (message), reply_part); @@ -178,5 +240,8 @@ notmuch_reply_command (void *ctx, int argc, char *argv[]) if (notmuch) notmuch_database_close (notmuch); + if (reply) + g_object_unref (G_OBJECT (reply)); + return ret; } diff --git a/notmuch.c b/notmuch.c index 224e4a6b..56f93a7c 100644 --- a/notmuch.c +++ b/notmuch.c @@ -212,6 +212,8 @@ main (int argc, char *argv[]) command_t *command; unsigned int i; + g_mime_init (0); + if (argc == 1) return notmuch_setup_command (local, 0, NULL); diff --git a/show-message.c b/show-message.c index 2d3189e3..79b02e2f 100644 --- a/show-message.c +++ b/show-message.c @@ -74,15 +74,9 @@ show_message_body (const char *filename, GMimeParser *parser = NULL; GMimeMessage *mime_message = NULL; notmuch_status_t ret = NOTMUCH_STATUS_SUCCESS; - static int initialized = 0; FILE *file = NULL; int part_count = 0; - if (! initialized) { - g_mime_init (0); - initialized = 1; - } - file = fopen (filename, "r"); if (! file) { fprintf (stderr, "Error opening %s: %s\n", filename, strerror (errno));