Small improvements to bash completions

spec/completion_spec.lua

@@ -13,8 +13,7 @@ _foo() {
     cmd="foo"
     opts="-h --help"
 
-    if [[ "$cur" = -* ]]
+    if [[ "$cur" = -* ]]; then
-    then
         COMPREPLY=($(compgen -W "$opts" -- "$cur"))
     fi
 }
@@ -41,8 +40,7 @@ _foo() {
             ;;
     esac
 
-    if [[ "$cur" = -* ]]
+    if [[ "$cur" = -* ]]; then
-    then
         COMPREPLY=($(compgen -W "$opts" -- "$cur"))
     fi
 }
@@ -70,8 +68,7 @@ _foo() {
             ;;
     esac
 
-    if [[ "$cur" = -* ]]
+    if [[ "$cur" = -* ]]; then
-    then
         COMPREPLY=($(compgen -W "$opts" -- "$cur"))
     fi
 }
@@ -92,8 +89,7 @@ _foo() {
     cmd="foo"
     opts=""
 
-    for arg in ${COMP_WORDS[@]:1}
+    for arg in ${COMP_WORDS[@]:1}; do
-    do
         case "$arg" in
             install)
                 cmd="install"
@@ -106,11 +102,9 @@ _foo() {
         foo)
             COMPREPLY=($(compgen -W "install" -- "$cur"))
             ;;
-
     esac
 
-    if [[ "$cur" = -* ]]
+    if [[ "$cur" = -* ]]; then
-    then
         COMPREPLY=($(compgen -W "$opts" -- "$cur"))
     fi
 }
@@ -132,8 +126,7 @@ _foo() {
     cmd="foo"
     opts=""
 
-    for arg in ${COMP_WORDS[@]:1}
+    for arg in ${COMP_WORDS[@]:1}; do
-    do
         case "$arg" in
             install)
                 cmd="install"
@@ -151,8 +144,7 @@ _foo() {
             ;;
     esac
 
-    if [[ "$cur" = -* ]]
+    if [[ "$cur" = -* ]]; then
-    then
         COMPREPLY=($(compgen -W "$opts" -- "$cur"))
     fi
 }
@@ -174,12 +166,8 @@ _foo() {
     cmd="foo"
     opts=""
 
-    for arg in ${COMP_WORDS[@]:1}
+    for arg in ${COMP_WORDS[@]:1}; do
-    do
         case "$arg" in
-            help)
-                break
-                ;;
             install)
                 cmd="install"
                 break
@@ -191,11 +179,9 @@ _foo() {
         foo)
             COMPREPLY=($(compgen -W "help install" -- "$cur"))
             ;;
-
     esac
 
-    if [[ "$cur" = -* ]]
+    if [[ "$cur" = -* ]]; then
-    then
         COMPREPLY=($(compgen -W "$opts" -- "$cur"))
     fi
 }

src/argparse.lua

@@ -1144,7 +1144,7 @@ function Parser:_bash_option_args(buf, indent)
             compreply = ('COMPREPLY=($(compgen -W "%s" -- "$cur"))')
                :format(table.concat(option._choices, " "))
          else
-            compreply = ('COMPREPLY=($(compgen -f "$cur"))')
+            compreply = 'COMPREPLY=($(compgen -f "$cur"))'
          end
          table.insert(opts, (" "):rep(indent + 4) .. pattern)
          table.insert(opts, (" "):rep(indent + 8) .. compreply)
@@ -1163,23 +1163,21 @@ end
 function Parser:_bash_get_cmd(buf)
    local cmds = {}
    for idx, command in ipairs(self._commands) do
-      table.insert(cmds, (" "):rep(12) .. ("%s)"):format(table.concat(command._aliases, "|")))
       if idx ~= self._help_command_idx then
+         table.insert(cmds, (" "):rep(12) .. ("%s)"):format(table.concat(command._aliases, "|")))
          table.insert(cmds, (" "):rep(16) .. ('cmd="%s"'):format(command._aliases[1]))
-      end
          table.insert(cmds, (" "):rep(16) .. "break")
          table.insert(cmds, (" "):rep(16) .. ";;")
       end
+   end
 
-   local get_cmd = ([[
+   if #cmds > 0 then
-    for arg in ${COMP_WORDS[@]:1}
+      table.insert(buf, (" "):rep(4) .. "for arg in ${COMP_WORDS[@]:1}; do")
-    do
+      table.insert(buf, (" "):rep(8) .. 'case "$arg" in')
-        case "$arg" in
+      table.insert(buf, table.concat(cmds, "\n"))
-%s
+      table.insert(buf, (" "):rep(8) .. "esac")
-        esac
+      table.insert(buf, (" "):rep(4) .. "done\n")
-    done
+   end
-]]):format(table.concat(cmds, "\n"))
-   table.insert(buf, get_cmd)
 end
 
 function Parser:_bash_cmd_completions(buf)
@@ -1193,15 +1191,14 @@ function Parser:_bash_cmd_completions(buf)
       end
    end
 
-   local cmd_completions = ([[
+   table.insert(buf, (" "):rep(4) .. 'case "$cmd" in')
-    case "$cmd" in
+   table.insert(buf, (" "):rep(8) .. self._name .. ")")
-        %s)
+   table.insert(buf, (" "):rep(12) .. ('COMPREPLY=($(compgen -W "%s" -- "$cur"))'):format(get_commands(self)))
-            COMPREPLY=($(compgen -W "%s" -- "$cur"))
+   table.insert(buf, (" "):rep(12) .. ";;")
-            ;;
+   if #subcmds > 0 then
-%s
+      table.insert(buf, table.concat(subcmds, "\n"))
-    esac
+   end
-]]):format(self._name, get_commands(self), table.concat(subcmds, "\n"))
+   table.insert(buf, (" "):rep(4) .. "esac\n")
-   table.insert(buf, cmd_completions)
 end
 
 function Parser:get_bash_complete()
@@ -1215,21 +1212,18 @@ _%s() {
 ]]):format(self._name, self._name, get_options(self))}
 
    self:_bash_option_args(buf, 4)
-
-   if #self._commands > 0 then
    self:_bash_get_cmd(buf)
+   if #self._commands > 0 then
       self:_bash_cmd_completions(buf)
    end
 
-   table.insert(buf, [=[
+   table.insert(buf, ([=[
-    if [[ "$cur" = -* ]]
+    if [[ "$cur" = -* ]]; then
-    then
         COMPREPLY=($(compgen -W "$opts" -- "$cur"))
-    fi]=])
+    fi
-   table.insert(buf, "}\n")
+}
-   local complete = ("complete -F _%s -o bashdefault -o default %s")
+
-      :format(self._name, self._name)
+complete -F _%s -o bashdefault -o default %s]=]):format(self._name, self._name))
-   table.insert(buf, complete)
 
    return table.concat(buf, "\n")
 end