From 2fbf80ea2fd946fb64accbfd0a2a91b57b05b882 Mon Sep 17 00:00:00 2001 From: David Bremner Date: Sun, 24 Jan 2016 10:59:58 +2000 Subject: [PATCH] [PATCH 5/5] CLI: optionally restore config data. --- c4/3d79b934d4dad71c5c84ab680421cfd9e2f596 | 172 ++++++++++++++++++++++ 1 file changed, 172 insertions(+) create mode 100644 c4/3d79b934d4dad71c5c84ab680421cfd9e2f596 diff --git a/c4/3d79b934d4dad71c5c84ab680421cfd9e2f596 b/c4/3d79b934d4dad71c5c84ab680421cfd9e2f596 new file mode 100644 index 000000000..35b10ebe5 --- /dev/null +++ b/c4/3d79b934d4dad71c5c84ab680421cfd9e2f596 @@ -0,0 +1,172 @@ +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 DBA066DE137E + for ; Sat, 23 Jan 2016 07:00:14 -0800 (PST) +X-Virus-Scanned: Debian amavisd-new at cworth.org +X-Spam-Flag: NO +X-Spam-Score: -0.31 +X-Spam-Level: +X-Spam-Status: No, score=-0.31 tagged_above=-999 required=5 tests=[AWL=0.241, + RP_MATCHES_RCVD=-0.55, SPF_PASS=-0.001] 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 aALqoui491bC for ; + Sat, 23 Jan 2016 07:00:13 -0800 (PST) +Received: from fethera.tethera.net (fethera.tethera.net [198.245.60.197]) + by arlo.cworth.org (Postfix) with ESMTPS id 735976DE091B + for ; Sat, 23 Jan 2016 07:00:12 -0800 (PST) +Received: from remotemail by fethera.tethera.net with local (Exim 4.84) + (envelope-from ) + id 1aMzf3-00016G-Au; Sat, 23 Jan 2016 09:59:33 -0500 +Received: (nullmailer pid 4051 invoked by uid 1000); + Sat, 23 Jan 2016 15:00:07 -0000 +From: David Bremner +To: notmuch@notmuchmail.org +Subject: [PATCH 5/5] CLI: optionally restore config data. +Date: Sat, 23 Jan 2016 10:59:58 -0400 +Message-Id: <1453561198-2893-6-git-send-email-david@tethera.net> +X-Mailer: git-send-email 2.6.4 +In-Reply-To: <1453561198-2893-1-git-send-email-david@tethera.net> +References: <1453561198-2893-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, 23 Jan 2016 15:00:14 -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.6.4 + -- 2.26.2