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     void *ctx_quote = talloc_new (ctx);
 108
 109     printf ("\"thread\": %s,\n"
 110             "\"timestamp\": %ld,\n"
 111             "\"matched\": %d,\n"
 112             "\"total\": %d,\n"
 113             "\"authors\": %s,\n"
 114             "\"subject\": %s,\n",
 115             json_quote_str (ctx_quote, thread_id),
 116             date,
 117             matched,
 118             total,
 119             json_quote_str (ctx_quote, authors),
 120             json_quote_str (ctx_quote, subject));
 121
 122     talloc_free (ctx_quote);
 123 }
 124
 125 static void
 126 do_search_threads (const void *ctx,
 127                    const search_format_t *format,
 128                    notmuch_query_t *query,
 129                    notmuch_sort_t sort)
 130 {
 131     notmuch_thread_t *thread;
 132     notmuch_threads_t *threads;
 133     notmuch_tags_t *tags;
 134     time_t date;
 135     int first_thread = 1;
 136
 137     fputs (format->results_start, stdout);
 138
 139     for (threads = notmuch_query_search_threads (query);
 140          notmuch_threads_valid (threads);
 141          notmuch_threads_move_to_next (threads))
 142     {
 143         int first_tag = 1;
 144
 145         if (! first_thread)
 146             fputs (format->thread_sep, stdout);
 147
 148         thread = notmuch_threads_get (threads);
 149
 150         if (sort == NOTMUCH_SORT_OLDEST_FIRST)
 151             date = notmuch_thread_get_oldest_date (thread);
 152         else
 153             date = notmuch_thread_get_newest_date (thread);
 154
 155         fputs (format->thread_start, stdout);
 156
 157         format->thread (ctx,
 158                         notmuch_thread_get_thread_id (thread),
 159                         date,
 160                         notmuch_thread_get_matched_messages (thread),
 161                         notmuch_thread_get_total_messages (thread),
 162                         notmuch_thread_get_authors (thread),
 163                         notmuch_thread_get_subject (thread));
 164
 165         fputs (format->tag_start, stdout);
 166
 167         for (tags = notmuch_thread_get_tags (thread);
 168              notmuch_tags_valid (tags);
 169              notmuch_tags_move_to_next (tags))
 170         {
 171             if (! first_tag)
 172                 fputs (format->tag_sep, stdout);
 173             printf (format->tag, notmuch_tags_get (tags));
 174             first_tag = 0;
 175         }
 176
 177         fputs (format->tag_end, stdout);
 178         fputs (format->thread_end, stdout);
 179
 180         first_thread = 0;
 181
 182         notmuch_thread_destroy (thread);
 183     }
 184
 185     fputs (format->results_end, stdout);
 186 }
 187
 188 int
 189 notmuch_search_command (void *ctx, int argc, char *argv[])
 190 {
 191     notmuch_config_t *config;
 192     notmuch_database_t *notmuch;
 193     notmuch_query_t *query;
 194     char *query_str;
 195     char *opt;
 196     notmuch_sort_t sort = NOTMUCH_SORT_NEWEST_FIRST;
 197     const search_format_t *format = &format_text;
 198     int i;
 199
 200     for (i = 0; i < argc && argv[i][0] == '-'; i++) {
 201         if (strcmp (argv[i], "--") == 0) {
 202             i++;
 203             break;
 204         }
 205         if (STRNCMP_LITERAL (argv[i], "--sort=") == 0) {
 206             opt = argv[i] + sizeof ("--sort=") - 1;
 207             if (strcmp (opt, "oldest-first") == 0) {
 208                 sort = NOTMUCH_SORT_OLDEST_FIRST;
 209             } else if (strcmp (opt, "newest-first") == 0) {
 210                 sort = NOTMUCH_SORT_NEWEST_FIRST;
 211             } else {
 212                 fprintf (stderr, "Invalid value for --sort: %s\n", opt);
 213                 return 1;
 214             }
 215         } else if (STRNCMP_LITERAL (argv[i], "--format=") == 0) {
 216             opt = argv[i] + sizeof ("--format=") - 1;
 217             if (strcmp (opt, "text") == 0) {
 218                 format = &format_text;
 219             } else if (strcmp (opt, "json") == 0) {
 220                 format = &format_json;
 221             } else {
 222                 fprintf (stderr, "Invalid value for --format: %s\n", opt);
 223                 return 1;
 224             }
 225         } else {
 226             fprintf (stderr, "Unrecognized option: %s\n", argv[i]);
 227             return 1;
 228         }
 229     }
 230
 231     argc -= i;
 232     argv += i;
 233
 234     config = notmuch_config_open (ctx, NULL, NULL);
 235     if (config == NULL)
 236         return 1;
 237
 238     notmuch = notmuch_database_open (notmuch_config_get_database_path (config),
 239                                      NOTMUCH_DATABASE_MODE_READ_ONLY);
 240     if (notmuch == NULL)
 241         return 1;
 242
 243     query_str = query_string_from_args (ctx, argc, argv);
 244     if (query_str == NULL) {
 245         fprintf (stderr, "Out of memory.\n");
 246         return 1;
 247     }
 248     if (*query_str == '\0') {
 249         fprintf (stderr, "Error: notmuch search requires at least one search term.\n");
 250         return 1;
 251     }
 252
 253     query = notmuch_query_create (notmuch, query_str);
 254     if (query == NULL) {
 255         fprintf (stderr, "Out of memory\n");
 256         return 1;
 257     }
 258
 259     notmuch_query_set_sort (query, sort);
 260
 261     do_search_threads (ctx, format, query, sort);
 262
 263     notmuch_query_destroy (query);
 264     notmuch_database_close (notmuch);
 265
 266     return 0;
 267 }