如何自动补全特定目录下的文件？

我创建了一个命令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 to 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

这是一个简单的例子：

_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