1 files changed, 157 insertions, 0 deletions
diff --git a/templates/.bin/mutt-notmuch-py b/templates/.bin/mutt-notmuch-py
new file mode 100755
index 0000000..40c7a76
--- /dev/null
+++ b/templates/.bin/mutt-notmuch-py
@@ -0,0 +1,157 @@
+#!/usr/bin/env python
+"""
+mutt-notmuch-py
+
+This is a Gmail-only version of the original mutt-notmuch script.
+
+It will interactively ask you for a search query and then symlink the matching
+messages to $HOME/.cache/mutt_results.
+
+Add this to your muttrc.
+
+macro index / "<enter-command>unset wait_key<enter><shell-escape>mutt-notmuch-py<enter><change-folder-readonly>~/.cache/mutt_results<enter>" \
+          "search mail (using notmuch)"
+
+This script overrides the $HOME/.cache/mutt_results each time you run a query.
+
+Install this by adding this file somewhere on your PATH.
+
+Tested on OSX Lion, OSX El Capitan, Fedora 23, and Arch Linux.
+
+(c) 2012-2016 --- Honza Pokorny
+Licensed under BSD
+"""
+
+from __future__ import print_function
+import os
+import hashlib
+
+try:
+    from shlex import quote
+except ImportError:
+    from pipes import quote
+
+import sys
+
+from mailbox import Maildir
+from optparse import OptionParser
+from collections import defaultdict
+import readline
+
+VERSION = '1.3.0'
+
+try:
+    from commands import getoutput
+except ImportError:
+    from subprocess import getoutput
+
+try:
+    input = raw_input
+except NameError:
+    pass
+
+
+def digest(filename):
+    with open(filename, 'rb') as f:
+        return hashlib.sha1(f.read()).hexdigest()
+
+
+def pick_all_mail(messages):
+    for m in messages:
+        if 'All Mail' in m:
+            return m
+
+
+def empty_dir(directory):
+    box = Maildir(directory)
+    box.clear()
+
+
+def command(cmd):
+    return getoutput(cmd)
+
+
+def normalize(path):
+    if path:
+      # Use expanduser() so that os.symlink() won't get weirded out by
+      # tildes.
+      return os.path.expanduser(path).rstrip('/')
+
+
+def main(dest_box, options):
+    is_gmail = options.gmail
+    filter_path = options.base_path
+    history_path = normalize(options.history_path)
+    if history_path and os.path.exists(history_path):
+        readline.read_history_file(history_path)
+    query = input('Query: ')
+    if history_path:
+      readline.write_history_file(history_path)
+
+    command('mkdir -p %s/cur' % dest_box)
+    command('mkdir -p %s/new' % dest_box)
+
+    empty_dir(dest_box)
+
+    files = command('notmuch search --output=files {}'.format(quote(query))).split('\n')
+
+    data = defaultdict(list)
+    messages = []
+
+    for f in files:
+        if not f:
+            continue
+
+        if filter_path is not None and filter_path not in f:
+            continue
+
+        try:
+            sha = digest(f)
+            data[sha].append(f)
+        except IOError:
+            print('File %s does not exist' % f)
+
+    for sha in data:
+        if is_gmail and len(data[sha]) > 1:
+            messages.append(pick_all_mail(data[sha]))
+        else:
+            messages.append(data[sha][0])
+
+    for m in messages:
+        if not m:
+            continue
+
+        target = os.path.join(dest_box, 'cur', os.path.basename(m))
+        if not os.path.exists(target):
+            os.symlink(m, target)
+
+
+if __name__ == '__main__':
+    p = OptionParser("usage: %prog [OPTIONS] [RESULTDIR]")
+    p.add_option('-g', '--gmail', dest='gmail',
+                 action='store_true', default=True,
+                 help='gmail-specific behavior')
+    p.add_option('-G', '--not-gmail', dest='gmail',
+                 action='store_false',
+                 help='Normal, non-gmail-specific behavior')
+    p.add_option('-p', '--base-path', dest='base_path',
+                 help='Only include messages in given base path')
+    p.add_option('--history-path', dest='history_path',
+                 help='Save/restore query history to given path')
+    p.add_option('-v', '--version', dest='version', action='store_true',
+                 help='Show version')
+    (options, args) = p.parse_args()
+
+    if args:
+        dest = args[0]
+    else:
+        dest = '~/.cache/mutt_results'
+
+    if options.version:
+        print(VERSION)
+        sys.exit(0)
+
+    try:
+        main(normalize(dest), options)
+    except KeyboardInterrupt:
+        pass