#!/bin/zsh
bindkey -e

stty -ixon ixany
stty erase '^H'

zstyle ':completion:*' matcher-list 'm:{a-zA-Z}={A-Za-z} r:|[/]=* r:|=*'
zstyle ':completion:*' use-cache true

if [ -d "$HOME/.zfunc" ]; then
    fpath=($HOME/.zfunc $fpath)
    autoload -U ~/.zfunc/*(:t)
    function r() {
        local f
        f=(~/.zfunc/*(.))
        unfunction $f:t 2> /dev/null
        autoload -U $f:t
    }
fi

if [ -f "$HOME/.ssh/known_hosts" ]; then
    _cache_hosts=(`perl -ne  'if (/^([a-zA-Z0-9.-]+)/) { print "$1\n";}' ~/.ssh/known_hosts`)
fi

autoload -U colors
colors

setopt no_beep
setopt auto_cd
setopt auto_list
setopt auto_menu
setopt auto_name_dirs
setopt auto_param_keys
setopt auto_param_slash
setopt auto_remove_slash
setopt c_bases
setopt chase_links
setopt complete_aliases
setopt correct
setopt equals
setopt glob_complete
setopt prompt_bang
setopt prompt_percent
setopt prompt_subst
setopt pushd_ignore_dups
setopt pushd_to_home
setopt sh_file_expansion
setopt always_last_prompt
setopt extended_glob
setopt hist_ignore_all_dups
setopt hist_ignore_space
setopt inc_append_history
setopt share_history

function _we_are_in_gnu_screen () {
    [[ -n "$WINDOW" ]]
}

function _we_are_in_ssh_session () {
    [[ -n "$SSH_CONNECTION" ]]
}

_prompt_base='%B[%n@%m] %{%(?.$fg[green].$fg[red])%}%#%{$reset_color%}%b '
if ( _we_are_in_gnu_screen ); then
    # GNU Screen has a capability to set its window title by "ESC k
    # {string} ESC \". We want to reset the title to `_' to show we
    # aren't running any command now.
    PROMPT=$'%{\ek_\e\\%}'$_prompt_base
else
    PROMPT=$_prompt_base
fi

function preexec() { # Hook
    if { _we_are_in_gnu_screen } then
        # Set the window title of GNU Screen.
        local cmd=${1[(wr)^(*=*|sudo|nice|env|time|ssh|-*|[0-9]*)]}
        printf "\ek$cmd\e\\"
    fi
}

function precmd() { # Hook
    local dirinfo="$(print_directory_info $(pwd))"

    case $dirinfo in
        "")
            RPROMPT='%U%~%u';;
        *)
            RPROMPT="%U%~%u %U($dirinfo)%u";;
    esac
}

function print_directory_info() {
    local absdir=$1

    if [[ -r $absdir/.git ]] then
        # Git
        git branch --no-color 2>/dev/null \
            | sed -e '/^[^*]/d' -e 's/* \(.*\)/\1/'

    elif [[ -r $absdir/.hg/branch ]] then
        # Hg
        local hgbranch=`cat $absdir/.hg/branch`
        if [[ -e $absdir/.hg/unpushed ]] then
            echo "*${hgbranch}*"
        else
            echo "$hgbranch"
        fi

    elif [[ -d $absdir/_darcs ]] then
        # Darcs
        basename $absdir

    elif [[ -r $absdir/CVS/Repository ]] then
        # CVS
        cat $absdir/CVS/Repository \
            | sed -e 's!\([^/]*\).*!\1!'

    else
        # Unknown.
        if [[ $absdir = "/" ]] then
            # This is the root directory so exit from the recursion.
            echo
        else
            # Recurse to the parent dir.
            print_directory_info $(dirname $absdir)
        fi
    fi
}

function nfgrep() {
    fgrep --with-filename --line-number --context=1 --colour=always "$@" | $PAGER
}

function ppgrep() {
    local -a percol_opts

    if [[ $# -gt 0 ]]; then
        percol_opts=("--query" "$1")
    fi

    ps aux | percol "${percol_opts[@]}" | awk '{ print $2 }'
}

function ppkill() {
    local -a ppgrep_opts

    if [[ $1 =~ "^[^-]" ]]; then
        ppgrep_opts=("$1")
        shift
    fi

    ppgrep "${ppgrep_opts[@]}" | xargs kill "$@"
}

function percol_select_history() {
    local -a tac_cmd

    if which gtac >&/dev/null; then
        tac_cmd=("gtac")
    elif which tac >&/dev/null; then
        tac_cmd=("tac")
    else
        tac_cmd=("cat")
    fi

    local selected
    selected=$("${tac_cmd[@]}" "$HISTFILE" | \
        sed 's/^: [0-9]*:[0-9]*;//' | \
        awk 'seen[$0] {next} {seen[$0]++; print}' | \
        percol --match-method regex --query "$LBUFFER")
    if [[ $? -eq 0 ]]; then
        BUFFER="$selected"
        CURSOR="$#BUFFER" # move cursor to the end of line
        zle redisplay
        #zle accept-line  # uncomment this to execute the selected one immediately.
    else
        zle redisplay
    fi
}
if which percol >&/dev/null; then
    zle -N percol_select_history
    bindkey '^R' percol_select_history
fi

HISTFILE="$HOME/.zhistory"
HISTSIZE=6000000
SAVEHIST=6000000

alias -g L="| $PAGER"
alias -g G="| grep"
alias -g H="| head"
alias -g T="| tail"
alias -g X="| hexdump -C"

alias ..='cd ..'
alias cd..='cd ..'
alias p='pushd'
alias o='popd'
alias d='dirs'
alias df='df -h'
alias du='du -k'
alias ed='ed -p "ed> "'
alias man="LANG=C man"
alias mwget="wget --user-agent='Mozilla/1.4b'"
alias nc='nc -vv'
alias pkgsrc="pushd /usr/pkgsrc"
alias pstow="pushd /usr/local/stow"
alias w3m="w3m -O UTF-8"
alias w3mb="w3m -O UTF-8 -B"

alias aria-emacs="ssh -f -X admin@aria.cielonegro.org emacsclient -c"
alias pho-dev-boinc="ssh -f -X pho@pho.dev.office.ymir.co.jp env LANG=C /home/pho/var/BOINC/run_manager"

if [ $(hostname) = "aria.cielonegro.org" ]; then
    alias safari="open -a Safari"
    alias unstuff='open -a StuffIt\ Expander.app'
    alias pa='open -a NiseRingo.app'
    alias heboris='open /Applications/HeborisUEMac/exe/Heboris\ OpenGL.app'
    alias hengband='open ~/Applications/hengband*/Hengband*'
    alias syssleep='sudo shutdown -s now'
