# to debug
#set -x
# redirect output to log file. this doesn't work. todo figure out why
#exec 1>>/a/tmp/bashlog
#exec 2>>/a/tmp/bashlog


# By default this file is sourced for ALL ssh commands. This is wonky.
# Normally, this file is not sourced when a script is run, but we can
# override that by having #!/bin/bash -l.  I want something similar for ssh
# commands. when a local script runs an ssh command, this file should not be
# sourced by default, but we should be able to override that.
#
# so here we test for conditions of a script under ssh and return if so. To test
# for an overriding condition, we have a few options. one is to use an
# environment variable.  env variables sent across ssh are strictly limited. ssh
# -t which sets $SSH_TTY, but within a script that won't work because tty
# allocation will fail.  We could override an obscure unused LC_var, like
# telephone, but I don't want to run into some edge case where that messes
# things up. we could transfer a file which we could test for, but I can't think
# of a way to make that inherently limited to a single ssh command.  I choose to
# set SendEnv and AcceptEnv ssh config vars to allow the environment variable
# BASH_LOGIN_SHELL to propagate across ssh.

# assume we want ssh commands to source this file if we are sourcing it,
# and we haven't specified otherwise already
[[ ! $BASH_LOGIN_SHELL ]] && export BASH_LOGIN_SHELL=true

# first conditions show that we are an ssh command without an interactive shell
if [[ $SSH_CONNECTION ]] \
       && [[ $- == *c* ]] \
       && [[ ! $SSH_TTY ]] \
       && [[ $- != *i* ]] \
       && [[ ! $BASH_LOGIN_SHELL == true ]]; then
    return
else
    source /etc/profile
fi



# note, to catch errors in functions but not outside, do:
# set -E -o pipefail
# trap return ERR
# trap 'trap ERR' RETURN



############
# settings #
############

CDPATH=.

# remove all aliases. aliases provided by the system tend to get in the way,
# for example, error happens if I try to define a function the same name as an alias
unalias -a

# remove gnome keyring warning messages
# there is probably a more proper way, but I didn't find any easily on google
# now using xfce+xmonad instead of vanilla xmonad, so disabling this
#unset GNOME_KEYRING_CONTROL

# use extra globing features.
shopt -s extglob
# include .files when globbing, but ignore files name . and ..
# setting this also sets dotglob
export GLOBIGNORE=*/.:*/..

# broken with bash_completion package. Saw a bug for this once. Don't anymore.
# still broken in wheezy
# still buggered in latest stable from the web, version 2.1
# perhaps its fixed in newer git version, which fails to make for me
# this note is from 6-2014.
# Also, enabling this before sourcing .bashrc makes PATH be empty.
#shopt -s nullglob

# make tab on an empty line do nothing
shopt -s no_empty_cmd_completion

# advanced completion
# http://bash-completion.alioth.debian.org/
# might be sourced by the system already, but I've noticed it not being sourced before
if ! type _init_completion &> /dev/null && [[ -r "/usr/share/bash-completion/bash_completion" ]]; then
    . /usr/share/bash-completion/bash_completion
fi


# fix spelling errors for cd, only in interactive shell
shopt -s cdspell
# append history instead of overwritting it
shopt -s histappend
# for compatibility, per gentoo/debian bashrc
shopt -s checkwinsize
# attempt to save multiline single commands as single history entries.
shopt -s cmdhist
shopt -s globstar


# inside emacs fixes
if [[ $INSIDE_EMACS ]]; then
    # EMACS is used by bash on startup, but we don't need it anymore.
    # plus I hit a bug in a makefile which inherited it
    unset EMACS
    export INSIDE_EMACS
    export PAGER=cat
    export MANPAGER=cat
    # scp completion does not work, but this doesn't fix it. todo, figure this out
    complete -r scp &> /dev/null
    # todo, remote file completion fails, figure out how to turn it off
fi


if [[ $- == *i* ]]; then
    # for readline-complete.el
    if [[ $INSIDE_EMACS ]]; then
        # all for readline-complete.el
        stty echo
        bind 'set horizontal-scroll-mode on'
        bind 'set print-completions-horizontally on'
        bind '"\C-i": self-insert'
    else
        # arrow keys. for other terminals, see http://unix.stackexchange.com/questions/10806/how-to-change-previous-next-word-shortcut-in-bash
        if [[ $TERM == "xterm" ]]; then
            bind '"\e[1;5C": shell-forward-word' 2>/dev/null
            bind '"\e[1;5D": shell-backward-word' 2>/dev/null
        else
            bind '"\eOc": shell-forward-word'
            bind '"\eOd": shell-backward-word'
        fi
        # terminal keys: C-c, C-z. the rest defined by stty -a are, at least in
        # gnome-terminal, overridden by bash, or disabled by the system
        stty werase undef lnext undef stop undef start undef

    fi

fi


