ClickHouse/programs/bash-completion/completions/clickhouse-bootstrap

189 lines
4.2 KiB
Plaintext
Raw Normal View History

#
# bash autocomplete, that can work with:
# a) --help of program
#
# Also you may like:
# $ bind "set completion-ignore-case on"
# $ bind "set show-all-if-ambiguous on"
#
# It uses bash-completion dynamic loader.
# Known to work with bash 3.* with programmable completion and extended
# pattern matching enabled (use 'shopt -s extglob progcomp' to enable
# these if they are not already enabled).
shopt -s extglob
export _CLICKHOUSE_COMPLETION_LOADED=1
CLICKHOUSE_logs_level=(
none
fatal
error
warning
information
debug
trace
test
)
CLICKHOUSE_QueryProcessingStage=(
complete
fetch_columns
with_mergeable_state
with_mergeable_state_after_aggregation
with_mergeable_state_after_aggregation_and_limit
)
CLICKHOUSE_Format=(
CapnProto
PostgreSQLWire
MySQLWire
JSONStringsEachRowWithProgress
JSONEachRowWithProgress
JSONCompact
JSON
CSV
Vertical
ODBCDriver2
PrettySpaceNoEscapes
Pretty
JSONCompactStrings
PrettyNoEscapes
ArrowStream
TabSeparatedWithNames
Parquet
Arrow
PrettyCompact
AvroConfluent
ORC
PrettyCompactNoEscapes
RawBLOB
Template
MsgPack
JSONCompactEachRow
CustomSeparated
TemplateIgnoreSpaces
Markdown
XML
ProtobufSingle
JSONCompactStringsEachRowWithNamesAndTypes
TSKV
TabSeparated
JSONStringEachRow
JSONStringsEachRow
TSVRaw
Values
TabSeparatedWithNamesAndTypes
PrettyCompactMonoBlock
TSVWithNamesAndTypes
Avro
RowBinaryWithNamesAndTypes
LineAsString
Native
JSONCompactEachRowWithNamesAndTypes
PrettySpace
Regexp
TSV
JSONEachRow
CustomSeparatedIgnoreSpaces
CSVWithNames
JSONStrings
Null
TabSeparatedRaw
TSVWithNames
Protobuf
RowBinary
JSONAsString
JSONCompactStringsEachRow
)
function _clickhouse_bin_exist()
{ [ -x "$1" ] || command -v "$1" >& /dev/null; }
function _clickhouse_quote()
{
local quoted=${1//\'/\'\\\'\'};
printf "'%s'" "$quoted"
}
# Extract every option (everything that starts with "-") from the --help dialog.
function _clickhouse_get_options()
{
"$@" --help 2>&1 | awk -F '[ ,=<>]' '{ for (i=1; i <= NF; ++i) { if (substr($i, 0, 1) == "-" && length($i) > 1) print $i; } }' | sort -u
}
function _complete_for_clickhouse_generic_bin_impl()
{
local prev=$1 && shift
case "$prev" in
-C|--config-file|--config)
return 1
;;
--stage)
COMPREPLY=( $(compgen -W "${CLICKHOUSE_QueryProcessingStage[*]}" -- "$cur") )
return 1
;;
--send_logs_level)
COMPREPLY=( $(compgen -W "${CLICKHOUSE_logs_level[*]}" -- "$cur") )
return 1
;;
--format|--input-format|--output-format)
COMPREPLY=( $(compgen -W "${CLICKHOUSE_Format[*]}" -- "$cur") )
return 1
;;
--host)
COMPREPLY=( $(compgen -A hostname -- "$cur") )
return 1
;;
# Argh... This looks like a bash bug...
# Redirections are passed to the completion function
# although it is managed by the shell directly...
'<'|'>'|'>>'|[12]'>'|[12]'>>')
return 1
;;
esac
return 0
}
function _complete_for_clickhouse_generic_bin()
{
local cur prev
eval local cmd="$( _clickhouse_quote "$1" )"
_clickhouse_bin_exist "$cmd" || return 0
COMPREPLY=()
_get_comp_words_by_ref cur prev
if _complete_for_clickhouse_generic_bin_impl "$prev"; then
COMPREPLY=( $(compgen -W "$(_clickhouse_get_options "$cmd")" -- "$cur") )
fi
return 0
}
function _complete_clickhouse_generic()
{
local bin=$1 && shift
local f=${1:-_complete_for_clickhouse_generic_bin}
local o=(
-o default
-o bashdefault
-o nospace
-F "$f"
"$bin"
)
complete "${o[@]}"
}
function _complete_clickhouse_bootstrap_main()
{
local runtime=/usr/share/bash-completion/bash_completion
if ! type _get_comp_words_by_ref >& /dev/null && [[ -f $runtime ]]; then
source $runtime
fi
type _get_comp_words_by_ref >& /dev/null || return 0
}
_complete_clickhouse_bootstrap_main "$@"