From b5e3f4ca2f7f854665a5adfac097d9c1c37d48f2 Mon Sep 17 00:00:00 2001
From: wickles <4229542+wickles@users.noreply.github.com>
Date: Fri, 14 Jul 2023 10:32:21 -0700
Subject: [PATCH] Fix filename completions for zsh and bash (#331)

Reapply commit accidentally reverted
---
 completions/_dust     | 1 +
 completions/dust.bash | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/completions/_dust b/completions/_dust
index ec51dd1..1ea8587 100644
--- a/completions/_dust
+++ b/completions/_dust
@@ -71,6 +71,7 @@ _dust() {
 '-V[Print version]' \
 '--version[Print version]' \
 '*::params:' \
+'*:inputs:_files' \
 && ret=0
 }
 
diff --git a/completions/dust.bash b/completions/dust.bash
index eddf14f..2247149 100644
--- a/completions/dust.bash
+++ b/completions/dust.bash
@@ -20,9 +20,12 @@ _dust() {
     case "${cmd}" in
         dust)
             opts="-d -n -p -X -I -L -x -s -r -c -b -B -z -R -f -i -v -e -t -w -H -P -D -F -S -h -V --depth --number-of-lines --full-paths --ignore-directory --ignore-all-in-file --dereference-links --limit-filesystem --apparent-size --reverse --no-colors --no-percent-bars --bars-on-right --min-size --screen-reader --skip-total --filecount --ignore_hidden --invert-filter --filter --file_types --terminal_width --si --no-progress --only-dir --only-file --stack-size --help --version [params]..."
-            if [[ ${cur} == -* || ${COMP_CWORD} -eq 1 ]] ; then
+            if [[ ${cur} == -* ]] ; then
                 COMPREPLY=( $(compgen -W "${opts}" -- "${cur}") )
                 return 0
+            elif [[ ${cur} == * ]] ; then
+                _filedir
+                return 0
             fi
             case "${prev}" in
                 --depth)