]> git.notmuchmail.org Git - notmuch/blob - lib/message.cc
b04efbea538ec42e10eccb5293475160a0bb69f9
[notmuch] / lib / 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 https://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 #include "message-private.h"
24
25 #include <stdint.h>
26
27 #include <gmime/gmime.h>
28
29 struct _notmuch_message {
30     notmuch_database_t *notmuch;
31     Xapian::docid doc_id;
32     int frozen;
33     char *message_id;
34     char *thread_id;
35     size_t thread_depth;
36     char *in_reply_to;
37     notmuch_string_list_t *tag_list;
38     notmuch_string_list_t *filename_term_list;
39     notmuch_string_list_t *filename_list;
40     char *maildir_flags;
41     char *author;
42     notmuch_message_file_t *message_file;
43     notmuch_string_list_t *property_term_list;
44     notmuch_string_map_t *property_map;
45     notmuch_string_list_t *reference_list;
46     notmuch_message_list_t *replies;
47     unsigned long flags;
48     /* For flags that are initialized on-demand, lazy_flags indicates
49      * if each flag has been initialized. */
50     unsigned long lazy_flags;
51
52     /* Message document modified since last sync */
53     bool modified;
54
55     /* last view of database the struct is synced with */
56     unsigned long last_view;
57
58     Xapian::Document doc;
59     Xapian::termcount termpos;
60 };
61
62 #define ARRAY_SIZE(arr) (sizeof (arr) / sizeof (arr[0]))
63
64 struct maildir_flag_tag {
65     char flag;
66     const char *tag;
67     bool inverse;
68 };
69
70 /* ASCII ordered table of Maildir flags and associated tags */
71 static const struct maildir_flag_tag flag2tag[] = {
72     { 'D', "draft",   false },
73     { 'F', "flagged", false },
74     { 'P', "passed",  false },
75     { 'R', "replied", false },
76     { 'S', "unread",  true }
77 };
78
79 /* We end up having to call the destructor explicitly because we had
80  * to use "placement new" in order to initialize C++ objects within a
81  * block that we allocated with talloc. So C++ is making talloc
82  * slightly less simple to use, (we wouldn't need
83  * talloc_set_destructor at all otherwise).
84  */
85 static int
86 _notmuch_message_destructor (notmuch_message_t *message)
87 {
88     message->doc.~Document ();
89
90     return 0;
91 }
92
93 #define LOG_XAPIAN_EXCEPTION(message, error) _log_xapian_exception (__location__, message, error)
94
95 static void
96 _log_xapian_exception (const char *where, notmuch_message_t *message,  const Xapian::Error error)
97 {
98     notmuch_database_t *notmuch = notmuch_message_get_database (message);
99
100     _notmuch_database_log (notmuch,
101                            "A Xapian exception occurred at %s: %s\n",
102                            where,
103                            error.get_msg ().c_str ());
104     notmuch->exception_reported = true;
105 }
106
107 static notmuch_message_t *
108 _notmuch_message_create_for_document (const void *talloc_owner,
109                                       notmuch_database_t *notmuch,
110                                       unsigned int doc_id,
111                                       Xapian::Document doc,
112                                       notmuch_private_status_t *status)
113 {
114     notmuch_message_t *message;
115
116     if (status)
117         *status = NOTMUCH_PRIVATE_STATUS_SUCCESS;
118
119     message = talloc (talloc_owner, notmuch_message_t);
120     if (unlikely (message == NULL)) {
121         if (status)
122             *status = NOTMUCH_PRIVATE_STATUS_OUT_OF_MEMORY;
123         return NULL;
124     }
125
126     message->notmuch = notmuch;
127     message->doc_id = doc_id;
128
129     message->frozen = 0;
130     message->flags = 0;
131     message->lazy_flags = 0;
132
133     /* the message is initially not synchronized with Xapian */
134     message->last_view = 0;
135
136     /* Calculated after the thread structure is computed */
137     message->thread_depth = 0;
138
139     /* Each of these will be lazily created as needed. */
140     message->message_id = NULL;
141     message->thread_id = NULL;
142     message->in_reply_to = NULL;
143     message->tag_list = NULL;
144     message->filename_term_list = NULL;
145     message->filename_list = NULL;
146     message->maildir_flags = NULL;
147     message->message_file = NULL;
148     message->author = NULL;
149     message->property_term_list = NULL;
150     message->property_map = NULL;
151     message->reference_list = NULL;
152
153     message->replies = _notmuch_message_list_create (message);
154     if (unlikely (message->replies == NULL)) {
155         if (status)
156             *status = NOTMUCH_PRIVATE_STATUS_OUT_OF_MEMORY;
157         return NULL;
158     }
159
160     /* This is C++'s creepy "placement new", which is really just an
161      * ugly way to call a constructor for a pre-allocated object. So
162      * it's really not an error to not be checking for OUT_OF_MEMORY
163      * here, since this "new" isn't actually allocating memory. This
164      * is language-design comedy of the wrong kind. */
165
166     new (&message->doc) Xapian::Document;
167
168     talloc_set_destructor (message, _notmuch_message_destructor);
169
170     message->doc = doc;
171     message->termpos = 0;
172     message->modified = false;
173
174     return message;
175 }
176
177 /* Create a new notmuch_message_t object for an existing document in
178  * the database.
179  *
180  * Here, 'talloc owner' is an optional talloc context to which the new
181  * message will belong. This allows for the caller to not bother
182  * calling notmuch_message_destroy on the message, and know that all
183  * memory will be reclaimed when 'talloc_owner' is freed. The caller
184  * still can call notmuch_message_destroy when finished with the
185  * message if desired.
186  *
187  * The 'talloc_owner' argument can also be NULL, in which case the
188  * caller *is* responsible for calling notmuch_message_destroy.
189  *
190  * If no document exists in the database with document ID of 'doc_id'
191  * then this function returns NULL and optionally sets *status to
192  * NOTMUCH_PRIVATE_STATUS_NO_DOCUMENT_FOUND.
193  *
194  * This function can also fail to due lack of available memory,
195  * returning NULL and optionally setting *status to
196  * NOTMUCH_PRIVATE_STATUS_OUT_OF_MEMORY.
197  *
198  * The caller can pass NULL for status if uninterested in
199  * distinguishing these two cases.
200  */
201 notmuch_message_t *
202 _notmuch_message_create (const void *talloc_owner,
203                          notmuch_database_t *notmuch,
204                          unsigned int doc_id,
205                          notmuch_private_status_t *status)
206 {
207     Xapian::Document doc;
208
209     try {
210         doc = notmuch->xapian_db->get_document (doc_id);
211     } catch (const Xapian::DocNotFoundError &error) {
212         if (status)
213             *status = NOTMUCH_PRIVATE_STATUS_NO_DOCUMENT_FOUND;
214         return NULL;
215     }
216
217     return _notmuch_message_create_for_document (talloc_owner, notmuch,
218                                                  doc_id, doc, status);
219 }
220
221 /* Create a new notmuch_message_t object for a specific message ID,
222  * (which may or may not already exist in the database).
223  *
224  * The 'notmuch' database will be the talloc owner of the returned
225  * message.
226  *
227  * This function returns a valid notmuch_message_t whether or not
228  * there is already a document in the database with the given message
229  * ID. These two cases can be distinguished by the value of *status:
230  *
231  *
232  *   NOTMUCH_PRIVATE_STATUS_SUCCESS:
233  *
234  *     There is already a document with message ID 'message_id' in the
235  *     database. The returned message can be used to query/modify the
236  *     document. The message may be a ghost message.
237  *
238  *   NOTMUCH_PRIVATE_STATUS_NO_DOCUMENT_FOUND:
239  *
240  *     No document with 'message_id' exists in the database. The
241  *     returned message contains a newly created document (not yet
242  *     added to the database) and a document ID that is known not to
243  *     exist in the database.  This message is "blank"; that is, it
244  *     contains only a message ID and no other metadata. The caller
245  *     can modify the message, and a call to _notmuch_message_sync
246  *     will add the document to the database.
247  *
248  * If an error occurs, this function will return NULL and *status
249  * will be set as appropriate. (The status pointer argument must
250  * not be NULL.)
251  */
252 notmuch_message_t *
253 _notmuch_message_create_for_message_id (notmuch_database_t *notmuch,
254                                         const char *message_id,
255                                         notmuch_private_status_t *status_ret)
256 {
257     notmuch_message_t *message;
258     Xapian::Document doc;
259     unsigned int doc_id;
260     char *term;
261
262     *status_ret = (notmuch_private_status_t) notmuch_database_find_message (notmuch,
263                                                                             message_id,
264                                                                             &message);
265     if (message)
266         return talloc_steal (notmuch, message);
267     else if (*status_ret)
268         return NULL;
269
270     /* If the message ID is too long, substitute its sha1 instead. */
271     if (strlen (message_id) > NOTMUCH_MESSAGE_ID_MAX)
272         message_id = _notmuch_message_id_compressed (message, message_id);
273
274     term = talloc_asprintf (NULL, "%s%s",
275                             _find_prefix ("id"), message_id);
276     if (term == NULL) {
277         *status_ret = NOTMUCH_PRIVATE_STATUS_OUT_OF_MEMORY;
278         return NULL;
279     }
280
281     if (_notmuch_database_mode (notmuch) == NOTMUCH_DATABASE_MODE_READ_ONLY)
282         INTERNAL_ERROR ("Failure to ensure database is writable.");
283
284     try {
285         doc.add_term (term, 0);
286         talloc_free (term);
287
288         doc.add_value (NOTMUCH_VALUE_MESSAGE_ID, message_id);
289
290         doc_id = _notmuch_database_generate_doc_id (notmuch);
291     } catch (const Xapian::Error &error) {
292         _notmuch_database_log (notmuch,
293                                "A Xapian exception occurred creating message: %s\n",
294                                error.get_msg ().c_str ());
295         notmuch->exception_reported = true;
296         *status_ret = NOTMUCH_PRIVATE_STATUS_XAPIAN_EXCEPTION;
297         return NULL;
298     }
299
300     message = _notmuch_message_create_for_document (notmuch, notmuch,
301                                                     doc_id, doc, status_ret);
302
303     /* We want to inform the caller that we had to create a new
304      * document. */
305     if (*status_ret == NOTMUCH_PRIVATE_STATUS_SUCCESS)
306         *status_ret = NOTMUCH_PRIVATE_STATUS_NO_DOCUMENT_FOUND;
307
308     return message;
309 }
310
311 static char *
312 _notmuch_message_get_term (notmuch_message_t *message,
313                            Xapian::TermIterator &i, Xapian::TermIterator &end,
314                            const char *prefix)
315 {
316     int prefix_len = strlen (prefix);
317     char *value;
318
319     i.skip_to (prefix);
320
321     if (i == end)
322         return NULL;
323
324     const std::string &term = *i;
325
326     if (strncmp (term.c_str (), prefix, prefix_len))
327         return NULL;
328
329     value = talloc_strdup (message, term.c_str () + prefix_len);
330
331 #if DEBUG_DATABASE_SANITY
332     i++;
333
334     if (i != end && strncmp ((*i).c_str (), prefix, prefix_len) == 0) {
335         INTERNAL_ERROR ("Mail (doc_id: %d) has duplicate %s terms: %s and %s\n",
336                         message->doc_id, prefix, value,
337                         (*i).c_str () + prefix_len);
338     }
339 #endif
340
341     return value;
342 }
343
344 static void
345 _notmuch_message_ensure_metadata (notmuch_message_t *message, void *field)
346 {
347     Xapian::TermIterator i, end;
348
349     if (field && (message->last_view >= message->notmuch->view))
350         return;
351
352     const char *thread_prefix = _find_prefix ("thread"),
353                *tag_prefix = _find_prefix ("tag"),
354                *id_prefix = _find_prefix ("id"),
355                *type_prefix = _find_prefix ("type"),
356                *filename_prefix = _find_prefix ("file-direntry"),
357                *property_prefix = _find_prefix ("property"),
358                *reference_prefix = _find_prefix ("reference"),
359                *replyto_prefix = _find_prefix ("replyto");
360
361     /* We do this all in a single pass because Xapian decompresses the
362      * term list every time you iterate over it.  Thus, while this is
363      * slightly more costly than looking up individual fields if only
364      * one field of the message object is actually used, it's a huge
365      * win as more fields are used. */
366     for (int count = 0; count < 3; count++) {
367         try {
368             i = message->doc.termlist_begin ();
369             end = message->doc.termlist_end ();
370
371             /* Get thread */
372             if (! message->thread_id)
373                 message->thread_id =
374                     _notmuch_message_get_term (message, i, end, thread_prefix);
375
376             /* Get tags */
377             assert (strcmp (thread_prefix, tag_prefix) < 0);
378             if (! message->tag_list) {
379                 message->tag_list =
380                     _notmuch_database_get_terms_with_prefix (message, i, end,
381                                                              tag_prefix);
382                 _notmuch_string_list_sort (message->tag_list);
383             }
384
385             /* Get id */
386             assert (strcmp (tag_prefix, id_prefix) < 0);
387             if (! message->message_id)
388                 message->message_id =
389                     _notmuch_message_get_term (message, i, end, id_prefix);
390
391             /* Get document type */
392             assert (strcmp (id_prefix, type_prefix) < 0);
393             if (! NOTMUCH_TEST_BIT (message->lazy_flags, NOTMUCH_MESSAGE_FLAG_GHOST)) {
394                 i.skip_to (type_prefix);
395                 /* "T" is the prefix "type" fields.  See
396                  * BOOLEAN_PREFIX_INTERNAL. */
397                 if (*i == "Tmail")
398                     NOTMUCH_CLEAR_BIT (&message->flags, NOTMUCH_MESSAGE_FLAG_GHOST);
399                 else if (*i == "Tghost")
400                     NOTMUCH_SET_BIT (&message->flags, NOTMUCH_MESSAGE_FLAG_GHOST);
401                 else
402                     INTERNAL_ERROR ("Message without type term");
403                 NOTMUCH_SET_BIT (&message->lazy_flags, NOTMUCH_MESSAGE_FLAG_GHOST);
404             }
405
406             /* Get filename list.  Here we get only the terms.  We lazily
407              * expand them to full file names when needed in
408              * _notmuch_message_ensure_filename_list. */
409             assert (strcmp (type_prefix, filename_prefix) < 0);
410             if (! message->filename_term_list && ! message->filename_list)
411                 message->filename_term_list =
412                     _notmuch_database_get_terms_with_prefix (message, i, end,
413                                                              filename_prefix);
414
415
416             /* Get property terms. Mimic the setup with filenames above */
417             assert (strcmp (filename_prefix, property_prefix) < 0);
418             if (! message->property_map && ! message->property_term_list)
419                 message->property_term_list =
420                     _notmuch_database_get_terms_with_prefix (message, i, end,
421                                                              property_prefix);
422
423             /* get references */
424             assert (strcmp (property_prefix, reference_prefix) < 0);
425             if (! message->reference_list) {
426                 message->reference_list =
427                     _notmuch_database_get_terms_with_prefix (message, i, end,
428                                                              reference_prefix);
429             }
430
431             /* Get reply to */
432             assert (strcmp (property_prefix, replyto_prefix) < 0);
433             if (! message->in_reply_to)
434                 message->in_reply_to =
435                     _notmuch_message_get_term (message, i, end, replyto_prefix);
436
437
438             /* It's perfectly valid for a message to have no In-Reply-To
439              * header. For these cases, we return an empty string. */
440             if (! message->in_reply_to)
441                 message->in_reply_to = talloc_strdup (message, "");
442
443             /* all the way without an exception */
444             break;
445         } catch (const Xapian::DatabaseModifiedError &error) {
446             notmuch_status_t status = notmuch_database_reopen (message->notmuch,
447                                                                NOTMUCH_DATABASE_MODE_READ_ONLY);
448             if (status != NOTMUCH_STATUS_SUCCESS)
449                 INTERNAL_ERROR ("unhandled error from notmuch_database_reopen: %s\n",
450                                 notmuch_status_to_string (status));
451         }
452     }
453     message->last_view = message->notmuch->view;
454 }
455
456 void
457 _notmuch_message_invalidate_metadata (notmuch_message_t *message,
458                                       const char *prefix_name)
459 {
460     if (strcmp ("thread", prefix_name) == 0) {
461         talloc_free (message->thread_id);
462         message->thread_id = NULL;
463     }
464
465     if (strcmp ("tag", prefix_name) == 0) {
466         talloc_unlink (message, message->tag_list);
467         message->tag_list = NULL;
468     }
469
470     if (strcmp ("type", prefix_name) == 0) {
471         NOTMUCH_CLEAR_BIT (&message->flags, NOTMUCH_MESSAGE_FLAG_GHOST);
472         NOTMUCH_CLEAR_BIT (&message->lazy_flags, NOTMUCH_MESSAGE_FLAG_GHOST);
473     }
474
475     if (strcmp ("file-direntry", prefix_name) == 0) {
476         talloc_free (message->filename_term_list);
477         talloc_free (message->filename_list);
478         message->filename_term_list = message->filename_list = NULL;
479     }
480
481     if (strcmp ("property", prefix_name) == 0) {
482
483         if (message->property_term_list)
484             talloc_free (message->property_term_list);
485         message->property_term_list = NULL;
486
487         if (message->property_map)
488             talloc_unlink (message, message->property_map);
489
490         message->property_map = NULL;
491     }
492
493     if (strcmp ("replyto", prefix_name) == 0) {
494         talloc_free (message->in_reply_to);
495         message->in_reply_to = NULL;
496     }
497 }
498
499 unsigned int
500 _notmuch_message_get_doc_id (notmuch_message_t *message)
501 {
502     return message->doc_id;
503 }
504
505 const char *
506 notmuch_message_get_message_id (notmuch_message_t *message)
507 {
508     try {
509         _notmuch_message_ensure_metadata (message, message->message_id);
510     } catch (const Xapian::Error &error) {
511         LOG_XAPIAN_EXCEPTION (message, error);
512         return NULL;
513     }
514
515     if (! message->message_id)
516         INTERNAL_ERROR ("Message with document ID of %u has no message ID.\n",
517                         message->doc_id);
518     return message->message_id;
519 }
520
521 static void
522 _notmuch_message_ensure_message_file (notmuch_message_t *message)
523 {
524     const char *filename;
525
526     if (message->message_file)
527         return;
528
529     filename = notmuch_message_get_filename (message);
530     if (unlikely (filename == NULL))
531         return;
532
533     message->message_file = _notmuch_message_file_open_ctx (
534         notmuch_message_get_database (message), message, filename);
535 }
536
537 const char *
538 notmuch_message_get_header (notmuch_message_t *message, const char *header)
539 {
540     Xapian::valueno slot = Xapian::BAD_VALUENO;
541
542     /* Fetch header from the appropriate xapian value field if
543      * available */
544     if (strcasecmp (header, "from") == 0)
545         slot = NOTMUCH_VALUE_FROM;
546     else if (strcasecmp (header, "subject") == 0)
547         slot = NOTMUCH_VALUE_SUBJECT;
548     else if (strcasecmp (header, "message-id") == 0)
549         slot = NOTMUCH_VALUE_MESSAGE_ID;
550
551     if (slot != Xapian::BAD_VALUENO) {
552         try {
553             std::string value = message->doc.get_value (slot);
554
555             /* If we have NOTMUCH_FEATURE_FROM_SUBJECT_ID_VALUES, then
556              * empty values indicate empty headers.  If we don't, then
557              * it could just mean we didn't record the header. */
558             if ((message->notmuch->features &
559                  NOTMUCH_FEATURE_FROM_SUBJECT_ID_VALUES) ||
560                 ! value.empty ())
561                 return talloc_strdup (message, value.c_str ());
562
563         } catch (Xapian::Error &error) {
564             LOG_XAPIAN_EXCEPTION (message, error);
565             return NULL;
566         }
567     }
568
569     /* Otherwise fall back to parsing the file */
570     _notmuch_message_ensure_message_file (message);
571     if (message->message_file == NULL)
572         return NULL;
573
574     return _notmuch_message_file_get_header (message->message_file, header);
575 }
576
577 /* Return the message ID from the In-Reply-To header of 'message'.
578  *
579  * Returns an empty string ("") if 'message' has no In-Reply-To
580  * header.
581  *
582  * Returns NULL if any error occurs.
583  */
584 const char *
585 _notmuch_message_get_in_reply_to (notmuch_message_t *message)
586 {
587     _notmuch_message_ensure_metadata (message, message->in_reply_to);
588     return message->in_reply_to;
589 }
590
591 const char *
592 notmuch_message_get_thread_id (notmuch_message_t *message)
593 {
594     try {
595         _notmuch_message_ensure_metadata (message, message->thread_id);
596     } catch (Xapian::Error &error) {
597         LOG_XAPIAN_EXCEPTION (message, error);
598         return NULL;
599     }
600     if (! message->thread_id)
601         INTERNAL_ERROR ("Message with document ID of %u has no thread ID.\n",
602                         message->doc_id);
603     return message->thread_id;
604 }
605
606 void
607 _notmuch_message_add_reply (notmuch_message_t *message,
608                             notmuch_message_t *reply)
609 {
610     _notmuch_message_list_add_message (message->replies, reply);
611 }
612
613 size_t
614 _notmuch_message_get_thread_depth (notmuch_message_t *message)
615 {
616     return message->thread_depth;
617 }
618
619 void
620 _notmuch_message_label_depths (notmuch_message_t *message,
621                                size_t depth)
622 {
623     message->thread_depth = depth;
624
625     for (notmuch_messages_t *messages = _notmuch_messages_create (message->replies);
626          notmuch_messages_valid (messages);
627          notmuch_messages_move_to_next (messages)) {
628         notmuch_message_t *child = notmuch_messages_get (messages);
629         _notmuch_message_label_depths (child, depth + 1);
630     }
631 }
632
633 const notmuch_string_list_t *
634 _notmuch_message_get_references (notmuch_message_t *message)
635 {
636     _notmuch_message_ensure_metadata (message, message->reference_list);
637     return message->reference_list;
638 }
639
640 static int
641 _cmpmsg (const void *pa, const void *pb)
642 {
643     notmuch_message_t **a = (notmuch_message_t **) pa;
644     notmuch_message_t **b = (notmuch_message_t **) pb;
645     time_t time_a = notmuch_message_get_date (*a);
646     time_t time_b = notmuch_message_get_date (*b);
647
648     if (time_a == time_b)
649         return 0;
650     else if (time_a < time_b)
651         return -1;
652     else
653         return 1;
654 }
655
656 notmuch_message_list_t *
657 _notmuch_message_sort_subtrees (void *ctx, notmuch_message_list_t *list)
658 {
659
660     size_t count = 0;
661     size_t capacity = 16;
662
663     if (! list)
664         return list;
665
666     void *local = talloc_new (NULL);
667     notmuch_message_list_t *new_list = _notmuch_message_list_create (ctx);
668     notmuch_message_t **message_array = talloc_zero_array (local, notmuch_message_t *, capacity);
669
670     for (notmuch_messages_t *messages = _notmuch_messages_create (list);
671          notmuch_messages_valid (messages);
672          notmuch_messages_move_to_next (messages)) {
673         notmuch_message_t *root = notmuch_messages_get (messages);
674         if (count >= capacity) {
675             capacity *= 2;
676             message_array = talloc_realloc (local, message_array, notmuch_message_t *, capacity);
677         }
678         message_array[count++] = root;
679         root->replies = _notmuch_message_sort_subtrees (root, root->replies);
680     }
681
682     qsort (message_array, count, sizeof (notmuch_message_t *), _cmpmsg);
683     for (size_t i = 0; i < count; i++) {
684         _notmuch_message_list_add_message (new_list, message_array[i]);
685     }
686
687     talloc_free (local);
688     talloc_free (list);
689     return new_list;
690 }
691
692 notmuch_messages_t *
693 notmuch_message_get_replies (notmuch_message_t *message)
694 {
695     return _notmuch_messages_create (message->replies);
696 }
697
698 void
699 _notmuch_message_remove_terms (notmuch_message_t *message, const char *prefix)
700 {
701     Xapian::TermIterator i;
702     size_t prefix_len = 0;
703
704     prefix_len = strlen (prefix);
705
706     while (1) {
707         i = message->doc.termlist_begin ();
708         i.skip_to (prefix);
709
710         /* Terminate loop when no terms remain with desired prefix. */
711         if (i == message->doc.termlist_end () ||
712             strncmp ((*i).c_str (), prefix, prefix_len))
713             break;
714
715         try {
716             message->doc.remove_term ((*i));
717             message->modified = true;
718         } catch (const Xapian::InvalidArgumentError) {
719             /* Ignore failure to remove non-existent term. */
720         }
721     }
722 }
723
724
725 /* Remove all terms generated by indexing, i.e. not tags or
726  * properties, along with any automatic tags*/
727 /* According to Xapian API docs, none of these calls throw
728  * exceptions */
729 static notmuch_private_status_t
730 _notmuch_message_remove_indexed_terms (notmuch_message_t *message)
731 {
732     Xapian::TermIterator i;
733
734     const std::string
735         id_prefix = _find_prefix ("id"),
736         property_prefix = _find_prefix ("property"),
737         tag_prefix = _find_prefix ("tag"),
738         type_prefix = _find_prefix ("type");
739
740     /* Make sure we have the data to restore to Xapian*/
741     _notmuch_message_ensure_metadata (message, NULL);
742
743     /* Empirically, it turns out to be faster to remove all the terms,
744      * and add back the ones we want. */
745     message->doc.clear_terms ();
746     message->modified = true;
747
748     /* still a mail message */
749     message->doc.add_term (type_prefix + "mail");
750
751     /* Put back message-id */
752     message->doc.add_term (id_prefix + message->message_id);
753
754     /* Put back non-automatic tags */
755     for (notmuch_tags_t *tags = notmuch_message_get_tags (message);
756          notmuch_tags_valid (tags);
757          notmuch_tags_move_to_next (tags)) {
758
759         const char *tag = notmuch_tags_get (tags);
760
761         if (strcmp (tag, "encrypted") != 0 &&
762             strcmp (tag, "signed") != 0 &&
763             strcmp (tag, "attachment") != 0) {
764             std::string term = tag_prefix + tag;
765             message->doc.add_term (term);
766         }
767     }
768
769     /* Put back properties */
770     notmuch_message_properties_t *list;
771
772     for (list = notmuch_message_get_properties (message, "", false);
773          notmuch_message_properties_valid (list); notmuch_message_properties_move_to_next (list)) {
774         std::string term = property_prefix +
775                            notmuch_message_properties_key (list) + "=" +
776                            notmuch_message_properties_value (list);
777
778         message->doc.add_term (term);
779     }
780
781     notmuch_message_properties_destroy (list);
782
783     return NOTMUCH_PRIVATE_STATUS_SUCCESS;
784 }
785
786
787 /* Return true if p points at "new" or "cur". */
788 static bool
789 is_maildir (const char *p)
790 {
791     return strcmp (p, "cur") == 0 || strcmp (p, "new") == 0;
792 }
793
794 /* Add "folder:" term for directory. */
795 static notmuch_status_t
796 _notmuch_message_add_folder_terms (notmuch_message_t *message,
797                                    const char *directory)
798 {
799     char *folder, *last;
800
801     folder = talloc_strdup (NULL, directory);
802     if (! folder)
803         return NOTMUCH_STATUS_OUT_OF_MEMORY;
804
805     /*
806      * If the message file is in a leaf directory named "new" or
807      * "cur", presume maildir and index the parent directory. Thus a
808      * "folder:" prefix search matches messages in the specified
809      * maildir folder, i.e. in the specified directory and its "new"
810      * and "cur" subdirectories.
811      *
812      * Note that this means the "folder:" prefix can't be used for
813      * distinguishing between message files in "new" or "cur". The
814      * "path:" prefix needs to be used for that.
815      *
816      * Note the deliberate difference to _filename_is_in_maildir(). We
817      * don't want to index different things depending on the existence
818      * or non-existence of all maildir sibling directories "new",
819      * "cur", and "tmp". Doing so would be surprising, and difficult
820      * for the user to fix in case all subdirectories were not in
821      * place during indexing.
822      */
823     last = strrchr (folder, '/');
824     if (last) {
825         if (is_maildir (last + 1))
826             *last = '\0';
827     } else if (is_maildir (folder)) {
828         *folder = '\0';
829     }
830
831     if (notmuch_status_t status = COERCE_STATUS (_notmuch_message_add_term (message, "folder",
832                                                                             folder),
833                                                  "adding folder term"))
834         return status;
835
836     talloc_free (folder);
837
838     message->modified = true;
839     return NOTMUCH_STATUS_SUCCESS;
840 }
841
842 #define RECURSIVE_SUFFIX "/**"
843
844 /* Add "path:" terms for directory. */
845 static notmuch_status_t
846 _notmuch_message_add_path_terms (notmuch_message_t *message,
847                                  const char *directory)
848 {
849     notmuch_status_t status;
850
851     /* Add exact "path:" term. */
852     status = COERCE_STATUS (_notmuch_message_add_term (message, "path", directory),
853                             "adding path term");
854     if (status)
855         return status;
856
857     if (strlen (directory)) {
858         char *path, *p;
859
860         path = talloc_asprintf (NULL, "%s%s", directory, RECURSIVE_SUFFIX);
861         if (! path)
862             return NOTMUCH_STATUS_OUT_OF_MEMORY;
863
864         /* Add recursive "path:" terms for directory and all parents. */
865         for (p = path + strlen (path) - 1; p > path; p--) {
866             if (*p == '/') {
867                 strcpy (p, RECURSIVE_SUFFIX);
868                 status = COERCE_STATUS (_notmuch_message_add_term (message, "path", path),
869                                         "adding path term");
870                 if (status)
871                     return status;
872             }
873         }
874
875         talloc_free (path);
876     }
877
878     /* Recursive all-matching path:** for consistency. */
879     status = COERCE_STATUS (_notmuch_message_add_term (message, "path", "**"),
880                             "adding path term");
881     if (status)
882         return status;
883
884     return NOTMUCH_STATUS_SUCCESS;
885 }
886
887 /* Add directory based terms for all filenames of the message. */
888 static notmuch_status_t
889 _notmuch_message_add_directory_terms (void *ctx, notmuch_message_t *message)
890 {
891     const char *direntry_prefix = _find_prefix ("file-direntry");
892     int direntry_prefix_len = strlen (direntry_prefix);
893     Xapian::TermIterator i = message->doc.termlist_begin ();
894     notmuch_status_t status = NOTMUCH_STATUS_SUCCESS;
895
896     for (i.skip_to (direntry_prefix); i != message->doc.termlist_end (); i++) {
897         unsigned int directory_id;
898         const char *direntry, *directory;
899         char *colon;
900         const std::string &term = *i;
901
902         /* Terminate loop at first term without desired prefix. */
903         if (strncmp (term.c_str (), direntry_prefix, direntry_prefix_len))
904             break;
905
906         /* Indicate that there are filenames remaining. */
907         status = NOTMUCH_STATUS_DUPLICATE_MESSAGE_ID;
908
909         direntry = term.c_str ();
910         direntry += direntry_prefix_len;
911
912         directory_id = strtol (direntry, &colon, 10);
913
914         if (colon == NULL || *colon != ':')
915             INTERNAL_ERROR ("malformed direntry");
916
917         directory = _notmuch_database_get_directory_path (ctx,
918                                                           message->notmuch,
919                                                           directory_id);
920
921         _notmuch_message_add_folder_terms (message, directory);
922         _notmuch_message_add_path_terms (message, directory);
923     }
924
925     return status;
926 }
927
928 /* Add an additional 'filename' for 'message'.
929  *
930  * This change will not be reflected in the database until the next
931  * call to _notmuch_message_sync. */
932 notmuch_status_t
933 _notmuch_message_add_filename (notmuch_message_t *message,
934                                const char *filename)
935 {
936     const char *relative, *directory;
937     notmuch_status_t status;
938     void *local = talloc_new (message);
939     char *direntry;
940
941     if (filename == NULL)
942         INTERNAL_ERROR ("Message filename cannot be NULL.");
943
944     if (! (message->notmuch->features & NOTMUCH_FEATURE_FILE_TERMS) ||
945         ! (message->notmuch->features & NOTMUCH_FEATURE_BOOL_FOLDER))
946         return NOTMUCH_STATUS_UPGRADE_REQUIRED;
947
948     relative = _notmuch_database_relative_path (message->notmuch, filename);
949
950     status = _notmuch_database_split_path (local, relative, &directory, NULL);
951     if (status)
952         return status;
953
954     status = _notmuch_database_filename_to_direntry (
955         local, message->notmuch, filename, NOTMUCH_FIND_CREATE, &direntry);
956     if (status)
957         return status;
958
959     /* New file-direntry allows navigating to this message with
960      * notmuch_directory_get_child_files() . */
961     status = COERCE_STATUS (_notmuch_message_add_term (message, "file-direntry", direntry),
962                             "adding file-direntry term");
963     if (status)
964         return status;
965
966     _notmuch_message_add_folder_terms (message, directory);
967     _notmuch_message_add_path_terms (message, directory);
968
969     talloc_free (local);
970
971     return NOTMUCH_STATUS_SUCCESS;
972 }
973
974 /* Remove a particular 'filename' from 'message'.
975  *
976  * This change will not be reflected in the database until the next
977  * call to _notmuch_message_sync.
978  *
979  * If this message still has other filenames, returns
980  * NOTMUCH_STATUS_DUPLICATE_MESSAGE_ID.
981  *
982  * Note: This function does not remove a document from the database,
983  * even if the specified filename is the only filename for this
984  * message. For that functionality, see
985  * notmuch_database_remove_message. */
986 notmuch_status_t
987 _notmuch_message_remove_filename (notmuch_message_t *message,
988                                   const char *filename)
989 {
990     void *local = talloc_new (message);
991     char *direntry;
992     notmuch_private_status_t private_status;
993     notmuch_status_t status;
994
995     if (! (message->notmuch->features & NOTMUCH_FEATURE_FILE_TERMS) ||
996         ! (message->notmuch->features & NOTMUCH_FEATURE_BOOL_FOLDER))
997         return NOTMUCH_STATUS_UPGRADE_REQUIRED;
998
999     status = _notmuch_database_filename_to_direntry (
1000         local, message->notmuch, filename, NOTMUCH_FIND_LOOKUP, &direntry);
1001     if (status || ! direntry)
1002         return status;
1003
1004     /* Unlink this file from its parent directory. */
1005     private_status = _notmuch_message_remove_term (message,
1006                                                    "file-direntry", direntry);
1007     status = COERCE_STATUS (private_status,
1008                             "Unexpected error from _notmuch_message_remove_term");
1009     if (status)
1010         return status;
1011
1012     /* Re-synchronize "folder:" and "path:" terms for this message. */
1013
1014     /* Remove all "folder:" terms. */
1015     _notmuch_message_remove_terms (message, _find_prefix ("folder"));
1016
1017     /* Remove all "path:" terms. */
1018     _notmuch_message_remove_terms (message, _find_prefix ("path"));
1019
1020     /* Add back terms for all remaining filenames of the message. */
1021     status = _notmuch_message_add_directory_terms (local, message);
1022
1023     talloc_free (local);
1024
1025     return status;
1026 }
1027
1028 /* Upgrade the "folder:" prefix from V1 to V2. */
1029 #define FOLDER_PREFIX_V1       "XFOLDER"
1030 #define ZFOLDER_PREFIX_V1      "Z" FOLDER_PREFIX_V1
1031 void
1032 _notmuch_message_upgrade_folder (notmuch_message_t *message)
1033 {
1034     /* Remove all old "folder:" terms. */
1035     _notmuch_message_remove_terms (message, FOLDER_PREFIX_V1);
1036
1037     /* Remove all old "folder:" stemmed terms. */
1038     _notmuch_message_remove_terms (message, ZFOLDER_PREFIX_V1);
1039
1040     /* Add new boolean "folder:" and "path:" terms. */
1041     _notmuch_message_add_directory_terms (message, message);
1042 }
1043
1044 char *
1045 _notmuch_message_talloc_copy_data (notmuch_message_t *message)
1046 {
1047     return talloc_strdup (message, message->doc.get_data ().c_str ());
1048 }
1049
1050 void
1051 _notmuch_message_clear_data (notmuch_message_t *message)
1052 {
1053     message->doc.set_data ("");
1054     message->modified = true;
1055 }
1056
1057 static void
1058 _notmuch_message_ensure_filename_list (notmuch_message_t *message)
1059 {
1060     notmuch_string_node_t *node;
1061
1062     if (message->filename_list)
1063         return;
1064
1065     _notmuch_message_ensure_metadata (message, message->filename_term_list);
1066
1067     message->filename_list = _notmuch_string_list_create (message);
1068     node = message->filename_term_list->head;
1069
1070     if (! node) {
1071         /* A message document created by an old version of notmuch
1072          * (prior to rename support) will have the filename in the
1073          * data of the document rather than as a file-direntry term.
1074          *
1075          * It would be nice to do the upgrade of the document directly
1076          * here, but the database is likely open in read-only mode. */
1077
1078         std::string datastr = message->doc.get_data ();
1079         const char *data = datastr.c_str ();
1080
1081         if (data == NULL)
1082             INTERNAL_ERROR ("message with no filename");
1083
1084         _notmuch_string_list_append (message->filename_list, data);
1085
1086         return;
1087     }
1088
1089     for (; node; node = node->next) {
1090         void *local = talloc_new (message);
1091         const char *db_path, *directory, *basename, *filename;
1092         char *colon, *direntry = NULL;
1093         unsigned int directory_id;
1094
1095         direntry = node->string;
1096
1097         directory_id = strtol (direntry, &colon, 10);
1098
1099         if (colon == NULL || *colon != ':')
1100             INTERNAL_ERROR ("malformed direntry");
1101
1102         basename = colon + 1;
1103
1104         *colon = '\0';
1105
1106         db_path = notmuch_config_get (message->notmuch, NOTMUCH_CONFIG_MAIL_ROOT);
1107
1108         directory = _notmuch_database_get_directory_path (local,
1109                                                           message->notmuch,
1110                                                           directory_id);
1111
1112         if (strlen (directory))
1113             filename = talloc_asprintf (message, "%s/%s/%s",
1114                                         db_path, directory, basename);
1115         else
1116             filename = talloc_asprintf (message, "%s/%s",
1117                                         db_path, basename);
1118
1119         _notmuch_string_list_append (message->filename_list, filename);
1120
1121         talloc_free (local);
1122     }
1123
1124     talloc_free (message->filename_term_list);
1125     message->filename_term_list = NULL;
1126 }
1127
1128 const char *
1129 notmuch_message_get_filename (notmuch_message_t *message)
1130 {
1131     try {
1132         _notmuch_message_ensure_filename_list (message);
1133     } catch (Xapian::Error &error) {
1134         LOG_XAPIAN_EXCEPTION (message, error);
1135         return NULL;
1136     }
1137
1138     if (message->filename_list == NULL)
1139         return NULL;
1140
1141     if (message->filename_list->head == NULL ||
1142         message->filename_list->head->string == NULL) {
1143         INTERNAL_ERROR ("message with no filename");
1144     }
1145
1146     return message->filename_list->head->string;
1147 }
1148
1149 notmuch_filenames_t *
1150 notmuch_message_get_filenames (notmuch_message_t *message)
1151 {
1152     try {
1153         _notmuch_message_ensure_filename_list (message);
1154     } catch (Xapian::Error &error) {
1155         LOG_XAPIAN_EXCEPTION (message, error);
1156         return NULL;
1157     }
1158
1159     return _notmuch_filenames_create (message, message->filename_list);
1160 }
1161
1162 int
1163 notmuch_message_count_files (notmuch_message_t *message)
1164 {
1165     try {
1166         _notmuch_message_ensure_filename_list (message);
1167     } catch (Xapian::Error &error) {
1168         LOG_XAPIAN_EXCEPTION (message, error);
1169         return -1;
1170     }
1171
1172     return _notmuch_string_list_length (message->filename_list);
1173 }
1174
1175 notmuch_status_t
1176 notmuch_message_get_flag_st (notmuch_message_t *message,
1177                              notmuch_message_flag_t flag,
1178                              notmuch_bool_t *is_set)
1179 {
1180     if (! is_set)
1181         return NOTMUCH_STATUS_NULL_POINTER;
1182
1183     try {
1184         if (flag == NOTMUCH_MESSAGE_FLAG_GHOST &&
1185             ! NOTMUCH_TEST_BIT (message->lazy_flags, flag))
1186             _notmuch_message_ensure_metadata (message, NULL);
1187     } catch (Xapian::Error &error) {
1188         LOG_XAPIAN_EXCEPTION (message, error);
1189         return NOTMUCH_STATUS_XAPIAN_EXCEPTION;
1190     }
1191
1192     *is_set = NOTMUCH_TEST_BIT (message->flags, flag);
1193     return NOTMUCH_STATUS_SUCCESS;
1194 }
1195
1196 notmuch_bool_t
1197 notmuch_message_get_flag (notmuch_message_t *message,
1198                           notmuch_message_flag_t flag)
1199 {
1200     notmuch_bool_t is_set;
1201     notmuch_status_t status;
1202
1203     status = notmuch_message_get_flag_st (message, flag, &is_set);
1204
1205     if (status)
1206         return FALSE;
1207     else
1208         return is_set;
1209 }
1210
1211 void
1212 notmuch_message_set_flag (notmuch_message_t *message,
1213                           notmuch_message_flag_t flag, notmuch_bool_t enable)
1214 {
1215     if (enable)
1216         NOTMUCH_SET_BIT (&message->flags, flag);
1217     else
1218         NOTMUCH_CLEAR_BIT (&message->flags, flag);
1219     NOTMUCH_SET_BIT (&message->lazy_flags, flag);
1220 }
1221
1222 time_t
1223 notmuch_message_get_date (notmuch_message_t *message)
1224 {
1225     std::string value;
1226
1227     try {
1228         value = message->doc.get_value (NOTMUCH_VALUE_TIMESTAMP);
1229     } catch (Xapian::Error &error) {
1230         LOG_XAPIAN_EXCEPTION (message, error);
1231         return 0;
1232     }
1233
1234     if (value.empty ())
1235         /* sortable_unserialise is undefined on empty string */
1236         return 0;
1237     return Xapian::sortable_unserialise (value);
1238 }
1239
1240 notmuch_tags_t *
1241 notmuch_message_get_tags (notmuch_message_t *message)
1242 {
1243     notmuch_tags_t *tags;
1244
1245     try {
1246         _notmuch_message_ensure_metadata (message, message->tag_list);
1247     } catch (Xapian::Error &error) {
1248         LOG_XAPIAN_EXCEPTION (message, error);
1249         return NULL;
1250     }
1251
1252     tags = _notmuch_tags_create (message, message->tag_list);
1253     /* _notmuch_tags_create steals the reference to the tag_list, but
1254      * in this case it's still used by the message, so we add an
1255      * *additional* talloc reference to the list.  As a result, it's
1256      * possible to modify the message tags (which talloc_unlink's the
1257      * current list from the message) while still iterating because
1258      * the iterator will keep the current list alive. */
1259     if (! talloc_reference (message, message->tag_list))
1260         return NULL;
1261
1262     return tags;
1263 }
1264
1265 const char *
1266 _notmuch_message_get_author (notmuch_message_t *message)
1267 {
1268     return message->author;
1269 }
1270
1271 void
1272 _notmuch_message_set_author (notmuch_message_t *message,
1273                              const char *author)
1274 {
1275     if (message->author)
1276         talloc_free (message->author);
1277     message->author = talloc_strdup (message, author);
1278     return;
1279 }
1280
1281 void
1282 _notmuch_message_set_header_values (notmuch_message_t *message,
1283                                     const char *date,
1284                                     const char *from,
1285                                     const char *subject)
1286 {
1287     time_t time_value;
1288
1289     /* GMime really doesn't want to see a NULL date, so protect its
1290      * sensibilities. */
1291     if (date == NULL || *date == '\0') {
1292         time_value = 0;
1293     } else {
1294         time_value = g_mime_utils_header_decode_date_unix (date);
1295         /*
1296          * Workaround for https://bugzilla.gnome.org/show_bug.cgi?id=779923
1297          */
1298         if (time_value < 0)
1299             time_value = 0;
1300     }
1301
1302     message->doc.add_value (NOTMUCH_VALUE_TIMESTAMP,
1303                             Xapian::sortable_serialise (time_value));
1304     message->doc.add_value (NOTMUCH_VALUE_FROM, from);
1305     message->doc.add_value (NOTMUCH_VALUE_SUBJECT, subject);
1306     message->modified = true;
1307 }
1308
1309 void
1310 _notmuch_message_update_subject (notmuch_message_t *message,
1311                                  const char *subject)
1312 {
1313     message->doc.add_value (NOTMUCH_VALUE_SUBJECT, subject);
1314     message->modified = true;
1315 }
1316
1317 /* Upgrade a message to support NOTMUCH_FEATURE_LAST_MOD.  The caller
1318  * must call _notmuch_message_sync. */
1319 void
1320 _notmuch_message_upgrade_last_mod (notmuch_message_t *message)
1321 {
1322     /* _notmuch_message_sync will update the last modification
1323      * revision; we just have to ask it to. */
1324     message->modified = true;
1325 }
1326
1327 /* Synchronize changes made to message->doc out into the database. */
1328 void
1329 _notmuch_message_sync (notmuch_message_t *message)
1330 {
1331     if (_notmuch_database_mode (message->notmuch) == NOTMUCH_DATABASE_MODE_READ_ONLY)
1332         return;
1333
1334     if (! message->modified)
1335         return;
1336
1337     /* Update the last modification of this message. */
1338     if (message->notmuch->features & NOTMUCH_FEATURE_LAST_MOD)
1339         /* sortable_serialise gives a reasonably compact encoding,
1340          * which directly translates to reduced IO when scanning the
1341          * value stream.  Since it's built for doubles, we only get 53
1342          * effective bits, but that's still enough for the database to
1343          * last a few centuries at 1 million revisions per second. */
1344         message->doc.add_value (NOTMUCH_VALUE_LAST_MOD,
1345                                 Xapian::sortable_serialise (
1346                                     _notmuch_database_new_revision (
1347                                         message->notmuch)));
1348
1349     message->notmuch->writable_xapian_db->
1350         replace_document (message->doc_id, message->doc);
1351     message->modified = false;
1352 }
1353
1354 /* Delete a message document from the database, leaving a ghost
1355  * message in its place */
1356 notmuch_status_t
1357 _notmuch_message_delete (notmuch_message_t *message)
1358 {
1359     notmuch_status_t status;
1360     const char *mid, *tid;
1361     notmuch_message_t *ghost;
1362     notmuch_private_status_t private_status;
1363     notmuch_database_t *notmuch;
1364     unsigned int count = 0;
1365     bool is_ghost;
1366
1367     mid = notmuch_message_get_message_id (message);
1368     tid = notmuch_message_get_thread_id (message);
1369     notmuch = message->notmuch;
1370
1371     status = _notmuch_database_ensure_writable (message->notmuch);
1372     if (status)
1373         return status;
1374
1375     message->notmuch->writable_xapian_db->delete_document (message->doc_id);
1376
1377     /* if this was a ghost to begin with, we are done */
1378     private_status = _notmuch_message_has_term (message, "type", "ghost", &is_ghost);
1379     if (private_status)
1380         return COERCE_STATUS (private_status,
1381                               "Error trying to determine whether message was a ghost");
1382     if (is_ghost)
1383         return NOTMUCH_STATUS_SUCCESS;
1384
1385     /* look for a non-ghost message in the same thread */
1386     try {
1387         Xapian::PostingIterator thread_doc, thread_doc_end;
1388         Xapian::PostingIterator mail_doc, mail_doc_end;
1389
1390         _notmuch_database_find_doc_ids (message->notmuch, "thread", tid, &thread_doc,
1391                                         &thread_doc_end);
1392         _notmuch_database_find_doc_ids (message->notmuch, "type", "mail", &mail_doc, &mail_doc_end);
1393
1394         while (count == 0 &&
1395                thread_doc != thread_doc_end &&
1396                mail_doc != mail_doc_end) {
1397             thread_doc.skip_to (*mail_doc);
1398             if (thread_doc != thread_doc_end) {
1399                 if (*thread_doc == *mail_doc) {
1400                     count++;
1401                 } else {
1402                     mail_doc.skip_to (*thread_doc);
1403                     if (mail_doc != mail_doc_end && *thread_doc == *mail_doc)
1404                         count++;
1405                 }
1406             }
1407         }
1408     } catch (Xapian::Error &error) {
1409         LOG_XAPIAN_EXCEPTION (message, error);
1410         return NOTMUCH_STATUS_XAPIAN_EXCEPTION;
1411     }
1412     if (count > 0) {
1413         /* reintroduce a ghost in its place because there are still
1414          * other active messages in this thread: */
1415         ghost = _notmuch_message_create_for_message_id (notmuch, mid, &private_status);
1416         if (private_status == NOTMUCH_PRIVATE_STATUS_NO_DOCUMENT_FOUND) {
1417             private_status = _notmuch_message_initialize_ghost (ghost, tid);
1418             if (! private_status)
1419                 _notmuch_message_sync (ghost);
1420         } else if (private_status == NOTMUCH_PRIVATE_STATUS_SUCCESS) {
1421             /* this is deeply weird, and we should not have gotten
1422              * into this state.  is there a better error message to
1423              * return here? */
1424             status = NOTMUCH_STATUS_DUPLICATE_MESSAGE_ID;
1425         }
1426
1427         notmuch_message_destroy (ghost);
1428         status = COERCE_STATUS (private_status, "Error converting to ghost message");
1429     } else {
1430         /* the thread now contains only ghosts: delete them */
1431         try {
1432             Xapian::PostingIterator doc, doc_end;
1433
1434             _notmuch_database_find_doc_ids (message->notmuch, "thread", tid, &doc, &doc_end);
1435
1436             for (; doc != doc_end; doc++) {
1437                 message->notmuch->writable_xapian_db->delete_document (*doc);
1438             }
1439         } catch (Xapian::Error &error) {
1440             LOG_XAPIAN_EXCEPTION (message, error);
1441             return NOTMUCH_STATUS_XAPIAN_EXCEPTION;
1442         }
1443
1444     }
1445     return status;
1446 }
1447
1448 /* Transform a blank message into a ghost message.  The caller must
1449  * _notmuch_message_sync the message. */
1450 notmuch_private_status_t
1451 _notmuch_message_initialize_ghost (notmuch_message_t *message,
1452                                    const char *thread_id)
1453 {
1454     notmuch_private_status_t status;
1455
1456     status = _notmuch_message_add_term (message, "type", "ghost");
1457     if (status)
1458         return status;
1459     status = _notmuch_message_add_term (message, "thread", thread_id);
1460     if (status)
1461         return status;
1462
1463     return NOTMUCH_PRIVATE_STATUS_SUCCESS;
1464 }
1465
1466 /* Ensure that 'message' is not holding any file object open. Future
1467  * calls to various functions will still automatically open the
1468  * message file as needed.
1469  */
1470 void
1471 _notmuch_message_close (notmuch_message_t *message)
1472 {
1473     if (message->message_file) {
1474         _notmuch_message_file_close (message->message_file);
1475         message->message_file = NULL;
1476     }
1477 }
1478
1479 /* Add a name:value term to 'message', (the actual term will be
1480  * encoded by prefixing the value with a short prefix). See
1481  * NORMAL_PREFIX and BOOLEAN_PREFIX arrays for the mapping of term
1482  * names to prefix values.
1483  *
1484  * This change will not be reflected in the database until the next
1485  * call to _notmuch_message_sync. */
1486 NODISCARD notmuch_private_status_t
1487 _notmuch_message_add_term (notmuch_message_t *message,
1488                            const char *prefix_name,
1489                            const char *value)
1490 {
1491
1492     char *term;
1493     notmuch_private_status_t status = NOTMUCH_PRIVATE_STATUS_SUCCESS;
1494
1495     if (value == NULL)
1496         return NOTMUCH_PRIVATE_STATUS_NULL_POINTER;
1497
1498     term = talloc_asprintf (message, "%s%s",
1499                             _find_prefix (prefix_name), value);
1500     if (strlen (term) > NOTMUCH_TERM_MAX) {
1501         status = NOTMUCH_PRIVATE_STATUS_TERM_TOO_LONG;
1502         goto DONE;
1503     }
1504
1505     try {
1506         message->doc.add_term (term, 0);
1507         message->modified = true;
1508         _notmuch_message_invalidate_metadata (message, prefix_name);
1509     } catch (Xapian::Error &error) {
1510         LOG_XAPIAN_EXCEPTION (message, error);
1511         status = NOTMUCH_PRIVATE_STATUS_XAPIAN_EXCEPTION;
1512     }
1513
1514   DONE:
1515     talloc_free (term);
1516     return status;
1517 }
1518
1519 /* Parse 'text' and add a term to 'message' for each parsed word. Each
1520  * term will be added with the appropriate prefix if prefix_name is
1521  * non-NULL.
1522  */
1523 notmuch_private_status_t
1524 _notmuch_message_gen_terms (notmuch_message_t *message,
1525                             const char *prefix_name,
1526                             const char *text)
1527 {
1528     Xapian::TermGenerator *term_gen = message->notmuch->term_gen;
1529
1530     if (text == NULL)
1531         return NOTMUCH_PRIVATE_STATUS_NULL_POINTER;
1532
1533     term_gen->set_document (message->doc);
1534     term_gen->set_termpos (message->termpos);
1535
1536     if (prefix_name) {
1537         const char *prefix = _notmuch_database_prefix (message->notmuch, prefix_name);
1538         if (prefix == NULL)
1539             return NOTMUCH_PRIVATE_STATUS_BAD_PREFIX;
1540
1541         _notmuch_message_invalidate_metadata (message, prefix_name);
1542         term_gen->index_text (text, 1, prefix);
1543     } else {
1544         term_gen->index_text (text);
1545     }
1546
1547     /* Create a gap between this an the next terms so they don't
1548      * appear to be a phrase. */
1549     message->termpos = term_gen->get_termpos () + 100;
1550
1551     return NOTMUCH_PRIVATE_STATUS_SUCCESS;
1552 }
1553
1554 /* Remove a name:value term from 'message', (the actual term will be
1555  * encoded by prefixing the value with a short prefix). See
1556  * NORMAL_PREFIX and BOOLEAN_PREFIX arrays for the mapping of term
1557  * names to prefix values.
1558  *
1559  * This change will not be reflected in the database until the next
1560  * call to _notmuch_message_sync. */
1561 NODISCARD notmuch_private_status_t
1562 _notmuch_message_remove_term (notmuch_message_t *message,
1563                               const char *prefix_name,
1564                               const char *value)
1565 {
1566     char *term;
1567
1568     if (value == NULL)
1569         return NOTMUCH_PRIVATE_STATUS_NULL_POINTER;
1570
1571     term = talloc_asprintf (message, "%s%s",
1572                             _find_prefix (prefix_name), value);
1573
1574     if (strlen (term) > NOTMUCH_TERM_MAX)
1575         return NOTMUCH_PRIVATE_STATUS_TERM_TOO_LONG;
1576
1577     try {
1578         message->doc.remove_term (term);
1579         message->modified = true;
1580     } catch (const Xapian::InvalidArgumentError &error) {
1581         /* We'll let the philosophers try to wrestle with the
1582          * question of whether failing to remove that which was not
1583          * there in the first place is failure. For us, we'll silently
1584          * consider it all good. */
1585         LOG_XAPIAN_EXCEPTION (message, error);
1586     }
1587
1588     talloc_free (term);
1589
1590     _notmuch_message_invalidate_metadata (message, prefix_name);
1591
1592     return NOTMUCH_PRIVATE_STATUS_SUCCESS;
1593 }
1594
1595 notmuch_private_status_t
1596 _notmuch_message_has_term (notmuch_message_t *message,
1597                            const char *prefix_name,
1598                            const char *value,
1599                            bool *result)
1600 {
1601     char *term;
1602     bool out = false;
1603     notmuch_private_status_t status = NOTMUCH_PRIVATE_STATUS_SUCCESS;
1604
1605     if (value == NULL)
1606         return NOTMUCH_PRIVATE_STATUS_NULL_POINTER;
1607
1608     term = talloc_asprintf (message, "%s%s",
1609                             _find_prefix (prefix_name), value);
1610
1611     if (strlen (term) > NOTMUCH_TERM_MAX)
1612         return NOTMUCH_PRIVATE_STATUS_TERM_TOO_LONG;
1613
1614     try {
1615         /* Look for the exact term */
1616         Xapian::TermIterator i = message->doc.termlist_begin ();
1617         i.skip_to (term);
1618         if (i != message->doc.termlist_end () &&
1619             ! strcmp ((*i).c_str (), term))
1620             out = true;
1621     } catch (Xapian::Error &error) {
1622         status = NOTMUCH_PRIVATE_STATUS_XAPIAN_EXCEPTION;
1623     }
1624     talloc_free (term);
1625
1626     *result = out;
1627     return status;
1628 }
1629
1630 notmuch_status_t
1631 notmuch_message_add_tag (notmuch_message_t *message, const char *tag)
1632 {
1633     notmuch_private_status_t private_status;
1634     notmuch_status_t status;
1635
1636     try {
1637         status = _notmuch_database_ensure_writable (message->notmuch);
1638         if (status)
1639             return status;
1640
1641         if (tag == NULL)
1642             return NOTMUCH_STATUS_NULL_POINTER;
1643
1644         if (strlen (tag) > NOTMUCH_TAG_MAX)
1645             return NOTMUCH_STATUS_TAG_TOO_LONG;
1646
1647         private_status = _notmuch_message_add_term (message, "tag", tag);
1648         if (private_status) {
1649             return COERCE_STATUS (private_status,
1650                                   "_notmuch_message_remove_term return unexpected value: %d\n",
1651                                   private_status);
1652         }
1653
1654         if (! message->frozen)
1655             _notmuch_message_sync (message);
1656
1657     } catch (Xapian::Error &error) {
1658         LOG_XAPIAN_EXCEPTION (message, error);
1659         return NOTMUCH_STATUS_XAPIAN_EXCEPTION;
1660     }
1661
1662     return NOTMUCH_STATUS_SUCCESS;
1663 }
1664
1665 notmuch_status_t
1666 notmuch_message_remove_tag (notmuch_message_t *message, const char *tag)
1667 {
1668     notmuch_private_status_t private_status;
1669     notmuch_status_t status;
1670
1671     try {
1672         status = _notmuch_database_ensure_writable (message->notmuch);
1673         if (status)
1674             return status;
1675
1676         if (tag == NULL)
1677             return NOTMUCH_STATUS_NULL_POINTER;
1678
1679         if (strlen (tag) > NOTMUCH_TAG_MAX)
1680             return NOTMUCH_STATUS_TAG_TOO_LONG;
1681
1682         private_status = _notmuch_message_remove_term (message, "tag", tag);
1683         if (private_status) {
1684             return COERCE_STATUS (private_status,
1685                                   "_notmuch_message_remove_term return unexpected value: %d\n",
1686                                   private_status);
1687         }
1688
1689         if (! message->frozen)
1690             _notmuch_message_sync (message);
1691     } catch (Xapian::Error &error) {
1692         LOG_XAPIAN_EXCEPTION (message, error);
1693         return NOTMUCH_STATUS_XAPIAN_EXCEPTION;
1694     }
1695
1696     return NOTMUCH_STATUS_SUCCESS;
1697 }
1698
1699 /* Is the given filename within a maildir directory?
1700  *
1701  * Specifically, is the final directory component of 'filename' either
1702  * "cur" or "new". If so, return a pointer to that final directory
1703  * component within 'filename'. If not, return NULL.
1704  *
1705  * A non-NULL return value is guaranteed to be a valid string pointer
1706  * pointing to the characters "new/" or "cur/", (but not
1707  * NUL-terminated).
1708  */
1709 static const char *
1710 _filename_is_in_maildir (const char *filename)
1711 {
1712     const char *slash, *dir = NULL;
1713
1714     /* Find the last '/' separating directory from filename. */
1715     slash = strrchr (filename, '/');
1716     if (slash == NULL)
1717         return NULL;
1718
1719     /* Jump back 4 characters to where the previous '/' will be if the
1720      * directory is named "cur" or "new". */
1721     if (slash - filename < 4)
1722         return NULL;
1723
1724     slash -= 4;
1725
1726     if (*slash != '/')
1727         return NULL;
1728
1729     dir = slash + 1;
1730
1731     if (STRNCMP_LITERAL (dir, "cur/") == 0 ||
1732         STRNCMP_LITERAL (dir, "new/") == 0) {
1733         return dir;
1734     }
1735
1736     return NULL;
1737 }
1738
1739 static notmuch_status_t
1740 _ensure_maildir_flags (notmuch_message_t *message, bool force)
1741 {
1742     const char *flags;
1743     notmuch_filenames_t *filenames;
1744     const char *filename, *dir;
1745     char *combined_flags = talloc_strdup (message, "");
1746     int seen_maildir_info = 0;
1747
1748     if (message->maildir_flags) {
1749         if (force) {
1750             talloc_free (message->maildir_flags);
1751             message->maildir_flags = NULL;
1752         }
1753     }
1754     filenames = notmuch_message_get_filenames (message);
1755     if (! filenames)
1756         return NOTMUCH_STATUS_XAPIAN_EXCEPTION;
1757     for (;
1758          notmuch_filenames_valid (filenames);
1759          notmuch_filenames_move_to_next (filenames)) {
1760         filename = notmuch_filenames_get (filenames);
1761         dir = _filename_is_in_maildir (filename);
1762
1763         if (! dir)
1764             continue;
1765
1766         flags = strstr (filename, ":2,");
1767         if (flags) {
1768             seen_maildir_info = 1;
1769             flags += 3;
1770             combined_flags = talloc_strdup_append (combined_flags, flags);
1771         } else if (STRNCMP_LITERAL (dir, "new/") == 0) {
1772             /* Messages are delivered to new/ with no "info" part, but
1773              * they effectively have default maildir flags.  According
1774              * to the spec, we should ignore the info part for
1775              * messages in new/, but some MUAs (mutt) can set maildir
1776              * flags on messages in new/, so we're liberal in what we
1777              * accept. */
1778             seen_maildir_info = 1;
1779         }
1780     }
1781     if (seen_maildir_info)
1782         message->maildir_flags = combined_flags;
1783     return NOTMUCH_STATUS_SUCCESS;
1784 }
1785
1786 notmuch_bool_t
1787 notmuch_message_has_maildir_flag (notmuch_message_t *message, char flag)
1788 {
1789     notmuch_status_t status;
1790     notmuch_bool_t ret;
1791
1792     status = notmuch_message_has_maildir_flag_st (message, flag, &ret);
1793     if (status)
1794         return FALSE;
1795
1796     return ret;
1797 }
1798
1799 notmuch_status_t
1800 notmuch_message_has_maildir_flag_st (notmuch_message_t *message,
1801                                      char flag,
1802                                      notmuch_bool_t *is_set)
1803 {
1804     notmuch_status_t status;
1805
1806     if (! is_set)
1807         return NOTMUCH_STATUS_NULL_POINTER;
1808
1809     status = _ensure_maildir_flags (message, false);
1810     if (status)
1811         return status;
1812
1813     *is_set =  message->maildir_flags && (strchr (message->maildir_flags, flag) != NULL);
1814     return NOTMUCH_STATUS_SUCCESS;
1815 }
1816
1817 notmuch_status_t
1818 notmuch_message_maildir_flags_to_tags (notmuch_message_t *message)
1819 {
1820     notmuch_status_t status;
1821     unsigned i;
1822
1823     status = _ensure_maildir_flags (message, true);
1824     if (status)
1825         return status;
1826     /* If none of the filenames have any maildir info field (not even
1827      * an empty info with no flags set) then there's no information to
1828      * go on, so do nothing. */
1829     if (! message->maildir_flags)
1830         return NOTMUCH_STATUS_SUCCESS;
1831
1832     status = notmuch_message_freeze (message);
1833     if (status)
1834         return status;
1835
1836     for (i = 0; i < ARRAY_SIZE (flag2tag); i++) {
1837         if ((strchr (message->maildir_flags, flag2tag[i].flag) != NULL)
1838             ^
1839             flag2tag[i].inverse) {
1840             status = notmuch_message_add_tag (message, flag2tag[i].tag);
1841         } else {
1842             status = notmuch_message_remove_tag (message, flag2tag[i].tag);
1843         }
1844         if (status)
1845             return status;
1846     }
1847     status = notmuch_message_thaw (message);
1848
1849     return status;
1850 }
1851
1852 /* From the set of tags on 'message' and the flag2tag table, compute a
1853  * set of maildir-flag actions to be taken, (flags that should be
1854  * either set or cleared).
1855  *
1856  * The result is returned as two talloced strings: to_set, and to_clear
1857  */
1858 static void
1859 _get_maildir_flag_actions (notmuch_message_t *message,
1860                            char **to_set_ret,
1861                            char **to_clear_ret)
1862 {
1863     char *to_set, *to_clear;
1864     notmuch_tags_t *tags;
1865     const char *tag;
1866     unsigned i;
1867
1868     to_set = talloc_strdup (message, "");
1869     to_clear = talloc_strdup (message, "");
1870
1871     /* First, find flags for all set tags. */
1872     for (tags = notmuch_message_get_tags (message);
1873          notmuch_tags_valid (tags);
1874          notmuch_tags_move_to_next (tags)) {
1875         tag = notmuch_tags_get (tags);
1876
1877         for (i = 0; i < ARRAY_SIZE (flag2tag); i++) {
1878             if (strcmp (tag, flag2tag[i].tag) == 0) {
1879                 if (flag2tag[i].inverse)
1880                     to_clear = talloc_asprintf_append (to_clear,
1881                                                        "%c",
1882                                                        flag2tag[i].flag);
1883                 else
1884                     to_set = talloc_asprintf_append (to_set,
1885                                                      "%c",
1886                                                      flag2tag[i].flag);
1887             }
1888         }
1889     }
1890
1891     /* Then, find the flags for all tags not present. */
1892     for (i = 0; i < ARRAY_SIZE (flag2tag); i++) {
1893         if (flag2tag[i].inverse) {
1894             if (strchr (to_clear, flag2tag[i].flag) == NULL)
1895                 to_set = talloc_asprintf_append (to_set, "%c", flag2tag[i].flag);
1896         } else {
1897             if (strchr (to_set, flag2tag[i].flag) == NULL)
1898                 to_clear = talloc_asprintf_append (to_clear, "%c", flag2tag[i].flag);
1899         }
1900     }
1901
1902     *to_set_ret = to_set;
1903     *to_clear_ret = to_clear;
1904 }
1905
1906 /* Given 'filename' and a set of maildir flags to set and to clear,
1907  * compute the new maildir filename.
1908  *
1909  * If the existing filename is in the directory "new", the new
1910  * filename will be in the directory "cur", except for the case when
1911  * no flags are changed and the existing filename does not contain
1912  * maildir info (starting with ",2:").
1913  *
1914  * After a sequence of ":2," in the filename, any subsequent
1915  * single-character flags will be added or removed according to the
1916  * characters in flags_to_set and flags_to_clear. Any existing flags
1917  * not mentioned in either string will remain. The final list of flags
1918  * will be in ASCII order.
1919  *
1920  * If the original flags seem invalid, (repeated characters or
1921  * non-ASCII ordering of flags), this function will return NULL
1922  * (meaning that renaming would not be safe and should not occur).
1923  */
1924 static char *
1925 _new_maildir_filename (void *ctx,
1926                        const char *filename,
1927                        const char *flags_to_set,
1928                        const char *flags_to_clear)
1929 {
1930     const char *info, *flags;
1931     unsigned int flag, last_flag;
1932     char *filename_new, *dir;
1933     char flag_map[128];
1934     int flags_in_map = 0;
1935     bool flags_changed = false;
1936     unsigned int i;
1937     char *s;
1938
1939     memset (flag_map, 0, sizeof (flag_map));
1940
1941     info = strstr (filename, ":2,");
1942
1943     if (info == NULL) {
1944         info = filename + strlen (filename);
1945     } else {
1946         /* Loop through existing flags in filename. */
1947         for (flags = info + 3, last_flag = 0;
1948              *flags;
1949              last_flag = flag, flags++) {
1950             flag = *flags;
1951
1952             /* Original flags not in ASCII order. Abort. */
1953             if (flag < last_flag)
1954                 return NULL;
1955
1956             /* Non-ASCII flag. Abort. */
1957             if (flag > sizeof (flag_map) - 1)
1958                 return NULL;
1959
1960             /* Repeated flag value. Abort. */
1961             if (flag_map[flag])
1962                 return NULL;
1963
1964             flag_map[flag] = 1;
1965             flags_in_map++;
1966         }
1967     }
1968
1969     /* Then set and clear our flags from tags. */
1970     for (flags = flags_to_set; *flags; flags++) {
1971         flag = *flags;
1972         if (flag_map[flag] == 0) {
1973             flag_map[flag] = 1;
1974             flags_in_map++;
1975             flags_changed = true;
1976         }
1977     }
1978
1979     for (flags = flags_to_clear; *flags; flags++) {
1980         flag = *flags;
1981         if (flag_map[flag]) {
1982             flag_map[flag] = 0;
1983             flags_in_map--;
1984             flags_changed = true;
1985         }
1986     }
1987
1988     /* Messages in new/ without maildir info can be kept in new/ if no
1989      * flags have changed. */
1990     dir = (char *) _filename_is_in_maildir (filename);
1991     if (dir && STRNCMP_LITERAL (dir, "new/") == 0 && ! *info && ! flags_changed)
1992         return talloc_strdup (ctx, filename);
1993
1994     filename_new = (char *) talloc_size (ctx,
1995                                          info - filename +
1996                                          strlen (":2,") + flags_in_map + 1);
1997     if (unlikely (filename_new == NULL))
1998         return NULL;
1999
2000     strncpy (filename_new, filename, info - filename);
2001     filename_new[info - filename] = '\0';
2002
2003     strcat (filename_new, ":2,");
2004
2005     s = filename_new + strlen (filename_new);
2006     for (i = 0; i < sizeof (flag_map); i++) {
2007         if (flag_map[i]) {
2008             *s = i;
2009             s++;
2010         }
2011     }
2012     *s = '\0';
2013
2014     /* If message is in new/ move it under cur/. */
2015     dir = (char *) _filename_is_in_maildir (filename_new);
2016     if (dir && STRNCMP_LITERAL (dir, "new/") == 0)
2017         memcpy (dir, "cur/", 4);
2018
2019     return filename_new;
2020 }
2021
2022 notmuch_status_t
2023 notmuch_message_tags_to_maildir_flags (notmuch_message_t *message)
2024 {
2025     notmuch_filenames_t *filenames;
2026     const char *filename;
2027     char *filename_new;
2028     char *to_set, *to_clear;
2029     notmuch_status_t status = NOTMUCH_STATUS_SUCCESS;
2030
2031     _get_maildir_flag_actions (message, &to_set, &to_clear);
2032
2033     for (filenames = notmuch_message_get_filenames (message);
2034          notmuch_filenames_valid (filenames);
2035          notmuch_filenames_move_to_next (filenames)) {
2036         filename = notmuch_filenames_get (filenames);
2037
2038         if (! _filename_is_in_maildir (filename))
2039             continue;
2040
2041         filename_new = _new_maildir_filename (message, filename,
2042                                               to_set, to_clear);
2043         if (filename_new == NULL)
2044             continue;
2045
2046         if (strcmp (filename, filename_new)) {
2047             int err;
2048             notmuch_status_t new_status;
2049
2050             err = rename (filename, filename_new);
2051             if (err)
2052                 continue;
2053
2054             new_status = _notmuch_message_remove_filename (message,
2055                                                            filename);
2056             /* Hold on to only the first error. */
2057             if (! status && new_status
2058                 && new_status != NOTMUCH_STATUS_DUPLICATE_MESSAGE_ID) {
2059                 status = new_status;
2060                 continue;
2061             }
2062
2063             new_status = _notmuch_message_add_filename (message,
2064                                                         filename_new);
2065             /* Hold on to only the first error. */
2066             if (! status && new_status) {
2067                 status = new_status;
2068                 continue;
2069             }
2070
2071             _notmuch_message_sync (message);
2072         }
2073
2074         talloc_free (filename_new);
2075     }
2076
2077     talloc_free (to_set);
2078     talloc_free (to_clear);
2079
2080     return status;
2081 }
2082
2083 notmuch_status_t
2084 notmuch_message_remove_all_tags (notmuch_message_t *message)
2085 {
2086     notmuch_private_status_t private_status;
2087     notmuch_status_t status;
2088     notmuch_tags_t *tags;
2089     const char *tag;
2090
2091     status = _notmuch_database_ensure_writable (message->notmuch);
2092     if (status)
2093         return status;
2094     tags = notmuch_message_get_tags (message);
2095     if (! tags)
2096         return NOTMUCH_STATUS_XAPIAN_EXCEPTION;
2097
2098     for (;
2099          notmuch_tags_valid (tags);
2100          notmuch_tags_move_to_next (tags)) {
2101         tag = notmuch_tags_get (tags);
2102
2103         private_status = _notmuch_message_remove_term (message, "tag", tag);
2104         if (private_status) {
2105             return COERCE_STATUS (private_status,
2106                                   "_notmuch_message_remove_term return unexpected value: %d\n",
2107                                   private_status);
2108         }
2109     }
2110
2111     if (! message->frozen)
2112         _notmuch_message_sync (message);
2113
2114     talloc_free (tags);
2115     return NOTMUCH_STATUS_SUCCESS;
2116 }
2117
2118 notmuch_status_t
2119 notmuch_message_freeze (notmuch_message_t *message)
2120 {
2121     notmuch_status_t status;
2122
2123     status = _notmuch_database_ensure_writable (message->notmuch);
2124     if (status)
2125         return status;
2126
2127     message->frozen++;
2128
2129     return NOTMUCH_STATUS_SUCCESS;
2130 }
2131
2132 notmuch_status_t
2133 notmuch_message_thaw (notmuch_message_t *message)
2134 {
2135     notmuch_status_t status;
2136
2137     status = _notmuch_database_ensure_writable (message->notmuch);
2138     if (status)
2139         return status;
2140
2141     if (message->frozen > 0) {
2142         message->frozen--;
2143         if (message->frozen == 0)
2144             _notmuch_message_sync (message);
2145         return NOTMUCH_STATUS_SUCCESS;
2146     } else {
2147         return NOTMUCH_STATUS_UNBALANCED_FREEZE_THAW;
2148     }
2149 }
2150
2151 void
2152 notmuch_message_destroy (notmuch_message_t *message)
2153 {
2154     talloc_free (message);
2155 }
2156
2157 notmuch_database_t *
2158 notmuch_message_get_database (const notmuch_message_t *message)
2159 {
2160     return message->notmuch;
2161 }
2162
2163 static void
2164 _notmuch_message_ensure_property_map (notmuch_message_t *message)
2165 {
2166     notmuch_string_node_t *node;
2167
2168     if (message->property_map)
2169         return;
2170
2171     _notmuch_message_ensure_metadata (message, message->property_term_list);
2172
2173     message->property_map = _notmuch_string_map_create (message);
2174
2175     for (node = message->property_term_list->head; node; node = node->next) {
2176         const char *key;
2177         char *value;
2178
2179         value = strchr (node->string, '=');
2180         if (! value)
2181             INTERNAL_ERROR ("malformed property term");
2182
2183         *value = '\0';
2184         value++;
2185         key = node->string;
2186
2187         _notmuch_string_map_append (message->property_map, key, value);
2188
2189     }
2190
2191     talloc_free (message->property_term_list);
2192     message->property_term_list = NULL;
2193 }
2194
2195 notmuch_string_map_t *
2196 _notmuch_message_property_map (notmuch_message_t *message)
2197 {
2198     _notmuch_message_ensure_property_map (message);
2199
2200     return message->property_map;
2201 }
2202
2203 bool
2204 _notmuch_message_frozen (notmuch_message_t *message)
2205 {
2206     return message->frozen;
2207 }
2208
2209 notmuch_status_t
2210 notmuch_message_reindex (notmuch_message_t *message,
2211                          notmuch_indexopts_t *indexopts)
2212 {
2213     notmuch_database_t *notmuch = NULL;
2214     notmuch_status_t ret = NOTMUCH_STATUS_SUCCESS;
2215     notmuch_private_status_t private_status;
2216     notmuch_filenames_t *orig_filenames = NULL;
2217     const char *orig_thread_id = NULL;
2218     notmuch_message_file_t *message_file = NULL;
2219
2220     int found = 0;
2221
2222     if (message == NULL)
2223         return NOTMUCH_STATUS_NULL_POINTER;
2224
2225     /* Save in case we need to delete message */
2226     orig_thread_id = notmuch_message_get_thread_id (message);
2227     if (! orig_thread_id) {
2228         /* the following is correct as long as there is only one reason
2229          * n_m_get_thread_id returns NULL
2230          */
2231         return NOTMUCH_STATUS_XAPIAN_EXCEPTION;
2232     }
2233
2234     /* strdup it because the metadata may be invalidated */
2235     orig_thread_id = talloc_strdup (message, orig_thread_id);
2236
2237     notmuch = notmuch_message_get_database (message);
2238
2239     ret = _notmuch_database_ensure_writable (notmuch);
2240     if (ret)
2241         return ret;
2242
2243     orig_filenames = notmuch_message_get_filenames (message);
2244
2245     private_status = _notmuch_message_remove_indexed_terms (message);
2246     if (private_status) {
2247         ret = COERCE_STATUS (private_status, "error removing terms");
2248         goto DONE;
2249     }
2250
2251     ret = notmuch_message_remove_all_properties_with_prefix (message, "index.");
2252     if (ret)
2253         goto DONE; /* XXX TODO: distinguish from other error returns above? */
2254     if (indexopts && notmuch_indexopts_get_decrypt_policy (indexopts) == NOTMUCH_DECRYPT_FALSE) {
2255         ret = notmuch_message_remove_all_properties (message, "session-key");
2256         if (ret)
2257             goto DONE;
2258     }
2259
2260     /* re-add the filenames with the associated indexopts */
2261     for (; notmuch_filenames_valid (orig_filenames);
2262          notmuch_filenames_move_to_next (orig_filenames)) {
2263
2264         const char *date;
2265         const char *from, *to, *subject;
2266         char *message_id = NULL;
2267         const char *thread_id = NULL;
2268
2269         const char *filename = notmuch_filenames_get (orig_filenames);
2270
2271         message_file = _notmuch_message_file_open (notmuch, filename);
2272         if (message_file == NULL)
2273             continue;
2274
2275         ret = _notmuch_message_file_get_headers (message_file,
2276                                                  &from, &subject, &to, &date,
2277                                                  &message_id);
2278         if (ret)
2279             goto DONE;
2280
2281         /* XXX TODO: deal with changing message id? */
2282
2283         _notmuch_message_add_filename (message, filename);
2284
2285         ret = _notmuch_database_link_message_to_parents (notmuch, message,
2286                                                          message_file,
2287                                                          &thread_id);
2288         if (ret)
2289             goto DONE;
2290
2291         if (thread_id == NULL)
2292             thread_id = orig_thread_id;
2293
2294         ret = COERCE_STATUS (_notmuch_message_add_term (message, "thread", thread_id),
2295                              "adding thread term");
2296         if (ret)
2297             goto DONE;
2298
2299         /* Take header values only from first filename */
2300         if (found == 0)
2301             _notmuch_message_set_header_values (message, date, from, subject);
2302
2303         ret = _notmuch_message_index_file (message, indexopts, message_file);
2304
2305         if (ret == NOTMUCH_STATUS_FILE_ERROR)
2306             continue;
2307         if (ret)
2308             goto DONE;
2309
2310         found++;
2311         _notmuch_message_file_close (message_file);
2312         message_file = NULL;
2313     }
2314     if (found == 0) {
2315         /* put back thread id to help cleanup */
2316         ret = COERCE_STATUS (_notmuch_message_add_term (message, "thread", orig_thread_id),
2317                              "adding thread term");
2318         if (ret)
2319             goto DONE;
2320
2321         ret = _notmuch_message_delete (message);
2322     } else {
2323         _notmuch_message_sync (message);
2324     }
2325
2326   DONE:
2327     if (message_file)
2328         _notmuch_message_file_close (message_file);
2329
2330     /* XXX TODO destroy orig_filenames? */
2331     return ret;
2332 }