]> git.notmuchmail.org Git - notmuch/blobdiff - notmuch-reply.c
Use notmuch_database_destroy instead of notmuch_database_close
[notmuch] / notmuch-reply.c
index bf6796019e71b3bbd295bf6c0bae555afdb9040e..da99a1374688befd36a859807862a25d7043a825 100644 (file)
  */
 
 #include "notmuch-client.h"
-#include "gmime-filter-reply.h"
 #include "gmime-filter-headers.h"
 
-static void
-reply_headers_message_part (GMimeMessage *message);
-
-static void
-reply_part_content (GMimeObject *part);
-
-static const notmuch_show_format_t format_reply = {
-    "",
-       "", NULL,
-           "", NULL, reply_headers_message_part, ">\n",
-           "",
-               NULL,
-               NULL,
-               NULL,
-               reply_part_content,
-               NULL,
-               "",
-           "",
-       "", "",
-    ""
-};
-
 static void
 show_reply_headers (GMimeMessage *message)
 {
@@ -66,85 +43,59 @@ show_reply_headers (GMimeMessage *message)
 }
 
 static void
-reply_headers_message_part (GMimeMessage *message)
-{
-    InternetAddressList *recipients;
-    const char *recipients_string;
-
-    printf ("> From: %s\n", g_mime_message_get_sender (message));
-    recipients = g_mime_message_get_recipients (message, GMIME_RECIPIENT_TYPE_TO);
-    recipients_string = internet_address_list_to_string (recipients, 0);
-    if (recipients_string)
-       printf ("> To: %s\n",
-               recipients_string);
-    recipients = g_mime_message_get_recipients (message, GMIME_RECIPIENT_TYPE_CC);
-    recipients_string = internet_address_list_to_string (recipients, 0);
-    if (recipients_string)
-       printf ("> Cc: %s\n",
-               recipients_string);
-    printf ("> Subject: %s\n", g_mime_message_get_subject (message));
-    printf ("> Date: %s\n", g_mime_message_get_date_as_string (message));
-}
-
-
-static void
-reply_part_content (GMimeObject *part)
+format_part_reply (mime_node_t *node)
 {
-    GMimeContentType *content_type = g_mime_object_get_content_type (GMIME_OBJECT (part));
-    GMimeContentDisposition *disposition = g_mime_object_get_content_disposition (part);
-
-    if (g_mime_content_type_is_type (content_type, "multipart", "*") ||
-       g_mime_content_type_is_type (content_type, "message", "rfc822"))
-    {
-       /* Output nothing, since multipart subparts will be handled individually. */
-    }
-    else if (g_mime_content_type_is_type (content_type, "application", "pgp-encrypted") ||
-            g_mime_content_type_is_type (content_type, "application", "pgp-signature"))
-    {
-       /* Ignore PGP/MIME cruft parts */
-    }
-    else if (g_mime_content_type_is_type (content_type, "text", "*") &&
-       !g_mime_content_type_is_type (content_type, "text", "html"))
-    {
-       GMimeStream *stream_stdout = NULL, *stream_filter = NULL;
-       GMimeDataWrapper *wrapper;
-       const char *charset;
+    int i;
 
-       charset = g_mime_object_get_content_type_parameter (part, "charset");
-       stream_stdout = g_mime_stream_file_new (stdout);
-       if (stream_stdout) {
+    if (node->envelope_file) {
+       printf ("On %s, %s wrote:\n",
+               notmuch_message_get_header (node->envelope_file, "date"),
+               notmuch_message_get_header (node->envelope_file, "from"));
+    } else if (GMIME_IS_MESSAGE (node->part)) {
+       GMimeMessage *message = GMIME_MESSAGE (node->part);
+       InternetAddressList *recipients;
+       const char *recipients_string;
+
+       printf ("> From: %s\n", g_mime_message_get_sender (message));
+       recipients = g_mime_message_get_recipients (message, GMIME_RECIPIENT_TYPE_TO);
+       recipients_string = internet_address_list_to_string (recipients, 0);
+       if (recipients_string)
+           printf ("> To: %s\n",
+                   recipients_string);
+       recipients = g_mime_message_get_recipients (message, GMIME_RECIPIENT_TYPE_CC);
+       recipients_string = internet_address_list_to_string (recipients, 0);
+       if (recipients_string)
+           printf ("> Cc: %s\n",
+                   recipients_string);
+       printf ("> Subject: %s\n", g_mime_message_get_subject (message));
+       printf ("> Date: %s\n", g_mime_message_get_date_as_string (message));
+       printf (">\n");
+    } else if (GMIME_IS_PART (node->part)) {
+       GMimeContentType *content_type = g_mime_object_get_content_type (node->part);
+       GMimeContentDisposition *disposition = g_mime_object_get_content_disposition (node->part);
+
+       if (g_mime_content_type_is_type (content_type, "application", "pgp-encrypted") ||
+           g_mime_content_type_is_type (content_type, "application", "pgp-signature")) {
+           /* Ignore PGP/MIME cruft parts */
+       } else if (g_mime_content_type_is_type (content_type, "text", "*") &&
+                  !g_mime_content_type_is_type (content_type, "text", "html")) {
+           GMimeStream *stream_stdout = g_mime_stream_file_new (stdout);
            g_mime_stream_file_set_owner (GMIME_STREAM_FILE (stream_stdout), FALSE);
-           stream_filter = g_mime_stream_filter_new(stream_stdout);
-           if (charset) {
-               g_mime_stream_filter_add(GMIME_STREAM_FILTER(stream_filter),
-                                        g_mime_filter_charset_new(charset, "UTF-8"));
-           }
-       }
-       g_mime_stream_filter_add(GMIME_STREAM_FILTER(stream_filter),
-                                g_mime_filter_reply_new(TRUE));
-       wrapper = g_mime_part_get_content_object (GMIME_PART (part));
-       if (wrapper && stream_filter)
-           g_mime_data_wrapper_write_to_stream (wrapper, stream_filter);
-       if (stream_filter)
-           g_object_unref(stream_filter);
-       if (stream_stdout)
+           show_text_part_content (node->part, stream_stdout, NOTMUCH_SHOW_TEXT_PART_REPLY);
            g_object_unref(stream_stdout);
-    }
-    else
-    {
-       if (disposition &&
-           strcmp (disposition->disposition, GMIME_DISPOSITION_ATTACHMENT) == 0)
-       {
-           const char *filename = g_mime_part_get_filename (GMIME_PART (part));
+       } else if (disposition &&
+                  strcmp (disposition->disposition, GMIME_DISPOSITION_ATTACHMENT) == 0) {
+           const char *filename = g_mime_part_get_filename (GMIME_PART (node->part));
            printf ("Attachment: %s (%s)\n", filename,
                    g_mime_content_type_to_string (content_type));
-       }
-       else
-       {
+       } else {
            printf ("Non-text part: %s\n",
                    g_mime_content_type_to_string (content_type));
        }
     }
+
+    for (i = 0; i < node->nchildren; i++)
+       format_part_reply (mime_node_child (node, i));
 }
 
 /* Is the given address configured as one of the user's "personal" or
@@ -505,6 +456,61 @@ guess_from_received_header (notmuch_config_t *config, notmuch_message_t *message
     return NULL;
 }
 
+static GMimeMessage *
+create_reply_message(void *ctx,
+                    notmuch_config_t *config,
+                    notmuch_message_t *message,
+                    notmuch_bool_t reply_all)
+{
+    const char *subject, *from_addr = NULL;
+    const char *in_reply_to, *orig_references, *references;
+
+    /* The 1 means we want headers in a "pretty" order. */
+    GMimeMessage *reply = g_mime_message_new (1);
+    if (reply == NULL) {
+       fprintf (stderr, "Out of memory\n");
+       return NULL;
+    }
+
+    subject = notmuch_message_get_header (message, "subject");
+    if (subject) {
+       if (strncasecmp (subject, "Re:", 3))
+           subject = talloc_asprintf (ctx, "Re: %s", subject);
+       g_mime_message_set_subject (reply, subject);
+    }
+
+    from_addr = add_recipients_from_message (reply, config,
+                                            message, reply_all);
+
+    if (from_addr == NULL)
+       from_addr = guess_from_received_header (config, message);
+
+    if (from_addr == NULL)
+       from_addr = notmuch_config_get_user_primary_email (config);
+
+    from_addr = talloc_asprintf (ctx, "%s <%s>",
+                                notmuch_config_get_user_name (config),
+                                from_addr);
+    g_mime_object_set_header (GMIME_OBJECT (reply),
+                             "From", from_addr);
+
+    in_reply_to = talloc_asprintf (ctx, "<%s>",
+                                  notmuch_message_get_message_id (message));
+
+    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);
+
+    return reply;
+}
+
 static int
 notmuch_reply_format_default(void *ctx,
                             notmuch_config_t *config,
@@ -515,9 +521,7 @@ notmuch_reply_format_default(void *ctx,
     GMimeMessage *reply;
     notmuch_messages_t *messages;
     notmuch_message_t *message;
-    const char *subject, *from_addr = NULL;
-    const char *in_reply_to, *orig_references, *references;
-    const notmuch_show_format_t *format = &format_reply;
+    mime_node_t *root;
 
     for (messages = notmuch_query_search_messages (query);
         notmuch_messages_valid (messages);
@@ -525,62 +529,74 @@ notmuch_reply_format_default(void *ctx,
     {
        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");
+       reply = create_reply_message (ctx, config, message, reply_all);
+
+       /* If reply creation failed, we're out of memory, so don't
+        * bother trying any more messages.
+        */
+       if (!reply) {
+           notmuch_message_destroy (message);
            return 1;
        }
 
-       subject = notmuch_message_get_header (message, "subject");
-       if (subject) {
-           if (strncasecmp (subject, "Re:", 3))
-               subject = talloc_asprintf (ctx, "Re: %s", subject);
-           g_mime_message_set_subject (reply, subject);
-       }
+       show_reply_headers (reply);
 
-       from_addr = add_recipients_from_message (reply, config, message,
-                                                reply_all);
+       g_object_unref (G_OBJECT (reply));
+       reply = NULL;
 
-       if (from_addr == NULL)
-           from_addr = guess_from_received_header (config, message);
+       if (mime_node_open (ctx, message, params->cryptoctx, params->decrypt,
+                           &root) == NOTMUCH_STATUS_SUCCESS) {
+           format_part_reply (root);
+           talloc_free (root);
+       }
 
-       if (from_addr == NULL)
-           from_addr = notmuch_config_get_user_primary_email (config);
+       notmuch_message_destroy (message);
+    }
+    return 0;
+}
 
-       from_addr = talloc_asprintf (ctx, "%s <%s>",
-                                    notmuch_config_get_user_name (config),
-                                    from_addr);
-       g_mime_object_set_header (GMIME_OBJECT (reply),
-                                 "From", from_addr);
+static int
+notmuch_reply_format_json(void *ctx,
+                         notmuch_config_t *config,
+                         notmuch_query_t *query,
+                         notmuch_show_params_t *params,
+                         notmuch_bool_t reply_all)
+{
+    GMimeMessage *reply;
+    notmuch_messages_t *messages;
+    notmuch_message_t *message;
+    mime_node_t *node;
 
-       in_reply_to = talloc_asprintf (ctx, "<%s>",
-                            notmuch_message_get_message_id (message));
+    if (notmuch_query_count_messages (query) != 1) {
+       fprintf (stderr, "Error: search term did not match precisely one message.\n");
+       return 1;
+    }
 
-       g_mime_object_set_header (GMIME_OBJECT (reply),
-                                 "In-Reply-To", in_reply_to);
+    messages = notmuch_query_search_messages (query);
+    message = notmuch_messages_get (messages);
+    if (mime_node_open (ctx, message, params->cryptoctx, params->decrypt,
+                       &node) != NOTMUCH_STATUS_SUCCESS)
+       return 1;
 
-       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 = create_reply_message (ctx, config, message, reply_all);
+    if (!reply)
+       return 1;
 
-       show_reply_headers (reply);
+    /* The headers of the reply message we've created */
+    printf ("{\"reply-headers\": ");
+    format_headers_json (ctx, reply, TRUE);
+    g_object_unref (G_OBJECT (reply));
+    reply = NULL;
 
-       g_object_unref (G_OBJECT (reply));
-       reply = NULL;
+    /* Start the original */
+    printf (", \"original\": ");
 
-       printf ("On %s, %s wrote:\n",
-               notmuch_message_get_header (message, "date"),
-               notmuch_message_get_header (message, "from"));
+    format_part_json (ctx, node, TRUE);
 
-       show_message_body (message, format, params);
+    /* End */
+    printf ("}\n");
+    notmuch_message_destroy (message);
 
-       notmuch_message_destroy (message);
-    }
     return 0;
 }
 
@@ -646,6 +662,7 @@ notmuch_reply_format_headers_only(void *ctx,
 
 enum {
     FORMAT_DEFAULT,
+    FORMAT_JSON,
     FORMAT_HEADERS_ONLY,
 };
 
@@ -661,18 +678,18 @@ notmuch_reply_command (void *ctx, int argc, char *argv[])
     notmuch_show_params_t params = { .part = -1 };
     int format = FORMAT_DEFAULT;
     int reply_all = TRUE;
-    notmuch_bool_t decrypt = FALSE;
 
     notmuch_opt_desc_t options[] = {
        { NOTMUCH_OPT_KEYWORD, &format, "format", 'f',
          (notmuch_keyword_t []){ { "default", FORMAT_DEFAULT },
+                                 { "json", FORMAT_JSON },
                                  { "headers-only", FORMAT_HEADERS_ONLY },
                                  { 0, 0 } } },
        { NOTMUCH_OPT_KEYWORD, &reply_all, "reply-to", 'r',
          (notmuch_keyword_t []){ { "all", TRUE },
                                  { "sender", FALSE },
                                  { 0, 0 } } },
-       { NOTMUCH_OPT_BOOLEAN, &decrypt, "decrypt", 'd', 0 },
+       { NOTMUCH_OPT_BOOLEAN, &params.decrypt, "decrypt", 'd', 0 },
        { 0, 0, 0, 0, 0 }
     };
 
@@ -684,10 +701,12 @@ notmuch_reply_command (void *ctx, int argc, char *argv[])
 
     if (format == FORMAT_HEADERS_ONLY)
        reply_format_func = notmuch_reply_format_headers_only;
+    else if (format == FORMAT_JSON)
+       reply_format_func = notmuch_reply_format_json;
     else
        reply_format_func = notmuch_reply_format_default;
 
-    if (decrypt) {
+    if (params.decrypt) {
 #ifdef GMIME_ATLEAST_26
        /* TODO: GMimePasswordRequestFunc */
        params.cryptoctx = g_mime_gpg_context_new (NULL, "gpg");
@@ -697,8 +716,8 @@ notmuch_reply_command (void *ctx, int argc, char *argv[])
 #endif
        if (params.cryptoctx) {
            g_mime_gpg_context_set_always_trust ((GMimeGpgContext*) params.cryptoctx, FALSE);
-           params.decrypt = TRUE;
        } else {
+           params.decrypt = FALSE;
            fprintf (stderr, "Failed to construct gpg context.\n");
        }
 #ifndef GMIME_ATLEAST_26
@@ -736,7 +755,7 @@ notmuch_reply_command (void *ctx, int argc, char *argv[])
        return 1;
 
     notmuch_query_destroy (query);
-    notmuch_database_close (notmuch);
+    notmuch_database_destroy (notmuch);
 
     if (params.cryptoctx)
        g_object_unref(params.cryptoctx);