Skip to content

Commit

Permalink
theme/powerline.base: lint base a bit
Browse files Browse the repository at this point in the history
  • Loading branch information
gaelicWizard committed Jan 17, 2022
1 parent 714dedb commit b7e8434
Showing 1 changed file with 94 additions and 83 deletions.
177 changes: 94 additions & 83 deletions themes/powerline/powerline.base.bash
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
# shellcheck shell=bash
# shellcheck disable=SC2034 # Expected behavior for themes.

#To set color for foreground and background
function set_color() {
local fg='' bg=''
if [[ "${1:-}" != "-" ]]; then
Expand All @@ -13,65 +14,68 @@ function set_color() {
echo -e "\[\033[${fg}${bg}m\]"
}

#Customising User Info Segment
function __powerline_user_info_prompt() {
local user_info=""
local color=${USER_INFO_THEME_PROMPT_COLOR}
local user_info="${SHORT_USER:-${USER}}"
local color=${USER_INFO_THEME_PROMPT_COLOR?}

if [[ "${THEME_CHECK_SUDO}" = true ]]; then
sudo -vn 1> /dev/null 2>&1 && color=${USER_INFO_THEME_PROMPT_COLOR_SUDO}
if [[ "${THEME_CHECK_SUDO:-false}" == true ]]; then
if sudo -vn 2> /dev/null; then
color=${USER_INFO_THEME_PROMPT_COLOR_SUDO?}
fi
fi

case "${POWERLINE_PROMPT_USER_INFO_MODE}" in
case "${POWERLINE_PROMPT_USER_INFO_MODE:-}" in
"sudo")
if [[ "${color}" = "${USER_INFO_THEME_PROMPT_COLOR_SUDO}" ]]; then
if [[ "${color}" == "${USER_INFO_THEME_PROMPT_COLOR_SUDO?}" ]]; then
user_info="!"
fi
;;
*)
local user=${SHORT_USER:-${USER}}
if [[ -n "${SSH_CLIENT}" ]] || [[ -n "${SSH_CONNECTION}" ]]; then
user_info="${USER_INFO_SSH_CHAR}${user}"
else
user_info="${user}"
if [[ -n "${SSH_CLIENT:-}" ]] || [[ -n "${SSH_CONNECTION:-}" ]]; then
user_info="${USER_INFO_SSH_CHAR?}${user_info}"
fi
;;
esac
[[ -n "${user_info}" ]] && echo "${user_info}|${color}"
echo "${user_info}|${color}"
}

function __powerline_terraform_prompt() {
local terraform_workspace=""

if [ -d .terraform ]; then
if [[ -d .terraform ]]; then
terraform_workspace="$(terraform_workspace_prompt)"
[[ -n "${terraform_workspace}" ]] && echo "${TERRAFORM_CHAR}${terraform_workspace}|${TERRAFORM_THEME_PROMPT_COLOR}"
[[ -n "${terraform_workspace}" ]] && echo "${TERRAFORM_CHAR?}${terraform_workspace}|${TERRAFORM_THEME_PROMPT_COLOR?}"
fi
}

function __powerline_gcloud_prompt() {
local active_gcloud_account=""

active_gcloud_account="$(active_gcloud_account_prompt)"
[[ -n "${active_gcloud_account}" ]] && echo "${GCLOUD_CHAR}${active_gcloud_account}|${GCLOUD_THEME_PROMPT_COLOR}"
[[ -n "${active_gcloud_account}" ]] && echo "${GCLOUD_CHAR?}${active_gcloud_account}|${GCLOUD_THEME_PROMPT_COLOR?}"
}

function __powerline_node_prompt() {
local node_version=""

node_version="$(node_version_prompt)"
[[ -n "${node_version}" ]] && echo "${NODE_CHAR}${node_version}|${NODE_THEME_PROMPT_COLOR}"
[[ -n "${node_version}" ]] && echo "${NODE_CHAR?}${node_version}|${NODE_THEME_PROMPT_COLOR?}"
}

#Customising Ruby Prompt
function __powerline_ruby_prompt() {
local ruby_version=""
local ruby_version

if _command_exists rvm; then
ruby_version="$(rvm_version_prompt)"
elif _command_exists rbenv; then
ruby_version=$(rbenv_version_prompt)
fi

[[ -n "${ruby_version}" ]] && echo "${RUBY_CHAR}${ruby_version}|${RUBY_THEME_PROMPT_COLOR}"
if [[ -n "${ruby_version:-}" ]]; then
echo "${RUBY_CHAR?}${ruby_version}|${RUBY_THEME_PROMPT_COLOR?}"
fi
}

