# Filename: prompt_adam3_setup # Purpose: a zsh-prompt that changes the color of your prompt based on # the hostname of your current host # Authors: Dennis Eriksen # Bug-Reports: Email # License: Distributed under the same BSD-ish license as zsh itself. ################################################################################ # adam3 prompt theme prompt_adam3_help () { cat <<'EOF' This prompt changes the color of your prompt based on the hostname of your current host. This can be quite nice if you log in to a lot of different hosts (if you can manage to distribute the prompt, of course). Global variables that can be used to change/manipulate the prompt, and their defaults: PROMPT_ADAM3_COLOR1=multi # host-color. 'multi' means color changes w/hostname PROMPT_ADAM3_COLOR2=white,blue # user color PROMPT_ADAM3_COLOR3=39,none,1 # color for directory PROMPT_ADAM3_COLOR4=66 # color for time PROMTP_ADAM3_COLOR6=208 # virtualenv color # PROMPT_ADAM3_COLOR{1,2,3,4,6} accept comma-separated values for # "foreground, background, bold". Fore- and background color takes both # color names and color values. Bold can be set with 1 for bold, and 0 for # not bold. For example, "blue,white,1" means bold white text on blue # background. Values can be dropped from the right, so "blue,white" means # blue text on white background, and "blue" means just blue text. PROMPT_ADAM3_COLOR5=white # default prompt-color. Only takes a single color. PROMPT_ADAM3_TIME= # Set to 'false' if you don't want time on RPS1 PROMPT_ADAM3_AOUH= # Set to 'true' if you want "user@host" to show on # localhost. Else it will only be shown when on remote host # (through ssh), or when you are root. PROMPT_ADAM3_PUSER='%n@' PROMPT_ADAM3_PHOST='%m' PROMPT_ADAM3_PdirS='%-40<..<%(5~|%-1~/../%3~|%4~)%<< ' PROMPT_ADAM3_PdirD='%-10<..<%~%<<' PROMPT_ADAM3_PTIME='%D{%H:%M:%S}' PROMPT_ADAM3_SHLVL='%(10L.%F{58}%L%f .)' PROMPT_ADAM3_PROMPTCHAR="%(?..[%F{red}%?%f] )%B%F{%(!.red.$PROMPT_ADAM3_COLOR5)}%#%f%b " EOF } prompt_adam3_setup () { emulate -L zsh setopt extendedglob autoload -Uz vcs_info # Colors # Prefer positional arguments. Use environment if posititionals are not # provided. # Colors are given in a comma-separated way. "foreground,bacground,bold". # Both fore- and background-colors may be given by name or number. Bold # should be given with 1 for bold, 0 for no bold. Arguments can be dropped # from the right. I.e. "blue,white,1" for bold blue text on white background. # "blue,white" for blue (non-bold) on white background. Or just "blue" for # just blue text and no given background. 'none' can be given as a color, and # indicates no color. So, bold blue text with no background is "blue,none,1". local -A prom color color[host]=${1:-${PROMPT_ADAM3_COLOR1:-multi}} # host color color[user]=${2:-${PROMPT_ADAM3_COLOR2:-white,blue}} # user color color[dirS]=${3:-${PROMPT_ADAM3_COLOR3:-39,none,1}} # dir (single-mode) color color[dirD]=${3:-${PROMPT_ADAM3_COLOR3:-39,none,1}} # dir (double mode) color color[time]=${4:-${PROMPT_ADAM3_TIME:-${PROMPT_ADAM3_COLOR4:-66}}} # time color color[char]=${5:-${PROMPT_ADAM3_COLOR5:-white}} # default promptchar-color color[venv]=${6:-${PROMPT_ADAM3_COLOR6:-208}} # virtualenv color # Username prom[user]=${PROMPT_ADAM3_PUSER:-'%n@'} # Hostname prom[host]=${PROMPT_ADAM3_PHOST:-'%m'} # DIR in single-line-mode prom[dirS]=${PROMPT_ADAM3_PdirS:-'%-40<..<%(5~|%-1~/../%3~|%~)%<< '} # DIR in double-line-mode prom[dirD]=${PROMPT_ADAM3_PdirD:-'%-10<..<%~%<<'} # Time [[ $color[time] != false ]] && prom[time]=${PROMPT_ADAM3_PTIME:-'%D{%H:%M:%S}'} # SHLVL prom[shlvl]=${PROMPT_ADAM3_SHLVL:-'%(10L.%F{58}%L%f .)'} # venv prom[venv]="%F{$color[venv]}(SUBSTITUTE)%f" # Promptchar (including exitcode) typeset -g PROMPT_ADAM3_PROMPTCHAR=${PROMPT_ADAM3_PROMPTCHAR:-"%(?..[%F{red}%?%f] )%B%(!.%F{red}.%F{$color[char]})%#%f%b "} # same as above, but with $ instead of % as promptchar for unprivileged users #typeset -g PROMPT_ADAM3_PROMPTCHAR='%B%(!.%F{red}#.%F{white}$)%f%b ' # Set some styles for vcs_info zstyle ':vcs_info:*' enable git svn zstyle ':vcs_info:*' check-for-changes true zstyle ':vcs_info:*' check-for-staged-changes true zstyle ':vcs_info:*' stagedstr ' S' zstyle ':vcs_info:*' unstagedstr ' U' zstyle ':vcs_info:*' formats '%s %F{green}%b%F{red}%u%c%f ' zstyle ':vcs_info:git*' actionformats '%s %r/%S %F{green}%b%F{red}%u%c%a%f ' local c # scope up a throwaway-variable # Change color of host, based on hostname if [[ $color[host] == 'multi' ]]; then # Here we take the charset-number of each character in the hostname, and # add them together. local -i hostnum i for i in {1..$#HOST}; do c=$HOST[$i] (( hostnum += #c )) done # color-combos. fg,bg local -a colors colors=( red,black green,black yellow,black blue,black \ magenta,black cyan,black white,red yellow,red \ cyan,red black,green blue,green black,yellow \ blue,yellow white,blue red,blue yellow,blue \ white,magenta yellow,magenta white,cyan blue,cyan \ black,white red,white blue,white magenta,white ) # Select color color[host]=${colors[$hostnum % $#colors + 1]} # Go bold? ((hostnum % 2)) && color[host]+=',1' # for testing purposes #color[host]=${colors[$RANDOM % $#colors + 1]} #((RANDOM % 2)) && color[host]+=',1' fi # colorize the prompt-parts. See footnotes for extra comments. for item in user host dirS dirD time; do c=(${(s.,.)color[$item]}) # cheaper to do this expansion just once [[ ${c[1]:-none} != none ]] && prom[$item]="%F{${c[1]}}$prom[${item}]%f" [[ ${c[2]:-none} != none ]] && prom[$item]="%K{${c[2]}}$prom[${item}]%k" (( ${c[3]} )) && prom[$item]="%B$prom[${item}]%b" done # Make each prompt-part global typeset -g PROMPT_ADAM3_PS1 # Show user@host if using ssh, PROMPT_ADAM3_AOUH is set, or you are root [[ -n $SSH_CLIENT || -n $SSH_TTY || -n $SSH_CONNECTION \ || -n $PROMPT_ADAM3_AOUH || $UID == 0 ]] \ && PROMPT_ADAM3_PS1="$prom[user]$prom[host] " typeset -g PROMPT_ADAM3_SLM_DIR=$prom[dirS] typeset -g PROMPT_ADAM3_DLM_DIR=$prom[dirD] typeset -g PROMPT_ADAM3_RPS1=$prom[shlvl]$prom[time] typeset -g PROMPT_ADAM3_VENV=$prom[venv] # Set up hook add-zsh-hook precmd prompt_adam3_precmd } prompt_adam3_precmd () { emulate -LR zsh setopt extendedglob # vcs vcs_info #$vcs_info_msg_0_ local prom venv_info (( ${+VIRTUAL_ENV} )) && venv_info="${PROMPT_ADAM3_VENV/SUBSTITUTE/${VIRTUAL_ENV:t}} " prom=${PROMPT_ADAM3_PS1} prom+=$PROMPT_ADAM3_SLM_DIR prom+=$venv_info prom+=$vcs_info_msg_0_ prom+=$PROMPT_ADAM3_PROMPTCHAR # Calculate prompt-length by first removing all zero-length characters. From # prompt_bart_setup in zsh. local zlchars='%([BSUbfksu]|[FK]{*})' # regex for zero-length chars local prompt_length=${#${(S%%)prom//$~zlchars/}} # If the prompt is over half the terminal, we go into multiline-mode. if (( prompt_length > ( COLUMNS / 2 ) )); then prom=${PROMPT_ADAM3_PS1} # reset prompt prom+=$PROMPT_ADAM3_DLM_DIR # double-line-mode dir # We need to recalculate prompt_length to know where to put time prompt_length=${#${(S%%)prom//$~zlchars/}} local rprompt_length=${#${(S%%)PROMPT_ADAM3_RPS1//$~zlchars/}} local timepadding=$((COLUMNS - prompt_length + $#PROMPT_ADAM3_RPS1 - rprompt_length - 3 )) # Add time. Left-pad with spaces. prom+=${(l:$timepadding:)PROMPT_ADAM3_RPS1} prom+=$'\n' # newline prom+=$venv_info # virtualenv info prom+=$vcs_info_msg_0_ # vcs prom+=$PROMPT_ADAM3_PROMPTCHAR # promptchar RPS1= # RPS1 does not support multiline, so reset it when we're in multiline else RPS1=$PROMPT_ADAM3_RPS1 fi PS1=$prom PS2="${PS1}%_> %b%f%k" PS3="${PS1}?# %b%f%k" } prompt_adam3_setup "${@}" # END OF FILE ################################################################# # vim: filetype=zsh