Bash completion for fig command

contrib/completion/bash/fig

@@ -0,0 +1,316 @@
+#!bash
+#
+# bash completion for fig commands
+# 
+# This work is based on the completion for the docker command.
+#
+# This script provides completion of:
+#  - commands and their options
+#  - service names
+#  - filepaths
+#
+# To enable the completions either:
+#  - place this file in /etc/bash_completion.d
+#  or
+#  - copy this file and add the line below to your .bashrc after
+#    bash completion features are loaded
+#     . docker.bash
+#
+# Note:
+# Some completions require the current user to have sufficient permissions
+# to execute the docker command.
+
+
+# Extracts all service names from the figfile.
+___fig_all_services_in_figfile() {
+	awk -F: '/^[a-zA-Z0-9]/{print $1}' "${fig_file:-fig.yml}"
+}
+
+# All services, even those without an existing container
+__fig_services_all() {
+	COMPREPLY=( $(compgen -W "$(___fig_all_services_in_figfile)" -- "$cur") )
+}
+
+# All services that have an entry with the given key in their figfile section
+___fig_services_with_key() {
+	# flatten sections to one line, then filter lines containing the key and return section name.
+	awk '/^[a-zA-Z0-9]/{printf "\n"};{printf $0;next;}' fig.yml | awk -F: -v key=": +$1:" '$0 ~ key {print $1}'
+}
+
+# All services that are defined by a Dockerfile reference
+__fig_services_from_build() {
+	COMPREPLY=( $(compgen -W "$(___fig_services_with_key build)" -- "$cur") )
+}
+
+# All services that are defined by an image
+__fig_services_from_image() {
+	COMPREPLY=( $(compgen -W "$(___fig_services_with_key image)" -- "$cur") )
+}
+
+# The services for which containers have been created, optionally filtered
+# by a boolean expression passed in as argument.
+__fig_services_with() {
+	local containers names
+	containers="$(fig 2>/dev/null ${fig_file:+-f $fig_file} ${fig_project:+-p $fig_project} ps -q)"
+	names=( $(docker 2>/dev/null inspect --format "{{if ${1:-true}}} {{ .Name }} {{end}}" $containers) )
+	names=( ${names[@]%_*} )  # strip trailing numbers
+	names=( ${names[@]#*_} )  # strip project name
+	COMPREPLY=( $(compgen -W "${names[*]}" -- "$cur") )
+}
+
+# The services for which at least one running container exists
+__fig_services_running() {
+	__fig_services_with '.State.Running'
+}
+
+# The services for which at least one stopped container exists
+__fig_services_stopped() {
+	__fig_services_with 'not .State.Running'
+}
+
+
+_fig_build() {
+	case "$cur" in
+		-*)
+			COMPREPLY=( $( compgen -W "--no-cache" -- "$cur" ) )
+			;;
+		*)
+			__fig_services_from_build
+			;;
+	esac
+}
+
+
+_fig_fig() {
+	case "$prev" in
+		--file|-f)
+			_filedir
+			return
+			;;
+		--project-name|-p)
+			return
+			;;
+	esac
+
+	case "$cur" in
+		-*)
+			COMPREPLY=( $( compgen -W "--help -h --verbose --version --file -f --project-name -p" -- "$cur" ) )
+			;;
+		*)
+			COMPREPLY=( $( compgen -W "${commands[*]}" -- "$cur" ) )
+			;;
+	esac
+}
+
+
+_fig_help() {
+	COMPREPLY=( $( compgen -W "${commands[*]}" -- "$cur" ) )
+}
+
+
+_fig_kill() {
+	case "$prev" in
+		-s)
+			COMPREPLY=( $( compgen -W "SIGHUP SIGINT SIGKILL SIGUSR1 SIGUSR2" -- "$(echo $cur | tr '[:lower:]' '[:upper:]')" ) )
+			return
+			;;
+	esac
+
+	case "$cur" in
+		-*)
+			COMPREPLY=( $( compgen -W "-s" -- "$cur" ) )
+			;;
+		*)
+			__fig_services_running
+			;;
+	esac
+}
+
+
+_fig_logs() {
+	case "$cur" in
+		-*)
+			COMPREPLY=( $( compgen -W "--no-color" -- "$cur" ) )
+			;;
+		*)
+			__fig_services_all
+			;;
+	esac
+}
+
+
+_fig_port() {
+	case "$prev" in
+		--protocol)
+			COMPREPLY=( $( compgen -W "tcp udp" -- "$cur" ) )
+			return;
+			;;
+		--index)
+			return;
+			;;
+	esac
+
+	case "$cur" in
+		-*)
+			COMPREPLY=( $( compgen -W "--protocol --index" -- "$cur" ) )
+			;;
+		*)
+			__fig_services_all
+			;;
+	esac
+}
+
+
+_fig_ps() {
+	case "$cur" in
+		-*)
+			COMPREPLY=( $( compgen -W "-q" -- "$cur" ) )
+			;;
+		*)
+			__fig_services_all
+			;;
+	esac
+}
+
+
+_fig_pull() {
+	case "$cur" in
+		-*)
+			COMPREPLY=( $( compgen -W "--allow-insecure-ssl" -- "$cur" ) )
+			;;
+		*)
+			__fig_services_from_image
+			;;
+	esac
+}
+
+
+_fig_restart() {
+	__fig_services_running
+}
+
+
+_fig_rm() {
+	case "$cur" in
+		-*)
+			COMPREPLY=( $( compgen -W "--force -v" -- "$cur" ) )
+			;;
+		*)
+			__fig_services_stopped
+			;;
+	esac
+}
+
+
+_fig_run() {
+	case "$prev" in
+		-e)
+			COMPREPLY=( $( compgen -e -- "$cur" ) )
+			compopt -o nospace
+			return
+			;;
+		--entrypoint)
+			return
+			;;	
+	esac
+
+	case "$cur" in
+		-*)
+			COMPREPLY=( $( compgen -W "--allow-insecure-ssl -d --entrypoint -e --no-deps --rm -T" -- "$cur" ) )
+			;;
+		*)
+			__fig_services_all
+			;;
+	esac
+}
+
+
+_fig_scale() {
+	case "$prev" in
+		=)
+			COMPREPLY=("$cur")
+			;;
+		*)
+			COMPREPLY=( $(compgen -S "=" -W "$(___fig_all_services_in_figfile)" -- "$cur") )
+			compopt -o nospace
+			;;
+	esac
+}
+
+
+_fig_start() {
+	__fig_services_stopped
+}
+
+
+_fig_stop() {
+	__fig_services_running
+}
+
+
+_fig_up() {
+	case "$cur" in
+		-*)
+			COMPREPLY=( $( compgen -W "--allow-insecure-ssl -d --no-build --no-color --no-deps --no-recreate" -- "$cur" ) )
+			;;
+		*)
+			__fig_services_all
+			;;
+	esac
+}
+
+
+_fig() {
+	local commands=(
+		build
+		help
+		kill
+		logs
+		port
+		ps
+		pull
+		restart
+		rm
+		run
+		scale
+		start
+		stop
+		up
+	)
+
+	COMPREPLY=()
+	local cur prev words cword
+	_get_comp_words_by_ref -n : cur prev words cword
+
+	# search subcommand and invoke its handler.
+	# special treatment of some top-level options
+	local command='fig'
+	local counter=1
+	local fig_file fig_project
+	while [ $counter -lt $cword ]; do
+		case "${words[$counter]}" in
+			-f|--file)
+				(( counter++ ))
+				fig_file="${words[$counter]}"
+				;;
+			-p|--project-name)
+				(( counter++ ))
+				fig_project="${words[$counter]}"
+				;;
+			-*)
+				;;
+			*)
+				command="${words[$counter]}"
+				break
+				;;
+		esac
+		(( counter++ ))
+	done
+
+	local completions_func=_fig_${command}
+	declare -F $completions_func >/dev/null && $completions_func
+
+	return 0
+}
+
+complete -F _fig fig

