# If not running interactively, don't do anything
[[ $- != *i* ]] && return

# {{{ settings
# disable the bloody ^S / ^Q, I use tmux all the time anyway
stty -ixon
setopt APPEND_HISTORY EXTENDED_HISTORY HIST_IGNORE_DUPS EXTENDED_GLOB AUTO_CD AUTO_PUSHD PRINT_EXIT_VALUE
unsetopt BEEP NO_MATCH NOTIFY MENU_COMPLETE AUTO_MENU

SAVEHIST=1000
HISTSIZE=1000
HISTFILE="$HOME/.histfile.$UID"

export EDITOR='vim'
export PAGER='less -R'
export TIME_STYLE='long-iso'
export MYSQL_PS1='mysql [\d]> '
export SSH_AUTH_SOCK="$HOME/.ssh/ssh_auth_sock"
export LS_COLORS='no=00:fi=00:di=34:ow=34;40:ln=35:pi=30;44:so=35;47:do=35;44:bd=33;44:cd=37;44:or=05;37;41:mi=05;37;41:ex=01;31:*.cmd=01;31:*.exe=01;31:*.com=01;31:*.bat=01;31:*.reg=01;31:*.app=01;31:*.txt=32:*.org=32:*.md=32:*.mkd=32:*.h=32:*.c=32:*.C=32:*.cc=32:*.cpp=32:*.cxx=32:*.objc=32:*.sh=32:*.csh=32:*.zsh=32:*.el=32:*.vim=32:*.java=32:*.pl=32:*.pm=32:*.py=32:*.rb=32:*.hs=32:*.php=32:*.htm=32:*.html=32:*.shtml=32:*.erb=32:*.haml=32:*.xml=32:*.rdf=32:*.css=32:*.sass=32:*.scss=32:*.less=32:*.js=32:*.coffee=32:*.man=32:*.0=32:*.1=32:*.2=32:*.3=32:*.4=32:*.5=32:*.6=32:*.7=32:*.8=32:*.9=32:*.l=32:*.n=32:*.p=32:*.pod=32:*.tex=32:*.bmp=33:*.cgm=33:*.dl=33:*.dvi=33:*.emf=33:*.eps=33:*.gif=33:*.jpeg=33:*.jpg=33:*.JPG=33:*.mng=33:*.pbm=33:*.pcx=33:*.pdf=33:*.pgm=33:*.png=33:*.ppm=33:*.pps=33:*.ppsx=33:*.ps=33:*.svg=33:*.svgz=33:*.tga=33:*.tif=33:*.tiff=33:*.xbm=33:*.xcf=33:*.xpm=33:*.xwd=33:*.xwd=33:*.yuv=33:*.aac=33:*.au=33:*.flac=33:*.mid=33:*.midi=33:*.mka=33:*.mp3=33:*.mpa=33:*.mpeg=33:*.mpg=33:*.ogg=33:*.ra=33:*.wav=33:*.anx=33:*.asf=33:*.avi=33:*.axv=33:*.flc=33:*.fli=33:*.flv=33:*.gl=33:*.m2v=33:*.m4v=33:*.mkv=33:*.mov=33:*.mp4=33:*.mp4v=33:*.mpeg=33:*.mpg=33:*.nuv=33:*.ogm=33:*.ogv=33:*.ogx=33:*.qt=33:*.rm=33:*.rmvb=33:*.swf=33:*.vob=33:*.wmv=33:*.doc=31:*.docx=31:*.rtf=31:*.dot=31:*.dotx=31:*.xls=31:*.xlsx=31:*.ppt=31:*.pptx=31:*.fla=31:*.psd=31:*.7z=1;35:*.apk=1;35:*.arj=1;35:*.bin=1;35:*.bz=1;35:*.bz2=1;35:*.cab=1;35:*.deb=1;35:*.dmg=1;35:*.gem=1;35:*.gz=1;35:*.iso=1;35:*.jar=1;35:*.msi=1;35:*.rar=1;35:*.rpm=1;35:*.tar=1;35:*.tbz=1;35:*.tbz2=1;35:*.tgz=1;35:*.tx=1;35:*.war=1;35:*.xpi=1;35:*.xz=1;35:*.z=1;35:*.Z=1;35:*.zip=1;35:*.ANSI-30-black=30:*.ANSI-01;30-brblack=01;30:*.ANSI-31-red=31:*.ANSI-01;31-brred=01;31:*.ANSI-32-green=32:*.ANSI-01;32-brgreen=01;32:*.ANSI-33-yellow=33:*.ANSI-01;33-bryellow=01;33:*.ANSI-34-blue=34:*.ANSI-01;34-brblue=01;34:*.ANSI-35-magenta=35:*.ANSI-01;35-brmagenta=01;35:*.ANSI-36-cyan=36:*.ANSI-01;36-brcyan=01;36:*.ANSI-37-white=37:*.ANSI-01;37-brwhite=01;37:*.log=01;32:*~=01;32:*#=01;32:*.bak=01;33:*.BAK=01;33:*.old=01;33:*.OLD=01;33:*.org_archive=01;33:*.off=01;33:*.OFF=01;33:*.dist=01;33:*.DIST=01;33:*.orig=01;33:*.ORIG=01;33:*.swp=01;33:*.swo=01;33:*,v=01;33:*.gpg=34:*.gpg=34:*.pgp=34:*.asc=34:*.3des=34:*.aes=34:*.enc=34:'

