#!/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