[PATCH 09/16] cli: add support for batch tagging operations to "notmuch tag"
authordavid <david@tethera.net>
Sun, 18 Nov 2012 19:04:51 +0000 (15:04 +2000)
committerW. Trevor King <wking@tremily.us>
Fri, 7 Nov 2014 17:50:42 +0000 (09:50 -0800)
45/0b4bfbfde44ef1adf26bcb83f92e124a3b692b [new file with mode: 0644]

diff --git a/45/0b4bfbfde44ef1adf26bcb83f92e124a3b692b b/45/0b4bfbfde44ef1adf26bcb83f92e124a3b692b
new file mode 100644 (file)
index 0000000..5d8b7d6
--- /dev/null
@@ -0,0 +1,342 @@
+Return-Path: <bremner@tethera.net>\r
+X-Original-To: notmuch@notmuchmail.org\r
+Delivered-To: notmuch@notmuchmail.org\r
+Received: from localhost (localhost [127.0.0.1])\r
+       by olra.theworths.org (Postfix) with ESMTP id E40F1431FBF\r
+       for <notmuch@notmuchmail.org>; Sun, 18 Nov 2012 11:05:38 -0800 (PST)\r
+X-Virus-Scanned: Debian amavisd-new at olra.theworths.org\r
+X-Spam-Flag: NO\r
+X-Spam-Score: 0\r
+X-Spam-Level: \r
+X-Spam-Status: No, score=0 tagged_above=-999 required=5 tests=[none]\r
+       autolearn=disabled\r
+Received: from olra.theworths.org ([127.0.0.1])\r
+       by localhost (olra.theworths.org [127.0.0.1]) (amavisd-new, port 10024)\r
+       with ESMTP id Sfnp1vpZ3h44 for <notmuch@notmuchmail.org>;\r
+       Sun, 18 Nov 2012 11:05:36 -0800 (PST)\r
+Received: from tesseract.cs.unb.ca (tesseract.cs.unb.ca [131.202.240.238])\r
+       (using TLSv1 with cipher AES256-SHA (256/256 bits))\r
+       (No client certificate requested)\r
+       by olra.theworths.org (Postfix) with ESMTPS id EB9CB429E26\r
+       for <notmuch@notmuchmail.org>; Sun, 18 Nov 2012 11:05:19 -0800 (PST)\r
+Received: from fctnnbsc30w-156034089108.dhcp-dynamic.fibreop.nb.bellaliant.net\r
+       ([156.34.89.108] helo=zancas.localnet)\r
+       by tesseract.cs.unb.ca with esmtpsa\r
+       (TLS1.0:DHE_RSA_AES_128_CBC_SHA1:16) (Exim 4.72)\r
+       (envelope-from <bremner@tethera.net>)\r
+       id 1TaABC-000289-8T; Sun, 18 Nov 2012 15:05:18 -0400\r
+Received: from bremner by zancas.localnet with local (Exim 4.80)\r
+       (envelope-from <bremner@tethera.net>)\r
+       id 1TaAB6-0001Io-O1; Sun, 18 Nov 2012 15:05:12 -0400\r
+From: david@tethera.net\r
+To: notmuch@notmuchmail.org\r
+Subject: [PATCH 09/16] cli: add support for batch tagging operations to\r
+       "notmuch tag"\r
+Date: Sun, 18 Nov 2012 15:04:51 -0400\r
+Message-Id: <1353265498-3839-10-git-send-email-david@tethera.net>\r
+X-Mailer: git-send-email 1.7.10.4\r
+In-Reply-To: <1353265498-3839-1-git-send-email-david@tethera.net>\r
+References: <1353265498-3839-1-git-send-email-david@tethera.net>\r
+X-Spam_bar: -\r
+X-BeenThere: notmuch@notmuchmail.org\r
+X-Mailman-Version: 2.1.13\r
+Precedence: list\r
+List-Id: "Use and development of the notmuch mail system."\r
+       <notmuch.notmuchmail.org>\r
+List-Unsubscribe: <http://notmuchmail.org/mailman/options/notmuch>,\r
+       <mailto:notmuch-request@notmuchmail.org?subject=unsubscribe>\r
+List-Archive: <http://notmuchmail.org/pipermail/notmuch>\r
+List-Post: <mailto:notmuch@notmuchmail.org>\r
+List-Help: <mailto:notmuch-request@notmuchmail.org?subject=help>\r
+List-Subscribe: <http://notmuchmail.org/mailman/listinfo/notmuch>,\r
+       <mailto:notmuch-request@notmuchmail.org?subject=subscribe>\r
+X-List-Received-Date: Sun, 18 Nov 2012 19:05:39 -0000\r
+\r
+From: Jani Nikula <jani@nikula.org>\r
+\r
+Add support for batch tagging operations through stdin to "notmuch\r
+tag". This can be enabled with the new --stdin command line option to\r
+"notmuch new". The input must consist of lines of the format:\r
+\r
++<tag>|-<tag> [...] [--] <search-terms>\r
+\r
+Each line is interpreted similarly to "notmuch tag" command line\r
+arguments. The delimiter is one or more spaces ' '. Any characters in\r
+<tag> and <search-terms> MAY be hex encoded with %NN where NN is the\r
+hexadecimal value of the character. Any ' ' and '%' characters in\r
+<tag> and <search-terms> MUST be hex encoded (using %20 and %25,\r
+respectively). Any characters that are not part of <tag> or\r
+<search-terms> MUST NOT be hex encoded.\r
+\r
+Leading and trailing space ' ' is ignored. Empty lines and lines\r
+beginning with '#' are ignored.\r
+\r
+Signed-off-by: Jani Nikula <jani@nikula.org>\r
+\r
+Hacked-like-crazy-by: David Bremner <david@tethera.net>\r
+---\r
+ notmuch-tag.c |  177 +++++++++++++++++++++++++++++++++++----------------------\r
+ 1 file changed, 109 insertions(+), 68 deletions(-)\r
+\r
+diff --git a/notmuch-tag.c b/notmuch-tag.c\r
+index 88d559b..ca120d5 100644\r
+--- a/notmuch-tag.c\r
++++ b/notmuch-tag.c\r
+@@ -19,6 +19,7 @@\r
+  */\r
+ \r
+ #include "notmuch-client.h"\r
++#include "tag-util.h"\r
+ \r
+ static volatile sig_atomic_t interrupted;\r
+ \r
+@@ -54,14 +55,9 @@ _escape_tag (char *buf, const char *tag)\r
+     return buf;\r
+ }\r
+ \r
+-typedef struct {\r
+-    const char *tag;\r
+-    notmuch_bool_t remove;\r
+-} tag_operation_t;\r
+-\r
+ static char *\r
+ _optimize_tag_query (void *ctx, const char *orig_query_string,\r
+-                   const tag_operation_t *tag_ops)\r
++                   const tag_op_list_t *list)\r
+ {\r
+     /* This is subtler than it looks.  Xapian ignores the '-' operator\r
+      * at the beginning both queries and parenthesized groups and,\r
+@@ -75,6 +71,7 @@ _optimize_tag_query (void *ctx, const char *orig_query_string,\r
+     const char *join = "";\r
+     int i;\r
+     unsigned int max_tag_len = 0;\r
++    const tag_operation_t *tag_ops = list->ops;\r
+ \r
+     /* Don't optimize if there are no tag changes. */\r
+     if (tag_ops[0].tag == NULL)\r
+@@ -116,12 +113,11 @@ _optimize_tag_query (void *ctx, const char *orig_query_string,\r
+  * element. */\r
+ static int\r
+ tag_query (void *ctx, notmuch_database_t *notmuch, const char *query_string,\r
+-         tag_operation_t *tag_ops, notmuch_bool_t synchronize_flags)\r
++         tag_op_list_t *tag_ops, tag_op_flag_t flags)\r
+ {\r
+     notmuch_query_t *query;\r
+     notmuch_messages_t *messages;\r
+     notmuch_message_t *message;\r
+-    int i;\r
+ \r
+     /* Optimize the query so it excludes messages that already have\r
+      * the specified set of tags. */\r
+@@ -144,21 +140,7 @@ tag_query (void *ctx, notmuch_database_t *notmuch, const char *query_string,\r
+        notmuch_messages_valid (messages) && ! interrupted;\r
+        notmuch_messages_move_to_next (messages)) {\r
+       message = notmuch_messages_get (messages);\r
+-\r
+-      notmuch_message_freeze (message);\r
+-\r
+-      for (i = 0; tag_ops[i].tag; i++) {\r
+-          if (tag_ops[i].remove)\r
+-              notmuch_message_remove_tag (message, tag_ops[i].tag);\r
+-          else\r
+-              notmuch_message_add_tag (message, tag_ops[i].tag);\r
+-      }\r
+-\r
+-      notmuch_message_thaw (message);\r
+-\r
+-      if (synchronize_flags)\r
+-          notmuch_message_tags_to_maildir_flags (message);\r
+-\r
++      tag_op_list_apply (message, tag_ops, flags);\r
+       notmuch_message_destroy (message);\r
+     }\r
+ \r
+@@ -170,15 +152,17 @@ tag_query (void *ctx, notmuch_database_t *notmuch, const char *query_string,\r
+ int\r
+ notmuch_tag_command (void *ctx, int argc, char *argv[])\r
+ {\r
+-    tag_operation_t *tag_ops;\r
+-    int tag_ops_count = 0;\r
+-    char *query_string;\r
++    tag_op_list_t *tag_ops = NULL;\r
++    char *query_string = NULL;\r
+     notmuch_config_t *config;\r
+     notmuch_database_t *notmuch;\r
+     struct sigaction action;\r
+-    notmuch_bool_t synchronize_flags;\r
+-    int i;\r
+-    int ret;\r
++    tag_op_flag_t synchronize_flags = TAG_FLAG_NONE;\r
++    notmuch_bool_t batch = FALSE;\r
++    FILE *input = stdin;\r
++    char *input_file_name = NULL;\r
++    int i, opt_index;\r
++    int ret = 0;\r
+ \r
+     /* Setup our handler for SIGINT */\r
+     memset (&action, 0, sizeof (struct sigaction));\r
+@@ -187,53 +171,76 @@ notmuch_tag_command (void *ctx, int argc, char *argv[])\r
+     action.sa_flags = SA_RESTART;\r
+     sigaction (SIGINT, &action, NULL);\r
+ \r
+-    argc--; argv++; /* skip subcommand argument */\r
++    notmuch_opt_desc_t options[] = {\r
++      { NOTMUCH_OPT_BOOLEAN, &batch, "batch", 0, 0 },\r
++      { NOTMUCH_OPT_STRING, &input_file_name, "input", 'i', 0 },\r
++      { 0, 0, 0, 0, 0 }\r
++    };\r
+ \r
+-    /* Array of tagging operations (add or remove), terminated with an\r
+-     * empty element. */\r
+-    tag_ops = talloc_array (ctx, tag_operation_t, argc + 1);\r
+-    if (tag_ops == NULL) {\r
+-      fprintf (stderr, "Out of memory.\n");\r
++    opt_index = parse_arguments (argc, argv, options, 1);\r
++    if (opt_index < 0)\r
+       return 1;\r
++\r
++    if (input_file_name) {\r
++      batch = TRUE;\r
++      input = fopen (input_file_name, "r");\r
++      if (input == NULL) {\r
++          fprintf (stderr, "Error opening %s for reading: %s\n",\r
++                   input_file_name, strerror (errno));\r
++          return 1;\r
++      }\r
+     }\r
+ \r
+-    for (i = 0; i < argc; i++) {\r
+-      if (strcmp (argv[i], "--") == 0) {\r
+-          i++;\r
+-          break;\r
++    if (batch) {\r
++      if (opt_index != argc) {\r
++          fprintf (stderr, "Can't specify both cmdline and stdin!\n");\r
++          return 1;\r
+       }\r
+-      if (argv[i][0] == '+' || argv[i][0] == '-') {\r
+-          if (argv[i][0] == '+' && argv[i][1] == '\0') {\r
+-              fprintf (stderr, "Error: tag names cannot be empty.\n");\r
+-              return 1;\r
++    } else {\r
++      /* Array of tagging operations (add or remove), terminated with an\r
++       * empty element. */\r
++      tag_ops = tag_op_list_create (ctx);\r
++      if (tag_ops == NULL) {\r
++          fprintf (stderr, "Out of memory.\n");\r
++          return 1;\r
++      }\r
++\r
++      for (i = opt_index; i < argc; i++) {\r
++          if (strcmp (argv[i], "--") == 0) {\r
++              i++;\r
++              break;\r
+           }\r
+-          if (argv[i][0] == '+' && argv[i][1] == '-') {\r
+-              /* This disallows adding the non-removable tag "-" and\r
+-               * enables notmuch tag to take long options in the\r
+-               * future. */\r
+-              fprintf (stderr, "Error: tag names must not start with '-'.\n");\r
+-              return 1;\r
++          if (argv[i][0] == '+' || argv[i][0] == '-') {\r
++              /* FIXME: checks should be consistent with those in batch tagging */\r
++              if (argv[i][0] == '+' && argv[i][1] == '\0') {\r
++                  fprintf (stderr, "Error: tag names cannot be empty.\n");\r
++                  return 1;\r
++              }\r
++              if (argv[i][0] == '+' && argv[i][1] == '-') {\r
++                  /* This disallows adding the non-removable tag "-" and\r
++                   * enables notmuch tag to take long options in the\r
++                   * future. */\r
++                  fprintf (stderr, "Error: tag names must not start with '-'.\n");\r
++                  return 1;\r
++              }\r
++              tag_op_list_append (ctx, tag_ops,\r
++                                  argv[i] + 1, (argv[i][0] == '-'));\r
++          } else {\r
++              break;\r
+           }\r
+-          tag_ops[tag_ops_count].tag = argv[i] + 1;\r
+-          tag_ops[tag_ops_count].remove = (argv[i][0] == '-');\r
+-          tag_ops_count++;\r
+-      } else {\r
+-          break;\r
+       }\r
+-    }\r
+-\r
+-    tag_ops[tag_ops_count].tag = NULL;\r
+ \r
+-    if (tag_ops_count == 0) {\r
+-      fprintf (stderr, "Error: 'notmuch tag' requires at least one tag to add or remove.\n");\r
+-      return 1;\r
+-    }\r
++      if (tag_op_list_empty (tag_ops)) {\r
++          fprintf (stderr, "Error: 'notmuch tag' requires at least one tag to add or remove.\n");\r
++          return 1;\r
++      }\r
+ \r
+-    query_string = query_string_from_args (ctx, argc - i, &argv[i]);\r
++      query_string = query_string_from_args (ctx, argc - i, &argv[i]);\r
+ \r
+-    if (*query_string == '\0') {\r
+-      fprintf (stderr, "Error: notmuch tag requires at least one search term.\n");\r
+-      return 1;\r
++      if (*query_string == '\0') {\r
++          fprintf (stderr, "Error: notmuch tag requires at least one search term.\n");\r
++          return 1;\r
++      }\r
+     }\r
+ \r
+     config = notmuch_config_open (ctx, NULL, NULL);\r
+@@ -244,11 +251,45 @@ notmuch_tag_command (void *ctx, int argc, char *argv[])\r
+                              NOTMUCH_DATABASE_MODE_READ_WRITE, &notmuch))\r
+       return 1;\r
+ \r
+-    synchronize_flags = notmuch_config_get_maildir_synchronize_flags (config);\r
++    if (notmuch_config_get_maildir_synchronize_flags (config))\r
++      synchronize_flags = TAG_FLAG_MAILDIR_SYNC;\r
++\r
++    if (batch) {\r
++\r
++      char *line = NULL;\r
++      size_t line_size = 0;\r
++      ssize_t line_len;\r
++      int ret = 0;\r
++      tag_op_list_t *tag_ops;\r
++\r
++      tag_ops = tag_op_list_create (ctx);\r
++      if (tag_ops == NULL) {\r
++          fprintf (stderr, "Out of memory.\n");\r
++          return 1;\r
++      }\r
++\r
++      while ((line_len = getline (&line, &line_size, input)) != -1 &&\r
++             ! interrupted) {\r
++\r
++          char *query_string;\r
++\r
++          ret =  parse_tag_line (ctx, line, &query_string, tag_ops);\r
+ \r
+-    ret = tag_query (ctx, notmuch, query_string, tag_ops, synchronize_flags);\r
++          if (ret > 0)\r
++              continue;\r
++\r
++          if (ret < 0 || tag_query (ctx, notmuch, query_string,\r
++                                    tag_ops, synchronize_flags))\r
++              break;\r
++      }\r
++\r
++      if (line)\r
++          free (line);\r
++    } else\r
++      ret = tag_query (ctx, notmuch, query_string, tag_ops, synchronize_flags);\r
+ \r
+     notmuch_database_destroy (notmuch);\r
+ \r
+-    return ret;\r
++    return ret || interrupted;\r
++\r
+ }\r
+-- \r
+1.7.10.4\r
+\r