git.cworth.org Git - notmuch/blob - notmuch-count.c

   1 /* notmuch - Not much of an email program, (just index and search)
   2  *
   3  * Copyright © 2009 Carl Worth
   4  * Copyright © 2009 Keith Packard
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see http://www.gnu.org/licenses/ .
  18  *
  19  * Author: Keith Packard <keithp@keithp.com>
  20  */
  21
  22 #include "notmuch-client.h"
  23
  24 enum {
  25     OUTPUT_THREADS,
  26     OUTPUT_MESSAGES,
  27 };
  28
  29 /* The following is to allow future options to be added more easily */
  30 enum {
  31     EXCLUDE_TRUE,
  32     EXCLUDE_FALSE,
  33 };
  34
  35 int
  36 notmuch_count_command (notmuch_config_t *config, int argc, char *argv[])
  37 {
  38     notmuch_database_t *notmuch;
  39     notmuch_query_t *query;
  40     char *query_str;
  41     int opt_index;
  42     int output = OUTPUT_MESSAGES;
  43     int exclude = EXCLUDE_TRUE;
  44     unsigned int i;
  45
  46     notmuch_opt_desc_t options[] = {
  47         { NOTMUCH_OPT_KEYWORD, &output, "output", 'o',
  48           (notmuch_keyword_t []){ { "threads", OUTPUT_THREADS },
  49                                   { "messages", OUTPUT_MESSAGES },
  50                                   { 0, 0 } } },
  51         { NOTMUCH_OPT_KEYWORD, &exclude, "exclude", 'x',
  52           (notmuch_keyword_t []){ { "true", EXCLUDE_TRUE },
  53                                   { "false", EXCLUDE_FALSE },
  54                                   { 0, 0 } } },
  55         { 0, 0, 0, 0, 0 }
  56     };
  57
  58     opt_index = parse_arguments (argc, argv, options, 1);
  59
  60     if (opt_index < 0) {
  61         return 1;
  62     }
  63
  64     if (notmuch_database_open (notmuch_config_get_database_path (config),
  65                                NOTMUCH_DATABASE_MODE_READ_ONLY, &notmuch))
  66         return 1;
  67
  68     query_str = query_string_from_args (config, argc-opt_index, argv+opt_index);
  69     if (query_str == NULL) {
  70         fprintf (stderr, "Out of memory.\n");
  71         return 1;
  72     }
  73
  74     if (*query_str == '\0') {
  75         query_str = talloc_strdup (config, "");
  76     }
  77
  78     query = notmuch_query_create (notmuch, query_str);
  79     if (query == NULL) {
  80         fprintf (stderr, "Out of memory\n");
  81         return 1;
  82     }
  83
  84     if (exclude == EXCLUDE_TRUE) {
  85         const char **search_exclude_tags;
  86         size_t search_exclude_tags_length;
  87
  88         search_exclude_tags = notmuch_config_get_search_exclude_tags
  89             (config, &search_exclude_tags_length);
  90         for (i = 0; i < search_exclude_tags_length; i++)
  91             notmuch_query_add_tag_exclude (query, search_exclude_tags[i]);
  92     }
  93
  94     switch (output) {
  95     case OUTPUT_MESSAGES:
  96         printf ("%u\n", notmuch_query_count_messages (query));
  97         break;
  98     case OUTPUT_THREADS:
  99         printf ("%u\n", notmuch_query_count_threads (query));
 100         break;
 101     }
 102
 103     notmuch_query_destroy (query);
 104     notmuch_database_destroy (notmuch);
 105
 106     return 0;
 107 }