1 /* notmuch - Not much of an email program, (just index and search)
3 * Copyright © 2009 Carl Worth
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.
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.
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/ .
18 * Author: Carl Worth <cworth@cworth.org>
21 #include "notmuch-client.h"
22 #include "dump-restore-private.h"
24 #include "string-util.h"
28 /* Non-zero return indicates an error in retrieving the message,
29 * or in applying the tags.
32 tag_message (unused (void *ctx),
33 notmuch_database_t *notmuch,
34 const char *message_id,
35 tag_op_list_t *tag_ops,
38 notmuch_status_t status;
39 notmuch_message_t *message = NULL;
42 status = notmuch_database_find_message (notmuch, message_id, &message);
43 if (status || message == NULL) {
44 fprintf (stderr, "Warning: cannot apply tags to %smessage: %s\n",
45 message ? "" : "missing ", message_id);
47 fprintf (stderr, "%s\n", notmuch_status_to_string (status));
51 /* In order to detect missing messages, this check/optimization is
52 * intentionally done *after* first finding the message. */
53 if ((flags & TAG_FLAG_REMOVE_ALL) || tag_op_list_size (tag_ops))
54 ret = tag_op_list_apply (message, tag_ops, flags);
56 notmuch_message_destroy (message);
61 /* Sup dump output is one line per message. We match a sequence of
62 * non-space characters for the message-id, then one or more
63 * spaces, then a list of space-separated tags as a sequence of
64 * characters within literal '(' and ')'. */
67 parse_sup_line (void *ctx, char *line,
68 char **query_str, tag_op_list_t *tag_ops)
75 tag_op_list_reset (tag_ops);
79 /* Silently ignore blank lines */
80 if (line[0] == '\0') {
84 rerr = xregexec (®ex, line, 3, match, 0);
85 if (rerr == REG_NOMATCH) {
86 fprintf (stderr, "Warning: Ignoring invalid sup format line: %s\n",
91 *query_str = talloc_strndup (ctx, line + match[1].rm_so,
92 match[1].rm_eo - match[1].rm_so);
93 file_tags = talloc_strndup (ctx, line + match[2].rm_so,
94 match[2].rm_eo - match[2].rm_so);
96 char *tok = file_tags;
99 tag_op_list_reset (tag_ops);
101 while ((tok = strtok_len (tok + tok_len, " ", &tok_len)) != NULL) {
103 if (*(tok + tok_len) != '\0') {
104 *(tok + tok_len) = '\0';
108 if (tag_op_list_append (ctx, tag_ops, tok, FALSE))
117 notmuch_restore_command (unused (void *ctx), int argc, char *argv[])
119 notmuch_config_t *config;
120 notmuch_database_t *notmuch;
121 notmuch_bool_t accumulate = FALSE;
122 tag_op_flag_t flags = 0;
123 tag_op_list_t *tag_ops;
125 char *input_file_name = NULL;
133 int input_format = DUMP_FORMAT_AUTO;
135 config = notmuch_config_open (ctx, NULL, NULL);
139 if (notmuch_database_open (notmuch_config_get_database_path (config),
140 NOTMUCH_DATABASE_MODE_READ_WRITE, ¬much))
143 if (notmuch_config_get_maildir_synchronize_flags (config))
144 flags |= TAG_FLAG_MAILDIR_SYNC;
146 notmuch_opt_desc_t options[] = {
147 { NOTMUCH_OPT_KEYWORD, &input_format, "format", 'f',
148 (notmuch_keyword_t []){ { "auto", DUMP_FORMAT_AUTO },
149 { "batch-tag", DUMP_FORMAT_BATCH_TAG },
150 { "sup", DUMP_FORMAT_SUP },
152 { NOTMUCH_OPT_STRING, &input_file_name, "input", 'i', 0 },
153 { NOTMUCH_OPT_BOOLEAN, &accumulate, "accumulate", 'a', 0 },
157 opt_index = parse_arguments (argc, argv, options, 1);
160 /* diagnostics already printed */
165 flags |= TAG_FLAG_REMOVE_ALL;
167 if (input_file_name) {
168 input = fopen (input_file_name, "r");
170 fprintf (stderr, "Error opening %s for reading: %s\n",
171 input_file_name, strerror (errno));
176 if (opt_index < argc) {
178 "Unused positional parameter: %s\n",
184 line_len = getline (&line, &line_size, input);
188 tag_ops = tag_op_list_create (ctx);
189 if (tag_ops == NULL) {
190 fprintf (stderr, "Out of memory.\n");
194 for (p = line; (input_format == DUMP_FORMAT_AUTO) && *p; p++) {
196 input_format = DUMP_FORMAT_SUP;
199 if (input_format == DUMP_FORMAT_AUTO)
200 input_format = DUMP_FORMAT_BATCH_TAG;
202 if (input_format == DUMP_FORMAT_SUP)
203 if ( xregcomp (®ex,
204 "^([^ ]+) \\(([^)]*)\\)$",
206 INTERNAL_ERROR ("compile time constant regex failed.");
211 if (input_format == DUMP_FORMAT_SUP) {
212 ret = parse_sup_line (ctx, line, &query_string, tag_ops);
214 ret = parse_tag_line (ctx, line, TAG_FLAG_BE_GENEROUS,
215 &query_string, tag_ops);
218 if (strncmp ("id:", query_string, 3) != 0) {
219 fprintf (stderr, "Warning: unsupported query: %s\n", query_string);
222 /* delete id: from front of string; tag_message
223 * expects a raw message-id.
225 * XXX: Note that query string id:foo and bar will be
226 * interpreted as a message id "foo and bar". This
227 * should eventually be fixed to give a better error
230 query_string = query_string + 3;
240 ret = tag_message (ctx, notmuch, query_string,
245 } while ((line_len = getline (&line, &line_size, input)) != -1);
247 if (input_format == DUMP_FORMAT_SUP)
253 notmuch_database_destroy (notmuch);