emacs: create notmuch-tag.el, and move appropriate functions from notmuch.el
[notmuch] / emacs / notmuch-tag.el
1 ;; notmuch-tag.el --- tag messages within emacs
2 ;;
3 ;; Copyright © Carl Worth
4 ;;
5 ;; This file is part of Notmuch.
6 ;;
7 ;; Notmuch is free software: you can redistribute it and/or modify it
8 ;; under the terms of the GNU General Public License as published by
9 ;; the Free Software Foundation, either version 3 of the License, or
10 ;; (at your option) any later version.
11 ;;
12 ;; Notmuch is distributed in the hope that it will be useful, but
13 ;; WITHOUT ANY WARRANTY; without even the implied warranty of
14 ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15 ;; General Public License for more details.
16 ;;
17 ;; You should have received a copy of the GNU General Public License
18 ;; along with Notmuch.  If not, see <http://www.gnu.org/licenses/>.
19 ;;
20 ;; Authors: Carl Worth <cworth@cworth.org>
21
22 (eval-when-compile (require 'cl))
23 (require 'crm)
24 (require 'notmuch-lib)
25
26 (defcustom notmuch-before-tag-hook nil
27   "Hooks that are run before tags of a message are modified.
28
29 'tags' will contain the tags that are about to be added or removed as
30 a list of strings of the form \"+TAG\" or \"-TAG\".
31 'query' will be a string containing the search query that determines
32 the messages that are about to be tagged"
33
34   :type 'hook
35   :options '(notmuch-hl-line-mode)
36   :group 'notmuch-hooks)
37
38 (defcustom notmuch-after-tag-hook nil
39   "Hooks that are run after tags of a message are modified.
40
41 'tags' will contain the tags that were added or removed as
42 a list of strings of the form \"+TAG\" or \"-TAG\".
43 'query' will be a string containing the search query that determines
44 the messages that were tagged"
45   :type 'hook
46   :options '(notmuch-hl-line-mode)
47   :group 'notmuch-hooks)
48
49 (defvar notmuch-select-tag-history nil
50   "Variable to store minibuffer history for
51 `notmuch-select-tag-with-completion' function.")
52
53 (defvar notmuch-read-tag-changes-history nil
54   "Variable to store minibuffer history for
55 `notmuch-read-tag-changes' function.")
56
57 (defun notmuch-tag-completions (&optional search-terms)
58   (if (null search-terms)
59       (setq search-terms (list "*")))
60   (split-string
61    (with-output-to-string
62      (with-current-buffer standard-output
63        (apply 'call-process notmuch-command nil t
64               nil "search" "--output=tags" "--exclude=false" search-terms)))
65    "\n+" t))
66
67 (defun notmuch-select-tag-with-completion (prompt &rest search-terms)
68   (let ((tag-list (notmuch-tag-completions search-terms)))
69     (completing-read prompt tag-list nil nil nil 'notmuch-select-tag-history)))
70
71 (defun notmuch-read-tag-changes (&optional initial-input &rest search-terms)
72   (let* ((all-tag-list (notmuch-tag-completions))
73          (add-tag-list (mapcar (apply-partially 'concat "+") all-tag-list))
74          (remove-tag-list (mapcar (apply-partially 'concat "-")
75                                   (if (null search-terms)
76                                       all-tag-list
77                                     (notmuch-tag-completions search-terms))))
78          (tag-list (append add-tag-list remove-tag-list))
79          (crm-separator " ")
80          ;; By default, space is bound to "complete word" function.
81          ;; Re-bind it to insert a space instead.  Note that <tab>
82          ;; still does the completion.
83          (crm-local-completion-map
84           (let ((map (make-sparse-keymap)))
85             (set-keymap-parent map crm-local-completion-map)
86             (define-key map " " 'self-insert-command)
87             map)))
88     (delete "" (completing-read-multiple "Tags (+add -drop): "
89                 tag-list nil nil initial-input
90                 'notmuch-read-tag-changes-history))))
91
92 (defun notmuch-update-tags (tags tag-changes)
93   "Return a copy of TAGS with additions and removals from TAG-CHANGES.
94
95 TAG-CHANGES must be a list of tags names, each prefixed with
96 either a \"+\" to indicate the tag should be added to TAGS if not
97 present or a \"-\" to indicate that the tag should be removed
98 from TAGS if present."
99   (let ((result-tags (copy-sequence tags)))
100     (dolist (tag-change tag-changes)
101       (let ((op (string-to-char tag-change))
102             (tag (unless (string= tag-change "") (substring tag-change 1))))
103         (case op
104           (?+ (unless (member tag result-tags)
105                 (push tag result-tags)))
106           (?- (setq result-tags (delete tag result-tags)))
107           (otherwise
108            (error "Changed tag must be of the form `+this_tag' or `-that_tag'")))))
109     (sort result-tags 'string<)))
110
111 (defun notmuch-tag (query &rest tag-changes)
112   "Add/remove tags in TAG-CHANGES to messages matching QUERY.
113
114 TAG-CHANGES should be a list of strings of the form \"+tag\" or
115 \"-tag\" and QUERY should be a string containing the
116 search-query.
117
118 Note: Other code should always use this function alter tags of
119 messages instead of running (notmuch-call-notmuch-process \"tag\" ..)
120 directly, so that hooks specified in notmuch-before-tag-hook and
121 notmuch-after-tag-hook will be run."
122   ;; Perform some validation
123   (mapc (lambda (tag-change)
124           (unless (string-match-p "^[-+]\\S-+$" tag-change)
125             (error "Tag must be of the form `+this_tag' or `-that_tag'")))
126         tag-changes)
127   (unless (null tag-changes)
128     (run-hooks 'notmuch-before-tag-hook)
129     (apply 'notmuch-call-notmuch-process "tag"
130            (append tag-changes (list "--" query)))
131     (run-hooks 'notmuch-after-tag-hook)))
132
133 ;;
134
135 (provide 'notmuch-tag)