Add kc script for kubectl context switching

This commit is contained in:
Pavle Portic 2022-04-18 12:05:43 +02:00
parent d637cac232
commit 3ce06883a7
Signed by: TheEdgeOfRage
GPG Key ID: 66AD4BA646FBC0D2
1 changed files with 128 additions and 0 deletions

128
dot_local/bin/executable_kc Normal file
View File

@ -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 "$@"