notmuch reply: Use GMime to construct the header for the reply.
authorCarl Worth <cworth@cworth.org>
Wed, 11 Nov 2009 18:54:12 +0000 (10:54 -0800)
committerCarl Worth <cworth@cworth.org>
Wed, 11 Nov 2009 18:54:12 +0000 (10:54 -0800)
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
notmuch.c
show-message.c

index 40ef4c64599c5599743be2449bc96c89d3405856..35a735f13f321924eec34ac9d0a2b26c74df04a9 100644 (file)
@@ -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);
 
-       subject = notmuch_message_get_header (message, "subject");
+       /* 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: 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: 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).
+        */
 
-       /* XXX: Should fold long recipient lists. */
-       printf ("To:");
-       has_recipient = 0;
+       subject = notmuch_message_get_header (message, "subject");
 
-       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;
+       if (strncasecmp (subject, "Re:", 3))
+           subject = talloc_asprintf (ctx, "Re: %s", subject);
+       g_mime_message_set_subject (reply, subject);
+
+       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);
        }
 
-       printf ("\n");
+       in_reply_to = talloc_asprintf (ctx, "<%s>",
+                            notmuch_message_get_message_id (message));
 
-       if (strncasecmp (subject, "Re:", 3))
-           subject = talloc_asprintf (ctx, "Re: %s", subject);
-       printf ("Subject: %s\n", subject);
+       g_mime_object_set_header (GMIME_OBJECT (reply),
+                                 "In-Reply-To", in_reply_to);
 
-       printf ("In-Reply-To: <%s>\n",
-               notmuch_message_get_message_id (message));
+       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);
 
-       /* 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));
+       reply_headers = g_mime_object_to_string (GMIME_OBJECT (reply));
+       printf ("%s", reply_headers);
+       free (reply_headers);
 
-       printf ("--text follows this line--\n");
+       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;
 }
index 224e4a6bfe5d4ea28e1d4477edd6942058940cb0..56f93a7c11b3eac8c089bba5aaf6d70ebe82ae81 100644 (file)
--- 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);
 
index 2d3189e33474b02e2e177827f4ab17b3459d24b3..79b02e2fd2688294591122cb44e75a868ab5183e 100644 (file)
@@ -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));