git.notmuchmail.org Git - notmuch/blob - notmuch-search.c

   1 /* notmuch - Not much of an email program, (just index and search)
   2  *
   3  * Copyright © 2009 Carl Worth
   4  *
   5  * This program is free software: you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation, either version 3 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program.  If not, see http://www.gnu.org/licenses/ .
  17  *
  18  * Author: Carl Worth <cworth@cworth.org>
  19  */
  20
  21 #include "notmuch-client.h"
  22
  23 typedef struct search_format {
  24     const char *results_start;
  25     const char *thread_start;
  26     void (*thread) (const void *ctx,
  27                     const char *thread_id,
  28                     const time_t date,
  29                     const int matched,
  30                     const int total,
  31                     const char *authors,
  32                     const char *subject);
  33     const char *tag_start;
  34     const char *tag;
  35     const char *tag_sep;
  36     const char *tag_end;
  37     const char *thread_sep;
  38     const char *thread_end;
  39     const char *results_end;
  40 } search_format_t;
  41
  42 static void
  43 format_thread_text (const void *ctx,
  44                     const char *thread_id,
  45                     const time_t date,
  46                     const int matched,
  47                     const int total,
  48                     const char *authors,
  49                     const char *subject);
  50 static const search_format_t format_text = {
  51     "",
  52         "",
  53             format_thread_text,
  54             " (",
  55                 "%s", " ",
  56             ")", "",
  57         "\n",
  58     "",
  59 };
  60
  61 static void
  62 format_thread_json (const void *ctx,
  63                     const char *thread_id,
  64                     const time_t date,
  65                     const int matched,
  66                     const int total,
  67                     const char *authors,
  68                     const char *subject);
  69 static const search_format_t format_json = {
  70     "[",
  71         "{",
  72             format_thread_json,
  73             "\"tags\": [",
  74                 "\"%s\"", ", ",
  75             "]", ",\n",
  76         "}",
  77     "]\n",
  78 };
  79
  80 static void
  81 format_thread_text (const void *ctx,
  82                     const char *thread_id,
  83                     const time_t date,
  84                     const int matched,
  85                     const int total,
  86                     const char *authors,
  87                     const char *subject)
  88 {
  89     printf ("thread:%s %12s [%d/%d] %s; %s",
  90             thread_id,
  91             notmuch_time_relative_date (ctx, date),
  92             matched,
  93             total,
  94             authors,
  95             subject);
  96 }
  97
  98 static void
  99 format_thread_json (const void *ctx,
 100                     const char *thread_id,
 101                     const time_t date,
 102                     const int matched,
 103                     const int total,
 104                     const char *authors,
 105                     const char *subject)
 106 {
 107     struct tm *tm;
 108     char timestamp[40];
 109     void *ctx_quote = talloc_new (ctx);
 110
 111     printf ("\"thread\": %s,\n"
 112             "\"timestamp\": %ld,\n"
 113             "\"matched\": %d,\n"
 114             "\"total\": %d,\n"
 115             "\"authors\": %s,\n"
 116             "\"subject\": %s,\n",
 117             json_quote_str (ctx_quote, thread_id),
 118             date,
 119             matched,
 120             total,
 121             json_quote_str (ctx_quote, authors),
 122             json_quote_str (ctx_quote, subject));
 123
 124     talloc_free (ctx_quote);
 125 }
 126
 127 static void
 128 do_search_threads (const void *ctx,
 129                    const search_format_t *format,
 130                    notmuch_query_t *query,
 131                    notmuch_sort_t sort)
 132 {
 133     notmuch_thread_t *thread;
 134     notmuch_threads_t *threads;
 135     notmuch_tags_t *tags;
 136     time_t date;
 137     int first_thread = 1;
 138
 139     fputs (format->results_start, stdout);
 140
 141     for (threads = notmuch_query_search_threads (query);
 142          notmuch_threads_valid (threads);
 143          notmuch_threads_move_to_next (threads))
 144     {
 145         int first_tag = 1;
 146
 147         if (! first_thread)
 148             fputs (format->thread_sep, stdout);
 149
 150         thread = notmuch_threads_get (threads);
 151
 152         if (sort == NOTMUCH_SORT_OLDEST_FIRST)
 153             date = notmuch_thread_get_oldest_date (thread);
 154         else
 155             date = notmuch_thread_get_newest_date (thread);
 156
 157         fputs (format->thread_start, stdout);
 158
 159         format->thread (ctx,
 160                         notmuch_thread_get_thread_id (thread),
 161                         date,
 162                         notmuch_thread_get_matched_messages (thread),
 163                         notmuch_thread_get_total_messages (thread),
 164                         notmuch_thread_get_authors (thread),
 165                         notmuch_thread_get_subject (thread));
 166
 167         fputs (format->tag_start, stdout);
 168
 169         for (tags = notmuch_thread_get_tags (thread);
 170              notmuch_tags_valid (tags);
 171              notmuch_tags_move_to_next (tags))
 172         {
 173             if (! first_tag)
 174                 fputs (format->tag_sep, stdout);
 175             printf (format->tag, notmuch_tags_get (tags));
 176             first_tag = 0;
 177         }
 178
 179         fputs (format->tag_end, stdout);
 180         fputs (format->thread_end, stdout);
 181
 182         first_thread = 0;
 183
 184         notmuch_thread_destroy (thread);
 185     }
 186
 187     fputs (format->results_end, stdout);
 188 }
 189
 190 int
 191 notmuch_search_command (void *ctx, int argc, char *argv[])
 192 {
 193     notmuch_config_t *config;
 194     notmuch_database_t *notmuch;
 195     notmuch_query_t *query;
 196     char *query_str;
 197     char *opt;
 198     notmuch_sort_t sort = NOTMUCH_SORT_NEWEST_FIRST;
 199     const search_format_t *format = &format_text;
 200     int i;
 201
 202     for (i = 0; i < argc && argv[i][0] == '-'; i++) {
 203         if (strcmp (argv[i], "--") == 0) {
 204             i++;
 205             break;
 206         }
 207         if (STRNCMP_LITERAL (argv[i], "--sort=") == 0) {
 208             opt = argv[i] + sizeof ("--sort=") - 1;
 209             if (strcmp (opt, "oldest-first") == 0) {
 210                 sort = NOTMUCH_SORT_OLDEST_FIRST;
 211             } else if (strcmp (opt, "newest-first") == 0) {
 212                 sort = NOTMUCH_SORT_NEWEST_FIRST;
 213             } else {
 214                 fprintf (stderr, "Invalid value for --sort: %s\n", opt);
 215                 return 1;
 216             }
 217         } else if (STRNCMP_LITERAL (argv[i], "--format=") == 0) {
 218             opt = argv[i] + sizeof ("--format=") - 1;
 219             if (strcmp (opt, "text") == 0) {
 220                 format = &format_text;
 221             } else if (strcmp (opt, "json") == 0) {
 222                 format = &format_json;
 223             } else {
 224                 fprintf (stderr, "Invalid value for --format: %s\n", opt);
 225                 return 1;
 226             }
 227         } else {
 228             fprintf (stderr, "Unrecognized option: %s\n", argv[i]);
 229             return 1;
 230         }
 231     }
 232
 233     argc -= i;
 234     argv += i;
 235
 236     config = notmuch_config_open (ctx, NULL, NULL);
 237     if (config == NULL)
 238         return 1;
 239
 240     notmuch = notmuch_database_open (notmuch_config_get_database_path (config),
 241                                      NOTMUCH_DATABASE_MODE_READ_ONLY);
 242     if (notmuch == NULL)
 243         return 1;
 244
 245     query_str = query_string_from_args (ctx, argc, argv);
 246     if (query_str == NULL) {
 247         fprintf (stderr, "Out of memory.\n");
 248         return 1;
 249     }
 250     if (*query_str == '\0') {
 251         fprintf (stderr, "Error: notmuch search requires at least one search term.\n");
 252         return 1;
 253     }
 254
 255     query = notmuch_query_create (notmuch, query_str);
 256     if (query == NULL) {
 257         fprintf (stderr, "Out of memory\n");
 258         return 1;
 259     }
 260
 261     notmuch_query_set_sort (query, sort);
 262
 263     do_search_threads (ctx, format, query, sort);
 264
 265     notmuch_query_destroy (query);
 266     notmuch_database_close (notmuch);
 267
 268     return 0;
 269 }