add: theme totvs and lib kubectl

This commit is contained in:
Franklin Moraes 2024-08-16 13:25:11 -03:00
commit daec6dfbae
2 changed files with 119 additions and 0 deletions

110
lib/kubectl.zsh Normal file
View file

@ -0,0 +1,110 @@
setopt prompt_subst
autoload -U add-zsh-hook
function() {
local namespace separator binary
# Specify the separator between context and namespace
zstyle -s ':zsh-kubectl-prompt:' separator separator
if [[ -z "$separator" ]]; then
zstyle ':zsh-kubectl-prompt:' separator '/'
fi
# Display the current namespace if `namespace` is true
zstyle -s ':zsh-kubectl-prompt:' namespace namespace
if [[ -z "$namespace" ]]; then
zstyle ':zsh-kubectl-prompt:' namespace true
fi
# Specify the binary to get the information from kubeconfig (e.g. `oc`)
zstyle -s ':zsh-kubectl-binary:' binary binary
if [[ -z "$binary" ]]; then
zstyle ':zsh-kubectl-prompt:' binary "kubectl"
fi
}
add-zsh-hook precmd _zsh_kubectl_prompt_precmd
function _zsh_kubectl_prompt_precmd() {
local kubeconfig config updated_at now context namespace ns separator modified_time_fmt binary
zstyle -s ':zsh-kubectl-prompt:' binary binary
if ! command -v "$binary" >/dev/null; then
ZSH_KUBECTL_PROMPT="${binary} command not found"
return 1
fi
kubeconfig="$HOME/.kube/config"
if [[ -n "$KUBECONFIG" ]]; then
kubeconfig="$KUBECONFIG"
fi
zstyle -s ':zsh-kubectl-prompt:' modified_time_fmt modified_time_fmt
if [[ -z "$modified_time_fmt" ]]; then
# Check the stat command because it has a different syntax between GNU coreutils and FreeBSD.
if stat --help >/dev/null 2>&1; then
modified_time_fmt='-c%y' # GNU coreutils
else
modified_time_fmt='-f%m' # FreeBSD
fi
zstyle ':zsh-kubectl-prompt:' modified_time_fmt $modified_time_fmt
fi
# KUBECONFIG environment variable can hold a list of kubeconfig files that is colon-delimited.
# Therefore, if KUBECONFIG has been held multiple files, each files need to be checked.
while read -d ":" config; do
if ! now="${now}$(stat -L $modified_time_fmt "$config" 2>/dev/null)"; then
ZSH_KUBECTL_PROMPT="$config doesn't exist"
return 1
fi
done <<< "${kubeconfig}:"
zstyle -s ':zsh-kubectl-prompt:' updated_at updated_at
if [[ "$updated_at" == "$now" ]]; then
return 0
fi
zstyle ':zsh-kubectl-prompt:' updated_at "$now"
# Set environment variable if context is not set
if ! context="$("$binary" config current-context 2>/dev/null)"; then
ZSH_KUBECTL_PROMPT="current-context is not set"
return 1
fi
ZSH_KUBECTL_USER="$("$binary" config view -o "jsonpath={.contexts[?(@.name==\"$context\")].context.user}")"
ZSH_KUBECTL_CONTEXT="${context}"
ns="$("$binary" config view -o "jsonpath={.contexts[?(@.name==\"$context\")].context.namespace}")"
[[ -z "$ns" ]] && ns="default"
ZSH_KUBECTL_NAMESPACE="${ns}"
function ZSH_K8S() {
local color="white"
if [[ "$ZSH_KUBECTL_CONTEXT" =~ "vmw-dev" ]]; then
color=blue
fi
if [[ "$ZSH_KUBECTL_CONTEXT" =~ "gcp-stg" ]]; then
color=yellow
fi
if [[ "$ZSH_KUBECTL_CONTEXT" =~ "gcp-prd" ]]; then
color=red
fi
echo "%{$fg[$color]%}$ZSH_KUBECTL_CONTEXT/$ZSH_KUBECTL_NAMESPACE%{$reset_color%}"
}
# Specify the entry before prompt (default empty)
zstyle -s ':zsh-kubectl-prompt:' preprompt preprompt
# Specify the entry after prompt (default empty)
zstyle -s ':zsh-kubectl-prompt:' postprompt postprompt
# Set environment variable without namespace
zstyle -s ':zsh-kubectl-prompt:' namespace namespace
if [[ "$namespace" != true ]]; then
ZSH_KUBECTL_PROMPT="${preprompt}${context}${postprompt}"
return 0
fi
zstyle -s ':zsh-kubectl-prompt:' separator separator
ZSH_KUBECTL_PROMPT="${preprompt}${context}${separator}${ns}${postprompt}"
return 0
}

9
themes/totvs.zsh-theme Normal file
View file

@ -0,0 +1,9 @@
PROMPT=$'
%{$bg[green]%}%{$fg_bold[black]%}%D{❪%H:%M:%S❫}%{$reset_color%} %{$bg[blue]%}%{$fg_bold[white]%}❪%n@%m❫%{$reset_color%}\
%{$bg[black]%}%{$fg_bold[white]%}❪$(ZSH_K8S)❫%{$reset_color%} %{$bg[black]%}%{$fg_bold[cyan]%}❪%~❫%{$reset_color%} $(git_prompt_info)\
%{$bg[black]%}%{$fg_no_bold[magenta]%}❱❱❱%{$reset_color%} '
ZSH_THEME_GIT_PROMPT_PREFIX="%{$bg[black]%}%{$fg_bold[yellow]%}❪git-branch%{$reset_color%}%{$fg_bold[blue]%}(%{$fg_bold[red]%}";
ZSH_THEME_GIT_PROMPT_SUFFIX="%{$reset_color%}";
ZSH_THEME_GIT_PROMPT_DIRTY="%{$bg[black]%}%{$fg_bold[blue]%})%{$reset_color%}%{$bg[black]%}%{$fg_bold[yellow]%}❫🚨";
ZSH_THEME_GIT_PROMPT_CLEAN="%{$bg[black]%}%{$fg_bold[blue]%})%{$reset_color%}%{$bg[black]%}%{$fg_bold[yellow]%}❫✅";