From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from localhost (localhost [127.0.0.1]) by olra.theworths.org (Postfix) with ESMTP id E6CDA429E26 for ; Sun, 18 Dec 2011 05:15:50 -0800 (PST) X-Virus-Scanned: Debian amavisd-new at olra.theworths.org X-Spam-Flag: NO X-Spam-Score: -2.3 X-Spam-Level: X-Spam-Status: No, score=-2.3 tagged_above=-999 required=5 tests=[RCVD_IN_DNSWL_MED=-2.3] autolearn=disabled Received: from olra.theworths.org ([127.0.0.1]) by localhost (olra.theworths.org [127.0.0.1]) (amavisd-new, port 10024) with ESMTP id A3Tmg-UfTR+I for ; Sun, 18 Dec 2011 05:15:49 -0800 (PST) Received: from tempo.its.unb.ca (tempo.its.unb.ca [131.202.1.21]) (using TLSv1 with cipher DHE-RSA-AES256-SHA (256/256 bits)) (No client certificate requested) by olra.theworths.org (Postfix) with ESMTPS id 5A65C429E32 for ; Sun, 18 Dec 2011 05:15:43 -0800 (PST) Received: from zancas.localnet (fctnnbsc36w-156034079193.pppoe-dynamic.High-Speed.nb.bellaliant.net [156.34.79.193]) (authenticated bits=0) by tempo.its.unb.ca (8.13.8/8.13.8) with ESMTP id pBIDFdgE017881 (version=TLSv1/SSLv3 cipher=AES256-SHA bits=256 verify=NO); Sun, 18 Dec 2011 09:15:40 -0400 Received: from bremner by zancas.localnet with local (Exim 4.77) (envelope-from ) id 1RcGaY-0008RS-S1; Sun, 18 Dec 2011 09:15:38 -0400 From: David Bremner To: notmuch@notmuchmail.org Subject: [PATCH Draft 2 6/9] notmuch-restore: add --format=notmuch support Date: Sun, 18 Dec 2011 09:15:08 -0400 Message-Id: <1324214111-32079-7-git-send-email-david@tethera.net> X-Mailer: git-send-email 1.7.7.3 In-Reply-To: <1324214111-32079-1-git-send-email-david@tethera.net> References: <1324214111-32079-1-git-send-email-david@tethera.net> Cc: David Bremner X-BeenThere: notmuch@notmuchmail.org X-Mailman-Version: 2.1.13 Precedence: list List-Id: "Use and development of the notmuch mail system." List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , X-List-Received-Date: Sun, 18 Dec 2011 13:15:51 -0000 From: David Bremner This is format is whitespace separated tokens, encoded by util/hex-escape.c --- notmuch-restore.c | 81 +++++++++++++++++++++++++++++++++++++++++------------ 1 files changed, 63 insertions(+), 18 deletions(-) diff --git a/notmuch-restore.c b/notmuch-restore.c index 87d9772..f618b47 100644 --- a/notmuch-restore.c +++ b/notmuch-restore.c @@ -19,6 +19,7 @@ */ #include "notmuch-client.h" +#include "dump-restore-private.h" int notmuch_restore_command (unused (void *ctx), int argc, char *argv[]) @@ -35,6 +36,7 @@ notmuch_restore_command (unused (void *ctx), int argc, char *argv[]) regex_t regex; int rerr; int opt_index; + int input_format = DUMP_FORMAT_SUP; config = notmuch_config_open (ctx, NULL, NULL); if (config == NULL) @@ -48,6 +50,7 @@ notmuch_restore_command (unused (void *ctx), int argc, char *argv[]) synchronize_flags = notmuch_config_get_maildir_synchronize_flags (config); notmuch_opt_desc_t options[] = { + FORMAT_DESC (input_format), { NOTMUCH_OPT_POSITION, &input_file_name, 0, 0, 0 }, { NOTMUCH_OPT_BOOLEAN, &accumulate, "accumulate", 'a', 0 }, { 0, 0, 0, 0, 0 } @@ -81,33 +84,63 @@ notmuch_restore_command (unused (void *ctx), int argc, char *argv[]) * non-space characters for the message-id, then one or more * spaces, then a list of space-separated tags as a sequence of * characters within literal '(' and ')'. */ - if ( xregcomp (®ex, - "^([^ ]+) \\(([^)]*)\\)$", - REG_EXTENDED) ) - INTERNAL_ERROR("compile time constant regex failed."); + if (input_format == DUMP_FORMAT_SUP) { + if ( xregcomp (®ex, + "^([^ ]+) \\(([^)]*)\\)$", + REG_EXTENDED) ) + INTERNAL_ERROR("compile time constant regex failed."); + } + + + /* These are out here to re-use the buffers with hex_decode */ + + char *message_id = NULL; + size_t message_id_size = 0; + char *tag = NULL; + size_t tag_size = 0; while ((line_len = getline (&line, &line_size, input)) != -1) { regmatch_t match[3]; - char *message_id, *file_tags, *tag, *next; + char *file_tags, *next; notmuch_message_t *message = NULL; + notmuch_status_t status; notmuch_tags_t *db_tags; char *db_tags_str; chomp_newline (line); - rerr = xregexec (®ex, line, 3, match, 0); - if (rerr == REG_NOMATCH) - { - fprintf (stderr, "Warning: Ignoring invalid input line: %s\n", - line); + /* Silently ignore blank lines */ + + if (line[0] == '\0') { continue; } - message_id = xstrndup (line + match[1].rm_so, - match[1].rm_eo - match[1].rm_so); - file_tags = xstrndup (line + match[2].rm_so, - match[2].rm_eo - match[2].rm_so); + if (input_format == DUMP_FORMAT_SUP) { + rerr = xregexec (®ex, line, 3, match, 0); + if (rerr == REG_NOMATCH) + { + fprintf (stderr, "Warning: Ignoring invalid input line: %s\n", + line); + continue; + } + message_id = talloc_strndup (notmuch, line + match[1].rm_so, + match[1].rm_eo - match[1].rm_so); + file_tags = talloc_strndup (notmuch, line + match[2].rm_so, + match[2].rm_eo - match[2].rm_so); + } else { + char *p = line; + char *raw_mid; + + raw_mid = strsep (&p, " \t"); + + if (hex_decode (notmuch, raw_mid, + &message_id, &message_id_size) != HEX_SUCCESS) + return 1; + + file_tags = xstrdup (p); + } + status = notmuch_database_find_message (notmuch, message_id, &message); if (status || message == NULL) { @@ -153,7 +186,16 @@ notmuch_restore_command (unused (void *ctx), int argc, char *argv[]) next = file_tags; while (next) { - tag = strsep (&next, " "); + char *raw_tag = strsep (&next, " "); + + if (input_format == DUMP_FORMAT_NOTMUCH) { + if (hex_decode (notmuch, raw_tag, + &tag, &tag_size) != HEX_SUCCESS) + return 1; + } else { + tag = talloc_strdup (notmuch, raw_tag); + } + if (*tag == '\0') continue; status = notmuch_message_add_tag (message, tag); @@ -175,11 +217,14 @@ notmuch_restore_command (unused (void *ctx), int argc, char *argv[]) if (message) notmuch_message_destroy (message); message = NULL; - free (message_id); - free (file_tags); + if (input_format == DUMP_FORMAT_SUP) { + talloc_free (message_id); + talloc_free (file_tags); + } } - regfree (®ex); + if (input_format == DUMP_FORMAT_SUP) + regfree (®ex); if (line) free (line); -- 1.7.7.3