autoload -Uz compinit edit-command-line
zle -N edit-command-line

# completion
compinit
zstyle ':completion:*' completer _list _complete _ignored
zstyle ':completion:*' insert-unambiguous true
zstyle ':completion:*' file-sort name
zstyle ':completion:*' format 'Completing %d'
zstyle ':completion:*' group-name ''
zstyle ':completion:*' list-colors ''
zstyle ':completion:*' list-prompt '%SAt %p: Hit TAB for more, or the character to insert%s'
zstyle ':completion:*' list-suffixes true
zstyle ':completion:*' original true
zstyle ':completion:*' select-prompt '%SScrolling active: current selection at %p%s'
zstyle ':completion:*' matcher-list 'm:{[:lower:][:upper:]}={[:upper:][:lower:]}'
zstyle ':completion:*' rehash true
zstyle ':completion:*:kill:*:processes' command 'ps --forest -A -o pid,user,cmd'
zstyle ':completion:*:processes-names' command 'ps axho command'
zstyle ':completion:*:default' list-colors ${(s.:.)LS_COLORS}
# }}}
# {{{ key bindings
bindkey -e
# urxvt
bindkey '^[[7~'   beginning-of-line    # home
bindkey '^[[8~'   end-of-line          # end
bindkey '^[Oc'    forward-word         # ctrl + right
bindkey '^[Od'    backward-word        # ctrl + left
bindkey '^[[3^'   delete-word          # ctrl + del
# screen
bindkey '^[[1~'   beginning-of-line    # home
bindkey '^[[4~'   end-of-line          # end
# xterm
bindkey '^[[H'    beginning-of-line    # home
bindkey '^[[F'    end-of-line          # end
# st
bindkey '^[[P'    delete-char          # del
bindkey '^[[M'    delete-word          # ctrl + del
# most of them (but not urxvt)
bindkey '^[[1;5C' forward-word         # ctrl + right
bindkey '^[[1;5D' backward-word        # ctrl + left
bindkey '^[[3;5~' delete-word          # ctrl + del
# all of them
bindkey '^[[5~'   backward-word        # page up
bindkey '^[[6~'   forward-word         # page down
bindkey '^[[3~'   delete-char          # del
bindkey '^[m'     copy-prev-shell-word # alt + m
bindkey -s '^j'   '^atime ^m'          # ctrl + j

