Merge branch 'cli-autocompletion' into 'master'

agneum · agneum · commit a23c541c16e1 · 2021-01-12T04:43:27.000Z
feat: enable bash autocompletion for Database Lab CLI

See merge request postgres-ai/database-lab!237
diff --git a/README.md b/README.md
@@ -29,6 +29,11 @@ Install Database Lab client CLI on your Linux machine (e.g., Ubuntu):
 curl https://gitlab.com/postgres-ai/database-lab/-/raw/master/scripts/cli_install.sh | bash
 ```
 
+### Activate CLI autocompletion:
+```bash
+sudo curl https://gitlab.com/postgres-ai/database-lab/-/raw/master/scripts/bash_autocomplete --output /etc/bash_completion.d/dblab 
+source /etc/bash_completion.d/dblab
+```
 
 ### How to use CLI
 - [How to install and initialize Database Lab CLI](https://postgres.ai/docs/guides/cli/cli-install-init)
diff --git a/cmd/cli/main.go b/cmd/cli/main.go
@@ -76,6 +76,7 @@ func main() {
 				EnvVars: []string{"DBLAB_CLI_DEBUG"},
 			},
 		},
+		EnableBashCompletion: true,
 	}
 
 	adoptTemplates()
diff --git a/scripts/bash_autocomplete b/scripts/bash_autocomplete
@@ -0,0 +1,23 @@
+#! /bin/bash
+
+: ${PROG:=$(basename ${BASH_SOURCE})}
+
+
+
+_cli_bash_autocomplete() {
+  if [[ "${COMP_WORDS[0]}" != "source" ]]; then
+    local cur opts base
+    COMPREPLY=()
+    cur="${COMP_WORDS[COMP_CWORD]}"
+    if [[ "$cur" == "-"* ]]; then
+      opts=$( ${COMP_WORDS[@]:0:$COMP_CWORD} ${cur} --generate-bash-completion )
+    else
+      opts=$( ${COMP_WORDS[@]:0:$COMP_CWORD} --generate-bash-completion )
+    fi
+    COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
+    return 0
+  fi
+}
+
+complete -o bashdefault -o default -o nospace -F _cli_bash_autocomplete $PROG
+unset PROG

Original file line number	Diff line number	Diff line change
`@@ -76,6 +76,7 @@ func main() {`
`76`	`76`	`EnvVars: []string{"DBLAB_CLI_DEBUG"},`
`77`	`77`	`},`
`78`	`78`	`},`
	`79`	`+ EnableBashCompletion: true,`
`79`	`80`	`}`
`80`	`81`
`81`	`82`	`adoptTemplates()`