# history number. History expansion is good.
PS4='$LINENO+ '
# history file size limit, set to unlimited.
# this needs to be different from the default because
# default HISTFILESIZE is 500 and could clobber our history
HISTFILESIZE=
# max commands 1 session can append/read from history
HISTSIZE=100000
# my own history size limit based on lines
HISTFILELINES=1000000
HISTFILE=$HOME/.bh
# the time format display when doing the history command
# also, setting this makes the history file record time
# of each command as seconds from the epoch
HISTTIMEFORMAT="%I:%M %p %m/%d "
# consecutive duplicate lines don't go in history
HISTCONTROL=ignoredups
# works in addition to HISTCONTROL to do more flexible things
# it could also do the same things as HISTCONTROL and thus replace it,
# but meh
HISTIGNORE='k *; *'

export BC_LINE_LENGTH=0


# note, if I use a machine I don't want files readable by all users, set
# umask 077  # If fewer than 4 digits are entered, leading zeros are assumed

C_DEFAULT_DIR=/a


###################
## include files ###
###################

for _x in /a/bin/distro-functions/src/* /a/bin/bash-programs-by-ian/repos/*/*-function; do
    source "$_x"
done
unset _x
source /a/bin/semi-private # so I can share my bashrc
source $(dirname $(readlink -f $BASH_SOURCE))/path_add-function
path_add --ifexists --end /a/opt/adt-bundle*/tools /a/opt/adt-bundle*/platform-tools
# todo, these need to be renamed to be less generic.
# sync overrode something else useful
#path_add $HOME/bin/bash-programs-by-ian/utils

[[ -e $HOME/mw_vars ]] && source $HOME/mw_vars

###############
### aliases ###
###############

# disabled for now, but these are generally good
# if [[ $- == *i* ]]; then
#     alias cp='cp -i'
#     alias mv='mv -i'
# fi

# remove any default aliases for these
alias ls > /dev/null 2>&1 && unalias ls
alias ll > /dev/null 2>&1 && unalias ll
alias grep > /dev/null 2>&1 && unalias grep


mkdir() {
    command mkdir -p "$@"
}


alias d='builtin bg'
complete -A stopped -P '"%' -S '"' d

alias hi='history'


# note: gksudo is recommended for X apps because it does not set the
# home directory to the same, and thus apps writing to ~ fuck things up
# with root owned files.

# background
# alias s='SUDOD="$PWD" sudo -i '
# because this is an alias, and the extra space at the end, it would allow
#  aliases to be used with it. but aliases aren't used in scripts,
#  better to eliminate inconsistencies. Plus, you can't do s=s; $s command
#  with an alias, which I like to do in some functions
# extra space at the end allows aliases to work
s() {
    if [[ $EUID != 0 || $1 == -* ]]; then
        SUDOD="$PWD" sudo -i "$@"
    else
        "$@"
    fi
}



if [[ $OS == Windows_NT ]]; then
    alias ffs='cygstart "/c/Program Files (x86)/Mozilla Firefox/firefox.exe" -P scratch'
    export DISPLAY=nt
    alias j='command cygpath'
    alias t='command cygstart'
    alias cygstart='echo be quick, use the alias "t" instead :\)'
    alias cygpath='echo be quick, use the alias "j" instead :\)'

fi





#####################
###  functions   ####
#####################


sdf() {
    c /sdx/test/sandbox/
}

debian_pick_mirror () {
    # netselect-apt finds a fast mirror.
    # but we need to replace the mirrors ourselves,
    # because it doesn't do that. best it can do is
    # output a basic sources file
    # here we get the server it found, get the main server we use
    # then substitute all instances of one for the other in the sources file
    # and backup original to /etc/apt/sources.list-original.
    # this is idempotent. the only way to identify debian sources is to
    # note the original server, so we put it in a comment so we can
    # identify it later.
    local file=$(mktemp -d)/f # safe way to get file name without creating one
    sudo netselect-apt -o "$file" || return 1
    url=$(grep ^\\w $file | head -n1 | awk '{print $2}')
    sudo cp -f /etc/apt/sources.list /etc/apt/sources.list-original

    sudo sed -ri "/http.us.debian.org/ s@( *[^ #]+ +)[^ ]+([^#]+).*@\1$url\2# http.us.debian.org@" /etc/apt/sources.list
    sudo apt-get update
}


a() {
    beet "${@}"
}

dus() {
    du -sh ${@:-*} | sort -h
}

t() {
    local x
    local -a args
    if type -t trash-put >/dev/null; then
        # skip args that don't exist, or else it's an err
        for x in "$@"; do [[ ! -e $x ]] || args+=("$x"); done
        [[ ! ${args[@]} ]] || trash-put "${args[@]}"
    else
        rm -rf "$@"
    fi
}

ccat () { # config cat. see a config without extra lines.
    grep '^\s*[^[:space:]#]' "$@"
}


if type ack-grep >/dev/null 2>&1; then
    alias ack=ack-grep
fi

