_writeimg_completion() {
    local cur prev opts

    cur="${COMP_WORDS[COMP_CWORD]}"
    prev="${COMP_WORDS[COMP_CWORD-1]}"

    opts="-v -d -h --help -n --noconfirm -V --version"

    # Devices
    if [[ "$prev" == "-d" ]]; then
        COMPREPLY=( $(compgen -W "$(ls -d /dev/sd* /dev/nvme* /dev/mmcblk* 2>/dev/null)" -- "$cur") )
        return 0
    fi

    # Flags
    if [[ "$cur" != -* ]]; then
        COMPREPLY=( $(compgen -f -- "$cur") )
        return 0
    fi

    # Files
    COMPREPLY=( $(compgen -W "$opts" -- "$cur") )
}

complete -F _writeimg_completion writeimg