git.notmuchmail.org Git - notmuch/blob - message.cc

   1 /* message.cc - Results of message-based searches from a notmuch database
   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-private.h"
  22 #include "database-private.h"
  23
  24 #include <xapian.h>
  25
  26 struct _notmuch_message {
  27     notmuch_database_t *notmuch;
  28     Xapian::docid doc_id;
  29     char *message_id;
  30     Xapian::Document doc;
  31 };
  32
  33 struct _notmuch_tags {
  34     Xapian::TermIterator iterator;
  35     Xapian::TermIterator iterator_end;
  36 };
  37
  38 struct _notmuch_thread_ids {
  39     char *current;
  40     char *next;
  41 };
  42
  43 #define ARRAY_SIZE(arr) (sizeof (arr) / sizeof (arr[0]))
  44
  45 /* These prefix values are specifically chosen to be compatible
  46  * with sup, (http://sup.rubyforge.org), written by
  47  * William Morgan <wmorgan-sup@masanjin.net>, and released
  48  * under the GNU GPL v2.
  49  */
  50
  51 typedef struct {
  52     const char *name;
  53     const char *prefix;
  54 } prefix_t;
  55
  56 prefix_t NORMAL_PREFIX[] = {
  57     { "subject", "S" },
  58     { "body", "B" },
  59     { "from_name", "FN" },
  60     { "to_name", "TN" },
  61     { "name", "N" },
  62     { "attachment", "A" }
  63 };
  64
  65 prefix_t BOOLEAN_PREFIX[] = {
  66     { "type", "K" },
  67     { "from_email", "FE" },
  68     { "to_email", "TE" },
  69     { "email", "E" },
  70     { "date", "D" },
  71     { "label", "L" },
  72     { "tag", "L" },
  73     { "source_id", "I" },
  74     { "attachment_extension", "O" },
  75     { "msgid", "Q" },
  76     { "thread", "H" },
  77     { "ref", "R" }
  78 };
  79
  80 const char *
  81 _find_prefix (const char *name)
  82 {
  83     unsigned int i;
  84
  85     for (i = 0; i < ARRAY_SIZE (NORMAL_PREFIX); i++)
  86         if (strcmp (name, NORMAL_PREFIX[i].name) == 0)
  87             return NORMAL_PREFIX[i].prefix;
  88
  89     for (i = 0; i < ARRAY_SIZE (BOOLEAN_PREFIX); i++)
  90         if (strcmp (name, BOOLEAN_PREFIX[i].name) == 0)
  91             return BOOLEAN_PREFIX[i].prefix;
  92
  93     return "";
  94 }
  95
  96 /* We end up having to call the destructor explicitly because we had
  97  * to use "placement new" in order to initialize C++ objects within a
  98  * block that we allocated with talloc. So C++ is making talloc
  99  * slightly less simple to use, (we wouldn't need
 100  * talloc_set_destructor at all otherwise).
 101  */
 102 static int
 103 _notmuch_message_destructor (notmuch_message_t *message)
 104 {
 105     message->doc.~Document ();
 106
 107     return 0;
 108 }
 109
 110 notmuch_message_t *
 111 _notmuch_message_create (const void *talloc_owner,
 112                          notmuch_database_t *notmuch,
 113                          unsigned int doc_id)
 114 {
 115     notmuch_message_t *message;
 116
 117     message = talloc (talloc_owner, notmuch_message_t);
 118     if (unlikely (message == NULL))
 119         return NULL;
 120
 121     message->notmuch = notmuch;
 122     message->doc_id = doc_id;
 123     message->message_id = NULL; /* lazily created */
 124     new (&message->doc) Xapian::Document;
 125
 126     talloc_set_destructor (message, _notmuch_message_destructor);
 127
 128     message->doc = notmuch->xapian_db->get_document (doc_id);
 129
 130     return message;
 131 }
 132
 133 const char *
 134 notmuch_message_get_message_id (notmuch_message_t *message)
 135 {
 136     Xapian::TermIterator i;
 137
 138     if (message->message_id)
 139         return message->message_id;
 140
 141     i = message->doc.termlist_begin ();
 142     i.skip_to (_find_prefix ("msgid"));
 143
 144     /* XXX: This should really be an internal error, but we'll need to
 145      * fix the add_message side of things first. */
 146     if (i == message->doc.termlist_end ())
 147         return NULL;
 148
 149     message->message_id = talloc_strdup (message, (*i).c_str () + 1);
 150     return message->message_id;
 151 }
 152
 153 /* We end up having to call the destructors explicitly because we had
 154  * to use "placement new" in order to initialize C++ objects within a
 155  * block that we allocated with talloc. So C++ is making talloc
 156  * slightly less simple to use, (we wouldn't need
 157  * talloc_set_destructor at all otherwise).
 158  */
 159 static int
 160 _notmuch_tags_destructor (notmuch_tags_t *tags)
 161 {
 162     tags->iterator.~TermIterator ();
 163     tags->iterator_end.~TermIterator ();
 164
 165     return 0;
 166 }
 167
 168 notmuch_tags_t *
 169 notmuch_message_get_tags (notmuch_message_t *message)
 170 {
 171     notmuch_tags_t *tags;
 172
 173     tags = talloc (message, notmuch_tags_t);
 174     if (unlikely (tags == NULL))
 175         return NULL;
 176
 177     new (&tags->iterator) Xapian::TermIterator;
 178     new (&tags->iterator_end) Xapian::TermIterator;
 179
 180     talloc_set_destructor (tags, _notmuch_tags_destructor);
 181
 182     tags->iterator = message->doc.termlist_begin ();
 183     tags->iterator.skip_to (_find_prefix ("tag"));
 184     tags->iterator_end = message->doc.termlist_end ();
 185
 186     return tags;
 187 }
 188
 189 notmuch_thread_ids_t *
 190 notmuch_message_get_thread_ids (notmuch_message_t *message)
 191 {
 192     notmuch_thread_ids_t *thread_ids;
 193     const char *id_str;
 194
 195     thread_ids = talloc (message, notmuch_thread_ids_t);
 196     if (unlikely (thread_ids == NULL))
 197         return NULL;
 198
 199     id_str = message->doc.get_value (NOTMUCH_VALUE_THREAD).c_str ();
 200     thread_ids->next = talloc_strdup (message, id_str);
 201
 202     /* Initialize thread_ids->current and terminate first ID. */
 203     notmuch_thread_ids_advance (thread_ids);
 204
 205     return thread_ids;
 206 }
 207
 208 /* Synchronize changes made to message->doc into the database. */
 209 static void
 210 _notmuch_message_sync (notmuch_message_t *message)
 211 {
 212     Xapian::WritableDatabase *db = message->notmuch->xapian_db;
 213
 214     db->replace_document (message->doc_id, message->doc);
 215 }
 216
 217 notmuch_private_status_t
 218 _notmuch_message_add_term (notmuch_message_t *message,
 219                            const char *prefix_name,
 220                            const char *value)
 221 {
 222
 223     char *term;
 224
 225     if (value == NULL)
 226         return NOTMUCH_PRIVATE_STATUS_NULL_POINTER;
 227
 228     term = talloc_asprintf (message, "%s%s",
 229                             _find_prefix (prefix_name), value);
 230
 231     if (strlen (term) > NOTMUCH_TERM_MAX)
 232         return NOTMUCH_PRIVATE_STATUS_TERM_TOO_LONG;
 233
 234     message->doc.add_term (term);
 235     _notmuch_message_sync (message);
 236
 237     talloc_free (term);
 238
 239     return NOTMUCH_PRIVATE_STATUS_SUCCESS;
 240 }
 241
 242 notmuch_private_status_t
 243 _notmuch_message_remove_term (notmuch_message_t *message,
 244                               const char *prefix_name,
 245                               const char *value)
 246 {
 247     char *term;
 248
 249     if (value == NULL)
 250         return NOTMUCH_PRIVATE_STATUS_NULL_POINTER;
 251
 252     term = talloc_asprintf (message, "%s%s",
 253                             _find_prefix (prefix_name), value);
 254
 255     if (strlen (term) > NOTMUCH_TERM_MAX)
 256         return NOTMUCH_PRIVATE_STATUS_TERM_TOO_LONG;
 257
 258     message->doc.remove_term (term);
 259     _notmuch_message_sync (message);
 260
 261     talloc_free (term);
 262
 263     return NOTMUCH_PRIVATE_STATUS_SUCCESS;
 264 }
 265
 266 notmuch_status_t
 267 notmuch_message_add_tag (notmuch_message_t *message, const char *tag)
 268 {
 269     notmuch_private_status_t status;
 270
 271     if (tag == NULL)
 272         return NOTMUCH_STATUS_NULL_POINTER;
 273
 274     if (strlen (tag) > NOTMUCH_TAG_MAX)
 275         return NOTMUCH_STATUS_TAG_TOO_LONG;
 276
 277     status = _notmuch_message_add_term (message, "tag", tag);
 278     if (status) {
 279         fprintf (stderr, "Internal error: _notmuch_message_add_term return unexpected value: %d\n",
 280                  status);
 281         exit (1);
 282     }
 283
 284     return NOTMUCH_STATUS_SUCCESS;
 285 }
 286
 287 notmuch_status_t
 288 notmuch_message_remove_tag (notmuch_message_t *message, const char *tag)
 289 {
 290     notmuch_private_status_t status;
 291
 292     if (tag == NULL)
 293         return NOTMUCH_STATUS_NULL_POINTER;
 294
 295     if (strlen (tag) > NOTMUCH_TAG_MAX)
 296         return NOTMUCH_STATUS_TAG_TOO_LONG;
 297
 298     status = _notmuch_message_remove_term (message, "tag", tag);
 299     if (status) {
 300         fprintf (stderr, "Internal error: _notmuch_message_remove_term return unexpected value: %d\n",
 301                  status);
 302         exit (1);
 303     }
 304
 305     return NOTMUCH_STATUS_SUCCESS;
 306 }
 307
 308 void
 309 notmuch_message_destroy (notmuch_message_t *message)
 310 {
 311     talloc_free (message);
 312 }
 313
 314 notmuch_bool_t
 315 notmuch_tags_has_more (notmuch_tags_t *tags)
 316 {
 317     std::string s;
 318
 319     if (tags->iterator == tags->iterator_end)
 320         return FALSE;
 321
 322     s = *tags->iterator;
 323     if (s.size () && s[0] == 'L')
 324         return TRUE;
 325     else
 326         return FALSE;
 327 }
 328
 329 const char *
 330 notmuch_tags_get (notmuch_tags_t *tags)
 331 {
 332     return talloc_strdup (tags, (*tags->iterator).c_str () + 1);
 333 }
 334
 335 void
 336 notmuch_tags_advance (notmuch_tags_t *tags)
 337 {
 338     tags->iterator++;
 339 }
 340
 341 void
 342 notmuch_tags_destroy (notmuch_tags_t *tags)
 343 {
 344     talloc_free (tags);
 345 }
 346
 347 notmuch_bool_t
 348 notmuch_thread_ids_has_more (notmuch_thread_ids_t *thread_ids)
 349 {
 350     if (thread_ids->current == NULL || *thread_ids->current == '\0')
 351         return FALSE;
 352     else
 353         return TRUE;
 354 }
 355
 356 const char *
 357 notmuch_thread_ids_get (notmuch_thread_ids_t *thread_ids)
 358 {
 359     return thread_ids->current;
 360 }
 361
 362 void
 363 notmuch_thread_ids_advance (notmuch_thread_ids_t *thread_ids)
 364 {
 365     thread_ids->current = strsep (&thread_ids->next, ",");
 366 }
 367
 368 void
 369 notmuch_thread_ids_destroy (notmuch_thread_ids_t *thread_ids)
 370 {
 371     talloc_free (thread_ids);
 372 }