postconfin() {
    local MAPFILE
    mapfile -t
    local s
    [[ $EUID == 0 ]] || s=s
    $s postconf -ev "${MAPFILE[@]}"
}


tclock() {
    clear
    date +%l:%_M
    len=60
    # this goes to full width
    #len=${1:-$((COLUMNS -7))}
    x=1
    while true; do
        if (( x == len )); then
            end=true
            d="$(date +%l:%_M)   "
        else
            end=false
            d=$(date +%l:%M:%_S)
        fi
        echo -en "\r"
        echo -n "$d"
        for ((i=0; i<x; i++)); do
            if (( i % 6 )); then
                echo -n _
            else
                echo -n .
            fi
        done
        if $end; then
            echo
            x=1
        else
            x=$((x+1))
        fi
        sleep 5
    done
}

gr() {
    grep -iIP --color=auto "$@"
}

grr() {
    if [[ ${#@} == 1 ]]; then
        grep -riIP --color=auto "$@" .
    else
        grep -riIP --color=auto "$@"
    fi
}

pub() {
    rld /a/h/_site/ li:/var/www/iankelling.org/public_html
}

mkc() {
    mkdir "$1"
    c "$1"
}

bashrcpush () {
    local startdir="$PWD"
    cd ~
    for x in "$@"; do
        ssh $x mkdir -p bin/distro-functions/src
        tar cz bin/bash-programs-by-ian bin/semi-private bin/distro-functions/src | ssh $x tar xz
    done
    cd $(mktemp -d)
    command cp /a/c/repos/bash/!(.git) ~/.gitconfig .
    for x in "$@"; do
        tar cz * | ssh $x tar xz
    done
    cd "$startdir"
}

virshrm() {
    for x in "$@"; do virsh destroy "$x"; virsh undefine "$x"; done
}

whatismyip() { curl ipecho.net/plain ; echo; }

# history search
k() { grep -P --binary-files=text "$@" ${HISTFILE:-~/.bash_history}  | tail -n 40; }

calc() { echo "scale=3; $*" | bc -l; }
# no having to type quotes, but also no command history:
clc() {
    local x
    read -r x
    echo "scale=3; $x" | bc -l
}

# diff config files,
# setup for format of postfix
# option = stuff[,]
#    [more stuff]
cdiff() {

    local pastline
    local unified="$(mktemp)"
    local f1="$(mktemp)"
    local f2="$(mktemp)"

    _cdiff-prep "$1" "$f1"
    _cdiff-prep "$2" "$f2"
    cat "$f1" "$f2" |  grep -Po '^[^=]+=' | sort | uniq > "$unified"
    while IFS= read -r line; do
        # the default bright red / blue doesn't work in emacs shell
        dwdiff -cblue,red -A best -d " 	," <(grep "^$line" "$f1" || echo ) <(grep "^$line" "$f2" || echo ) | colordiff
    done < "$unified"
}

cim() {
    git commit -m "$*"
}

cam() {
    git commit -am "$*"
}

shellck() {
    # 2086 = unquoted $var
    # 2046 = unquoted $(cmd)
    # i had -x as an arg, but debian testing(stretch) doesn't support it
    shellcheck -e 2086,2046,2068,2006,2119 "$@"
}

# join options which are continued to multiples lines onto one line
_cdiff-prep() {
    local first=true
    grep -vE '^([ \t]*#|^[ \t]*$)' "$1" | while IFS= read -r line; do
        # remove leading spaces/tabs. assumes extglob
        if [[ $line == "[ 	]*" ]]; then
            line="${line##+( )}"
        fi
        if $first; then
            pastline="$line"
            first=false
        elif [[ $line == *=* ]]; then
            echo "$pastline" >> "$2"
            pastline="$line"
        else
            pastline="$pastline $line"
        fi
    done
    echo "$pastline" >> "$2"
}

# makes it so chown -R symlink affects the symlink and its target.
chown() {
    if [[ $1 == -R ]]; then
	shift
	command chown -h "$@"
	command chown "$@"
	command chown -RH "$@"
    else
	command chown "$@"
    fi
}



cgpl ()
{
    if [[ $# == 0 ]]; then
        cp /a/bin/data/COPYING .
    else
        cp /a/bin/data/COPYING "$@"
    fi
}


dc() {
    diff --strip-trailing-cr -w "$@"   # diff content
}


dt() {
    date "+%A, %B %d, %r" "$@"
}


e() { echo "$@"; }


envload() { # load environment from a previous: export > file
    local file=${1:-$HOME/.${USER}_env}
    eval "$(export | sed 's/^declare -x/export -n/')"
    while IFS= read -r line; do
	# declare -x makes variables local to a function
	eval ${line/#declare -x/export}
    done < "$file"
}



# havn't tested these:
#file cut copy and paste, like the text buffers :)
_fbufferinit() { # internal use by
    ! [[ $my_f_tempdir ]] && my_f_tempdir=$(mktemp -d)
    rm -rf "$my_f_tempdir"/*
}
fcp() { # file cp
    _fbufferinit
    cp "$@" "$my_f_tempdir"/
}
fct() { # file cut
    _fbufferinit
    mv "$@" "$my_f_tempdir"/
}
fpst() { # file paste
    [[ $2 ]] && { echo too many arguments; return 1; }
    target=${1:-.}
    cp "$my_f_tempdir"/* "$target"
}


# find array. make an array of file names found by find into $x
# argument: find arguments
# return: find results in an array $x
fa() {
    while read -rd ''; do
	x+=("$REPLY");
    done < <(find "$@" -print0);
}


git_empty_branch() { # start an empty git branch. carefull, it deletes untracked files.
    [[ $# == 1 ]] || { echo 'need a branch name!'; return 1;}
    local gitroot
    gitroot || return 1 # function to set gitroot
    builtin cd "$gitroot"
    git symbolic-ref HEAD refs/heads/$1
    rm .git/index
    git clean -fdx
}


ff() {
    if type -P firefox &>/dev/null; then
        firefox "$@"
    else
        iceweasel "$@"
    fi
}

fw() {
    firefox -P default "$@" >/dev/null 2>&1
}

fn() {
    firefox -P alt "$@" >/dev/null 2>&1
}





# horizontal row. used to break up output
hr() { printf "$(tput setaf 5)█$(tput sgr0)%.0s" $(seq $COLUMNS); }


i() {
    git "$@"
}
# modified from ~/local/bin/git-completion.bash
# other completion commands are mostly taken from bash_completion package
complete -o bashdefault -o default -o nospace -F _git i 2>/dev/null \
    || complete -o default -o nospace -F _git i

# fast commit all
ic() {
    git commit -am "$*"
}

# insensitive find
ifn () {
    find -L . -iname "*$**" 2>/dev/null
}




l() {
    if [[ $PWD == /[iap] ]]; then
	command ls -A --color=auto -I lost+found "$@"
    else
	command ls -A --color=auto "$@"
    fi
}



lld() { ll -d "$@"; }


low() {  # make filenames all lowercase
    local x y
    for x in "$@"; do
	y=$(tr "[A-Z]" "[a-z]" <<<"$x")
	[[ $y != $x ]] && mv "$x" "$y"
    done
}


lower() { # make first letter of filenames lowercase.
    local x
    for x in "$@"; do
	if [[ ${x::1} == [A-Z] ]]; then
	    y=$(tr "[A-Z]" "[a-z]" <<<"${x::1}")"${x:1}"
            safe_rename "$x" "$y"
	fi
    done
}

safe_rename() {
    if [[ $# != 2 ]]; then
        echo safe_rename error: $# args, need 2 >2
        return 1
    elif [[ $1 != $2 ]]; then
        if [[ -e $2 ]]; then
            echo Cannot rename "$1" to "$2" as it already exists.
        else
            mv "$1" "$2"
        fi
    fi
}

despace() {
    local x y
    for x in "$@"; do
        y="${x// /_}"
        safe_rename "$x" "$y"
    done
}





# test existence / exists
te() {
    local ret=0
    for x in "$@"; do
        [[ -e "$x" || -L "$x" ]] || ret=1
    done
    return $ret
}

# show make targets, via http://stackoverflow.com/questions/3063507/list-goals-targets-in-gnu-make
make-targets() {
    make -qp | awk -F':' '/^[a-zA-Z0-9][^$#\/\t=]*:([^=]|$)/ {split($1,A,/ /);for(i in A)print A[i]}'
}


# fix root file ownership for FILE argument.
# check if parent or grandparent is not root and if the dir of FILE is also
# owned by that user, and change ownership to that user
perm_fix() {
    local parent
    if [[ $EUID == 0 ]]; then
	te "$1" || touch "$1"
	if [[ $(stat -c "%u" "$1") == 0 ]] ; then
	    argdir=$(dirname "$1")
	    if [[ $(stat -c "%u" "$argdir") != 0 ]] ; then
		if ! chown "--reference=$argdir" "$1"; then
		    echo failed to fix bad ownership file permissons
                    return 1
                fi
	    fi
	fi
    fi
}

# see notes for usage
fsdiff () {
    local missing=false
    local dname="${PWD##*/}"
    local m="/a/tmp/$dname-missing"
    local d="/a/tmp/$dname-diff"
    [[ -e $d ]] && rm "$d"
    [[ -e $m ]] && rm "$m"
    local msize=0
    local fsfile
    while read -r line; do
        fsfile="$1${line#.}"
        if [[ -e "$fsfile" ]]; then
            md5diff "$line" "$fsfile" && tee -a "/a/tmp/$dname-diff" <<< "$fsfile $line"
        else
            missing=true
            echo "$line" >> "$m"
            msize=$((msize + 1))
        fi
    done < <(find -type f )
    if $missing; then
        echo "$m"
        (( msize <= 100 )) && cat $m
    fi
}

fsdiff-test() {
    cd $(mktemp -d)
    echo ok > a
    echo nok > b
    mkdir c
    echo ok > c/d
    local x=$(mktemp -d)
    mkdir $x/c
    echo different > $x/c/d
    echo ok > $x/a
    fsdiff $x
}
# expected output, with different tmp dirs
# /tmp/tmp.HDPbwMqdC9/c/d ./c/d
# /a/tmp/tmp.qLDkYxBYPM-missing
# ./b


# test whether missing files were renamed, generally for use with fsdiff
# $1 = fsdiff output file, $2 = directory to compare to. pwd = fsdiff dir
# echos non-renamed files
rename-test() {
    local x y found
    unset sums
    for x in "$2"/*; do
        { sums+=( "$(md5sum < "$x")" ) ; } 2>/dev/null
    done
    while read -r line; do
        { missing_sum=$(md5sum < "$line") ; } 2>/dev/null
        renamed=false
        for x in "${sums[@]}"; do
            if [[ $missing_sum == "$x" ]]; then
                renamed=true
                break
            fi
        done
        $renamed || echo "$line"
    done < "$1"
    return 0
}


# get a random string
#par2rbase=$(head -c 50 /dev/urandom | tr -cd '[:alpha:]')
par2r() {
    [[ $par2rbase ]] || { echo "set \$par2rbase first"; return 1; }
    d="$PWD"
    for x in **; do
        if [[ -d $x ]]; then
            cd "$x"
            par2 c QDLDeDJ6z4.par2 *
            cd "$d"
        fi
    done

}

md5diff() {
    [[ $(md5sum < "$1") != $(md5sum < "$2") ]]
}


pfind() { #find *$1* in $PATH
    [[ $# != 1 ]] && { echo requires 1 argument; return 1; }
    local pathArray
    IFS=: pathArray=($PATH); unset IFS
    find "${pathArray[@]}" -iname "*$1*"
}

# do pwd + some other info.
#pwd() { echo "$(ll -d "$PWD")		$USER@$HOSTNAME		$(date +%r)"; }

q() { # start / launch a program in the backround and redir output to null
    "$@" &> /dev/null &
}

pwgen() {
    apg -s -m 10 -x 14 -t
}

r() {
    exit "$@" 2>/dev/null
}

# connect to vms made with virt-install
vspicy() {
    spicy -p $(sudo virsh dumpxml "$1"|grep "<graphics.*type='spice'"|\
                   sed -r "s/.*port='([0-9]+).*/\1/")
}
rspicy() { # HOST DOMAIN
    local port=$(ssh $1<<EOF
sudo virsh dumpxml $2|grep "<graphics.*type='spice'" | \
       sed -rn "s/.*port='([0-9]+).*/\1/p"
EOF
          )
    if [[ $port ]]; then
        spicy -h $1 -p $port
    else
        echo "error: no port found. check that the domain is running."
    fi
}


# trash-restore lists everything that has been trashed at or below CWD
# This picks out files just in CWD, not subdirectories,
# which also match grep $1, usually use $1 for a time string
# which you get from running restore-trash once first
pick-trash() {
    local name x ask
    local nth=1
    # last condition is to not ask again for ones we skipped
    while name="$( echo | restore-trash | gr "$PWD/[^/]\+$" | gr "$1" )" \
            && [[ $name ]] && (( $(wc -l <<<"$name") >= nth )); do
        name="$(echo "$name" | head -n $nth | tail -n 1 )"
        read -p "$name [Y/n] " ask
        if [[ ! $ask || $ask == [Yy] ]]; then
            x=$( echo "$name"  | gr -o "^\s*[0-9]*" )
            echo $x | restore-trash > /dev/null
        elif [[ $ask == [Nn] ]]; then
            nth=$((nth+1))
        else
            return
        fi
    done
}

# rsync, root is required to keep permissions right.
# rsync  --archive --human-readable --verbose --itemize-changes --checksum \(-ahvic\) \
    #					--no-times --delete
# basically, make an exact copy, use checksums instead of file times to be more accurate
rl() { rsync -ahvic --delete "$@"; }
# don't delete files on the target end which do not exist on the original end:
rld() { rsync -ahvic "$@"; }
# useful for selectively sending dirs which have been synced with unison,
# where the path is the same on both hosts.
rlu() { # [OPTS] HOST PATH
    # eg rlu -opts frodo testpath
    o=o
    opts=("${@:1:$#-2}") #  1 to last -2
    path="${@:$#}"
    host="${@:$#-1:1}"
    if [[ $host == lj ]]; then
        o=
    fi
    # rync options are without -g for group
    s rsync -rlpthvi$o "${opts[@]}" "$path" "root@$host:$path";
}

lcn() { locate -i "*$**"; }

vpn() {
    s systemctl start openvpn@client&
    journalctl --unit=openvpn@client -f -n0
}
vpnoff() {
    s systemctl stop openvpn@client
}


complete -F _rsync -o nospace rld rlt fl
# rl without preserving modification time. for some reason I had this as default before.
# perhaps that reason will come up again and I will document it.
rlt() { rsync -ahvic --delete --no-t "$@"; }

if ! type service &>/dev/null; then
    service() {
        echo actually running: systemctl $2 $1
        systemctl $2 $1
    }
fi

# use sb instead of s is for sudo redirections, eg. sb 'echo "ok fine" > /etc/file'
sb() {
    local SUDOD="$PWD"
    sudo -i bash -c "$@"
}
complete -F _root_command s sb

# use -ll, less secure but faster.
srm () {
    command srm -ll "$@"
}

# sudo redo. be aware, this command may not work right on strange distros or earlier software
sr() {
    if [[ $# == 0 ]]; then
	sudo -E bash -c -l "$(history -p '!!')"
    else
	echo this command redos last history item. no argument is accepted
    fi
}

rbpipe() { rbt post -o --diff-filename=- "$@"; }
rbp() { rbt post -o "$@"; }


# log with script. timing is $1.t and script is $1.s
# -l to save to ~/typescripts/
# -t to add a timestamp to the filenames
slog() {
    local logdir do_stamp arg_base
    (( $# >= 1 )) || { echo "arguments wrong"; return 1; }
    logdir="/a/dt/"
    do_stamp=false
    while getopts "lt" option
    do
	case $option in
	    l ) arg_base=$logdir ;;
            t ) do_stamp=true ;;
        esac
    done
    shift $(($OPTIND - 1))
    arg_base+=$1
    [[ -e $logdir ]] || mkdir -p $logdir
    $do_stamp && arg_base+=$(date +%F.%T%z)
    script -t $arg_base.s 2> $arg_base.t
}
splay() { # script replay
    #logRoot="$HOME/typescripts/"
    #scriptreplay "$logRoot$1.t" "$logRoot$1.s"
    scriptreplay "$1.t" "$1.s"
}


# like -e for functions. returns on error.
# at the end of the function, disable with:
# trap ERR
funce() {
    trap 'echo "${BASH_COMMAND:+BASH_COMMAND=\"$BASH_COMMAND\" }
${FUNCNAME:+FUNCNAME=\"$FUNCNAME\" }${LINENO:+LINENO=\"$LINENO\"  }\$?=$?"
trap ERR
return' ERR
}

# timer in minutes
tm() {
    (sleep $(calc "$@ * 60") && mpv --volume 50 /a/bin/data/alarm.mp3) > /dev/null 2>&1 &
}

ser() {
    local s; [[ $EUID != 0 ]] && s=sudo
    if type -p systemctl &>/dev/null; then
        $s systemctl $1 $2
    else
        $s service $2 $1
    fi
}

sgo() { # service go
    service=$1
    ser start $service
    if type -p systemctl &>/dev/null; then
        ser enable $service
    fi
}


vm-set-listen(){
    local t=$(mktemp)
    local vm=$1
    local ip=$2
    s virsh dumpxml $vm | sed -r "s/(<listen.*address=')([^']+)/\1$ip/" | \
        sed -r "s/listen='[^']+/listen='$ip/"> $t
    s virsh undefine $vm
    s virsh define $t
}


vmshare() {
    vm-set-listen $1 0.0.0.0
}

vmunshare() {
    vm-set-listen $1 127.0.0.1
}


tu() {
    local s;
    local dir="$(dirname "$1")"
    if [[ -e $1 && ! -w $1 || ! -w $(dirname "$1") ]]; then
        s=s;
    fi
    $s teeu "$@"
}
ts() { # start editing a new file
    [[ $# != 1 ]] && echo "I need a filename." && return 1
    local quiet
    if [[ $- != *i* ]]; then
        quiet=true
    fi
    if [[ $1 == *.c ]]; then
	e '#include <stdio.h>' >"$1"
	e '#include <stdlib.h>' >>"$1"
	e 'int main(int argc, char * argv[]) {' >>"$1"
	e '	printf( "hello world\n");' >>"$1"
	e '	return 0;' >>"$1"
	e '}' >>"$1"
	e "${1%.c}: $1" > Makefile
	e "	g++ -ggdb -std=gnu99 -o ${1%.c} $<" >> Makefile
	e "#!/bin/bash" >run.sh
	e "./${1%.c}" >>run.sh
	chmod +x run.sh
    elif [[ $1 == *.java ]]; then
	e "public class ${1%.*} {" >"$1"
	e '	public static void main(String[] args) {' >>"$1"
	e '		System.out.println("Hello, world!");' >>"$1"
	e '	}' >>"$1"
	e '}' >>"$1"

    else
	echo "#!/bin/bash" > "$1"
	chmod +x "$1"
    fi
    [[ $quiet ]] || g "$1"

}

ediff() {
    [[ ${#@} == 2 ]] || { echo "error: ediff requires 2 arguments"; return 1; }
    emacs --eval "(ediff-files \"$1\" \"$2\")"
}

tx() { # toggle set -x, and the prompt so it doesn't spam
    if [[ $- == *x* ]]; then
	set +x
        PROMPT_COMMAND=prompt_command
    else
        unset PROMPT_COMMAND
        PS1="\w \$ "
	set -x
    fi
}


dat() { # do all tee, for more complex scripts
    tee >(ssh frodo bash -l) >(bash -l) >(ssh x2 bash -l) >(ssh tp bash -l)
}
da() { # do all
    local host
    "$@"
    for host in x2 tp treetowl; do
        ssh $host "$@"
    done

}


istext() {
    grep -Il "" "$@" &>/dev/null
}

if [[ $OS == Windows_NT ]]; then
    # cygstart wrapper
    cs() {
	cygstart "$@" &
    }
    xp() {
	explorer.exe .
    }
    # launch
    o() {
	local x=(*$1*)
	(( ${#x[#]} > 1 )) && { echo "warning ${#x[#]} matches found"; sleep 1; }
	cygstart *$1* &
    }
else
    o() {
	if type gvfs-open &> /dev/null ; then
	    gvfs-open "$@"
	else
	    xdg-open "$@"
	fi
        # another alternative is run-mailcap
    }
fi

khfix() { # known hosts fix
    ssh-keygen -R $1
    local x=$(host $1)
    ssh-keygen -R ${x##* }
    ssh $1 :
}


# todo, update this
complete -F _longopt la lower low rlt rld rl lld ts ll dircp ex fcp fct fpst gr



hl() { # history limit. Write extra history to archive file.
    local max_lines linecount tempfile prune_lines x
    local harchive="${HISTFILE}_archive"
    for x in "$HISTFILE" "$harchive"; do
        [[ -e $x ]] || { touch "$x" && echo "notice from hl(): creating $x"; }
        if  [[ ! $x || ! -e $x || ! -w $x || $(stat -c "%u" "$x") != $EUID  ]];  then
            echo "error in hl: history file \$x:$x no good"
	    return 1
        fi
    done
    history -a # save history
    max_lines=$HISTFILELINES
    [[ $max_lines =~ ^[0-9]+$ ]] || { echo "error in hl: failed to get max line count"; return 1; }
    linecount=$(wc -l < $HISTFILE) # pipe so it doesn't output a filename
    [[ $linecount =~ ^[0-9]+$ ]] || { echo "error in hl: wc failed"; return 1; }
    if (($linecount > $max_lines)); then
	prune_lines=$(($linecount - $max_lines))
	head -n $prune_lines "$HISTFILE" >> "$harchive" \
	    && sed -ie "1,${prune_lines}d"  $HISTFILE
    fi
}

############## work stuff #############

rn() {
    pushd /sdx/test/sandbox/;
    ./setup.sh && ./run.sh
    popd
}

vrm() {
    virsh destroy $1
    virsh undefine $1
}




############# some other section
if [[ $- == *i* ]]; then
    # commands to run when bash exits normally
    trap "hl; _smh" EXIT
fi


# temporary variables to test colorization
# some copied from gentoo /etc/bash/bashrc,
use_color=false
# dircolors --print-database uses its own built-in database
# instead of using /etc/DIR_COLORS.  Try to use the external file
# first to take advantage of user additions.
safe_term=${TERM//[^[:alnum:]]/?}   # sanitize TERM
match_lhs=""
[[ -f ~/.dir_colors   ]] && match_lhs="${match_lhs}$(<~/.dir_colors)"
[[ -f /etc/DIR_COLORS ]] && match_lhs="${match_lhs}$(</etc/DIR_COLORS)"
[[ -z ${match_lhs}    ]] \
    && type -P dircolors >/dev/null \
    && match_lhs=$(dircolors --print-database)
# test if our $TERM is in the TERM values in dircolor
[[ $'\n'${match_lhs} == *$'\n'"TERM "${safe_term}* ]] && use_color=true


if ${use_color} && [[ $- == *i* ]]; then

    if [[ $XTERM_VERSION == Cygwin* ]]; then
        get_term_color() {
            for x in "$@"; do
                case $x in
	            underl) echo -n $'\E[4m' ;;
	            bold) echo -n $'\E[1m' ;;
	            red) echo -n $'\E[31m' ;;
	            green) echo -n $'\E[32m' ;;
	            blue) echo -n $'\E[34m' ;;
	            cyan) echo -n $'\E[36m' ;;
	            yellow) echo -n $'\E[33m' ;;
	            purple) echo -n $'\E[35m' ;;
	            nocolor) echo -n $'\E(B\E[m' ;;
                esac
            done
        }

    else
        get_term_color() {
            for x in "$@"; do
                case $x in
	            underl) echo -n $(tput smul) ;;
	            bold) echo -n $(tput bold) ;;
	            red) echo -n $(tput setaf 1) ;;
	            green) echo -n $(tput setaf 2) ;;
	            blue) echo -n $(tput setaf 4) ;;
	            cyan) echo -n $(tput setaf 6) ;;
	            yellow) echo -n $(tput setaf 3) ;;
	            purple) echo -n $(tput setaf 5) ;;
	            nocolor) echo -n $(tput sgr0) ;; # no font attributes
                esac
            done
        }
    fi