# command line editing
bindkey '^x^e'    edit-command-line
# }}}
# {{{ prompt
prompt_state_file="/tmp/zsh_gitstatus_$$.tmp"
prompt_ln1='%K{white}%F{black}[ %F{%(!.red.blue)}%n%F{black}@%m %F{green}%2~%F{black} '
prompt_ln2=']%f%k '
PROMPT="$prompt_ln1$prompt_ln2"
PROMPT2='%K{white}%F{black}[ %_ ]%f%k '
PROMPT3='%K{white}%F{black}[ ?# ]%f%k '
PROMPT4='%K{white}%F{black}[ +%N:%i ]%f%k '
precmd.title() {
    case $TERM in
        (screen*) printf '\033k%s\033\'  ${HOST%%.*};;
        (*)       printf '\033]2;%s\007' ${HOST%%.*};;
    esac
}
precmd.is_git_repo() {
    typeset curr_dir=$PWD
    while [[ -n $curr_dir ]]; do
        if [[ -r $curr_dir/.git/HEAD ]]; then
            [[ -e $curr_dir/.git/nozsh ]] && return 1
            return 0
        else
            curr_dir=${curr_dir%/*}
        fi
    done
    return 1
}
precmd.git() {
    typeset raw_status
    raw_status="$(flock -w 0 $prompt_state_file git --no-optional-locks status --porcelain -bu 2>/dev/null)"
    (($?)) && return 0

    typeset branch_info full_status git_status= IFS=
    typeset staged_count=0 unstaged_count=0 untracked_count=0 unmerged_count=0

    branch_info=$(git rev-parse --abbrev-ref HEAD)
    while read line; do
        [[ $line[1,2] == '??'     ]] && (( untracked_count++ ))
        [[ $line[1,2] =~ .[MD]    ]] && (( unstaged_count++  ))
        [[ $line[1,2] =~ [MDARC]. ]] && (( staged_count++    ))
        [[ $line[1,2] =~ [ADU]{2} ]] && (( unmerged_count++  ))
    done <<< $raw_status

    (( unstaged_count  )) && git_status+="%F{3}~$unstaged_count"
    (( staged_count    )) && git_status+="%F{4}+$staged_count"
    (( untracked_count )) && git_status+="%F{1}-$untracked_count"
    (( unmerged_count  )) && git_status+="%F{5}*$unmerged_count"
    [[ -z $git_status  ]] && git_status="%F{2}ok"

    printf '\ue0a0 %s:%s%%F{black} ' $branch_info[1,16] $git_status > $prompt_state_file
}
precmd.prompt() {
    PROMPT="$prompt_ln1$1$prompt_ln2"
}
precmd.git_update() {
    precmd.git
    kill -s USR1 $$
}
precmd() {
    precmd.title
    if precmd.is_git_repo; then
        precmd.prompt $'\ue0a0 '
        precmd.git_update &!
    else
        precmd.prompt
    fi
}
TRAPUSR1() {
    precmd.prompt "$(<$prompt_state_file)"
    zle && zle reset-prompt
}
TRAPEXIT() {
    [[ -f $prompt_state_file ]] && rm $prompt_state_file
}
# }}}
# {{{ aliases
alias beep='printf "\007"'
alias cower='command cower -c'
alias fixterm='printf "c"'
alias less='command less -R'
alias mysql='command mysql --sigint-ignore'
alias pacman='command pacman --color=auto'
alias rgrep='command grep --exclude-dir=\.git -R'
alias ggrep='command git grep'
alias tailf='command less -R +F'
alias diff='command diff --color'

# ls
alias ls='command ls --color=auto --group-directories-first '
alias ll='ls -lha'

# git
alias gci='command git commit'
alias gsl='command git stash list'
alias gss='command git status -sbu'
alias gup='command git pull'
alias groot='cd $(command git rev-parse --show-cdup)'
alias gsi='command tig status'

# tmux
alias tmux='command tmux -2'
alias atmux='command tmux -2 attach'

# screen
alias rscreen='command screen -Dr'
alias scr='command screen sudo -Es'

# termcompat
if [[ -x $(whence termcompat) ]]; then
    alias s='command termcompat ssh'
else
    alias s='command ssh'
fi

# }}}
# {{{ plugins
# grc
if [[ -x $(whence grc) ]]; then
    cmds=(\
        cc configure cvs df dig gcc gmake id ip last lsof make mount \
        mtr netstat ping ping6 ps tcpdump traceroute traceroute6 \
    )
    for cmd in $cmds[@]; do
        alias $cmd="command grc -es --colour=auto $cmd"
    done
    unset cmds cmd
fi
# some cool git stuff
gdiff() { /usr/bin/git diff --color "$@"; }
gdf() {
    typeset difftool
    if difftool=$(whence diff-so-fancy); then
        gdiff "$@" | $difftool | less --tabs=4 -RSFX
    else
        gdiff "$@"
    fi
}
greset() {
    echo "OK to reset and clean teh repo?"
    read -sq _
    (( $? )) && return 1
    /usr/bin/git clean -fd
    /usr/bin/git reset --hard
}
# }}}