# mutt completion                                          -*- shell-script -*-
#
# Mutt doesn't have an "addressbook" like Pine, but it has aliases and
# a "query" function to retrieve addresses, so that's what we use here.

# @param $1 (cur) Current word to complete
proc _muttaddr {
    _muttaliases $1
    _muttquery $1

    setvar COMPREPLY = ''( $( compgen -u -- "$1" ) )

    return 0
}


# Find muttrc to use
# @output  muttrc filename
proc _muttrc {
    # Search COMP_WORDS for '-F muttrc' or '-Fmuttrc' argument
    set -- ${words[@]} {
        if [[ $1 == -F* ]] {
            if [[ ${#1} -gt 2 ]] {
                setvar muttrc = "$(dequote "${1:2}")"
            } else {
                shift
                [[ $1 ]] && setvar muttrc = "$(dequote "$1")"
            }
            break
        }
        shift
    }

    if [[ -z $muttrc ]] {
        if [[ -f ~/.${muttcmd}rc ]] {
            setvar muttrc = ""~/.${muttcmd}rc""
        } elif [[ -f ~/.${muttcmd}/${muttcmd}rc ]] {
            setvar muttrc = ""~/.${muttcmd}/${muttcmd}rc""
        }
    }
    printf "%s" $muttrc
}


# Recursively build list of sourced config files
# @param $1  List of config files found so far
# @param $2  Config file to process
# @output  List of config files
proc _muttconffiles {
    local file sofar
    local -a newconffiles

    setvar sofar = "" $1 ""
    shift {
        setvar newconffiles = ''( $(sed -n 's|^source[[:space:]]\{1,\}\([^[:space:]]\{1,\}\).*$|\1|p' $(eval printf %s $1) ) )
        for file in "${newconffiles[@]}" {
            __expand_tilde_by_ref file
            [[ ! -f "$file" || $sofar == *\ $file\ * ]] && continue
            setvar sofar = "" $file""
            setvar sofar = "" $(eval _muttconffiles \"$sofar\" $file) ""
        }
        shift
    }
    printf '%s\n' $sofar
}


# @param $1 (cur) Current word to complete
proc _muttaliases {
    local cur=$1 muttrc muttcmd=${words[0]}
    local -a conffiles aliases

    setvar muttrc = $(_muttrc)
    [[ -z $muttrc ]] && return 0

    setvar conffiles = ''( $(eval _muttconffiles $muttrc $muttrc) )
    setvar aliases = ''( $( sed -n 's|^alias[[:space:]]\{1,\}\([^[:space:]]\{1,\}\).*$|\1|p' \
        $(eval echo "${conffiles[@]}") ) )
    setvar COMPREPLY = ''( $( compgen -W "${aliases[*]}" -- "$cur" ) )

    return 0
}


# @param $1 (cur) Current word to complete
proc _muttquery {
    local cur=$1 querycmd muttcmd=${words[0]}
    local -a queryresults

    setvar querycmd = "$( $muttcmd -Q query_command 2>/dev/null | sed -e 's|^query_command=\"\(.*\)\"$|\1|' -e 's|%s|'$cur'|' )"
    if [[ -z "$cur" || -z "$querycmd" ]] {
        setvar queryresults = ''()
    } else {
        __expand_tilde_by_ref querycmd
        setvar queryresults = ''( $( $querycmd | \
            sed -n '2,$s|^\([^[:space:]]\{1,\}\).*|\1|p' ) )
    }

    setvar COMPREPLY = ''( $( compgen -W "${queryresults[*]}" -- "$cur" ) )

    return 0
}


# @param $1 (cur) Current word to complete
proc _muttfiledir {
    local cur=$1 folder muttrc spoolfile muttcmd=${words[0]}

    setvar muttrc = $(_muttrc)
    if [[ $cur == [=+]* ]] {
        setvar folder = "$( $muttcmd -F "$muttrc" -Q folder 2>/dev/null | sed -e 's|^folder=\"\(.*\)\"$|\1|' )"
        : folder:=~/Mail

        # Match any file in $folder beginning with $cur
        # (minus the leading '=' sign).
        compopt -o filenames
        setvar COMPREPLY = ''( $( compgen -f -- "$folder/${cur:1}" ) )
        setvar COMPREPLY = ''( ${COMPREPLY[@]#$folder/} )
        return 0
    } elif [[ $cur == !* ]] {
        setvar spoolfile = "$( $muttcmd -F "$muttrc" -Q spoolfile 2>/dev/null | \
            sed -e 's|^spoolfile=\"\(.*\)\"$|\1|' )"
        [[ ! -z $spoolfile ]] && eval cur="${cur/^!/$spoolfile}"
    }
    _filedir

    return 0
}


proc _mutt {
    local cur prev words cword
    _init_completion -n =+! || return

    case (cur) {
    -* {
        setvar COMPREPLY = ''( $( compgen -W '-A -a -b -c -e -f -F -H -i -m -n -p -Q -R -s
            -v -x -y -z -Z -h' -- "$cur" ) )
        return 0
        }
    * {
        case (prev) {
        -a|-f|-F|-H|-i {
            _muttfiledir $cur
            return 0
            }
        -A {
            _muttaliases $cur
            return 0
            }
        -e|-m|-Q|-s|-h|-p|-R|-v|-y|-z|-Z {
            return 0
            }
        * {
            _muttaddr $cur
            return 0
            }
        }
        }
    }
} &&
complete -F _mutt -o default mutt muttng

# ex: ts=4 sw=4 et filetype=sh