X-Git-Url: https://git.notmuchmail.org/git?p=notmuch;a=blobdiff_plain;f=notmuch-search.c;h=93724dd2994422cfd0365eff0ebe7d9fe70bbe4d;hp=482c6e8ea8c061d7865728c857a30f1dfe31497a;hb=6dcb7592e32ed5140ea0c0357ce78d6a37af6066;hpb=6ce2bf68f53fec4302121763e588c33c7cd86a9c diff --git a/notmuch-search.c b/notmuch-search.c index 482c6e8e..93724dd2 100644 --- a/notmuch-search.c +++ b/notmuch-search.c @@ -20,28 +20,43 @@ #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 *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 *id, + const char *thread_id, const time_t date, const int matched, const int total, @@ -50,17 +65,23 @@ 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 *id, + const char *thread_id, const time_t date, const int matched, const int total, @@ -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,9 +99,17 @@ 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 *id, + const char *thread_id, const time_t date, const int matched, const int total, @@ -87,7 +117,7 @@ format_thread_text (const void *ctx, const char *subject) { printf ("thread:%s %12s [%d/%d] %s; %s", - id, + thread_id, notmuch_time_relative_date (ctx, date), matched, total, @@ -95,34 +125,38 @@ 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 *id, + const char *thread_id, const time_t date, const int matched, const int total, const char *authors, const char *subject) { - struct tm *tm; - char timestamp[40]; void *ctx_quote = talloc_new (ctx); - tm = gmtime (&date); - if (tm == NULL) - INTERNAL_ERROR ("gmtime failed on thread %s.", id); - - if (strftime (timestamp, sizeof (timestamp), "%s", tm) == 0) - INTERNAL_ERROR ("strftime failed on thread %s.", id); - - printf ("\"id\": %s,\n" - "\"timestamp\": %s,\n" + printf ("\"thread\": %s,\n" + "\"timestamp\": %ld,\n" "\"matched\": %d,\n" "\"total\": %d,\n" "\"authors\": %s,\n" "\"subject\": %s,\n", - json_quote_str (ctx_quote, id), - timestamp, + json_quote_str (ctx_quote, thread_id), + date, matched, total, json_quote_str (ctx_quote, authors), @@ -131,11 +165,12 @@ format_thread_json (const void *ctx, talloc_free (ctx_quote); } -static void +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; @@ -143,55 +178,143 @@ 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; - for (threads = notmuch_query_search_threads (query); - notmuch_threads_has_more (threads); - notmuch_threads_advance (threads)) + for (; + notmuch_threads_valid (threads); + notmuch_threads_move_to_next (threads)) { 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_has_more (tags); - notmuch_tags_advance (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->tag_end, stdout); - fputs (format->thread_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; } int @@ -204,7 +327,8 @@ notmuch_search_command (void *ctx, int argc, char *argv[]) char *opt; notmuch_sort_t sort = NOTMUCH_SORT_NEWEST_FIRST; const search_format_t *format = &format_text; - int i; + int i, ret; + output_t output = OUTPUT_SUMMARY; for (i = 0; i < argc && argv[i][0] == '-'; i++) { if (strcmp (argv[i], "--") == 0) { @@ -231,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; @@ -267,10 +405,26 @@ notmuch_search_command (void *ctx, int argc, char *argv[]) notmuch_query_set_sort (query, sort); - 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); - return 0; + return ret; }