From edd515d19b3dc921436d0ee06910aee44449988c Mon Sep 17 00:00:00 2001 From: David Bremner Date: Sun, 13 Mar 2016 08:31:29 +2000 Subject: [PATCH] [PATCH 5/6] CLI: optionally restore config data. --- 94/de0ea3123b76e07ae67ed0c7703137843282a7 | 174 ++++++++++++++++++++++ 1 file changed, 174 insertions(+) create mode 100644 94/de0ea3123b76e07ae67ed0c7703137843282a7 diff --git a/94/de0ea3123b76e07ae67ed0c7703137843282a7 b/94/de0ea3123b76e07ae67ed0c7703137843282a7 new file mode 100644 index 000000000..efb540c01 --- /dev/null +++ b/94/de0ea3123b76e07ae67ed0c7703137843282a7 @@ -0,0 +1,174 @@ +Return-Path: +X-Original-To: notmuch@notmuchmail.org +Delivered-To: notmuch@notmuchmail.org +Received: from localhost (localhost [127.0.0.1]) + by arlo.cworth.org (Postfix) with ESMTP id 7CF0C6DE189C + for ; Sat, 12 Mar 2016 04:31:40 -0800 (PST) +X-Virus-Scanned: Debian amavisd-new at cworth.org +X-Spam-Flag: NO +X-Spam-Score: -0.033 +X-Spam-Level: +X-Spam-Status: No, score=-0.033 tagged_above=-999 required=5 + tests=[AWL=-0.022, SPF_PASS=-0.001, T_RP_MATCHES_RCVD=-0.01] + autolearn=disabled +Received: from arlo.cworth.org ([127.0.0.1]) + by localhost (arlo.cworth.org [127.0.0.1]) (amavisd-new, port 10024) + with ESMTP id bcW7UzuvYYzg for ; + Sat, 12 Mar 2016 04:31:38 -0800 (PST) +Received: from fethera.tethera.net (fethera.tethera.net [198.245.60.197]) + by arlo.cworth.org (Postfix) with ESMTPS id 1D4606DE1534 + for ; Sat, 12 Mar 2016 04:31:37 -0800 (PST) +Received: from remotemail by fethera.tethera.net with local (Exim 4.84) + (envelope-from ) + id 1aeiiL-0004yp-Qv; Sat, 12 Mar 2016 07:32:13 -0500 +Received: (nullmailer pid 17183 invoked by uid 1000); + Sat, 12 Mar 2016 12:31:33 -0000 +From: David Bremner +To: David Bremner , notmuch@notmuchmail.org +Subject: [PATCH 5/6] CLI: optionally restore config data. +Date: Sat, 12 Mar 2016 08:31:29 -0400 +Message-Id: <1457785890-17058-6-git-send-email-david@tethera.net> +X-Mailer: git-send-email 2.7.0 +In-Reply-To: <1457785890-17058-1-git-send-email-david@tethera.net> +References: <1453561198-2893-1-git-send-email-david@tethera.net> + <1457785890-17058-1-git-send-email-david@tethera.net> +X-BeenThere: notmuch@notmuchmail.org +X-Mailman-Version: 2.1.20 +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: Sat, 12 Mar 2016 12:31:40 -0000 + +Like the corresponding change to dump, this needs documentation +--- + notmuch-restore.c | 53 ++++++++++++++++++++++++++++++++++++++++++++++++++ + test/T590-libconfig.sh | 11 +++++++++++ + 2 files changed, 64 insertions(+) + +diff --git a/notmuch-restore.c b/notmuch-restore.c +index 9abc64f..e06fbde 100644 +--- a/notmuch-restore.c ++++ b/notmuch-restore.c +@@ -24,6 +24,38 @@ + #include "string-util.h" + #include "zlib-extra.h" + ++static int ++process_config_line(notmuch_database_t *notmuch, const char* line){ ++ const char *key_p, *val_p; ++ char *key, *val; ++ size_t key_len,val_len; ++ const char *delim=" \t\n"; ++ int ret = EXIT_FAILURE; ++ ++ void *local = talloc_new(NULL); ++ ++ key_p = strtok_len_c(line, delim, &key_len); ++ val_p = strtok_len_c(key_p+key_len, delim, &val_len); ++ ++ key = talloc_strndup(local, key_p, key_len); ++ val = talloc_strndup(local, val_p, val_len); ++ if (hex_decode_inplace (key) != HEX_SUCCESS || ++ hex_decode_inplace (val) != HEX_SUCCESS ) { ++ fprintf (stderr, "hex decoding failure on line %s\n", line); ++ goto DONE; ++ } ++ ++ if (print_status_database ("notmuch restore", notmuch, ++ notmuch_database_set_config (notmuch, key, val))) ++ goto DONE; ++ ++ ret = EXIT_SUCCESS; ++ ++ DONE: ++ talloc_free (local); ++ return ret; ++} ++ + static regex_t regex; + + /* Non-zero return indicates an error in retrieving the message, +@@ -137,6 +169,7 @@ notmuch_restore_command (notmuch_config_t *config, int argc, char *argv[]) + + int ret = 0; + int opt_index; ++ int include=0; + int input_format = DUMP_FORMAT_AUTO; + + if (notmuch_database_open (notmuch_config_get_database_path (config), +@@ -152,6 +185,10 @@ notmuch_restore_command (notmuch_config_t *config, int argc, char *argv[]) + { "batch-tag", DUMP_FORMAT_BATCH_TAG }, + { "sup", DUMP_FORMAT_SUP }, + { 0, 0 } } }, ++ { NOTMUCH_OPT_KEYWORD_FLAGS, &include, "include", 'I', ++ (notmuch_keyword_t []){ { "config", DUMP_INCLUDE_CONFIG }, ++ { "tags", DUMP_INCLUDE_TAGS} } }, ++ + { NOTMUCH_OPT_STRING, &input_file_name, "input", 'i', 0 }, + { NOTMUCH_OPT_BOOLEAN, &accumulate, "accumulate", 'a', 0 }, + { NOTMUCH_OPT_INHERIT, (void *) ¬much_shared_options, NULL, 0, 0 }, +@@ -167,6 +204,10 @@ notmuch_restore_command (notmuch_config_t *config, int argc, char *argv[]) + notmuch_process_shared_options (argv[0]); + notmuch_exit_if_unmatched_db_uuid (notmuch); + ++ if (include == 0) { ++ include = DUMP_INCLUDE_CONFIG | DUMP_INCLUDE_TAGS; ++ } ++ + name_for_error = input_file_name ? input_file_name : "stdin"; + + if (! accumulate) +@@ -225,11 +266,23 @@ notmuch_restore_command (notmuch_config_t *config, int argc, char *argv[]) + ret = EXIT_FAILURE; + goto DONE; + } ++ ++ if ((include & DUMP_INCLUDE_CONFIG) && line_len >= 2 && line[0] == '#' && line[1] == '@') { ++ ret = process_config_line(notmuch, line+2); ++ if (ret) ++ goto DONE; ++ } ++ + } while ((line_len == 0) || + (line[0] == '#') || + /* the cast is safe because we checked about for line_len < 0 */ + (strspn (line, " \t\n") == (unsigned)line_len)); + ++ if (! (include & DUMP_INCLUDE_TAGS)) { ++ ret = EXIT_SUCCESS; ++ goto DONE; ++ } ++ + char *p; + for (p = line; (input_format == DUMP_FORMAT_AUTO) && *p; p++) { + if (*p == '(') +diff --git a/test/T590-libconfig.sh b/test/T590-libconfig.sh +index 5ea5300..9c1e566 100755 +--- a/test/T590-libconfig.sh ++++ b/test/T590-libconfig.sh +@@ -132,4 +132,15 @@ cat <<'EOF' >EXPECTED + EOF + test_expect_equal_file EXPECTED OUTPUT + ++test_begin_subtest "restore config" ++notmuch dump --include=config >EXPECTED ++cat c_head - c_tail <<'EOF' | test_C ${MAIL_DIR} ++{ ++ RUN(notmuch_database_set_config (db, "testkey1", "mutatedvalue")); ++} ++EOF ++notmuch restore --include=config OUTPUT ++test_expect_equal_file EXPECTED OUTPUT ++ + test_done +-- +2.7.0 + -- 2.26.2