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