From fb50cc41fe6f508118697d65a44e8f742a466ca0 Mon Sep 17 00:00:00 2001 From: David Bremner Date: Mon, 12 Nov 2012 21:54:16 -0400 Subject: tag-util.[ch]: New files for common tagging routines These are meant to be shared between notmuch-tag and notmuch-restore. The bulk of the routines implement a "tag operation list" abstract data type act as a structured representation of a set of tag operations (typically coming from a single tag command or line of input). --- tag-util.c | 305 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 305 insertions(+) create mode 100644 tag-util.c (limited to 'tag-util.c') diff --git a/tag-util.c b/tag-util.c new file mode 100644 index 00000000..6d33ef8f --- /dev/null +++ b/tag-util.c @@ -0,0 +1,305 @@ +#include +#include "string-util.h" +#include "tag-util.h" +#include "hex-escape.h" + +#define TAG_OP_LIST_INITIAL_SIZE 10 + +struct _tag_operation_t { + const char *tag; + notmuch_bool_t remove; +}; + +struct _tag_op_list_t { + tag_operation_t *ops; + size_t count; + size_t size; +}; + +static tag_parse_status_t +line_error (tag_parse_status_t status, + const char *line, + const char *format, ...) +{ + va_list va_args; + + va_start (va_args, format); + + fprintf (stderr, status < 0 ? "Error: " : "Warning: "); + vfprintf (stderr, format, va_args); + fprintf (stderr, " [%s]\n", line); + return status; +} + +tag_parse_status_t +parse_tag_line (void *ctx, char *line, + tag_op_flag_t flags, + char **query_string, + tag_op_list_t *tag_ops) +{ + char *tok = line; + size_t tok_len = 0; + char *line_for_error; + int ret = 0; + + chomp_newline (line); + + line_for_error = talloc_strdup (ctx, line); + if (line_for_error == NULL) { + fprintf (stderr, "Error: out of memory\n"); + return -1; + } + + /* remove leading space */ + while (*tok == ' ' || *tok == '\t') + tok++; + + /* Skip empty and comment lines. */ + if (*tok == '\0' || *tok == '#') { + ret = TAG_PARSE_SKIPPED; + goto DONE; + } + + tag_op_list_reset (tag_ops); + + /* Parse tags. */ + while ((tok = strtok_len (tok + tok_len, " ", &tok_len)) != NULL) { + notmuch_bool_t remove; + char *tag; + + /* Optional explicit end of tags marker. */ + if (tok_len == 2 && strncmp (tok, "--", tok_len) == 0) { + tok = strtok_len (tok + tok_len, " ", &tok_len); + if (tok == NULL) { + ret = line_error (TAG_PARSE_INVALID, line_for_error, + "no query string after --"); + goto DONE; + } + break; + } + + /* Implicit end of tags. */ + if (*tok != '-' && *tok != '+') + break; + + /* If tag is terminated by NUL, there's no query string. */ + if (*(tok + tok_len) == '\0') { + ret = line_error (TAG_PARSE_INVALID, line_for_error, + "no query string"); + goto DONE; + } + + /* Terminate, and start next token after terminator. */ + *(tok + tok_len++) = '\0'; + + remove = (*tok == '-'); + tag = tok + 1; + + /* Maybe refuse empty tags. */ + if (! (flags & TAG_FLAG_BE_GENEROUS) && *tag == '\0') { + ret = line_error (TAG_PARSE_INVALID, line_for_error, + "empty tag"); + goto DONE; + } + + /* Decode tag. */ + if (hex_decode_inplace (tag) != HEX_SUCCESS) { + ret = line_error (TAG_PARSE_INVALID, line_for_error, + "hex decoding of tag %s failed", tag); + goto DONE; + } + + if (tag_op_list_append (ctx, tag_ops, tag, remove)) { + ret = line_error (TAG_PARSE_OUT_OF_MEMORY, line_for_error, + "aborting"); + goto DONE; + } + } + + if (tok == NULL) { + /* use a different error message for testing */ + ret = line_error (TAG_PARSE_INVALID, line_for_error, + "missing query string"); + goto DONE; + } + + /* tok now points to the query string */ + if (hex_decode_inplace (tok) != HEX_SUCCESS) { + ret = line_error (TAG_PARSE_INVALID, line_for_error, + "hex decoding of query %s failed", tok); + goto DONE; + } + + *query_string = tok; + + DONE: + talloc_free (line_for_error); + return ret; +} + +static inline void +message_error (notmuch_message_t *message, + notmuch_status_t status, + const char *format, ...) +{ + va_list va_args; + + va_start (va_args, format); + + vfprintf (stderr, format, va_args); + fprintf (stderr, "Message-ID: %s\n", notmuch_message_get_message_id (message)); + fprintf (stderr, "Status: %s\n", notmuch_status_to_string (status)); +} + +notmuch_status_t +tag_op_list_apply (notmuch_message_t *message, + tag_op_list_t *list, + tag_op_flag_t flags) +{ + size_t i; + notmuch_status_t status = 0; + tag_operation_t *tag_ops = list->ops; + + status = notmuch_message_freeze (message); + if (status) { + message_error (message, status, "freezing message"); + return status; + } + + if (flags & TAG_FLAG_REMOVE_ALL) { + status = notmuch_message_remove_all_tags (message); + if (status) { + message_error (message, status, "removing all tags"); + return status; + } + } + + for (i = 0; i < list->count; i++) { + if (tag_ops[i].remove) { + status = notmuch_message_remove_tag (message, tag_ops[i].tag); + if (status) { + message_error (message, status, "removing tag %s", tag_ops[i].tag); + return status; + } + } else { + status = notmuch_message_add_tag (message, tag_ops[i].tag); + if (status) { + message_error (message, status, "adding tag %s", tag_ops[i].tag); + return status; + } + + } + } + + status = notmuch_message_thaw (message); + if (status) { + message_error (message, status, "thawing message"); + return status; + } + + + if (flags & TAG_FLAG_MAILDIR_SYNC) { + status = notmuch_message_tags_to_maildir_flags (message); + if (status) { + message_error (message, status, "synching tags to maildir"); + return status; + } + } + + return NOTMUCH_STATUS_SUCCESS; + +} + + +/* Array of tagging operations (add or remove. Size will be increased + * as necessary. */ + +tag_op_list_t * +tag_op_list_create (void *ctx) +{ + tag_op_list_t *list; + + list = talloc (ctx, tag_op_list_t); + if (list == NULL) + return NULL; + + list->size = TAG_OP_LIST_INITIAL_SIZE; + list->count = 0; + + list->ops = talloc_array (ctx, tag_operation_t, list->size); + if (list->ops == NULL) + return NULL; + + return list; +} + + +int +tag_op_list_append (void *ctx, + tag_op_list_t *list, + const char *tag, + notmuch_bool_t remove) +{ + /* Make room if current array is full. This should be a fairly + * rare case, considering the initial array size. + */ + + if (list->count == list->size) { + list->size *= 2; + list->ops = talloc_realloc (ctx, list->ops, tag_operation_t, + list->size); + if (list->ops == NULL) { + fprintf (stderr, "Out of memory.\n"); + return 1; + } + } + + /* add the new operation */ + + list->ops[list->count].tag = tag; + list->ops[list->count].remove = remove; + list->count++; + return 0; +} + +/* + * Is the i'th tag operation a remove? + */ + +notmuch_bool_t +tag_op_list_isremove (const tag_op_list_t *list, size_t i) +{ + assert (i < list->count); + return list->ops[i].remove; +} + +/* + * Reset a list to contain no operations + */ + +void +tag_op_list_reset (tag_op_list_t *list) +{ + list->count = 0; +} + +/* + * Return the number of operations in a list + */ + +size_t +tag_op_list_size (const tag_op_list_t *list) +{ + return list->count; +} + +/* + * return the i'th tag in the list + */ + +const char * +tag_op_list_tag (const tag_op_list_t *list, size_t i) +{ + assert (i < list->count); + return list->ops[i].tag; +} -- cgit v1.2.3