Add internal functions to search for alternate doc types
[notmuch] / lib / notmuch-private.h
1 /* notmuch-private.h - Internal interfaces for notmuch.
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 #ifndef NOTMUCH_PRIVATE_H
22 #define NOTMUCH_PRIVATE_H
23
24 #ifndef _GNU_SOURCE
25 #define _GNU_SOURCE /* For getline and asprintf */
26 #endif
27 #include <stdio.h>
28
29 #include "compat.h"
30
31 #include "notmuch.h"
32
33 NOTMUCH_BEGIN_DECLS
34
35 #include <stdlib.h>
36 #include <stdarg.h>
37 #include <sys/types.h>
38 #include <sys/stat.h>
39 #include <sys/mman.h>
40 #include <string.h>
41 #include <errno.h>
42 #include <fcntl.h>
43 #include <unistd.h>
44 #include <ctype.h>
45 #include <assert.h>
46
47 #include <talloc.h>
48
49 #include <gmime/gmime.h>
50
51 #include "xutil.h"
52 #include "error_util.h"
53 #include "string-util.h"
54
55 #pragma GCC visibility push(hidden)
56
57 #ifdef DEBUG
58 # define DEBUG_DATABASE_SANITY 1
59 # define DEBUG_QUERY 1
60 #endif
61
62 #define COMPILE_TIME_ASSERT(pred) ((void)sizeof(char[1 - 2*!(pred)]))
63
64 #define STRNCMP_LITERAL(var, literal) \
65     strncmp ((var), (literal), sizeof (literal) - 1)
66
67 /* Robust bit test/set/reset macros */
68 #define _NOTMUCH_VALID_BIT(bit) \
69     ((bit) >= 0 && ((unsigned long) bit) < CHAR_BIT * sizeof (unsigned long long))
70 #define NOTMUCH_TEST_BIT(val, bit) \
71     (_NOTMUCH_VALID_BIT(bit) ? !!((val) & (1ull << (bit))) : 0)
72 #define NOTMUCH_SET_BIT(valp, bit) \
73     (_NOTMUCH_VALID_BIT(bit) ? (*(valp) |= (1ull << (bit))) : *(valp))
74 #define NOTMUCH_CLEAR_BIT(valp,  bit) \
75     (_NOTMUCH_VALID_BIT(bit) ? (*(valp) &= ~(1ull << (bit))) : *(valp))
76
77 #define unused(x) x __attribute__ ((unused))
78
79 #ifdef __cplusplus
80 # define visible __attribute__((visibility("default")))
81 #else
82 # define visible
83 #endif
84
85 /* Thanks to Andrew Tridgell's (SAMBA's) talloc for this definition of
86  * unlikely. The talloc source code comes to us via the GNU LGPL v. 3.
87  */
88 /* these macros gain us a few percent of speed on gcc */
89 #if (__GNUC__ >= 3)
90 /* the strange !! is to ensure that __builtin_expect() takes either 0 or 1
91    as its first argument */
92 #ifndef likely
93 #define likely(x)   __builtin_expect(!!(x), 1)
94 #endif
95 #ifndef unlikely
96 #define unlikely(x) __builtin_expect(!!(x), 0)
97 #endif
98 #else
99 #ifndef likely
100 #define likely(x) (x)
101 #endif
102 #ifndef unlikely
103 #define unlikely(x) (x)
104 #endif
105 #endif
106
107 typedef enum {
108     NOTMUCH_VALUE_TIMESTAMP = 0,
109     NOTMUCH_VALUE_MESSAGE_ID,
110     NOTMUCH_VALUE_FROM,
111     NOTMUCH_VALUE_SUBJECT,
112     NOTMUCH_VALUE_LAST_MOD,
113 } notmuch_value_t;
114
115 /* Xapian (with flint backend) complains if we provide a term longer
116  * than this, but I haven't yet found a way to query the limit
117  * programmatically. */
118 #define NOTMUCH_TERM_MAX 245
119
120 #define NOTMUCH_METADATA_THREAD_ID_PREFIX "thread_id_"
121
122 /* For message IDs we have to be even more restrictive. Beyond fitting
123  * into the term limit, we also use message IDs to construct
124  * metadata-key values. And the documentation says that these should
125  * be restricted to about 200 characters. (The actual limit for the
126  * chert backend at least is 252.)
127  */
128 #define NOTMUCH_MESSAGE_ID_MAX (200 - sizeof (NOTMUCH_METADATA_THREAD_ID_PREFIX))
129
130 typedef enum _notmuch_private_status {
131     /* First, copy all the public status values. */
132     NOTMUCH_PRIVATE_STATUS_SUCCESS = NOTMUCH_STATUS_SUCCESS,
133     NOTMUCH_PRIVATE_STATUS_OUT_OF_MEMORY = NOTMUCH_STATUS_OUT_OF_MEMORY,
134     NOTMUCH_PRIVATE_STATUS_READ_ONLY_DATABASE = NOTMUCH_STATUS_READ_ONLY_DATABASE,
135     NOTMUCH_PRIVATE_STATUS_XAPIAN_EXCEPTION = NOTMUCH_STATUS_XAPIAN_EXCEPTION,
136     NOTMUCH_PRIVATE_STATUS_FILE_NOT_EMAIL = NOTMUCH_STATUS_FILE_NOT_EMAIL,
137     NOTMUCH_PRIVATE_STATUS_NULL_POINTER = NOTMUCH_STATUS_NULL_POINTER,
138     NOTMUCH_PRIVATE_STATUS_TAG_TOO_LONG = NOTMUCH_STATUS_TAG_TOO_LONG,
139     NOTMUCH_PRIVATE_STATUS_UNBALANCED_FREEZE_THAW = NOTMUCH_STATUS_UNBALANCED_FREEZE_THAW,
140
141     /* Then add our own private values. */
142     NOTMUCH_PRIVATE_STATUS_TERM_TOO_LONG = NOTMUCH_STATUS_LAST_STATUS,
143     NOTMUCH_PRIVATE_STATUS_NO_DOCUMENT_FOUND,
144
145     NOTMUCH_PRIVATE_STATUS_LAST_STATUS
146 } notmuch_private_status_t;
147
148 /* Coerce a notmuch_private_status_t value to a notmuch_status_t
149  * value, generating an internal error if the private value is equal
150  * to or greater than NOTMUCH_STATUS_LAST_STATUS. (The idea here is
151  * that the caller has previously handled any expected
152  * notmuch_private_status_t values.)
153  *
154  * Note that the function _internal_error does not return. Evaluating
155  * to NOTMUCH_STATUS_SUCCESS is done purely to appease the compiler.
156  */
157 #define COERCE_STATUS(private_status, format, ...)                      \
158     ((private_status >= (notmuch_private_status_t) NOTMUCH_STATUS_LAST_STATUS)\
159      ?                                                                  \
160      _internal_error (format " (%s).\n",                                \
161                       ##__VA_ARGS__,                                    \
162                       __location__),                                    \
163      (notmuch_status_t) NOTMUCH_PRIVATE_STATUS_SUCCESS                  \
164      :                                                                  \
165      (notmuch_status_t) private_status)
166
167 /* Flags shared by various lookup functions. */
168 typedef enum _notmuch_find_flags {
169     /* Lookup without creating any documents.  This is the default
170      * behavior. */
171     NOTMUCH_FIND_LOOKUP = 0,
172     /* If set, create the necessary document (or documents) if they
173      * are missing.  Requires a read/write database. */
174     NOTMUCH_FIND_CREATE = 1<<0,
175 } notmuch_find_flags_t;
176
177 typedef struct _notmuch_doc_id_set notmuch_doc_id_set_t;
178
179 /* database.cc */
180
181 /* Lookup a prefix value by name.
182  *
183  * XXX: This should really be static inside of message.cc, and we can
184  * do that once we convert database.cc to use the
185  * _notmuch_message_add/remove_term functions. */
186 const char *
187 _find_prefix (const char *name);
188
189 char *
190 _notmuch_message_id_compressed (void *ctx, const char *message_id);
191
192 notmuch_status_t
193 _notmuch_database_ensure_writable (notmuch_database_t *notmuch);
194
195 void
196 _notmuch_database_log (notmuch_database_t *notmuch,
197                        const char *format, ...);
198
199 unsigned long
200 _notmuch_database_new_revision (notmuch_database_t *notmuch);
201
202 const char *
203 _notmuch_database_relative_path (notmuch_database_t *notmuch,
204                                  const char *path);
205
206 notmuch_status_t
207 _notmuch_database_split_path (void *ctx,
208                               const char *path,
209                               const char **directory,
210                               const char **basename);
211
212 const char *
213 _notmuch_database_get_directory_db_path (const char *path);
214
215 unsigned int
216 _notmuch_database_generate_doc_id (notmuch_database_t *notmuch);
217
218 notmuch_private_status_t
219 _notmuch_database_find_unique_doc_id (notmuch_database_t *notmuch,
220                                       const char *prefix_name,
221                                       const char *value,
222                                       unsigned int *doc_id);
223
224 notmuch_status_t
225 _notmuch_database_find_directory_id (notmuch_database_t *database,
226                                      const char *path,
227                                      notmuch_find_flags_t flags,
228                                      unsigned int *directory_id);
229
230 const char *
231 _notmuch_database_get_directory_path (void *ctx,
232                                       notmuch_database_t *notmuch,
233                                       unsigned int doc_id);
234
235 notmuch_status_t
236 _notmuch_database_filename_to_direntry (void *ctx,
237                                         notmuch_database_t *notmuch,
238                                         const char *filename,
239                                         notmuch_find_flags_t flags,
240                                         char **direntry);
241
242 /* directory.cc */
243
244 notmuch_directory_t *
245 _notmuch_directory_create (notmuch_database_t *notmuch,
246                            const char *path,
247                            notmuch_find_flags_t flags,
248                            notmuch_status_t *status_ret);
249
250 unsigned int
251 _notmuch_directory_get_document_id (notmuch_directory_t *directory);
252
253 /* message.cc */
254
255 notmuch_message_t *
256 _notmuch_message_create (const void *talloc_owner,
257                          notmuch_database_t *notmuch,
258                          unsigned int doc_id,
259                          notmuch_private_status_t *status);
260
261 notmuch_message_t *
262 _notmuch_message_create_for_message_id (notmuch_database_t *notmuch,
263                                         const char *message_id,
264                                         notmuch_private_status_t *status);
265
266 unsigned int
267 _notmuch_message_get_doc_id (notmuch_message_t *message);
268
269 const char *
270 _notmuch_message_get_in_reply_to (notmuch_message_t *message);
271
272 notmuch_private_status_t
273 _notmuch_message_add_term (notmuch_message_t *message,
274                            const char *prefix_name,
275                            const char *value);
276
277 notmuch_private_status_t
278 _notmuch_message_remove_term (notmuch_message_t *message,
279                               const char *prefix_name,
280                               const char *value);
281
282 notmuch_private_status_t
283 _notmuch_message_gen_terms (notmuch_message_t *message,
284                             const char *prefix_name,
285                             const char *text);
286
287 void
288 _notmuch_message_upgrade_filename_storage (notmuch_message_t *message);
289
290 void
291 _notmuch_message_upgrade_folder (notmuch_message_t *message);
292
293 notmuch_status_t
294 _notmuch_message_add_filename (notmuch_message_t *message,
295                                const char *filename);
296
297 notmuch_status_t
298 _notmuch_message_remove_filename (notmuch_message_t *message,
299                                   const char *filename);
300
301 notmuch_status_t
302 _notmuch_message_rename (notmuch_message_t *message,
303                          const char *new_filename);
304
305 void
306 _notmuch_message_ensure_thread_id (notmuch_message_t *message);
307
308 void
309 _notmuch_message_set_header_values (notmuch_message_t *message,
310                                     const char *date,
311                                     const char *from,
312                                     const char *subject);
313
314 void
315 _notmuch_message_upgrade_last_mod (notmuch_message_t *message);
316
317 void
318 _notmuch_message_sync (notmuch_message_t *message);
319
320 notmuch_status_t
321 _notmuch_message_delete (notmuch_message_t *message);
322
323 notmuch_private_status_t
324 _notmuch_message_initialize_ghost (notmuch_message_t *message,
325                                    const char *thread_id);
326
327 void
328 _notmuch_message_close (notmuch_message_t *message);
329
330 /* Get a copy of the data in this message document.
331  *
332  * Caller should talloc_free the result when done.
333  *
334  * This function is intended to support database upgrade and really
335  * shouldn't be used otherwise. */
336 char *
337 _notmuch_message_talloc_copy_data (notmuch_message_t *message);
338
339 /* Clear the data in this message document.
340  *
341  * This function is intended to support database upgrade and really
342  * shouldn't be used otherwise. */
343 void
344 _notmuch_message_clear_data (notmuch_message_t *message);
345
346 /* Set the author member of 'message' - this is the representation used
347  * when displaying the message */
348 void
349 _notmuch_message_set_author (notmuch_message_t *message, const char *author);
350
351 /* Get the author member of 'message' */
352 const char *
353 _notmuch_message_get_author (notmuch_message_t *message);
354
355 /* message-file.c */
356
357 /* XXX: I haven't decided yet whether these will actually get exported
358  * into the public interface in notmuch.h
359  */
360
361 typedef struct _notmuch_message_file notmuch_message_file_t;
362
363 /* Open a file containing a single email message.
364  *
365  * The caller should call notmuch_message_close when done with this.
366  *
367  * Returns NULL if any error occurs.
368  */
369 notmuch_message_file_t *
370 _notmuch_message_file_open (notmuch_database_t *notmuch, const char *filename);
371
372 /* Like notmuch_message_file_open but with 'ctx' as the talloc owner. */
373 notmuch_message_file_t *
374 _notmuch_message_file_open_ctx (notmuch_database_t *notmuch,
375                                 void *ctx, const char *filename);
376
377 /* Close a notmuch message previously opened with notmuch_message_open. */
378 void
379 _notmuch_message_file_close (notmuch_message_file_t *message);
380
381 /* Parse the message.
382  *
383  * This will be done automatically as necessary on other calls
384  * depending on it, but an explicit call allows for better error
385  * status reporting.
386  */
387 notmuch_status_t
388 _notmuch_message_file_parse (notmuch_message_file_t *message);
389
390 /* Get the gmime message of a message file.
391  *
392  * The message file is parsed as necessary.
393  *
394  * The GMimeMessage* is set to *mime_message on success (which the
395  * caller must not unref).
396  *
397  * XXX: Would be nice to not have to expose GMimeMessage here.
398  */
399 notmuch_status_t
400 _notmuch_message_file_get_mime_message (notmuch_message_file_t *message,
401                                         GMimeMessage **mime_message);
402
403 /* Get the value of the specified header from the message as a UTF-8 string.
404  *
405  * The message file is parsed as necessary.
406  *
407  * The header name is case insensitive.
408  *
409  * The Received: header is special - for it all Received: headers in
410  * the message are concatenated
411  *
412  * The returned value is owned by the notmuch message and is valid
413  * only until the message is closed. The caller should copy it if
414  * needing to modify the value or to hold onto it for longer.
415  *
416  * Returns NULL on errors, empty string if the message does not
417  * contain a header line matching 'header'.
418  */
419 const char *
420 _notmuch_message_file_get_header (notmuch_message_file_t *message,
421                                  const char *header);
422
423 /* index.cc */
424
425 notmuch_status_t
426 _notmuch_message_index_file (notmuch_message_t *message,
427                              notmuch_message_file_t *message_file);
428
429 /* messages.c */
430
431 typedef struct _notmuch_message_node {
432     notmuch_message_t *message;
433     struct _notmuch_message_node *next;
434 } notmuch_message_node_t;
435
436 typedef struct _notmuch_message_list {
437     notmuch_message_node_t *head;
438     notmuch_message_node_t **tail;
439 } notmuch_message_list_t;
440
441 /* There's a rumor that there's an alternate struct _notmuch_messages
442  * somewhere with some nasty C++ objects in it. We'll try to maintain
443  * ignorance of that here. (See notmuch_mset_messages_t in query.cc)
444  */
445 struct visible _notmuch_messages {
446     notmuch_bool_t is_of_list_type;
447     notmuch_doc_id_set_t *excluded_doc_ids;
448     notmuch_message_node_t *iterator;
449 };
450
451 notmuch_message_list_t *
452 _notmuch_message_list_create (const void *ctx);
453
454 void
455 _notmuch_message_list_add_message (notmuch_message_list_t *list,
456                                    notmuch_message_t *message);
457
458 notmuch_messages_t *
459 _notmuch_messages_create (notmuch_message_list_t *list);
460
461 /* query.cc */
462
463 notmuch_bool_t
464 _notmuch_mset_messages_valid (notmuch_messages_t *messages);
465
466 notmuch_message_t *
467 _notmuch_mset_messages_get (notmuch_messages_t *messages);
468
469 void
470 _notmuch_mset_messages_move_to_next (notmuch_messages_t *messages);
471
472 notmuch_bool_t
473 _notmuch_doc_id_set_contains (notmuch_doc_id_set_t *doc_ids,
474                               unsigned int doc_id);
475
476 void
477 _notmuch_doc_id_set_remove (notmuch_doc_id_set_t *doc_ids,
478                             unsigned int doc_id);
479
480 /* querying xapian documents by type (e.g. "mail" or "ghost"): */
481 notmuch_status_t
482 _notmuch_query_search_documents (notmuch_query_t *query,
483                                  const char *type,
484                                  notmuch_messages_t **out);
485
486 notmuch_status_t
487 _notmuch_query_count_documents (notmuch_query_t *query,
488                                 const char *type,
489                                 unsigned *count_out);
490
491 /* message.cc */
492
493 void
494 _notmuch_message_add_reply (notmuch_message_t *message,
495                             notmuch_message_t *reply);
496 notmuch_database_t *
497 _notmuch_message_database (notmuch_message_t *message);
498
499 /* sha1.c */
500
501 char *
502 _notmuch_sha1_of_string (const char *str);
503
504 char *
505 _notmuch_sha1_of_file (const char *filename);
506
507 /* string-list.c */
508
509 typedef struct _notmuch_string_node {
510     char *string;
511     struct _notmuch_string_node *next;
512 } notmuch_string_node_t;
513
514 typedef struct visible _notmuch_string_list {
515     int length;
516     notmuch_string_node_t *head;
517     notmuch_string_node_t **tail;
518 } notmuch_string_list_t;
519
520 notmuch_string_list_t *
521 _notmuch_string_list_create (const void *ctx);
522
523 /* Add 'string' to 'list'.
524  *
525  * The list will create its own talloced copy of 'string'.
526  */
527 void
528 _notmuch_string_list_append (notmuch_string_list_t *list,
529                              const char *string);
530
531 void
532 _notmuch_string_list_sort (notmuch_string_list_t *list);
533
534 /* tags.c */
535
536 notmuch_tags_t *
537 _notmuch_tags_create (const void *ctx, notmuch_string_list_t *list);
538
539 /* filenames.c */
540
541 /* The notmuch_filenames_t iterates over a notmuch_string_list_t of
542  * file names */
543 notmuch_filenames_t *
544 _notmuch_filenames_create (const void *ctx,
545                            notmuch_string_list_t *list);
546
547 /* thread.cc */
548
549 notmuch_thread_t *
550 _notmuch_thread_create (void *ctx,
551                         notmuch_database_t *notmuch,
552                         unsigned int seed_doc_id,
553                         notmuch_doc_id_set_t *match_set,
554                         notmuch_string_list_t *excluded_terms,
555                         notmuch_exclude_t omit_exclude,
556                         notmuch_sort_t sort);
557
558 NOTMUCH_END_DECLS
559
560 #ifdef __cplusplus
561 /* Implicit typecast from 'void *' to 'T *' is okay in C, but not in
562  * C++. In talloc_steal, an explicit cast is provided for type safety
563  * in some GCC versions. Otherwise, a cast is required. Provide a
564  * template function for this to maintain type safety, and redefine
565  * talloc_steal to use it.
566  */
567 #if !(__GNUC__ >= 3)
568 template <class T> T *
569 _notmuch_talloc_steal (const void *new_ctx, const T *ptr)
570 {
571     return static_cast<T *> (talloc_steal (new_ctx, ptr));
572 }
573 #undef talloc_steal
574 #define talloc_steal _notmuch_talloc_steal
575 #endif
576 #endif
577
578 #pragma GCC visibility pop
579
580 #endif