else
    get_term_color() {
        :
    }
fi
# Try to keep environment pollution down, EPA loves us.
unset safe_term match_lhs use_color






###############
# prompt ######
###############


if [[ $- == *i* ]]; then
    # git branch/status prompt function
    if [[ $OS != Windows_NT ]]; then
        GIT_PS1_SHOWDIRTYSTATE=true
    fi
    # arch source lopip show -fcation
    [[ -r /usr/share/git/git-prompt.sh ]] && source /usr/share/git/git-prompt.sh
    # fedora/debian source
    [[ -r /usr/share/git-core/contrib/completion/git-prompt.sh ]] && source /usr/share/git-core/contrib/completion/git-prompt.sh

    # in case we didn't source git-prompt.sh
    if ! declare -f __git_ps1 > /dev/null; then
        __git_ps1() {
            :
        }
    fi

    # this needs to come before next ps1 stuff
    # this stuff needs bash 4, feb 2009,
    # old enough to no longer condition on $BASH_VERSION anymore
    shopt -s autocd
    shopt -s dirspell
    PS1='\w'
    if [[ $- == *i* ]]  && [[ ! $INSIDE_EMACS ]]; then
        PROMPT_DIRTRIM=2
	bind -m vi-command B:shell-backward-word
	bind -m vi-command W:shell-forward-word
    fi

    if [[ $SSH_CLIENT ]]; then
        PS1="\h $PS1"
    fi

    prompt_command() {
        local return=$? # this MUST COME FIRST
        local psc pst
        local ps_char ps_color
        unset IFS
        history -a # save history
        if [[ ! $DESKTOP_SESSION == xmonad && $TERM == *(screen*|xterm*|rxvt*) ]]; then
            # from the screen man page
            if [[ $TERM == screen* ]]; then
                local title_escape="\033]..2;"
            else
                local title_escape="\033]0;"
            fi
	    echo -ne "$title_escape${PWD/#$HOME/~}  $USER@$HOSTNAME\007"
	fi

        case $return in
	    0) ps_color="$(get_term_color blue)"
               ps_char='\$'
               ;;
	    1) ps_color="$(get_term_color green)"
               ps_char="$return \\$"
               ;;
	    *) ps_color="$(get_term_color yellow)"
               ps_char="$return \\$"
               ;;
        esac
        if [[ ! -O . ]]; then # not owner
            if [[ -w . ]]; then # writable
	        ps_color="$(get_term_color bold red)"
            else
	        ps_color="$(get_term_color bold green)"
            fi
        fi
        PS1="${PS1%"${PS1#*[wW]}"} \[$ps_color\]$ps_char\[$(get_term_color nocolor)\] "
        # emacs completion doesn't like the git prompt atm, so disabling it.
        #PS1="${PS1%"${PS1#*[wW]}"}$(__git_ps1 ' (%s)') \[$ps_color\]$ps_char\[$(get_term_color nocolor)\] "
    }
    PROMPT_COMMAND=prompt_command
