@ -1,16 +1,15 @@
#!/bin/z sh
#!/bin/ba sh
# Kubernetes prompt helper for bash/zsh
# ported to oh-my-zsh
# Displays current context and namespace
# Copyright 2018 Jon Mosco
# Copyright 2019 Jon Mosco
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https ://www.apache.org/licenses/LICENSE-2.0
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
@ -21,39 +20,150 @@
# Debug
[ [ -n $DEBUG ] ] && set -x
setopt PROMPT_SUBST
autoload -U add-zsh-hook
add-zsh-hook precmd _kube_ps1_update_cache
zmodload zsh/stat
zmodload zsh/datetime
# Default values for the prompt
# Override these values in ~/.zshrc
# Override these values in ~/.zshrc or ~/.bashrc
KUBE_PS1_BINARY = " ${ KUBE_PS1_BINARY :- kubectl } "
KUBE_PS1_SYMBOL_ENABLE = " ${ KUBE_PS1_SYMBOL_ENABLE :- true } "
KUBE_PS1_SYMBOL_DEFAULT = " ${ KUBE_PS1_SYMBOL_DEFAULT :- \u 2388 } "
KUBE_PS1_SYMBOL_DEFAULT = ${ KUBE_PS1_SYMBOL_DEFAULT :- $'\u2388 ' }
KUBE_PS1_SYMBOL_USE_IMG = " ${ KUBE_PS1_SYMBOL_USE_IMG :- false } "
KUBE_PS1_NS_ENABLE = " ${ KUBE_PS1_NS_ENABLE :- true } "
KUBE_PS1_CONTEXT_ENABLE = " ${ KUBE_PS1_CONTEXT_ENABLE :- true } "
KUBE_PS1_PREFIX = " ${ KUBE_PS1_PREFIX -( } "
KUBE_PS1_SEPARATOR = " ${ KUBE_PS1_SEPARATOR -| } "
KUBE_PS1_DIVIDER = " ${ KUBE_PS1_DIVIDER - : } "
KUBE_PS1_PREFIX = " ${ KUBE_PS1_PREFIX -( } "
KUBE_PS1_SUFFIX = " ${ KUBE_PS1_SUFFIX -) } "
KUBE_PS1_SYMBOL_COLOR = " ${ KUBE_PS1_SYMBOL_COLOR -blue } "
KUBE_PS1_CTX_COLOR = " ${ KUBE_PS1_CTX_COLOR -red } "
KUBE_PS1_NS_COLOR = " ${ KUBE_PS1_NS_COLOR -cyan } "
KUBE_PS1_BG_COLOR = " ${ KUBE_PS1_BG_COLOR } "
KUBE_PS1_KUBECONFIG_CACHE = " ${ KUBECONFIG } "
KUBE_PS1_DISABLE_PATH = " ${ HOME } /.kube/kube-ps1/disabled "
KUBE_PS1_LAST_TIME = 0
KUBE_PS1_ENABLED = true
KUBE_PS1_CLUSTER_FUNCTION = " ${ KUBE_PS1_CLUSTER_FUNCTION } "
KUBE_PS1_NAMESPACE_FUNCTION = " ${ KUBE_PS1_NAMESPACE_FUNCTION } "
# Determine our shell
if [ " ${ ZSH_VERSION - } " ] ; then
KUBE_PS1_SHELL = "zsh"
elif [ " ${ BASH_VERSION - } " ] ; then
KUBE_PS1_SHELL = "bash"
fi
_kube_ps1_init( ) {
[ [ -f " ${ KUBE_PS1_DISABLE_PATH } " ] ] && KUBE_PS1_ENABLED = off
KUBE_PS1_COLOR_SYMBOL = " %{ $fg [blue]%} "
KUBE_PS1_COLOR_CONTEXT = " %{ $fg [red]%} "
KUBE_PS1_COLOR_NS = " %{ $fg [cyan]%} "
case " ${ KUBE_PS1_SHELL } " in
"zsh" )
_KUBE_PS1_OPEN_ESC = "%{"
_KUBE_PS1_CLOSE_ESC = "%}"
_KUBE_PS1_DEFAULT_BG = "%k"
_KUBE_PS1_DEFAULT_FG = "%f"
setopt PROMPT_SUBST
autoload -U add-zsh-hook
add-zsh-hook precmd _kube_ps1_update_cache
zmodload -F zsh/stat b:zstat
zmodload zsh/datetime
; ;
"bash" )
_KUBE_PS1_OPEN_ESC = $'\001'
_KUBE_PS1_CLOSE_ESC = $'\002'
_KUBE_PS1_DEFAULT_BG = $'\033[49m'
_KUBE_PS1_DEFAULT_FG = $'\033[39m'
[ [ $PROMPT_COMMAND = ~ _kube_ps1_update_cache ] ] || PROMPT_COMMAND = " _kube_ps1_update_cache; ${ PROMPT_COMMAND :- : } "
; ;
esac
}
_kube_ps1_color_fg( ) {
local KUBE_PS1_FG_CODE
case " ${ 1 } " in
black) KUBE_PS1_FG_CODE = 0; ;
red) KUBE_PS1_FG_CODE = 1; ;
green) KUBE_PS1_FG_CODE = 2; ;
yellow) KUBE_PS1_FG_CODE = 3; ;
blue) KUBE_PS1_FG_CODE = 4; ;
magenta) KUBE_PS1_FG_CODE = 5; ;
cyan) KUBE_PS1_FG_CODE = 6; ;
white) KUBE_PS1_FG_CODE = 7; ;
# 256
[ 0-9] | [ 1-9] [ 0-9] | [ 1] [ 0-9] [ 0-9] | [ 2] [ 0-4] [ 0-9] | [ 2] [ 5] [ 0-6] ) KUBE_PS1_FG_CODE = " ${ 1 } " ; ;
*) KUBE_PS1_FG_CODE = default
esac
if [ [ " ${ KUBE_PS1_FG_CODE } " = = "default" ] ] ; then
KUBE_PS1_FG_CODE = " ${ _KUBE_PS1_DEFAULT_FG } "
return
elif [ [ " ${ KUBE_PS1_SHELL } " = = "zsh" ] ] ; then
KUBE_PS1_FG_CODE = " %F{ $KUBE_PS1_FG_CODE } "
elif [ [ " ${ KUBE_PS1_SHELL } " = = "bash" ] ] ; then
if tput setaf 1 & > /dev/null; then
KUBE_PS1_FG_CODE = " $( tput setaf ${ KUBE_PS1_FG_CODE } ) "
elif [ [ $KUBE_PS1_FG_CODE -ge 0 ] ] && [ [ $KUBE_PS1_FG_CODE -le 256 ] ] ; then
KUBE_PS1_FG_CODE = " \033[38;5; ${ KUBE_PS1_FG_CODE } m "
else
KUBE_PS1_FG_CODE = " ${ _KUBE_PS1_DEFAULT_FG } "
fi
fi
echo ${ _KUBE_PS1_OPEN_ESC } ${ KUBE_PS1_FG_CODE } ${ _KUBE_PS1_CLOSE_ESC }
}
_kube_ps1_color_bg( ) {
local KUBE_PS1_BG_CODE
case " ${ 1 } " in
black) KUBE_PS1_BG_CODE = 0; ;
red) KUBE_PS1_BG_CODE = 1; ;
green) KUBE_PS1_BG_CODE = 2; ;
yellow) KUBE_PS1_BG_CODE = 3; ;
blue) KUBE_PS1_BG_CODE = 4; ;
magenta) KUBE_PS1_BG_CODE = 5; ;
cyan) KUBE_PS1_BG_CODE = 6; ;
white) KUBE_PS1_BG_CODE = 7; ;
# 256
[ 0-9] | [ 1-9] [ 0-9] | [ 1] [ 0-9] [ 0-9] | [ 2] [ 0-4] [ 0-9] | [ 2] [ 5] [ 0-6] ) KUBE_PS1_BG_CODE = " ${ 1 } " ; ;
*) KUBE_PS1_BG_CODE = $'\033[0m' ; ;
esac
if [ [ " ${ KUBE_PS1_BG_CODE } " = = "default" ] ] ; then
KUBE_PS1_FG_CODE = " ${ _KUBE_PS1_DEFAULT_BG } "
return
elif [ [ " ${ KUBE_PS1_SHELL } " = = "zsh" ] ] ; then
KUBE_PS1_BG_CODE = " %K{ $KUBE_PS1_BG_CODE } "
elif [ [ " ${ KUBE_PS1_SHELL } " = = "bash" ] ] ; then
if tput setaf 1 & > /dev/null; then
KUBE_PS1_BG_CODE = " $( tput setab ${ KUBE_PS1_BG_CODE } ) "
elif [ [ $KUBE_PS1_BG_CODE -ge 0 ] ] && [ [ $KUBE_PS1_BG_CODE -le 256 ] ] ; then
KUBE_PS1_BG_CODE = " \033[48;5; ${ KUBE_PS1_BG_CODE } m "
else
KUBE_PS1_BG_CODE = " ${ DEFAULT_BG } "
fi
fi
echo ${ OPEN_ESC } ${ KUBE_PS1_BG_CODE } ${ CLOSE_ESC }
}
_kube_ps1_binary_check( ) {
command -v " $1 " >/dev/null
command -v $1 >/dev/null
}
_kube_ps1_symbol( ) {
[ [ " ${ KUBE_PS1_SYMBOL_ENABLE } " = = false ] ] && return
KUBE_PS1_SYMBOL = " ${ KUBE_PS1_SYMBOL_DEFAULT } "
KUBE_PS1_SYMBOL_IMG = "\u2638 "
case " ${ KUBE_PS1_SHELL } " in
bash)
if ( ( BASH_VERSINFO[ 0] >= 4) ) && [ [ $'\u2388 ' != "\\u2388 " ] ] ; then
KUBE_PS1_SYMBOL = " ${ KUBE_PS1_SYMBOL_DEFAULT } "
# KUBE_PS1_SYMBOL=$'\u2388 '
KUBE_PS1_SYMBOL_IMG = $'\u2638 '
else
KUBE_PS1_SYMBOL = $'\xE2\x8E\x88 '
KUBE_PS1_SYMBOL_IMG = $'\xE2\x98\xB8 '
fi
; ;
zsh)
KUBE_PS1_SYMBOL = " ${ KUBE_PS1_SYMBOL_DEFAULT } "
KUBE_PS1_SYMBOL_IMG = "\u2638 " ; ;
*)
KUBE_PS1_SYMBOL = "k8s"
esac
if [ [ " ${ KUBE_PS1_SYMBOL_USE_IMG } " = = true ] ] ; then
KUBE_PS1_SYMBOL = " ${ KUBE_PS1_SYMBOL_IMG } "
@ -73,22 +183,24 @@ _kube_ps1_file_newer_than() {
local file = $1
local check_time = $2
zmodload -e "zsh/stat"
if [ [ " $? " -eq 0 ] ] ; then
mtime = $( stat +mtime " ${ file } " )
if [ [ " ${ KUBE_PS1_SHELL } " = = "zsh" ] ] ; then
mtime = $( zstat -L +mtime " ${ file } " )
elif stat -c "%s" /dev/null & > /dev/null; then
# GNU stat
mtime = $( stat -c %Y " ${ file } " )
mtime = $( stat -L - c %Y " ${ file } " )
else
# BSD stat
mtime = $( stat -f %m " $file " )
mtime = $( stat -L - f %m " $file " )
fi
[ [ " ${ mtime } " -gt " ${ check_time } " ] ]
}
_kube_ps1_update_cache( ) {
KUBECONFIG = " ${ KUBECONFIG : = $HOME /.kube/config } "
local return_code = $?
[ [ " ${ KUBE_PS1_ENABLED } " = = "off" ] ] && return $return_code
if ! _kube_ps1_binary_check " ${ KUBE_PS1_BINARY } " ; then
# No ability to fetch context/namespace; display N/A.
KUBE_PS1_CONTEXT = "BINARY-N/A"
@ -113,47 +225,147 @@ _kube_ps1_update_cache() {
return
fi
done
return $return_code
}
_kube_ps1_get_context_ns( ) {
_kube_ps1_get_context( ) {
if [ [ " ${ KUBE_PS1_CONTEXT_ENABLE } " = = true ] ] ; then
KUBE_PS1_CONTEXT = " $( ${ KUBE_PS1_BINARY } config current-context 2>/dev/null) "
# Set namespace to 'N/A' if it is not defined
KUBE_PS1_CONTEXT = " ${ KUBE_PS1_CONTEXT :- N /A } "
# Set the command time
KUBE_PS1_LAST_TIME = $EPOCHSECONDS
if [ [ ! -z " ${ KUBE_PS1_CLUSTER_FUNCTION } " ] ] ; then
KUBE_PS1_CONTEXT = $( $KUBE_PS1_CLUSTER_FUNCTION $KUBE_PS1_CONTEXT )
fi
fi
}
KUBE_PS1_CONTEXT = " $( ${ KUBE_PS1_BINARY } config current-context 2>/dev/null) "
if [ [ -z " ${ KUBE_PS1_CONTEXT } " ] ] ; then
KUBE_PS1_CONTEXT = "N/A"
KUBE_PS1_NAMESPACE = "N/A"
return
elif [ [ " ${ KUBE_PS1_NS_ENABLE } " = = true ] ] ; then
_kube_ps1_get_ns( ) {
if [ [ " ${ KUBE_PS1_NS_ENABLE } " = = true ] ] ; then
KUBE_PS1_NAMESPACE = " $( ${ KUBE_PS1_BINARY } config view --minify --output 'jsonpath={..namespace}' 2>/dev/null) "
# Set namespace to 'default' if it is not defined
KUBE_PS1_NAMESPACE = " ${ KUBE_PS1_NAMESPACE :- default } "
if [ [ ! -z " ${ KUBE_PS1_NAMESPACE_FUNCTION } " ] ] ; then
KUBE_PS1_NAMESPACE = $( $KUBE_PS1_NAMESPACE_FUNCTION $KUBE_PS1_NAMESPACE )
fi
fi
}
# function to disable the prompt on the current shell
kubeon( ) {
KUBE_PS1_ENABLED = true
_kube_ps1_get_context_ns( ) {
# Set the command time
if [ [ " ${ KUBE_PS1_SHELL } " = = "bash" ] ] ; then
if ( ( BASH_VERSINFO[ 0] >= 4 && BASH_VERSINFO[ 1] >= 2) ) ; then
KUBE_PS1_LAST_TIME = $( printf '%(%s)T' )
else
KUBE_PS1_LAST_TIME = $( date +%s)
fi
elif [ [ " ${ KUBE_PS1_SHELL } " = = "zsh" ] ] ; then
KUBE_PS1_LAST_TIME = $EPOCHSECONDS
fi
_kube_ps1_get_context
_kube_ps1_get_ns
}
# function to disable the prompt on the current shell
kubeoff( ) {
KUBE_PS1_ENABLED = false
# Set kube-ps1 shell defaults
_kube_ps1_init
_kubeon_usage( ) {
cat <<"EOF"
Toggle kube-ps1 prompt on
Usage: kubeon [ -g | --global] [ -h | --help]
With no arguments, turn off kube-ps1 status for this shell instance ( default) .
-g --global turn on kube-ps1 status globally
-h --help print this message
EOF
}
_kubeoff_usage( ) {
cat <<"EOF"
Toggle kube-ps1 prompt off
Usage: kubeoff [ -g | --global] [ -h | --help]
With no arguments, turn off kube-ps1 status for this shell instance ( default) .
-g --global turn off kube-ps1 status globally
-h --help print this message
EOF
}
kubeon( ) {
if [ [ " ${ 1 } " = = '-h' || " ${ 1 } " = = '--help' ] ] ; then
_kubeon_usage
elif [ [ " ${ 1 } " = = '-g' || " ${ 1 } " = = '--global' ] ] ; then
rm -f -- " ${ KUBE_PS1_DISABLE_PATH } "
elif [ [ " $# " -ne 0 ] ] ; then
echo -e " error: unrecognized flag ${ 1 } \\n "
_kubeon_usage
return
fi
KUBE_PS1_ENABLED = on
}
kubeoff( ) {
if [ [ " ${ 1 } " = = '-h' || " ${ 1 } " = = '--help' ] ] ; then
_kubeoff_usage
elif [ [ " ${ 1 } " = = '-g' || " ${ 1 } " = = '--global' ] ] ; then
mkdir -p -- " $( dirname " ${ KUBE_PS1_DISABLE_PATH } " ) "
touch -- " ${ KUBE_PS1_DISABLE_PATH } "
elif [ [ $# -ne 0 ] ] ; then
echo " error: unrecognized flag ${ 1 } " >& 2
_kubeoff_usage
return
fi
KUBE_PS1_ENABLED = off
}
# Build our prompt
kube_ps1 ( ) {
local reset_color = " %{ $reset_color %} "
[ [ ${ KUBE_PS1_ENABLED } != 'true' ] ] && return
KUBE_PS1 = " ${ reset_color } $KUBE_PS1_PREFIX "
KUBE_PS1 += " ${ KUBE_PS1_COLOR_SYMBOL } $( _kube_ps1_symbol) "
KUBE_PS1 += " ${ reset_color } $KUBE_PS1_SEPERATOR "
KUBE_PS1 += " ${ KUBE_PS1_COLOR_CONTEXT } $KUBE_PS1_CONTEXT ${ reset_color } "
KUBE_PS1 += " $KUBE_PS1_DIVIDER "
KUBE_PS1 += " ${ KUBE_PS1_COLOR_NS } $KUBE_PS1_NAMESPACE ${ reset_color } "
KUBE_PS1 += " $KUBE_PS1_SUFFIX "
kube_ps1( ) {
[ [ " ${ KUBE_PS1_ENABLED } " = = "off" ] ] && return
[ [ -z " ${ KUBE_PS1_CONTEXT } " ] ] && [ [ " ${ KUBE_PS1_CONTEXT_ENABLE } " = = true ] ] && return
local KUBE_PS1
local KUBE_PS1_RESET_COLOR = " ${ _KUBE_PS1_OPEN_ESC } ${ _KUBE_PS1_DEFAULT_FG } ${ _KUBE_PS1_CLOSE_ESC } "
# Background Color
[ [ -n " ${ KUBE_PS1_BG_COLOR } " ] ] && KUBE_PS1 += " $( _kube_ps1_color_bg ${ KUBE_PS1_BG_COLOR } ) "
# Prefix
[ [ -n " ${ KUBE_PS1_PREFIX } " ] ] && KUBE_PS1 += " ${ KUBE_PS1_PREFIX } "
# Symbol
KUBE_PS1 += " $( _kube_ps1_color_fg $KUBE_PS1_SYMBOL_COLOR ) $( _kube_ps1_symbol) ${ KUBE_PS1_RESET_COLOR } "
if [ [ -n " ${ KUBE_PS1_SEPARATOR } " ] ] && [ [ " ${ KUBE_PS1_SYMBOL_ENABLE } " = = true ] ] ; then
KUBE_PS1 += " ${ KUBE_PS1_SEPARATOR } "
fi
# Context
if [ [ " ${ KUBE_PS1_CONTEXT_ENABLE } " = = true ] ] ; then
KUBE_PS1 += " $( _kube_ps1_color_fg $KUBE_PS1_CTX_COLOR ) ${ KUBE_PS1_CONTEXT } ${ KUBE_PS1_RESET_COLOR } "
fi
# Namespace
if [ [ " ${ KUBE_PS1_NS_ENABLE } " = = true ] ] ; then
if [ [ -n " ${ KUBE_PS1_DIVIDER } " ] ] && [ [ " ${ KUBE_PS1_CONTEXT_ENABLE } " = = true ] ] ; then
KUBE_PS1 += " ${ KUBE_PS1_DIVIDER } "
fi
KUBE_PS1 += " $( _kube_ps1_color_fg ${ KUBE_PS1_NS_COLOR } ) ${ KUBE_PS1_NAMESPACE } ${ KUBE_PS1_RESET_COLOR } "
fi
# Suffix
[ [ -n " ${ KUBE_PS1_SUFFIX } " ] ] && KUBE_PS1 += " ${ KUBE_PS1_SUFFIX } "
# Close Background color if defined
[ [ -n " ${ KUBE_PS1_BG_COLOR } " ] ] && KUBE_PS1 += " ${ _KUBE_PS1_OPEN_ESC } ${ _KUBE_PS1_DEFAULT_BG } ${ _KUBE_PS1_CLOSE_ESC } "
echo " ${ KUBE_PS1 } "
}
Marc Cornellà
6 years ago