docs/completion.md

@@ -0,0 +1,33 @@
+---
+layout: default
+title: Command Completion
+---
+
+Command Completion
+==================
+
+Fig comes with [command completion](http://en.wikipedia.org/wiki/Command-line_completion)
+for the bash shell.
+
+Installing Command Completion
+-----------------------------
+
+Make sure bash completion is installed. If you use a current Linux in a non-minimal installation, bash completion should be available.
+On a Mac, install with `brew install bash-completion`
+
+Place the completion script in `/etc/bash_completion.d/` (`/usr/local/etc/bash_completion.d/` on a Mac), using e.g. 
+
+     curl -L https://raw.githubusercontent.com/docker/fig/master/contrib/completion/bash/fig > /etc/bash_completion.d/fig
+
+Completion will be available upon next login.
+
+Available completions
+---------------------
+Depending on what you typed on the command line so far, it will complete
+
+ - available fig commands
+ - options that are available for a particular command
+ - service names that make sense in a given context (e.g. services with running or stopped instances or services based on images vs. services based on Dockerfiles). For `fig scale`, completed service names will automatically have "=" appended.
+ - arguments for selected options, e.g. `fig kill -s` will complete some signals like SIGHUP and SIGUSR1.
+
+Enjoy working with fig faster and with less typos!

docs/install.md

@@ -20,6 +20,8 @@ Next, install Fig:
 
     curl -L https://github.com/docker/fig/releases/download/1.0.1/fig-`uname -s`-`uname -m` > /usr/local/bin/fig; chmod +x /usr/local/bin/fig
 
+Optionally, install [command completion](completion.html) for the bash shell.
+
 Releases are available for OS X and 64-bit Linux. Fig is also available as a Python package if you're on another platform (or if you prefer that sort of thing):
 
     $ sudo pip install -U fig