# umount(8) completion                                     -*- shell-script -*-

# Just like COMPREPLY=(`compgen -W "${COMPREPLY[*]}" -- "$cur"`), only better!
#
# This will correctly escape special characters in COMPREPLY.
proc _reply_compgen_array {
    # Create the argument for compgen -W by escaping twice.
    #
    # One round of escape is because we want to reply with escaped arguments. A
    # second round is required because compgen -W will helpfully expand it's
    # argument.
    local i wlist
    for i in ${!COMPREPLY[*]} {
        local q=$(quote "$(printf %q "${COMPREPLY[$i]}")")
        setvar wlist = "$q$'\n"'
    }

    # We also have to add another round of escaping to $cur.
    local ecur="$cur"
    setvar ecur = ${ecur//\\/\\\\}
    setvar ecur = ${ecur//\'/\\\'}

    # Actually generate completions.
    local oldifs=$IFS'
    IFS=$'\n'' eval 'COMPREPLY=(`compgen -W "$wlist" -- "${ecur}"`)'
    setvar IFS = "$oldifs"
}

# Unescape strings in the linux fstab(5) format (with octal escapes).
proc __linux_fstab_unescape {
    eval $1="'${!1//\'/\\047}'"
    eval $1="'${!1/%\\/\\\\}'"
    eval "$1=$'${!1}'"
}

# Complete linux fstab entries.
#
# Reads a file from stdin in the linux fstab(5) format; as used by /etc/fstab
# and /proc/mounts.
proc _linux_fstab {
    setvar COMPREPLY = ''()

    # Read and unescape values into COMPREPLY
    local fs_spec fs_file fs_other
    local oldifs="$IFS"
    while read -r fs_spec fs_file fs_other {
        if [[ $fs_spec == [#]* ]] { continue; }
        if [[ $1 == -L ]] {
            local fs_label=${fs_spec/#LABEL=}
            if [[ $fs_label != "$fs_spec" ]] {
                __linux_fstab_unescape fs_label
                setvar IFS = '$'\0''
                setvar COMPREPLY = ''("$fs_label")
                setvar IFS = "$oldifs"
            }
        } else {
            __linux_fstab_unescape fs_spec
            __linux_fstab_unescape fs_file
            setvar IFS = '$'\0''
            [[ $fs_spec == */* ]] && setvar COMPREPLY = ''("$fs_spec")
            [[ $fs_file == */* ]] && setvar COMPREPLY = ''("$fs_file")
            setvar IFS = "$oldifs"
        }
    }

    # Add relative paths to COMPREPLY
    if [[ $cur && $cur != /* ]] {
        local realcur
        [[ $cur == */ ]] && # don't let readlink drop last / from path
            setvar realcur = ""$( readlink -f "$cur." 2> /dev/null )/"" ||
            setvar realcur = $( readlink -f "$cur" 2> /dev/null )
        if [[ $realcur ]] {
            local dirrealcur= dircur= basecur
            if [[ $cur == */* ]] {
               setvar dirrealcur = ""${realcur%/*}/""
               setvar dircur = ""${cur%/*}/""
            }
            setvar basecur = ${cur#"$dircur"}
            local i n=${#COMPREPLY[@]}
            for (( i=0; i < $n; i++ )); do
                [[ "${COMPREPLY[i]}" == "$realcur"* ]] &&
                    COMPREPLY+=( $( cd "$dircur" 2> /dev/null &&
                        compgen -f -d -P "$dircur" \
                        -X "!${COMPREPLY[i]##"$dirrealcur"}" -- "$basecur" ) )
            done
        }
    }

    _reply_compgen_array
}

proc _umount {
    local cur prev words cword
    _init_completion || return

    case (prev) {
        -t {
            # FIXME: no<fstype>
            local split=false
            if [[ "$cur" == ?*,* ]] {
                setvar prev = "${cur%,*}"
                setvar cur = "${cur##*,}"
                setvar split = 'true'
            }
            setvar COMPREPLY = ''( $( compgen -W 'adfs affs autofs btrfs cifs coda
                cramfs debugfs devpts efs ext2 ext3 ext4 fuse hfs hfsplus hpfs
                iso9660 jfs minix msdos ncpfs nfs nfs4 ntfs ntfs-3g proc qnx4
                ramfs reiserfs romfs squashfs smbfs sysv tmpfs ubifs udf ufs
                umsdos usbfs vfat xfs' -- "$cur" ) )
            _fstypes
            $split && setvar COMPREPLY = ''( ${COMPREPLY[@]/#/$prev,} )
            return
            }
        -O {
            # argument required but no completions available
            return
            }
    }

    if [[ "$cur" == -* ]] {
        setvar COMPREPLY = ''( $( compgen -W '-V -h -v -n -r -d -i -a -t -O -f -l
            --no-canonicalize --fake' -- "$cur" ) )
        [[ $COMPREPLY ]] && return
    }

    if [[ -r /proc/mounts ]] {
        # Linux /proc/mounts is properly quoted. This is important when
        # unmounting usb devices with pretty names.
        _linux_fstab < /proc/mounts
    } else {
        local IFS=$'\n'
        setvar COMPREPLY = ''( $( compgen -W '$( mount | cut -d" " -f 3 )' -- "$cur" ) )
    }
} &&
complete -F _umount -o dirnames umount

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