fi





###########################################
# stuff that makes sense to be at the end #
###########################################
if [[ "$SUDOD" ]]; then
    cd "$SUDOD"
    unset SUDOD
elif [[ -d /a ]] && [[ $PWD == $HOME ]] && [[ $- == *i* ]]; then
    cd /a
fi


# best practice
unset IFS


# if someone exported $SOE, catch errors
if [[ $SOE ]]; then
    errcatch
fi

# I'd prefer to have system-wide, plus user ruby, due to bug in it
# https://github.com/rubygems/rubygems/pull/1002
# further problems: installing multi-user ruby and user ruby,
# you don't get multi-user ruby when you sudo to root, unless its sudo -i.
# There a third hybrid form, which passenger error suggested I use,
# but it didn't actually work.

# in cased I never need this
# rvm for non-interactive shell: modified from https://rvm.io/rvm/basics
#if [[ $(type -t rvm) == file && ! $(type -t ruby) ]]; then
#    source $(rvm 1.9.3 do rvm env --path)
#fi

# based on warning from rvmsudo
export rvmsudo_secure_path=1


if [[ -s "/usr/local/rvm/scripts/rvm" ]]; then
    source "/usr/local/rvm/scripts/rvm"
elif [[ -s $HOME/.rvm/scripts/rvm ]]; then
    source $HOME/.rvm/scripts/rvm
fi

# https://wiki.archlinux.org/index.php/Xinitrc#Autostart_X_at_login
# i added an extra condition as gentoo xorg guide says depending on
# $DISPLAY is fragile.
if [[ ! $DISPLAY && $XDG_VTNR == 1 ]] && shopt -q login_shell && isarch; then
    exec startx
fi
# ensure no bad programs appending to this file will have an affect
return 0