#!/bin/bash
BASH_MAJMINVERSION=${BASH_VERSION%.*}
BASH_MAJVERSION=${BASH_VERSION%%.*}
BASH_MINVERSION=${BASH_MAJMINVERSION#${BASH_MAJVERSION}.}

FULLHOST=${HOSTNAME%%.*}

# source local configuration
if [ -f ~/.bash_local ]; then
  source ~/.bash_local
fi

#Start ssh in a screen session
sssh () {
    if ! ssh-add -l; then
        ssh-add
    fi
    screen -rd "$1" || screen -S "$1" "command ssh" "$1"
}

#Get a Kerberos ticket before logging into lxp
lxp () {
    if which klist &>/dev/null; then
        if klist -s; then
            command ssh lxp "$@"
        else
            kinit && command ssh lxp "$@"
        fi
    else
        command ssh lxp "$@"
    fi
}

#Check an ssh key is unlocked before sshing
ssh () {
    title=""
    for token in "$@"; do
        if [ "${token:0:1}" == "-" ]; then
            : #NOP
        elif [[ "$token" =~ @ ]]; then
            title="$token"
        else
            title="@$token"
        fi
        if [ "$token" == "lxp" -o "$token" == "lxplus" -o "$token" == "lxplus.cern.ch" ]; then
            lxp "${@/$token}"
            return
        fi
    done
    print_titlebar "$title"
    if [ -z "${HOSTNAME/*cern.ch}" ]; then
        command ssh "$@"
    elif [[ $HOSTNAME == "pi3" || $HOSTNAME =~ .+\.cern\.ch$ ]]; then
        ssh-add -l &> /dev/null || ssh-add -t 16h && command ssh "$@"
    else
        ssh-add -l &> /dev/null || ssh-add && command ssh "$@"
    fi
}

# Sanely search for process details
psgrep () {
    ps aux | {
        read titles
        echo "${titles}"
        grep -v grep | grep --color -i "$1"
    }
}

#Turn off echoing of ^C when interrupting a process - saves mangling the following prompt
stty -ctlecho

# A few aliases to improve the console and save time
export SKOORB=skoorb.net
export SERVER=server.skoorb.net
export DESKTOP=desktop.skoorb.net
export PI=pi.skoorb.net
export PI2=pi2.skoorb.net
export LAP=laptop.skoorb.net
export LXP=lxplus.cern.ch
export LA0=linappserv0.pp.rhul.ac.uk
export LA1=linappserv1.pp.rhul.ac.uk
export LA2=linappserv2.pp.rhul.ac.uk
export LA5=linappserv5.pp.rhul.ac.uk
export AFSHOME=/afs/cern.ch/user/b/brooks
export SVNOFF=svn+ssh://svn.cern.ch/reps/atlasoff
export SVNUSR=svn+ssh://svn.cern.ch/reps/atlasusr
export SVNGRP=svn+ssh://svn.cern.ch/reps/atlasgrp
export SVNTDAQ=svn+ssh://svn.cern.ch/reps/atlastdaq
export SVNINST=svn+ssh://svn.cern.ch/reps/atlasinst
alias s="screen -dR"
alias j="jobs"
alias g="git"
alias py="python"
alias py2="python2"
alias py3="python3"
alias ipy="ipython"
alias ipy3="ipython3"
alias ls="ls --color=auto"
alias sl="ls -r"
alias cda='cd $AFSHOME'
alias dir="ls *.*[^~]"
alias pdf="evince"
alias skoorb='ssh $SKOORB'
alias server='ssh $SERVER'
alias desktop='ssh $DESKTOP'
alias pi='ssh $PI'
alias pi2='ssh $PI2'
alias lap='ssh $LAP'
alias la0='ssh $LA0'
alias la1='ssh $LA1'
alias la2='ssh $LA2'
alias la5='ssh $LA5'
alias tree="tree -C"
alias grep='grep --color=auto'
alias fgrep='grep -F --color=auto'
alias egrep='grep -E --color=auto'
alias screen='env screen' # Forward through env vars to get X11 DISPLAY working
alias tbrowse='python -i ~/bin/browser.py '
export tmplog=/tmp/$USER/tmp.log
alias mailme='tee $tmplog;cat $tmplog | mail -s "Job done" "morphit2k@googlemail.com";rm $tmplog'
alias Date='date --rfc-3339=date' # YYYY-mm-dd format date

define () {
    expression=$(printf "/ && /%s" "$@")
    grep_expression=$(printf "\|%s" "$@")
    grep '^.\{1,50\}[^ ]:: ' "${HOME}"/vimwiki/*.wiki | gawk "${expression:5}/" IGNORECASE=1 | sed -e's/: *[*-] /:/' -e's/\[\[\(.*|\)\{0,1\}\([^]|]*\)\]\]/\2/' -e's/\$\([^$]*\)\$/\1/' -e's/.*\/\(.*\)\.wiki:\(.*\):: \(.*\)/\2 [\1]: \3/' | grep -i -e "$grep_expression"
}

todo () {
    IFS="
"
    if [ -z "$1" ]; then
        file_list=($(grep "\[ \]" "${HOME}"/vimwiki/*-*-*.wiki))
    else
        file_list=($(grep "\[ \]" "${HOME}"/vimwiki/*-*-*.wiki | grep -i --colour=always "$1"))
    fi
    select file_name in ${file_list[*]}
    do
        if [ -n "$file_name" ]; then
            vim -w ~/.vim/log "${file_name%%:*}"
        fi
        break
    done
}

paper () {
    files=()
    while IFS= read -d $'\0' -r file; do
        files=("${files[@]}" "$file")
    done < <(find ~/papers -xtype f -iname "*$1*" -print0)

    case "${#files[@]}" in
        0)
        echo "${RED}No candidates found${RESET}"
        return;;
        1)
        xdg-open "${files[0]}"
        return;;
    esac

    select file_name in "${files[@]}"; do
        if [ -f "$file_name" ]; then
            xdg-open "$file_name"
        elif [ -z "$file_name" ]; then
            return
        else
            echo "${RED}Could not open $file_name${RESET}"
        fi
        break
    done
}

tar () {
    # https://xkcd.com/1168/
    command tar "$@" || { echo "BOOM!"; false; }
}

scp () {
    # wrapper function to prevent scp of local-only files
    options="1246BCpqrvc:F:i:l:o:P:S:"
    while getopts $options option
    do
        case $option in
            1 | 2 | 4 | 6 | B | C | p | q | r | v)
                args+=($option)
                ;;
            c | F | i | l | o | P | S)
                args+=($option "$OPTARG")
                ;;
        esac
    done
    if [[ $1 != *:* && $2 != *:* ]]
    then
        echo "Are you sure you want to scp local to local?"
        echo "use 'command scp ARGS' if you're sure"
        return 1
    fi
    command scp "${args[@]}" "$1" "$2"
}

wordsearch () {
    pat=$1
    grep -e "^${pat}\$" /usr/share/dict/british-english-large
}

# Env vars for a few things
export EDITOR=vim
export PAGER=less
export LESS=-aiRsx4
# -a : search from end of this screen
# -i : ignore case when searching
# -R : output ANSI color codes as raw control characters (use ^notation for other control codes)
# -s : truncate multiple blank lines to a single one
# -x4 : set tab stops to multiples of 4 characters

alias vim='vim -w ~/.vim/log'

# Set the LS_COLORS variable
if [ "${HOSTNAME}" != "linappserv0.pp.rhul.ac.uk" ]; then
    eval "$(dircolors "${HOME}"/.config/colors)"
else
    eval "$(dircolors <(grep -v "RESET\|MULTIHARDLINK\|CAPABILITY" .config/colors))"
fi

# define some colors which will be used in the prompt
export NO_COLOUR=$'\033[0m'
export RED=$'\033[0;31m'
export LIGHT_RED=$'\033[1;31m'
export GREEN=$'\033[0;32m'
export LIGHT_GREEN=$'\033[1;32m'
export YELLOW=$'\033[0;33m'
export LIGHT_YELLOW=$'\033[1;33m'
export BLUE=$'\033[0;34m'
export LIGHT_BLUE=$'\033[1;34m'
export MAGENTA=$'\033[0;35m'
export LIGHT_MAGENTA=$'\033[1;35m'
export CYAN=$'\033[0;36m'
export LIGHT_CYAN=$'\033[1;36m'
export WHITE=$'\033[0;37m'
export LIGHT_WHITE=$'\033[1;37m'

if [[ "$TERM" =~ screen ]]; then
    START_TITLE=$"\033k"
    END_TITLE=$"\033\\"
else
    START_TITLE=$"\033]0;"
    END_TITLE=$"\007"
fi

print_titlebar () {
  # Echo title command sequence to the terminal
  if [ -n "$1" ]; then
    printf "${START_TITLE}%s${END_TITLE}" "$1"
  fi
}

newline () {
    # Return if cursor is not at col 1, see: http://unix.stackexchange.com/questions/88296/get-vertical-cursor-position/183121#183121
    IFS=';' read -sdR -p $'\E[6n' _ COL
    if [ "$COL" != 1 ]; then
        echo
    fi
}

get_titlebar () {
  TTY=$(tty)
  TTY=${TTY##*/}
  if [ -n "$SSH_CLIENT" ] || [ -n "$SSH_TTY" ]; then
    local _host=${FULLHOST#*-}

    if [ -n "$_host" ]; then
      _host="@${_host}"
    fi
  fi
  echo "${TTY}${TERM_TEXT}${_host}:\$(PWD)"
}

# List the screen id if this is a screen session
if [ -n "$STY" ]; then
    TERM_TEXT="${STY#*.}"
    TERM_TEXT="[${TERM_TEXT%.$HOSTNAME}.${WINDOW}]"
fi

# Don't junk up simple linux terminals
if [[ "$TERM" =~ linux ]];then #-o "$STY" == "" ]; then
  export PROMPT_COMMAND=""
else
  titlebar=$(get_titlebar)
  export PROMPT_COMMAND="newline;print_titlebar ${titlebar}"
  export TERM=xterm-256color
fi

# Set up the prompt to display user@first_part_of_server_name
# and the pwd relative to our home directory and Testarea
PWD () {
  es=$?
  local ps1="${PWD/$TestArea/$}"
  if [ "$BASH_MAJVERSION" -ge 4 ] && [ "$BASH_MINVERSION" -ge 3 ]; then
    ps1="${ps1/$HOME/\~}"   #Bash 4.3 now expands in substitutions
  else
    ps1="${ps1/$HOME/~}"    #But prior versions didn't
  fi
  ps1="${ps1/$AFSHOME/@}"
  echo "$ps1"
  return $es
}

prompt_exit () {
  #Set colour red or green based on exit code of last command
  es=$?
  if [ $es -eq 0 ]; then
    status_color="$LIGHT_GREEN"
  else
    status_color="$LIGHT_RED"
  fi
  echo $status_color
  return $es
}

# don't put duplicate lines in the history. See bash(1) for more options
# ... or force ignoredups and ignorespace
HISTCONTROL=ignoredups:ignorespace

# save infinite history
if [ "$BASH_MAJVERSION" -ge 4 ] && [ "$BASH_MINVERSION" -ge 3 ]; then
    HISTSIZE=-1
    HISTFILESIZE=-1
else
    HISTSIZE=
    HISTFILESIZE=
fi

# Use Vi shortcuts
set -o vi

# Expand variables in paths in bash v4
if [ "$BASH_MAJVERSION" == 4 ] && [ "$BASH_MINVERSION" -ge 2 ]; then
    shopt -s direxpand
fi

# append to the history file, don't overwrite it
shopt -s histappend

# check the window size after each command and, if necessary,
# update the values of LINES and COLUMNS.
shopt -s checkwinsize

# enable programmable completion features (you don't need to enable
# this, if it's already enabled in /etc/bash.bashrc and /etc/profile
# sources /etc/bash.bashrc).
if [ -f /etc/bash_completion ] && ! shopt -oq posix; then
    . /etc/bash_completion
    # Override dynamic completion loading to resolve aliases
    _completion_loader()
    {
        # Resolve alias
        name=${BASH_ALIASES[$1]:-$1}
        local compfile=/usr/share/bash-completion/completions
        compfile+="/${name##*/}"
        # echo " -completing $1 as $name with $compfile"

        # Avoid trying to source dirs; https://bugzilla.redhat.com/903540
        # Set alias completion to real completion
        if [[ -f "$compfile" ]] && . "$compfile" &>/dev/null; then
            # Find the completion set by $compfile and substitute in the alias
            local comp_line
            comp_line=$(complete | grep "${name}\$" | sed "s/${name}\$/${1}/")
            eval "$comp_line"
            # Tell bash to retry completion now this alias has been set
            return 124
        fi

        # Need to define *something*, otherwise there will be no completion at all.
        complete -F _minimal "$1" && return 124
    }
fi

# Set prompt
# See http://www.gnu.org/software/bash/manual/bash.html#Controlling-the-Prompt

# Set colour to red or green based on last command status
PS1="\[\$(prompt_exit)\]"
# Time in HH:MM:SS form
PS1+="\t"
# Number of jobs managed by this shell
PS1+="\[$NO_COLOUR\]<\[$BLUE\]\j"
# TTY number for this shell
PS1+="\[$NO_COLOUR\]/\[$BLUE\]\l"
# Screen name
PS1+="\[$YELLOW\]$TERM_TEXT"
# Hostname
if [ -n "$SSH_CLIENT" ] || [ -n "$SSH_TTY" ]; then
    PS1+="\[$NO_COLOUR\]@\[$BLUE\]${FULLHOST#*-}"
fi
# Present working directory, with some abbreviations
PS1+="\[$NO_COLOUR\]:\[$LIGHT_WHITE\]\$(PWD)"
# Git branch, if available
if type __git_ps1 &> /dev/null; then
    PS1+="\$(__git_ps1 '\[$NO_COLOUR\]<\[$GREEN\]%s')"
fi
# End of prompt
PS1+="\[$NO_COLOUR\]> "

if go version &> /dev/null; then
    export GOROOT
    GOROOT=$(go env GOROOT)
fi

# Set up virtualenv
export WORKON_HOME=$HOME/.virtualenvs
export PROJECT_HOME=$HOME/dev

# Set up environment variables for Python and ROOT (Allowing pyROOT)
export PYTHONDIR=/usr/bin/python
export PYTHONSTARTUP=$HOME/.pythonrc

export LD_LIBRARY_PATH=$ROOTSYS/lib:$PYTHONDIR/lib:$MINUIT2DIR/lib:$PYTHIA8/lib
export PYTHONPATH=$ROOTSYS/lib:$PYTHONDIR/lib
export PATH=$HOME/.local/bin:$HOME/.cargo/bin:$ROOTSYS/bin:$PATH
alias root="root -l"

# Set up ATLAS computing environment
export ATLAS_LOCAL_ROOT_BASE=/cvmfs/atlas.cern.ch/repo/ATLASLocalRootBase
alias atlsetup='source ${ATLAS_LOCAL_ROOT_BASE}/user/atlasLocalSetup.sh'

# Store ssh settings in a hostname specific script
SSH_ENV="$HOME/.ssh/${HOSTNAME%%.*}.environment"

start_agent () {
     echo "Initialising new SSH agent..."
     /usr/bin/ssh-agent | sed 's/^echo/#echo/' > "${SSH_ENV}"
     chmod 600 "${SSH_ENV}"
     . "${SSH_ENV}" > /dev/null
     # Import identities immediately
#     /usr/bin/ssh-add;
}

# Source SSH settings, if applicable

if [ -f "${SSH_ENV}" ]; then
     . "${SSH_ENV}" > /dev/null
     #ps ${SSH_AGENT_PID} doesn't work under cywgin, hopefully pgrep does
     pgrep -u "$USER" ssh-agent > /dev/null || {
         start_agent;
     }
else
    start_agent;
fi

#Move environment between sessions, i.e. into an old screen
ENV_FILE=${HOME}/.env
push_env() {
    if [ -f "$ENV_FILE" ]; then
        # echo "Removing $ENV_FILE"
        rm "$ENV_FILE"
    fi
    # mkfifo $ENV_FILE

    for var in "$@"; do
        # eval echo ${var}=\$${var}
        eval echo "export ${var}=\$${var} >> $ENV_FILE"
    done
}
pull_env() {
    if [ ! -e "$ENV_FILE" ]; then
        echo "$ENV_FILE not found!"
        return 1
    fi
    while read i; do
        # echo -n "receiving: \"$i\""
        # echo $i
        eval "$i"
    done < "$ENV_FILE"
}
#If this is a new shell, not a screen; save out the X DISPLAY variable
if [ -z "$STY" ]; then
    push_env DISPLAY
fi

# End up in /home/$USER
#cd ~