-
-
Save QuinnyPig/fdc52bec44cbc825255b to your computer and use it in GitHub Desktop.
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
diff --git a/.zsh/functions/gitstatus.py b/.zsh/functions/gitstatus.py | |
new file mode 100755 | |
index 0000000..d14420e | |
--- /dev/null | |
+++ b/.zsh/functions/gitstatus.py | |
@@ -0,0 +1,65 @@ | |
+#!/usr/bin/env python | |
+from __future__ import print_function | |
+ | |
+# change this symbol to whatever you prefer | |
+prehash = ':' | |
+ | |
+from subprocess import Popen, PIPE | |
+ | |
+import sys | |
+gitsym = Popen(['git', 'symbolic-ref', 'HEAD'], stdout=PIPE, stderr=PIPE) | |
+branch, error = gitsym.communicate() | |
+ | |
+error_string = error.decode('utf-8') | |
+ | |
+if 'fatal: Not a git repository' in error_string: | |
+ sys.exit(0) | |
+ | |
+branch = branch.decode("utf-8").strip()[11:] | |
+ | |
+res, err = Popen(['git','diff','--name-status'], stdout=PIPE, stderr=PIPE).communicate() | |
+err_string = err.decode('utf-8') | |
+if 'fatal' in err_string: | |
+ sys.exit(0) | |
+changed_files = [namestat[0] for namestat in res.decode("utf-8").splitlines()] | |
+staged_files = [namestat[0] for namestat in Popen(['git','diff', '--staged','--name-status'], stdout=PIPE).communicate()[0].splitlines()] | |
+nb_changed = len(changed_files) - changed_files.count('U') | |
+nb_U = staged_files.count('U') | |
+nb_staged = len(staged_files) - nb_U | |
+staged = str(nb_staged) | |
+conflicts = str(nb_U) | |
+changed = str(nb_changed) | |
+nb_untracked = len([0 for status in Popen(['git','status','--porcelain',],stdout=PIPE).communicate()[0].decode("utf-8").splitlines() if status.startswith('??')]) | |
+untracked = str(nb_untracked) | |
+ | |
+ahead, behind = 0,0 | |
+ | |
+if not branch: # not on any branch | |
+ branch = prehash + Popen(['git','rev-parse','--short','HEAD'], stdout=PIPE).communicate()[0].decode("utf-8")[:-1] | |
+else: | |
+ remote_name = Popen(['git','config','branch.%s.remote' % branch], stdout=PIPE).communicate()[0].decode("utf-8").strip() | |
+ if remote_name: | |
+ merge_name = Popen(['git','config','branch.%s.merge' % branch], stdout=PIPE).communicate()[0].decode("utf-8").strip() | |
+ if remote_name == '.': # local | |
+ remote_ref = merge_name | |
+ else: | |
+ remote_ref = 'refs/remotes/%s/%s' % (remote_name, merge_name[11:]) | |
+ revgit = Popen(['git', 'rev-list', '--left-right', '%s...HEAD' % remote_ref],stdout=PIPE, stderr=PIPE) | |
+ revlist = revgit.communicate()[0] | |
+ if revgit.poll(): # fallback to local | |
+ revlist = Popen(['git', 'rev-list', '--left-right', '%s...HEAD' % merge_name],stdout=PIPE, stderr=PIPE).communicate()[0] | |
+ behead = revlist.decode("utf-8").splitlines() | |
+ ahead = len([x for x in behead if x[0]=='>']) | |
+ behind = len(behead) - ahead | |
+ | |
+out = '\n'.join([ | |
+ branch, | |
+ str(ahead), | |
+ str(behind), | |
+ staged, | |
+ conflicts, | |
+ changed, | |
+ untracked, | |
+ ]) | |
+print(out) | |
+ | |
diff --git a/.zsh/functions/zshrc.sh b/.zsh/functions/zshrc.sh | |
new file mode 100755 | |
index 0000000..c3871bf | |
--- /dev/null | |
+++ b/.zsh/functions/zshrc.sh | |
@@ -0,0 +1,109 @@ | |
+# To install source this file from your .zshrc file | |
+ | |
+# see documentation at http://linux.die.net/man/1/zshexpn | |
+# A: finds the absolute path, even if this is symlinked | |
+# h: equivalent to dirname | |
+export __GIT_PROMPT_DIR=${0:A:h} | |
+ | |
+export GIT_PROMPT_EXECUTABLE=${GIT_PROMPT_USE_PYTHON:-"python"} | |
+ | |
+# Initialize colors. | |
+autoload -U colors | |
+colors | |
+ | |
+# Allow for functions in the prompt. | |
+setopt PROMPT_SUBST | |
+ | |
+autoload -U add-zsh-hook | |
+ | |
+add-zsh-hook chpwd chpwd_update_git_vars | |
+add-zsh-hook preexec preexec_update_git_vars | |
+add-zsh-hook precmd precmd_update_git_vars | |
+ | |
+## Function definitions | |
+function preexec_update_git_vars() { | |
+ case "$2" in | |
+ git*|hub*|gh*|stg*) | |
+ __EXECUTED_GIT_COMMAND=1 | |
+ ;; | |
+ esac | |
+} | |
+ | |
+function precmd_update_git_vars() { | |
+ if [ -n "$__EXECUTED_GIT_COMMAND" ] || [ ! -n "$ZSH_THEME_GIT_PROMPT_CACHE" ]; then | |
+ update_current_git_vars | |
+ unset __EXECUTED_GIT_COMMAND | |
+ fi | |
+} | |
+ | |
+function chpwd_update_git_vars() { | |
+ update_current_git_vars | |
+} | |
+ | |
+function update_current_git_vars() { | |
+ unset __CURRENT_GIT_STATUS | |
+ | |
+ if [[ "$GIT_PROMPT_EXECUTABLE" == "python" ]]; then | |
+ local gitstatus="$__GIT_PROMPT_DIR/gitstatus.py" | |
+ _GIT_STATUS=`python ${gitstatus} 2>/dev/null` | |
+ fi | |
+ if [[ "$GIT_PROMPT_EXECUTABLE" == "haskell" ]]; then | |
+ local gitstatus="$__GIT_PROMPT_DIR/dist/build/gitstatus/gitstatus" | |
+ _GIT_STATUS=`${gitstatus}` | |
+ fi | |
+ __CURRENT_GIT_STATUS=("${(@f)_GIT_STATUS}") | |
+ GIT_BRANCH=$__CURRENT_GIT_STATUS[1] | |
+ GIT_AHEAD=$__CURRENT_GIT_STATUS[2] | |
+ GIT_BEHIND=$__CURRENT_GIT_STATUS[3] | |
+ GIT_STAGED=$__CURRENT_GIT_STATUS[4] | |
+ GIT_CONFLICTS=$__CURRENT_GIT_STATUS[5] | |
+ GIT_CHANGED=$__CURRENT_GIT_STATUS[6] | |
+ GIT_UNTRACKED=$__CURRENT_GIT_STATUS[7] | |
+} | |
+ | |
+ | |
+git_super_status() { | |
+ precmd_update_git_vars | |
+ if [ -n "$__CURRENT_GIT_STATUS" ]; then | |
+ STATUS="$ZSH_THEME_GIT_PROMPT_PREFIX$ZSH_THEME_GIT_PROMPT_BRANCH$GIT_BRANCH%{${reset_color}%}" | |
+ if [ "$GIT_BEHIND" -ne "0" ]; then | |
+ STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_BEHIND$GIT_BEHIND%{${reset_color}%}" | |
+ fi | |
+ if [ "$GIT_AHEAD" -ne "0" ]; then | |
+ STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_AHEAD$GIT_AHEAD%{${reset_color}%}" | |
+ fi | |
+ STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_SEPARATOR" | |
+ if [ "$GIT_STAGED" -ne "0" ]; then | |
+ STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_STAGED$GIT_STAGED%{${reset_color}%}" | |
+ fi | |
+ if [ "$GIT_CONFLICTS" -ne "0" ]; then | |
+ STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_CONFLICTS$GIT_CONFLICTS%{${reset_color}%}" | |
+ fi | |
+ if [ "$GIT_CHANGED" -ne "0" ]; then | |
+ STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_CHANGED$GIT_CHANGED%{${reset_color}%}" | |
+ fi | |
+ if [ "$GIT_UNTRACKED" -ne "0" ]; then | |
+ STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_UNTRACKED%{${reset_color}%}" | |
+ fi | |
+ if [ "$GIT_CHANGED" -eq "0" ] && [ "$GIT_CONFLICTS" -eq "0" ] && [ "$GIT_STAGED" -eq "0" ] && [ "$GIT_UNTRACKED" -eq "0" ]; then | |
+ STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_CLEAN" | |
+ fi | |
+ STATUS="$STATUS%{${reset_color}%}$ZSH_THEME_GIT_PROMPT_SUFFIX" | |
+ echo "$STATUS" | |
+ fi | |
+} | |
+ | |
+# Default values for the appearance of the prompt. Configure at will. | |
+ZSH_THEME_GIT_PROMPT_PREFIX="(" | |
+ZSH_THEME_GIT_PROMPT_SUFFIX=")" | |
+ZSH_THEME_GIT_PROMPT_SEPARATOR="|" | |
+ZSH_THEME_GIT_PROMPT_BRANCH="%{$fg_bold[magenta]%}" | |
+ZSH_THEME_GIT_PROMPT_STAGED="%{$fg[red]%}%{●%G%}" | |
+ZSH_THEME_GIT_PROMPT_CONFLICTS="%{$fg[red]%}%{✖%G%}" | |
+ZSH_THEME_GIT_PROMPT_CHANGED="%{$fg[blue]%}%{✚%G%}" | |
+ZSH_THEME_GIT_PROMPT_BEHIND="%{↓%G%}" | |
+ZSH_THEME_GIT_PROMPT_AHEAD="%{↑%G%}" | |
+ZSH_THEME_GIT_PROMPT_UNTRACKED="%{…%G%}" | |
+ZSH_THEME_GIT_PROMPT_CLEAN="%{$fg_bold[green]%}%{✔%G%}" | |
+ | |
+ | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment