如何自动完成特定目录下的文件？

Question

我创建了一个命令memo如下：

memo() {
  vi $HOME/memo/$1
}

我想将 bash-completion 应用到我memo要打开的目录中的$HOME/memo文件：

$ memo [TAB] # to show files in $HOME/memo

$HOME/memo包含目录，因此列出其下的文件memo是不够的。换句话说，我想将ls命令中使用的内容应用$HOME/memo到memo：

$ ls [TAB]
foo.md bar/

我尝试了以下但它不适用于嵌套目录：

_memo() {
    local cur
    local files
    _get_comp_words_by_ref -n : cur
    files=$(ls $MEMODIR)
    COMPREPLY=( $(compgen -W "${files}" -- "${cur}") )
}
complete -F _memo memo

MEMODIR=$HOME/memo

Answer 1

这是一个简单的例子：

_memo()
{
    local MEMO_DIR=$HOME/memo
    local cmd=$1 cur=$2 pre=$3
    local arr i file

    arr=( $( cd "$MEMO_DIR" && compgen -f -- "$cur" ) )
    COMPREPLY=()
    for ((i = 0; i < ${#arr[@]}; ++i)); do
        file=${arr[i]}
        if [[ -d $MEMO_DIR/$file ]]; then
            file=$file/
        fi
        COMPREPLY[i]=$file
    done
}
complete -F _memo -o nospace memo