function __powerline_k8s_context_prompt() {
Expand All @@ -81,7 +85,7 @@ function __powerline_k8s_context_prompt() {
kubernetes_context="$(k8s_context_prompt)"
fi

[[ -n "${kubernetes_context}" ]] && echo "${KUBERNETES_CONTEXT_THEME_CHAR}${kubernetes_context}|${KUBERNETES_CONTEXT_THEME_PROMPT_COLOR}"
[[ -n "${kubernetes_context}" ]] && echo "${KUBERNETES_CONTEXT_THEME_CHAR?}${kubernetes_context}|${KUBERNETES_CONTEXT_THEME_PROMPT_COLOR?}"
}

function __powerline_k8s_namespace_prompt() {
Expand All @@ -91,65 +95,69 @@ function __powerline_k8s_namespace_prompt() {
kubernetes_namespace="$(k8s_namespace_prompt)"
fi

[[ -n "${kubernetes_namespace}" ]] && echo "${KUBERNETES_NAMESPACE_THEME_CHAR}${kubernetes_namespace}|${KUBERNETES_NAMESPACE_THEME_PROMPT_COLOR}"
[[ -n "${kubernetes_namespace}" ]] && echo "${KUBERNETES_NAMESPACE_THEME_CHAR?}${kubernetes_namespace}|${KUBERNETES_NAMESPACE_THEME_PROMPT_COLOR?}"
}

#Customising Python (venv) Prompt
function __powerline_python_venv_prompt() {
local python_venv=""

if [[ -n "${CONDA_DEFAULT_ENV:-}" ]]; then
python_venv="${CONDA_DEFAULT_ENV}"
PYTHON_VENV_CHAR=${CONDA_PYTHON_VENV_CHAR}
PYTHON_VENV_CHAR=${CONDA_PYTHON_VENV_CHAR?}
elif [[ -n "${VIRTUAL_ENV:-}" ]]; then
python_venv="${VIRTUAL_ENV##*/}"
fi

[[ -n "${python_venv}" ]] && echo "${PYTHON_VENV_CHAR}${python_venv}|${PYTHON_VENV_THEME_PROMPT_COLOR}"
[[ -n "${python_venv}" ]] && echo "${PYTHON_VENV_CHAR?}${python_venv}|${PYTHON_VENV_THEME_PROMPT_COLOR?}"
}

#Customising SCM(GIT) Prompt
function __powerline_scm_prompt() {
local color=""
local scm_prompt=""

scm_prompt_vars

if [[ "${SCM_NONE_CHAR}" != "${SCM_CHAR}" ]]; then
if [[ "${SCM_DIRTY}" -eq 3 ]]; then
color=${SCM_THEME_PROMPT_STAGED_COLOR}
elif [[ "${SCM_DIRTY}" -eq 2 ]]; then
color=${SCM_THEME_PROMPT_UNSTAGED_COLOR}
elif [[ "${SCM_DIRTY}" -eq 1 ]]; then
color=${SCM_THEME_PROMPT_DIRTY_COLOR}
if [[ "${SCM_NONE_CHAR?}" != "${SCM_CHAR?}" ]]; then
if [[ "${SCM_DIRTY?}" -eq 3 ]]; then
color=${SCM_THEME_PROMPT_STAGED_COLOR?}
elif [[ "${SCM_DIRTY?}" -eq 2 ]]; then
color=${SCM_THEME_PROMPT_UNSTAGED_COLOR?}
elif [[ "${SCM_DIRTY?}" -eq 1 ]]; then
color=${SCM_THEME_PROMPT_DIRTY_COLOR?}
elif [[ "${SCM_DIRTY?}" -eq 0 ]]; then
color=${SCM_THEME_PROMPT_CLEAN_COLOR?}
else
color=${SCM_THEME_PROMPT_CLEAN_COLOR}
color=${SCM_THEME_PROMPT_COLOR?}
fi
if [[ "${SCM_GIT_CHAR}" == "${SCM_CHAR}" ]]; then
scm_prompt+="${SCM_CHAR}${SCM_BRANCH}${SCM_STATE}"
elif [[ "${SCM_P4_CHAR}" == "${SCM_CHAR}" ]]; then
scm_prompt+="${SCM_CHAR}${SCM_BRANCH}${SCM_STATE}"
elif [[ "${SCM_HG_CHAR}" == "${SCM_CHAR}" ]]; then
scm_prompt+="${SCM_CHAR}${SCM_BRANCH}${SCM_STATE}"
elif [[ "${SCM_SVN_CHAR}" == "${SCM_CHAR}" ]]; then
scm_prompt+="${SCM_CHAR}${SCM_BRANCH}${SCM_STATE}"
if [[ "${SCM_GIT_CHAR?}" == "${SCM_CHAR?}" ]]; then
scm_prompt+="${SCM_CHAR}${SCM_BRANCH?}${SCM_STATE?}"
elif [[ "${SCM_P4_CHAR?}" == "${SCM_CHAR}" ]]; then
scm_prompt+="${SCM_CHAR}${SCM_BRANCH?}${SCM_STATE?}"
elif [[ "${SCM_HG_CHAR?}" == "${SCM_CHAR}" ]]; then
scm_prompt+="${SCM_CHAR}${SCM_BRANCH?}${SCM_STATE?}"
elif [[ "${SCM_SVN_CHAR?}" == "${SCM_CHAR}" ]]; then
scm_prompt+="${SCM_CHAR}${SCM_BRANCH?}${SCM_STATE?}"
fi
echo "${scm_prompt?}|${color}"
fi
}

function __powerline_cwd_prompt() {
echo "\w|${CWD_THEME_PROMPT_COLOR}"
echo "\w|${CWD_THEME_PROMPT_COLOR?}"
}

function __powerline_hostname_prompt() {
echo "${SHORT_HOSTNAME:-$(hostname -s)}|${HOST_THEME_PROMPT_COLOR}"
echo "\h|${HOST_THEME_PROMPT_COLOR?}"
}

function __powerline_wd_prompt() {
echo "\W|${CWD_THEME_PROMPT_COLOR}"
}

function __powerline_clock_prompt() {
echo "$(date +"${THEME_CLOCK_FORMAT}")|${CLOCK_THEME_PROMPT_COLOR}"
echo "$(date +"${THEME_CLOCK_FORMAT?}")|${CLOCK_THEME_PROMPT_COLOR?}"
}

function __powerline_battery_prompt() {
Expand All @@ -160,146 +168,149 @@ function __powerline_battery_prompt() {
true
else
if [[ "$((10#${battery_status}))" -le 5 ]]; then
color="${BATTERY_STATUS_THEME_PROMPT_CRITICAL_COLOR}"
color="${BATTERY_STATUS_THEME_PROMPT_CRITICAL_COLOR?}"
elif [[ "$((10#${battery_status}))" -le 25 ]]; then
color="${BATTERY_STATUS_THEME_PROMPT_LOW_COLOR}"
color="${BATTERY_STATUS_THEME_PROMPT_LOW_COLOR?}"
else
color="${BATTERY_STATUS_THEME_PROMPT_GOOD_COLOR}"
color="${BATTERY_STATUS_THEME_PROMPT_GOOD_COLOR?}"
fi
ac_adapter_connected && battery_status="${BATTERY_AC_CHAR}${battery_status}"
ac_adapter_connected && battery_status="${BATTERY_AC_CHAR?}${battery_status}"
echo "${battery_status}%|${color}"
fi
}

function __powerline_in_vim_prompt() {
if [[ -n "$VIMRUNTIME" ]]; then
echo "${IN_VIM_THEME_PROMPT_TEXT}|${IN_VIM_THEME_PROMPT_COLOR}"
if [[ -n "${VIMRUNTIME:-}" ]]; then
echo "${IN_VIM_THEME_PROMPT_TEXT?}|${IN_VIM_THEME_PROMPT_COLOR?}"
fi
}

function __powerline_aws_profile_prompt() {
if [[ -n "${AWS_PROFILE}" ]]; then
echo "${AWS_PROFILE_CHAR}${AWS_PROFILE}|${AWS_PROFILE_PROMPT_COLOR}"
if [[ -n "${AWS_PROFILE:-}" ]]; then
echo "${AWS_PROFILE_CHAR?}${AWS_PROFILE?}|${AWS_PROFILE_PROMPT_COLOR?}"
fi
}

function __powerline_in_toolbox_prompt() {
if [ -f /run/.containerenv ] && [ -f /run/.toolboxenv ]; then
echo "${IN_TOOLBOX_THEME_PROMPT_TEXT}|${IN_TOOLBOX_THEME_PROMPT_COLOR}"
if [[ -f /run/.containerenv ]] && [[ -f /run/.toolboxenv ]]; then
echo "${IN_TOOLBOX_THEME_PROMPT_TEXT?}|${IN_TOOLBOX_THEME_PROMPT_COLOR?}"
fi
}

function __powerline_shlvl_prompt() {
if [[ "${SHLVL}" -gt 1 ]]; then
local prompt="${SHLVL_THEME_PROMPT_CHAR}"
local prompt="${SHLVL_THEME_PROMPT_CHAR?}"
local level=$((SHLVL - 1))
echo "${prompt}${level}|${SHLVL_THEME_PROMPT_COLOR}"
echo "${prompt}${level}|${SHLVL_THEME_PROMPT_COLOR?}"
fi
}

function __powerline_dirstack_prompt() {
if [[ "${#DIRSTACK[@]}" -gt 1 ]]; then
local depth=$((${#DIRSTACK[@]} - 1))
local prompt="${DIRSTACK_THEME_PROMPT_CHAR}"
local prompt="${DIRSTACK_THEME_PROMPT_CHAR?}"
if [[ "${depth}" -ge 2 ]]; then
prompt+="${depth}"
fi
echo "${prompt}|${DIRSTACK_THEME_PROMPT_COLOR}"
echo "${prompt}|${DIRSTACK_THEME_PROMPT_COLOR?}"
fi
}

function __powerline_history_number_prompt() {
echo "${HISTORY_NUMBER_THEME_PROMPT_CHAR}\!|${HISTORY_NUMBER_THEME_PROMPT_COLOR}"
echo "${HISTORY_NUMBER_THEME_PROMPT_CHAR?}\!|${HISTORY_NUMBER_THEME_PROMPT_COLOR?}"
}

function __powerline_command_number_prompt() {
echo "${COMMAND_NUMBER_THEME_PROMPT_CHAR}\#|${COMMAND_NUMBER_THEME_PROMPT_COLOR}"
echo "${COMMAND_NUMBER_THEME_PROMPT_CHAR?}\#|${COMMAND_NUMBER_THEME_PROMPT_COLOR?}"
}

function __powerline_duration_prompt() {
local duration
duration=$(_command_duration)
[[ -n "$duration" ]] && echo "${duration}|${COMMAND_DURATION_PROMPT_COLOR}"
[[ -n "$duration" ]] && echo "${duration}|${COMMAND_DURATION_PROMPT_COLOR?}"
}

function __powerline_left_segment() {
local params
local -a params
IFS="|" read -ra params <<< "${1}"
local pad_before_segment=" "

if [[ "${SEGMENTS_AT_LEFT}" -eq 0 ]]; then
if [[ "${POWERLINE_COMPACT_BEFORE_FIRST_SEGMENT}" -ne 0 ]]; then
#for seperator character
if [[ "${SEGMENTS_AT_LEFT?}" -eq 0 ]]; then
if [[ "${POWERLINE_COMPACT_BEFORE_FIRST_SEGMENT:-}" -ne 0 ]]; then
pad_before_segment=""
fi
else
if [[ "${POWERLINE_COMPACT_AFTER_SEPARATOR}" -ne 0 ]]; then
if [[ "${POWERLINE_COMPACT_AFTER_SEPARATOR:-}" -ne 0 ]]; then
pad_before_segment=""
fi
# Since the previous segment wasn't the last segment, add padding, if needed
#
if [[ "${POWERLINE_COMPACT_BEFORE_SEPARATOR}" -eq 0 ]]; then
LEFT_PROMPT+="$(set_color - "${LAST_SEGMENT_COLOR}") ${normal?}"
if [[ "${POWERLINE_COMPACT_BEFORE_SEPARATOR:-0}" -eq 0 ]]; then
LEFT_PROMPT+="$(set_color - "${LAST_SEGMENT_COLOR?}") ${normal?}"
fi
if [[ "${LAST_SEGMENT_COLOR}" -eq "${params[1]}" ]]; then
LEFT_PROMPT+="$(set_color - "${LAST_SEGMENT_COLOR}")${POWERLINE_LEFT_SEPARATOR_SOFT}${normal?}"
if [[ "${LAST_SEGMENT_COLOR?}" -eq "${params[1]:-}" ]]; then
LEFT_PROMPT+="$(set_color - "${LAST_SEGMENT_COLOR?}")${POWERLINE_LEFT_SEPARATOR_SOFT?}${normal?}"
else
LEFT_PROMPT+="$(set_color "${LAST_SEGMENT_COLOR}" "${params[1]}")${POWERLINE_LEFT_SEPARATOR}${normal?}"
LEFT_PROMPT+="$(set_color "${LAST_SEGMENT_COLOR?}" "${params[1]:-}")${POWERLINE_LEFT_SEPARATOR?}${normal?}"
fi
fi

LEFT_PROMPT+="$(set_color - "${params[1]}")${pad_before_segment}${params[0]}${normal}"
LAST_SEGMENT_COLOR=${params[1]}
#change here to cahnge fg color
LEFT_PROMPT+="$(set_color - "${params[1]:-}")${pad_before_segment}${params[0]}${normal?}"
#seperator char color == current bg
LAST_SEGMENT_COLOR="${params[1]:-}"
((SEGMENTS_AT_LEFT += 1))
}

function __powerline_left_last_segment_padding() {
LEFT_PROMPT+="$(set_color - "${LAST_SEGMENT_COLOR}") ${normal?}"
LEFT_PROMPT+="$(set_color - "${LAST_SEGMENT_COLOR?}") ${normal?}"
}

function __powerline_last_status_prompt() {
[[ "$1" -ne 0 ]] && echo "${1}|${LAST_STATUS_THEME_PROMPT_COLOR}"
[[ "$1" -ne 0 ]] && echo "${1}|${LAST_STATUS_THEME_PROMPT_COLOR?}"
}

function __powerline_prompt_command() {
local last_status="$?" ## always the first
local separator_char="${POWERLINE_PROMPT_CHAR}" info prompt_color
local separator_char="${POWERLINE_PROMPT_CHAR?}" info prompt_color

LEFT_PROMPT=""
SEGMENTS_AT_LEFT=0
LAST_SEGMENT_COLOR=""

_save-and-reload-history "${HISTORY_AUTOSAVE:-0}"

if [[ -n "${POWERLINE_PROMPT_DISTRO_LOGO}" ]]; then
LEFT_PROMPT+="$(set_color "${PROMPT_DISTRO_LOGO_COLOR}" "${PROMPT_DISTRO_LOGO_COLORBG}")${PROMPT_DISTRO_LOGO}$(set_color - -)"
if [[ -n "${POWERLINE_PROMPT_DISTRO_LOGO:-}" ]]; then
LEFT_PROMPT+="$(set_color "${PROMPT_DISTRO_LOGO_COLOR?}" "${PROMPT_DISTRO_LOGO_COLORBG?}")${PROMPT_DISTRO_LOGO?}$(set_color - -)"
fi

## left prompt ##
for segment in $POWERLINE_PROMPT; do
info="$(__powerline_"${segment}"_prompt)"
for segment in ${POWERLINE_PROMPT?}; do
info="$("__powerline_${segment}_prompt")"
[[ -n "${info}" ]] && __powerline_left_segment "${info}"
done

[[ "${last_status}" -ne 0 ]] && __powerline_left_segment "$(__powerline_last_status_prompt "${last_status}")"

if [[ -n "${LEFT_PROMPT}" ]] && [[ "${POWERLINE_COMPACT_AFTER_LAST_SEGMENT:-}" -eq 0 ]]; then
if [[ -n "${LEFT_PROMPT:-}" ]] && [[ "${POWERLINE_COMPACT_AFTER_LAST_SEGMENT:-0}" -eq 0 ]]; then
__powerline_left_last_segment_padding
fi

# By default we try to match the prompt to the adjacent segment's background color,
# but when part of the prompt exists within that segment, we instead match the foreground color.
prompt_color="$(set_color "${LAST_SEGMENT_COLOR}" -)"
if [[ -n "${LEFT_PROMPT}" ]] && [[ -n "${POWERLINE_LEFT_LAST_SEGMENT_PROMPT_CHAR}" ]]; then
LEFT_PROMPT+="$(set_color - "${LAST_SEGMENT_COLOR}")${POWERLINE_LEFT_LAST_SEGMENT_PROMPT_CHAR}"
prompt_color="$(set_color "${LAST_SEGMENT_COLOR?}" -)"
if [[ -n "${LEFT_PROMPT:-}" ]] && [[ -n "${POWERLINE_LEFT_LAST_SEGMENT_PROMPT_CHAR:-}" ]]; then
LEFT_PROMPT+="$(set_color - "${LAST_SEGMENT_COLOR?}")${POWERLINE_LEFT_LAST_SEGMENT_PROMPT_CHAR}"
prompt_color="${normal?}"
fi
[[ -n "${LEFT_PROMPT}" ]] && LEFT_PROMPT+="${prompt_color}${separator_char}${normal?}"
[[ -n "${LEFT_PROMPT:-}" ]] && LEFT_PROMPT+="${prompt_color}${separator_char}${normal?}"

if [[ "${POWERLINE_COMPACT_PROMPT:-}" -eq 0 ]]; then
if [[ "${POWERLINE_COMPACT_PROMPT:-0}" -eq 0 ]]; then
LEFT_PROMPT+=" "
fi

PS1="${LEFT_PROMPT}"
PS1="${LEFT_PROMPT?}"

## cleanup ##
unset LAST_SEGMENT_COLOR \
Expand Down

0 comments on commit b7e8434

Please sign in to comment.