fi

if [[ $(uname) = "Darwin" ]]; then
    alias ldd="otool -L"

    alias ls='ls -Fw'
    alias la='ls -aw'
    alias l='ls -lw'
    alias lls='ls -law'

    alias top='/usr/bin/top -X -o cpu'
else
    alias ls='ls -F'
    alias la='ls -a'
    alias l='ls -l'
    alias lls='ls -la'
fi

function osname () {
    case $(uname) in
        "Darwin" | "FreeBSD" | "NetBSD")
            uname -v;;
        "Linux")
            if [[ -f "/etc/redhat-release" ]]; then
                cat "/etc/redhat-release"
            elif [[ -f "/etc/SuSE-release" ]]; then
                cat "/etc/SuSE-release"
            else
                echo -n "$0: I know this is "$(uname -o)" but " >&2
                echo    "have no idea about its distribution."  >&2
                return 1
            fi;;
        *)
            echo "$0: Failed to detect the OS name." >&2
            return 1
    esac
}

# The following lines were added by compinstall

zstyle :compinstall filename "$HOME/.zshrc"

autoload -U compinit
compinit
# End of lines added by compinstall

if ( _we_are_in_gnu_screen ); then
    # When we are in a GNU Screen session...

    if ( _we_are_in_ssh_session ); then
        # Propagate remote session's bells to the local one.
        screen -X vbell off
    fi
else
    if [ -e ".motd" ]; then
        cat ".motd"
    fi

    # Run "fortune" if we aren't in a GNU Screen session.
    if which fortune 2>&1 >/dev/null; then
        fortune
    fi
fi

if which ssh-agent-manager 2>&1 >/dev/null; then
    eval `ssh-agent-manager -s`
fi