167 lines
3.4 KiB
Bash
167 lines
3.4 KiB
Bash
# -*- mode: shell-script; sh-basic-offset: 8; indent-tabs-mode: t -*-
|
|
# ex: ts=8 sw=8 noet filetype=sh
|
|
#
|
|
# bash completion for Postgresql
|
|
|
|
have psql && {
|
|
_pg_databases()
|
|
{
|
|
return # intentionally disabled in 7ebed6af (TODO: why?)
|
|
COMPREPLY=( $( compgen -W "$( psql -l 2>/dev/null | \
|
|
sed -e '1,/^-/d' -e '/^(/,$d' | \
|
|
awk '{print $1}' )" -- $cur ) )
|
|
}
|
|
|
|
_pg_users()
|
|
{
|
|
#COMPREPLY=( $( psql -qtc 'select usename from pg_user' template1 2>/dev/null | \
|
|
# grep "^ $cur" ) )
|
|
#[ ${#COMPREPLY[@]} -eq 0 ] && COMPREPLY=( $( compgen -u -- $cur ) )
|
|
COMPREPLY=( $( compgen -u -- $cur ) )
|
|
}
|
|
|
|
# createdb(1) completion
|
|
#
|
|
_createdb()
|
|
{
|
|
local cur prev split=false
|
|
|
|
COMPREPLY=()
|
|
cur=`_get_cword`
|
|
prev=${COMP_WORDS[COMP_CWORD-1]}
|
|
|
|
_split_longopt && split=true
|
|
|
|
case "$prev" in
|
|
-h|--host)
|
|
_known_hosts_real "$cur"
|
|
return 0
|
|
;;
|
|
-U|--username|-O|--owner)
|
|
_pg_users
|
|
return 0
|
|
;;
|
|
-p|--port|-D|--tablespace|-E|--encoding|-T|--template)
|
|
# argument required but no completions available
|
|
return 0
|
|
;;
|
|
--help|--version)
|
|
# all other arguments are noop with these
|
|
return 0
|
|
;;
|
|
esac
|
|
|
|
$split && return 0
|
|
|
|
if [[ "$cur" == -* ]]; then
|
|
COMPREPLY=( $( compgen -W '-D -T -E -h -p -U -W -e -q \
|
|
--tablespace --template --encoding --host --port \
|
|
--username --password --echo --quiet --help --version' \
|
|
-- $cur ))
|
|
else
|
|
_pg_databases
|
|
fi
|
|
}
|
|
complete -F _createdb $default createdb
|
|
|
|
# dropdb(1) completion
|
|
#
|
|
_dropdb()
|
|
{
|
|
local cur prev split=false
|
|
|
|
COMPREPLY=()
|
|
cur=`_get_cword`
|
|
prev=${COMP_WORDS[COMP_CWORD-1]}
|
|
|
|
_split_longopt && split=true
|
|
|
|
case "$prev" in
|
|
-h|--host)
|
|
_known_hosts_real "$cur"
|
|
return 0
|
|
;;
|
|
-U|--username)
|
|
_pg_users
|
|
return 0
|
|
;;
|
|
--help|--version)
|
|
# all other arguments are noop with these
|
|
return 0
|
|
;;
|
|
esac
|
|
|
|
$split && return 0
|
|
|
|
if [[ "$cur" == -* ]]; then
|
|
COMPREPLY=( $( compgen -W '-h -p -U -W -i -e -q \
|
|
--host --port --username --password --interactive \
|
|
--echo --quiet --help --version' -- $cur ) )
|
|
else
|
|
_pg_databases
|
|
fi
|
|
}
|
|
complete -F _dropdb $default dropdb
|
|
|
|
# psql(1) completion
|
|
#
|
|
_psql()
|
|
{
|
|
local cur prev split=false
|
|
|
|
COMPREPLY=()
|
|
cur=`_get_cword`
|
|
prev=${COMP_WORDS[COMP_CWORD-1]}
|
|
|
|
_split_longopt && split=true
|
|
|
|
case "$prev" in
|
|
-h|--host)
|
|
_known_hosts_real "$cur"
|
|
return 0
|
|
;;
|
|
-U|--username)
|
|
_pg_users
|
|
return 0
|
|
;;
|
|
-d|--dbname)
|
|
_pg_databases
|
|
return 0
|
|
;;
|
|
-o|--output|-f|--file|-L|--log-file)
|
|
_filedir
|
|
return 0
|
|
;;
|
|
-c|--command|-F|--field-separator|-p|--port|-P|--pset|\
|
|
-R|--record-separator|-T|--table-attr|-v|--set|--variable)
|
|
# argument required but no completions available
|
|
return 0
|
|
;;
|
|
-\?|--help|-V|--version)
|
|
# all other arguments are noop with these
|
|
return 0
|
|
;;
|
|
esac
|
|
|
|
$split && return 0
|
|
|
|
if [[ "$cur" == -* ]]; then
|
|
# return list of available options
|
|
COMPREPLY=( $( compgen -W '-a --echo-all -A --no-align \
|
|
-c --command -d --dbname -e --echo-queries \
|
|
-E --echo-hidden -f --file -F --field-separator \
|
|
-h --host -H --html -l --list -L --log-file -n \
|
|
-o --output -p --port -P --pset -q --quiet \
|
|
-R --record-separator -s --single-step \
|
|
-S --single-line -t --tuples-only -T --table-attr \
|
|
-U --username -v --set --variable -V --version \
|
|
-W --password -x --expanded -X --no-psqlrc \
|
|
-1 --single-transaction -? --help' -- $cur ) )
|
|
else
|
|
# return list of available databases
|
|
_pg_databases
|
|
fi
|
|
}
|
|
complete -F _psql $filenames psql
|
|
}
|