From: Carl Worth Date: Thu, 28 Oct 2010 18:19:53 +0000 (-0700) Subject: notmuch search: Add a new --output=(summary|threads|messages|tags) option. X-Git-Tag: 0.4~38 X-Git-Url: http://git.tremily.us/?a=commitdiff_plain;h=6dcb7592;p=notmuch.git notmuch search: Add a new --output=(summary|threads|messages|tags) option. This can be handy for scripts which need to extract just a thread ID from a search term, for example. --- diff --git a/TODO b/TODO index 626ece43..08f0882b 100644 --- a/TODO +++ b/TODO @@ -89,15 +89,6 @@ option (or similar) to "notmuch show".) For now, this is being worked around in the emacs interface by noticing that "notmuch show" returns nothing and re-rerunning the command without the extra arguments. -Teach "notmuch search" to return many different kinds of results. Some -ideas: - - notmuch search --output=threads # Default if no --output is given - notmuch search --output=messages - notmuch search --output=tags - notmuch search --output=addresses - notmuch search --output=terms - Add a "--format" option to "notmuch search", (something printf-like for selecting what gets printed). diff --git a/notmuch-search.c b/notmuch-search.c index 704aa43b..93724dd2 100644 --- a/notmuch-search.c +++ b/notmuch-search.c @@ -20,25 +20,40 @@ #include "notmuch-client.h" +typedef enum { + OUTPUT_SUMMARY, + OUTPUT_THREADS, + OUTPUT_MESSAGES, + OUTPUT_TAGS +} output_t; + typedef struct search_format { const char *results_start; - const char *thread_start; - void (*thread) (const void *ctx, - const char *thread_id, - const time_t date, - const int matched, - const int total, - const char *authors, - const char *subject); + const char *item_start; + void (*item_id) (const void *ctx, + const char *item_type, + const char *item_id); + void (*thread_summary) (const void *ctx, + const char *thread_id, + const time_t date, + const int matched, + const int total, + const char *authors, + const char *subject); const char *tag_start; const char *tag; const char *tag_sep; const char *tag_end; - const char *thread_sep; - const char *thread_end; + const char *item_sep; + const char *item_end; const char *results_end; } search_format_t; +static void +format_item_id_text (const void *ctx, + const char *item_type, + const char *item_id); + static void format_thread_text (const void *ctx, const char *thread_id, @@ -50,14 +65,20 @@ format_thread_text (const void *ctx, static const search_format_t format_text = { "", "", + format_item_id_text, format_thread_text, " (", "%s", " ", - ")", "", - "\n", - "", + ")", "\n", + "", + "\n", }; +static void +format_item_id_json (const void *ctx, + const char *item_type, + const char *item_id); + static void format_thread_json (const void *ctx, const char *thread_id, @@ -69,6 +90,7 @@ format_thread_json (const void *ctx, static const search_format_t format_json = { "[", "{", + format_item_id_json, format_thread_json, "\"tags\": [", "\"%s\"", ", ", @@ -77,6 +99,14 @@ static const search_format_t format_json = { "]\n", }; +static void +format_item_id_text (unused (const void *ctx), + const char *item_type, + const char *item_id) +{ + printf ("%s%s", item_type, item_id); +} + static void format_thread_text (const void *ctx, const char *thread_id, @@ -95,6 +125,19 @@ format_thread_text (const void *ctx, subject); } +static void +format_item_id_json (const void *ctx, + unused (const char *item_type), + const char *item_id) +{ + void *ctx_quote = talloc_new (ctx); + + printf ("%s", json_quote_str (ctx_quote, item_id)); + + talloc_free (ctx_quote); + +} + static void format_thread_json (const void *ctx, const char *thread_id, @@ -126,7 +169,8 @@ static int do_search_threads (const void *ctx, const search_format_t *format, notmuch_query_t *query, - notmuch_sort_t sort) + notmuch_sort_t sort, + output_t output) { notmuch_thread_t *thread; notmuch_threads_t *threads; @@ -134,8 +178,6 @@ do_search_threads (const void *ctx, time_t date; int first_thread = 1; - fputs (format->results_start, stdout); - threads = notmuch_query_search_threads (query); if (threads == NULL) return 1; @@ -147,46 +189,130 @@ do_search_threads (const void *ctx, int first_tag = 1; if (! first_thread) - fputs (format->thread_sep, stdout); + fputs (format->item_sep, stdout); thread = notmuch_threads_get (threads); - if (sort == NOTMUCH_SORT_OLDEST_FIRST) - date = notmuch_thread_get_oldest_date (thread); - else - date = notmuch_thread_get_newest_date (thread); - - fputs (format->thread_start, stdout); - - format->thread (ctx, - notmuch_thread_get_thread_id (thread), - date, - notmuch_thread_get_matched_messages (thread), - notmuch_thread_get_total_messages (thread), - notmuch_thread_get_authors (thread), - notmuch_thread_get_subject (thread)); - - fputs (format->tag_start, stdout); - - for (tags = notmuch_thread_get_tags (thread); - notmuch_tags_valid (tags); - notmuch_tags_move_to_next (tags)) - { - if (! first_tag) - fputs (format->tag_sep, stdout); - printf (format->tag, notmuch_tags_get (tags)); - first_tag = 0; - } + if (output == OUTPUT_THREADS) { + format->item_id (ctx, "thread:", + notmuch_thread_get_thread_id (thread)); + } else { /* output == OUTPUT_SUMMARY */ + fputs (format->item_start, stdout); + + if (sort == NOTMUCH_SORT_OLDEST_FIRST) + date = notmuch_thread_get_oldest_date (thread); + else + date = notmuch_thread_get_newest_date (thread); + + format->thread_summary (ctx, + notmuch_thread_get_thread_id (thread), + date, + notmuch_thread_get_matched_messages (thread), + notmuch_thread_get_total_messages (thread), + notmuch_thread_get_authors (thread), + notmuch_thread_get_subject (thread)); + + fputs (format->tag_start, stdout); + + for (tags = notmuch_thread_get_tags (thread); + notmuch_tags_valid (tags); + notmuch_tags_move_to_next (tags)) + { + if (! first_tag) + fputs (format->tag_sep, stdout); + printf (format->tag, notmuch_tags_get (tags)); + first_tag = 0; + } - fputs (format->tag_end, stdout); - fputs (format->thread_end, stdout); + fputs (format->tag_end, stdout); + + fputs (format->item_end, stdout); + } first_thread = 0; notmuch_thread_destroy (thread); } - fputs (format->results_end, stdout); + return 0; +} + +static int +do_search_messages (const void *ctx, + const search_format_t *format, + notmuch_query_t *query) +{ + notmuch_message_t *message; + notmuch_messages_t *messages; + int first_message = 1; + + messages = notmuch_query_search_messages (query); + if (messages == NULL) + return 1; + + for (; + notmuch_messages_valid (messages); + notmuch_messages_move_to_next (messages)) + { + message = notmuch_messages_get (messages); + + if (! first_message) + fputs (format->item_sep, stdout); + + format->item_id (ctx, "id:", notmuch_message_get_message_id (message)); + + first_message = 0; + + notmuch_message_destroy (message); + } + + notmuch_messages_destroy (messages); + + return 0; +} + +static int +do_search_tags (const void *ctx, + notmuch_database_t *notmuch, + const search_format_t *format, + notmuch_query_t *query) +{ + notmuch_messages_t *messages = NULL; + notmuch_tags_t *tags; + const char *tag; + int first_tag = 1; + + /* Special-case query of "*" for better performance. */ + if (strcmp (notmuch_query_get_query_string (query), "*") == 0) { + tags = notmuch_database_get_all_tags (notmuch); + } else { + messages = notmuch_query_search_messages (query); + if (messages == NULL) + return 1; + + tags = notmuch_messages_collect_tags (messages); + } + if (tags == NULL) + return 1; + + for (; + notmuch_tags_valid (tags); + notmuch_tags_move_to_next (tags)) + { + tag = notmuch_tags_get (tags); + + if (! first_tag) + fputs (format->item_sep, stdout); + + format->item_id (ctx, "", tag); + + first_tag = 0; + } + + notmuch_tags_destroy (tags); + + if (messages) + notmuch_messages_destroy (messages); return 0; } @@ -202,6 +328,7 @@ notmuch_search_command (void *ctx, int argc, char *argv[]) notmuch_sort_t sort = NOTMUCH_SORT_NEWEST_FIRST; const search_format_t *format = &format_text; int i, ret; + output_t output = OUTPUT_SUMMARY; for (i = 0; i < argc && argv[i][0] == '-'; i++) { if (strcmp (argv[i], "--") == 0) { @@ -228,6 +355,20 @@ notmuch_search_command (void *ctx, int argc, char *argv[]) fprintf (stderr, "Invalid value for --format: %s\n", opt); return 1; } + } else if (STRNCMP_LITERAL (argv[i], "--output=") == 0) { + opt = argv[i] + sizeof ("--output=") - 1; + if (strcmp (opt, "summary") == 0) { + output = OUTPUT_SUMMARY; + } else if (strcmp (opt, "threads") == 0) { + output = OUTPUT_THREADS; + } else if (strcmp (opt, "messages") == 0) { + output = OUTPUT_MESSAGES; + } else if (strcmp (opt, "tags") == 0) { + output = OUTPUT_TAGS; + } else { + fprintf (stderr, "Invalid value for --output: %s\n", opt); + return 1; + } } else { fprintf (stderr, "Unrecognized option: %s\n", argv[i]); return 1; @@ -264,7 +405,23 @@ notmuch_search_command (void *ctx, int argc, char *argv[]) notmuch_query_set_sort (query, sort); - ret = do_search_threads (ctx, format, query, sort); + fputs (format->results_start, stdout); + + switch (output) { + default: + case OUTPUT_SUMMARY: + case OUTPUT_THREADS: + ret = do_search_threads (ctx, format, query, sort, output); + break; + case OUTPUT_MESSAGES: + ret = do_search_messages (ctx, format, query); + break; + case OUTPUT_TAGS: + ret = do_search_tags (ctx, notmuch, format, query); + break; + } + + fputs (format->results_end, stdout); notmuch_query_destroy (query); notmuch_database_close (notmuch); diff --git a/notmuch.1 b/notmuch.1 index f803f908..1919176e 100644 --- a/notmuch.1 +++ b/notmuch.1 @@ -150,6 +150,46 @@ include Presents the results in either JSON or plain-text (default). .RE + +.RS 4 +.TP 4 +.B \-\-output=(summary|threads|messages|tags) + +.RS 4 +.TP 4 +.B summary + +Output a summary of each thread with any message matching the search +terms. The summary includes the thread ID, date, the number of +messages in the thread (both the number matched and the total number), +the authors of the thread and the subject. +.RE +.RS 4 +.TP 4 +.B threads + +Output the thread IDs of all threads with any message matching the +search terms, either one per line (--format=text) or as a JSON array +(--format=json). +.RE +.RS 4 +.TP 4 +.B messages + +Output the message IDs of all messages matching the search terms, +either one per line (--format=text) or as a JSON array +(--format=json). +.RE +.RS 4 +.TP 4 +.B tags + +Output all tags that appear on any message matching the search terms, +either one per line (--format=text) or as a JSON array +(--format=json). +.RE +.RE + .RS 4 .TP 4 .BR \-\-sort= ( newest\-first | oldest\-first ) @@ -214,7 +254,6 @@ will be delimited by easily-parsed markers. Each marker consists of a Control-L character (ASCII decimal 12), the name of the marker, and then either an opening or closing brace, ('{' or '}'), to either open or close the component. - .RE .RS 4 .TP 4 @@ -240,7 +279,6 @@ beginning with "From " (preceded by zero or more '>' characters) have an additional '>' character added. This reversible escaping is termed "mboxrd" format and described in detail here: http://homepage.ntlworld.com/jonathan.deboynepollard/FGA/mail-mbox-formats.html - .RE A common use of .B notmuch show diff --git a/notmuch.c b/notmuch.c index 326aa4c8..e92f14c5 100644 --- a/notmuch.c +++ b/notmuch.c @@ -165,6 +165,33 @@ command_t commands[] = { "\t\tPresents the results in either JSON or\n" "\t\tplain-text (default)\n" "\n" + "\t--output=(summary|threads|messages|tags)\n" + "\n" + "\t\tsummary (default)\n" + "\n" + "\t\tOutput a summary of each thread with any message matching the\n" + "\t\tsearch terms. The summary includes the thread ID, date, the\n" + "\t\tnumber of messages in the thread (both the number matched and\n" + "\t\tthe total number), the authors of the thread and the subject.\n" + "\n" + "\t\tthreads\n" + "\n" + "\t\tOutput the thread IDs of all threads with any message matching\n" + "\t\tthe search terms, either one per line (--format=text) or as a\n" + "\t\tJSON array (--format=json).\n" + "\n" + "\t\tmessages\n" + "\n" + "\t\tOutput the message IDs of all messages matching the search\n" + "\t\tterms, either one per line (--format=text) or as a JSON array\n" + "\t\t(--format=json).\n" + "\n" + "\t\ttags\n" + "\n" + "\t\tOutput all tags that appear on any message matching the search\n" + "\t\tterms, either one per line (--format=text) or as a JSON array\n" + "\t\t(--format=json).\n" + "\n" "\t--sort=(newest-first|oldest-first)\n" "\n" "\t\tPresent results in either chronological order\n"