Last active
February 27, 2024 06:07
-
-
Save shrinath-shenoy/126b57cb8d7123f95875d3a995fd15e3 to your computer and use it in GitHub Desktop.
Oh My ZSH theme with virtualenv support
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
function theme_precmd { | |
local TERMWIDTH=$(( COLUMNS - ${ZLE_RPROMPT_INDENT:-1} )) | |
PR_FILLBAR="" | |
PR_PWDLEN="" | |
PR_VIRTUALENV="" | |
local promptsize=${#${(%):-(%n@%m:%l)-()-}} | |
local rubypromptsize=${#${(%)$(ruby_prompt_info)}} | |
local timepromtsize=${#${(%):-%D{%H:%M:%S}-()-}} | |
local datepromtsize=${#${(%):-%D{%a%b%d},-()-}} | |
local pwdsize=${#${(%):-%~}} | |
local venvsize=${#${VIRTUAL_ENV_PROMPT:q}} | |
export VIRTUAL_ENV_DISABLE_PROMPT=true | |
# check if venv is not zero | |
if [[ $venvsize -ne 0 ]]; then | |
(( venvsize += 3 )) | |
PR_VIRTUALENV="($VIRTUAL_ENV_PROMPT:q)-" | |
fi | |
# Truncate the path if it's too long. | |
if (( rubypromptsize + venvsize + timepromtsize + datepromtsize > TERMWIDTH )); then | |
(( PR_PWDLEN = TERMWIDTH - promptsize + rubypromptsize + venvsize + pwdsize )) | |
elif [[ "${langinfo[CODESET]}" = UTF-8 ]]; then | |
PR_FILLBAR="\${(l:$(( TERMWIDTH - (rubypromptsize + venvsize + timepromtsize + datepromtsize) ))::${PR_HBAR}:)}" | |
else | |
PR_FILLBAR="${PR_SHIFT_IN}\${(l:$(( TERMWIDTH - (rubypromptsize + venvsize + timepromtsize + datepromtsize) ))::${altchar[q]:--}:)}${PR_SHIFT_OUT}" | |
fi | |
} | |
function theme_preexec { | |
setopt local_options extended_glob | |
if [[ "$TERM" = "screen" ]]; then | |
local CMD=${1[(wr)^(*=*|sudo|-*)]} | |
echo -n "\ek$CMD\e\\" | |
fi | |
} | |
autoload -U add-zsh-hook | |
add-zsh-hook precmd theme_precmd | |
add-zsh-hook preexec theme_preexec | |
# Set the prompt | |
# Need this so the prompt will work. | |
setopt prompt_subst | |
# See if we can use colors. | |
autoload zsh/terminfo | |
for color in RED GREEN YELLOW BLUE MAGENTA CYAN WHITE GREY; do | |
typeset -g PR_$color="%{$terminfo[bold]$fg[${(L)color}]%}" | |
typeset -g PR_LIGHT_$color="%{$fg[${(L)color}]%}" | |
done | |
PR_NO_COLOUR="%{$terminfo[sgr0]%}" | |
# Modify Git prompt | |
ZSH_THEME_GIT_PROMPT_PREFIX=" on %{$fg[green]%}" | |
ZSH_THEME_GIT_PROMPT_SUFFIX="%{$reset_color%}" | |
ZSH_THEME_GIT_PROMPT_DIRTY="" | |
ZSH_THEME_GIT_PROMPT_CLEAN="" | |
ZSH_THEME_GIT_PROMPT_ADDED="%{$fg[green]%} %{%G✚%}" | |
ZSH_THEME_GIT_PROMPT_MODIFIED="%{$fg[blue]%} %{%G✹%}" | |
ZSH_THEME_GIT_PROMPT_DELETED="%{$fg[red]%} %{%G✖%}" | |
ZSH_THEME_GIT_PROMPT_RENAMED="%{$fg[magenta]%} %{%G➜%}" | |
ZSH_THEME_GIT_PROMPT_UNMERGED="%{$fg[yellow]%} %{%G═%}" | |
ZSH_THEME_GIT_PROMPT_UNTRACKED="%{$fg[cyan]%} %{%G✭%}" | |
# Use extended characters to look nicer if supported. | |
if [[ "${langinfo[CODESET]}" = UTF-8 ]]; then | |
PR_SET_CHARSET="" | |
PR_HBAR="─" | |
PR_ULCORNER="┌" | |
PR_LLCORNER="└" | |
PR_URCORNER="┐" | |
PR_LRCORNER="┘" | |
else | |
typeset -g -A altchar | |
set -A altchar ${(s..)terminfo[acsc]} | |
# Some stuff to help us draw nice lines | |
PR_SET_CHARSET="%{$terminfo[enacs]%}" | |
PR_SHIFT_IN="%{$terminfo[smacs]%}" | |
PR_SHIFT_OUT="%{$terminfo[rmacs]%}" | |
PR_HBAR="${PR_SHIFT_IN}${altchar[q]:--}${PR_SHIFT_OUT}" | |
PR_ULCORNER="${PR_SHIFT_IN}${altchar[l]:--}${PR_SHIFT_OUT}" | |
PR_LLCORNER="${PR_SHIFT_IN}${altchar[m]:--}${PR_SHIFT_OUT}" | |
PR_LRCORNER="${PR_SHIFT_IN}${altchar[j]:--}${PR_SHIFT_OUT}" | |
PR_URCORNER="${PR_SHIFT_IN}${altchar[k]:--}${PR_SHIFT_OUT}" | |
fi | |
# Decide if we need to set titlebar text. | |
case $TERM in | |
xterm*) | |
PR_TITLEBAR=$'%{\e]0;%(!.-=*[ROOT]*=- | .)%n@%m:%~ | ${COLUMNS}x${LINES} | %y\a%}' | |
;; | |
screen) | |
PR_TITLEBAR=$'%{\e_screen \005 (\005t) | %(!.-=[ROOT]=- | .)%n@%m:%~ | ${COLUMNS}x${LINES} | %y\e\\%}' | |
;; | |
*) | |
PR_TITLEBAR="" | |
;; | |
esac | |
# Decide whether to set a screen title | |
if [[ "$TERM" = "screen" ]]; then | |
PR_STITLE=$'%{\ekzsh\e\\%}' | |
else | |
PR_STITLE="" | |
fi | |
# Finally, the prompt. | |
PROMPT='${PR_SET_CHARSET}${PR_STITLE}${(e)PR_TITLEBAR}\ | |
${PR_CYAN}${PR_ULCORNER}${PR_HBAR}${PR_VIRTUALENV}${PR_LIGHT_BLUE}[\ | |
${PR_YELLOW}%D{%H:%M:%S}\ | |
${PR_LIGHT_BLUE}${PR_BLUE}]${PR_CYAN}${(e)PR_FILLBAR}\ | |
${PR_BLUE}[${PR_YELLOW}%D{%a,%b%d}${PR_BLUE}]\ | |
${PR_CYAN}${PR_HBAR}${PR_URCORNER}\ | |
${PR_CYAN}${PR_LLCORNER}${PR_HBAR}${PR_GREY}(\ | |
${PR_GREEN}%${PR_PWDLEN}<...<%~%<<\ | |
$(ruby_prompt_info)${PR_BLUE}$(git_prompt_info)$(git_prompt_status)${PR_GREY})${PR_CYAN}${PR_HBAR}\ | |
${PR_HBAR}\ | |
>${PR_NO_COLOUR} ' | |
# display exitcode on the right when > 0 | |
return_code="%(?..%{$fg[red]%}%? ↵ %{$reset_color%})" | |
RPROMPT=' $return_code${PR_CYAN}>_${PR_GREY}(\ | |
${PR_CYAN}%(!.%SROOT%s.%n)${PR_GREY})${PR_CYAN}${PR_LRCORNER}${PR_NO_COLOUR}' | |
PS2='${PR_CYAN}${PR_HBAR}\ | |
${PR_BLUE}${PR_HBAR}(\ | |
${PR_LIGHT_GREEN}%_${PR_BLUE})${PR_HBAR}\ | |
${PR_CYAN}${PR_HBAR}${PR_NO_COLOUR} ' |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
This is based on top of https://github.com/ohmyzsh/ohmyzsh/wiki/Themes#jonathan theme with
virtualenvwrapper
highlight brough within the theme