Support compopt in completion functions

iterative · Aug 5, 2022 · 6a7ba97 · 6a7ba97
1 parent 60a8525
commit 6a7ba97
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/shtab/__init__.py b/shtab/__init__.py
@@ -421,10 +421,16 @@ def complete_bash(parser, root_prefix=None, preamble="", choice_functions=None):
     COMPREPLY=( $(compgen -W "${current_option_strings[*]}" -- "${completing_word}") )
   else
     # use choices & compgen
+    # call current_action_compgen without a subshell to support compopt
     local IFS=$'\\n'
+    tmpfile="$(mktemp)"
+    [ -n "${current_action_compgen}" ] && \\
+        "${current_action_compgen}" "${completing_word}" > "$tmpfile"
+    mapfile current_action_compreply < "$tmpfile"
+    rm "$tmpfile"
+
     COMPREPLY=( $(compgen -W "${current_action_choices}" -- "${completing_word}") \\
-                $([ -n "${current_action_compgen}" ] \\
-                  && "${current_action_compgen}" "${completing_word}") )
+        ${current_action_compreply[@]} )
   fi
 
   return 0