X-Git-Url: https://git.madduck.net/etc/zsh.git/blobdiff_plain/f9a91601a570cea7ff13a5f9c1894a8f63fbbae6..e7b57bc5c5383a6eac9b7d8d0af4f1c63723b948:/.zsh/themes/prompt_madduck_setup diff --git a/.zsh/themes/prompt_madduck_setup b/.zsh/themes/prompt_madduck_setup new file mode 100644 index 0000000..0b6dddc --- /dev/null +++ b/.zsh/themes/prompt_madduck_setup @@ -0,0 +1,369 @@ +# +# My zsh prompt theme +# +# Copyright © 1994–2017 martin f. krafft +# Released under the terms of the Artistic Licence 2.0 +# +# Source repository: http://git.madduck.net/v/etc/zsh.git +# +# vcs stuff shamelessly based on http://glandium.org/blog/?p=170 +# + +zstyle -m :madduck:prompt:default path-maxlen '*' \ + || zstyle :madduck:prompt:default path-maxlen 25 +zstyle -m :madduck:prompt:default path-minlen '*' \ + || zstyle :madduck:prompt:default path-minlen 10 + +__on_networkfs() +{ + emulate -L zsh + case $(df -T . | sed -rne '$s,^[^[:space:]]+[[:space:]]+([^[:space:]]+).*,\1,p') in + (cifs|nfs) return 0;; + esac + return 1 +} + +__git_get_reporoot() +{ + emulate -L zsh + # return the full path to the root of the current git repository + [ -d "$GIT_DIR" ] && echo "$GIT_DIR" && return 0 + local dir; dir="$PWD/$(git rev-parse --show-cdup)" + # do not use --show-toplevel because it resolves symlinks + echo $dir:a +} + +__git_get_branch() +{ + emulate -L zsh + # use oh-my-zsh prompt info function if it exists + $(command -v git_prompt_info) && return + + # return the name of the git branch we're on + local ref gitdir + gitdir="$(git rev-parse --git-dir)" + ref=$(git --git-dir="$gitdir" symbolic-ref -q HEAD 2>/dev/null \ + || git --git-dir="$gitdir" name-rev --name-only HEAD 2>/dev/null) || return 1 + echo "${ref#refs/heads/}" +} + +__git_print_preprompt() +{ + emulate -L zsh + [ "$(git config --get core.bare)" = false ] || return + __on_networkfs && return + + local COLUMNS=${COLUMNS:-80} + local LINES=${LINES:-25} + + function output() { + emulate -L zsh + local title="$@" + local output=(${(f)"$(cat)"}) + + [[ ${#output} -ge 1 ]] || return + + local statl="$(echo ${output[-1]} | sed -re 's@^\s*([0-9]+)[^,]+(, ([0-9]+) [^(]+\(([-+])\))(, ([0-9]+) [^(]+\(([-+])\))?@\1/\4\3/\7\6@')" + + if [[ ${output[-2]## } = '...' ]]; then + print "${title} (${statl%/}, abbrev.):" + print "${(F)output[1,-3]}" + print " …" + else + print "${title} (${statl%/}):" + print "${(F)output[1,-2]}" + fi + } + + function gitdiffstat() { + emulate -L zsh + local common_options="--stat=$((COLUMNS/2-1)),$((COLUMNS/4-2)),$(($LINES/3)) --relative" + eval git diff $common_options "$@" 2>/dev/null + } + + local cached=(${(f)"$(gitdiffstat --cached | output cached)"}) + local changed=(${(f)"$(gitdiffstat | output changed)"}) + + local max=${#changed} + [[ $max -lt ${#cached} ]] && max=${#cached} + + ((max == 0)) && return + + local width=$(((COLUMNS-3)/2)) + + if (( ${#cached} > 0 && ${#changed} > 0 )); then + local i + for (( i=1 ; i <= max ; i++ )) do + printf "%-${width}s │ %-${width}s\n" "${cached[$i]}" "${changed[$i]}" + done + else + print ${(F)cached}${(F)changed} + fi +} + +__hg_get_reporoot() +{ + emulate -L zsh + hg root +} + +__hg_get_branch() +{ + emulate -L zsh + echo "hg:$(hg branch)" +} + +__bzr_get_reporoot() +{ + emulate -L zsh + local reporoot + reporoot="$(bzr info | sed -rne 's, *branch root: ,,p')" + case "$reporoot" in + .) echo "$PWD";; + *) echo "$reporoot";; + esac +} + +__bzr_get_branch() +{ + emulate -L zsh + local branch revno + bzr version-info | while read i j; do + case "$i" in + revno:) revno="$j";; + branch-nick:) branch="$j";; + esac + done + echo "bzr:${branch}@$revno" +} + +__vcs_get_repo_type() +{ + emulate -L zsh + # return the type of the closest repository in the path hierarchy + local dir + while true; do + [ -d ${dir}.git ] && echo git && break + [ -d "$GIT_DIR" ] && echo git && break + [ -d ${dir}.bzr ] && echo bzr && break + [ -d ${dir}.hg ] && echo hg && break + [ "$(readlink -f ${dir:-.})" = / ] && echo NONE && break + dir="../$dir" + done +} + +__get_prompt_path_len() { + emulate -L zsh + local result + zstyle -s ":madduck:prompt:$PWD" path-${1}len result + [ -z "$result" ] && zstyle -s ':madduck:prompt:default' path-${1}len result + echo $result +} + +__vcs_get_prompt_path_components() +{ + emulate -L zsh + # return formatted path components (prefix branch postfix) given + # the repository root and the branch. + + local MAXLEN MINLEN + MAXLEN=$(__get_prompt_path_len max) + MINLEN=$(__get_prompt_path_len min) + + # shortcut: if there are no arguments, return a default prompt + if [ -z "${1:-}" ]; then + pwdnamed="${(%):-%${MAXLEN}<…<%~%<<}" + echo "$pwdnamed" + return + fi + + local reporoot branch + reporoot="${1%%/}" + branch="$2" + + # replace named directories in the PWD, we need thi for the proper component + # count later + local pwdnamed + pwdnamed="${(%):-%~}" + + # store paths in arrays for component count calculation + typeset -la apwd apwdnamed areporoot + apwd=(${(s:/:)PWD}) + apwdnamed=(${(s:/:)pwdnamed}) + areporoot=(${(s:/:)reporoot}) + + # get the number of leading and trailing path components. Since we're using + # %~ later and then /home/madduck suddenly becomes ~, which is 1, not + # 2 components, we calculate the leading component count by using the named + # path and the number of post components + local precomps postcomps + postcomps=$(($#apwd - $#areporoot)) + precomps=$(($#apwdnamed - $postcomps)) + + local postfix + (( $postcomps > 0 )) && postfix="${(%):-%${postcomps}~}" + + # we don't want the prompt to get too long, so keep the total prompt length + # under $MAXLEN, but ensure that the prefix is not shorter + # than $MINLEN, no matter what + local prelen minlen prefix + prelen=$((${MAXLEN} - $#branch - $#postfix)) + minlen=${MINLEN} + (( $prelen < $minlen )) && prelen=$minlen + prefix="${(%):-%${prelen}<…<%-${precomps}~%<<}" + + echo "'$prefix'" "'$branch'" "'$postfix'" +} + +__vcs_set_prompt_variables() +{ + emulate -L zsh + # set psvar[1..3] depending on repo type, or just psvar[1] if no repo found + local reporoot branch repotype + repotype="${1:-$(__vcs_get_repo_type)}" + + case "$repotype" in + git) + reporoot="$(__git_get_reporoot)" || + { zerror "could not determine git repository root"; return 1 } + branch="$(__git_get_branch)" || + { zerror "could not determine git branch"; return 1 } + if [ -n "$VCSH_REPO_NAME" ]; then + # if vcsh is used to get a subshell, then the repo root is the home + # directory, but we want to indicate the vcsh context too: + eval set -- $(__vcs_get_prompt_path_components "$HOME" "$branch") + set -- "vcsh:$VCSH_REPO_NAME" "$2" "$1${3:+/$3}" + else + eval set -- $(__vcs_get_prompt_path_components "$reporoot" "$branch") + if [ -d "$GIT_DIR" ]; then + # poor man's replace until I find out how to do named dirs properly + # here: + local _D="${GIT_DIR/$HOME/~}" + set -- "$_D" "$2" "${${1#$_D}%/}" + fi + fi + ;; + hg) + reporoot="$(__hg_get_reporoot)" || + { zerror "could not determine hg repository root"; return 1 } + branch="$(__hg_get_branch)" || + { zerror "could not determine hg branch"; return 1 } + eval set -- $(__vcs_get_prompt_path_components "$reporoot" "$branch") + ;; + bzr) + reporoot="$(__bzr_get_reporoot)" || + { zerror "could not determine bzr repository root"; return 1 } + branch="$(__bzr_get_branch)" || + { zerror "could not determine bzr branch"; return 1 } + eval set -- $(__vcs_get_prompt_path_components "$reporoot" "$branch") + ;; + *) + case "$repotype" in + NONE) :;; + *) warn "$repotype repositories not (yet) supported in the prompt";; + esac + local MAXLEN MINLEN + MAXLEN=$(__get_prompt_path_len max) + local p="%${MAXLEN}<…<%~%<<" + #TODO find a better way so we don't have to nuke $psvar, but since the + # %(nv.true.false) check for prompts checks element count, not + # content, that's all we get for now + psvar=("${(%)p}") + return + esac + + psvar[1,3]=($1 $2 $3) +} + +__vcs_print_preprompt() +{ + emulate -L zsh + local repotype="${1:-$(__vcs_get_repo_type)}" + + case "$repotype" in + git) + __git_print_preprompt + ;; + esac +} + +if ! is_root; then + # too dangerous to be run as root + autoload -U add-zsh-hook + + _update_vcs_prompt_vars_if_vcs_ran() { + local vcs="$(__vcs_get_repo_type)" + case "$(history $(($HISTCMD - 1)))" in + # $vcs appeared in last command, so be sure to update + *${vcs}*) __vcs_set_prompt_variables "$vcs" + esac + } + add-zsh-hook precmd _update_vcs_prompt_vars_if_vcs_ran + + _update_vcs_prompt_vars() { + __vcs_set_prompt_variables + } + add-zsh-hook chpwd _update_vcs_prompt_vars + + _print_preprompt() { + [[ $? -eq 0 ]] && __vcs_print_preprompt + } + add-zsh-hook precmd _print_preprompt + + # call it once + _update_vcs_prompt_vars +fi + +function make_ps1() { + # start with '+' if in a subshell + echo -n '%(2L.+.)' + + # the machine name, bold or underlined based on non-root/root + local ps1_hl=B + is_root && ps1_hl=U + echo -n "%${ps1_hl:=B}%m%${(L)ps1_hl}" + + # if we're in a Debian chroot, make that stand out + echo -n "${DEBIAN_CHROOT:+/%S$DEBIAN_CHROOT%s}" + + # we end this with a : + echo -n : + + # now comes the working directory, composed from parts in $psvar, + # which is managed by $ZDOTDIR/zshrc/06-vcsprompt + echo -n '%1v%(2v.|%B%2v%b|.)%(3v.%3v.)' + + # and we finish with #/% for root/non-root, and a space + echo -n '%# ' + echo +} +PS1=$(make_ps1) +unfunction make_ps1 +PS2="%{$fg[red]%}%_>%{$reset_color%}" + +function make_rps1() { + # First, a comment character and parens + echo -n '#(' + + # Next, if the returncode was non-zero, make it stand-out + # and include a trailing space + echo -n "%(0?..%{$fg[red]%}%S%?%s%{$reset_color%} )" + + # If there are background jobs, print their number, followed by + # '@': + echo -n '%(1j.%j@.)' + + # and then the terminal line we're using + echo -n '%l' + + # this concludes the first part, but there's more + echo -n ') ' + + # the timestamp will finish it off: + echo -n '%D{%d %H:%M:%S.%.}' + echo +} +RPS1=$(make_rps1) +unfunction make_rps1 + +# vim:ft=zsh