Add kc script for kubectl context switching
This commit is contained in:
parent
d637cac232
commit
3ce06883a7
|
@ -0,0 +1,128 @@
|
||||||
|
#!/usr/bin/env bash
|
||||||
|
|
||||||
|
[[ -n $DEBUG ]] && set -x
|
||||||
|
|
||||||
|
set -eou pipefail
|
||||||
|
IFS=$'\n\t'
|
||||||
|
|
||||||
|
|
||||||
|
usage() {
|
||||||
|
local SELF
|
||||||
|
SELF="kc"
|
||||||
|
|
||||||
|
cat <<EOF
|
||||||
|
USAGE:
|
||||||
|
$SELF : list the configs
|
||||||
|
$SELF <NAME> : switch to context <NAME>
|
||||||
|
$SELF -c, --current : show the current context name
|
||||||
|
$SELF -u, --unset : unset the current context, better use: Ctrl+d
|
||||||
|
|
||||||
|
$SELF -h,--help : show this message
|
||||||
|
EOF
|
||||||
|
}
|
||||||
|
|
||||||
|
exit_err() {
|
||||||
|
echo >&2 "${1}"
|
||||||
|
exit 1
|
||||||
|
}
|
||||||
|
|
||||||
|
current_context() {
|
||||||
|
kubectl config view -o=jsonpath='{.current-context}'
|
||||||
|
}
|
||||||
|
|
||||||
|
get_configs() {
|
||||||
|
#kubectl config get-contexts -o=name | sort -n
|
||||||
|
ls -p --color=never -1 $HOME/.kube | grep -v '/$'
|
||||||
|
}
|
||||||
|
|
||||||
|
list_configs() {
|
||||||
|
set -u pipefail
|
||||||
|
local cur ctx_list
|
||||||
|
cur="$(current_context)" || exit_err "error getting current context"
|
||||||
|
ctx_list=$(get_configs) || exit_err "error getting context list"
|
||||||
|
|
||||||
|
local yellow darkbg normal
|
||||||
|
yellow=$(tput setaf 3 || true)
|
||||||
|
darkbg=$(tput setab 0 || true)
|
||||||
|
normal=$(tput sgr0 || true)
|
||||||
|
|
||||||
|
local cur_ctx_fg cur_ctx_bg
|
||||||
|
cur_ctx_fg=${KUBECTX_CURRENT_FGCOLOR:-$yellow}
|
||||||
|
cur_ctx_bg=${KUBECTX_CURRENT_BGCOLOR:-$darkbg}
|
||||||
|
|
||||||
|
for c in $ctx_list; do
|
||||||
|
if [[ -n "${_KUBECTX_FORCE_COLOR:-}" || \
|
||||||
|
-t 1 && -z "${NO_COLOR:-}" ]]; then
|
||||||
|
# colored output mode
|
||||||
|
if [[ "${c}" = "${cur}" ]]; then
|
||||||
|
echo "${cur_ctx_bg}${cur_ctx_fg}${c}${normal}"
|
||||||
|
else
|
||||||
|
echo "${c}"
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "${c}"
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
choose_context_interactive() {
|
||||||
|
local choice
|
||||||
|
choice="$(get_configs | _KUBECTX_FORCE_COLOR=1 fzf --ansi --no-preview || true)"
|
||||||
|
if [[ -z "${choice}" ]]; then
|
||||||
|
echo 2>&1 "error: you did not choose any of the options"
|
||||||
|
exit 1
|
||||||
|
else
|
||||||
|
set_context "${choice}"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
set_context() {
|
||||||
|
echo "Set context to ${1}." >&2
|
||||||
|
export KUBECONFIG="$HOME/.kube/${1}"
|
||||||
|
echo -e "\n\nYou are in a subshell. Press Ctrl+d for exit!\n\n"
|
||||||
|
$SHELL
|
||||||
|
}
|
||||||
|
|
||||||
|
unset_context() {
|
||||||
|
echo "Unsetting current context." >&2
|
||||||
|
kubectl config unset current-context
|
||||||
|
echo "Ctrl+d for exiting subshell." >&2
|
||||||
|
}
|
||||||
|
|
||||||
|
main() {
|
||||||
|
if [[ "$#" -eq 0 ]]; then
|
||||||
|
if [[ -t 1 && -z "${KUBECTX_IGNORE_FZF:-}" && "$(type fzf &>/dev/null; echo $?)" -eq 0 ]]; then
|
||||||
|
choose_context_interactive
|
||||||
|
else
|
||||||
|
list_configs
|
||||||
|
fi
|
||||||
|
elif [[ "$#" -gt 1 ]]; then
|
||||||
|
echo "error: too many arguments" >&2
|
||||||
|
usage
|
||||||
|
exit 1
|
||||||
|
elif [[ "$#" -eq 1 ]]; then
|
||||||
|
if [[ "${1}" == "-" ]]; then
|
||||||
|
swap_context
|
||||||
|
elif [[ "${1}" == '-c' || "${1}" == '--current' ]]; then
|
||||||
|
# we don't call current_context here for two reasons:
|
||||||
|
# - it does not fail when current-context property is not set
|
||||||
|
# - it does not return a trailing newline
|
||||||
|
kubectl config current-context
|
||||||
|
elif [[ "${1}" == '-u' || "${1}" == '--unset' ]]; then
|
||||||
|
unset_context
|
||||||
|
elif [[ "${1}" == '-h' || "${1}" == '--help' ]]; then
|
||||||
|
usage
|
||||||
|
elif [[ "${1}" =~ ^-(.*) ]]; then
|
||||||
|
echo "error: unrecognized flag \"${1}\"" >&2
|
||||||
|
usage
|
||||||
|
exit 1
|
||||||
|
else
|
||||||
|
set_context "${1}"
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
usage
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
main "$@"
|
Loading…
Reference in New Issue