#!/bin/bash

# If not running interactively, don't do anything
case $- in
    *i*) ;;
      *) return;;
esac

__SCRIPT=$BASH_SOURCE
RBASH_HOME="$HOME/.rbash"
RBASH_SOURCES="$RBASH_HOME/sources"
RBASH_CONFIG="$RBASH_HOME/config"
RBASH_ENV="$RBASH_HOME/env"
RBASH_REMOTE="https://git.k8s.astropenguin.net"
RBASH_REPO="penguin/utils"
RBASH_PATH="bash"
RBASH_MERGED="$RBASH_SOURCES/merged"

RHOSTNAME="<HOSTNAME>"
RDOMAIN="<DOMAIN>"
RCOLOR="<RCOLOR>"

if [ $RHOSTNAME == "<HOSTNAME"">" ]; then
	RHOSTNAME="\u"
	RDOMAIN="@"$( hostname -s )
	RCOLOR=$(( $RANDOM * 6 / 32767 + 1 ))
fi

# Source global definitions
if [ -f /etc/bashrc ]; then
    . /etc/bashrc
fi

# don't put duplicate lines or lines starting with space in the history.
# See bash(1) for more options
HISTCONTROL=ignoreboth

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

# for setting history length see HISTSIZE and HISTFILESIZE in bash(1)
HISTSIZE=1000
HISTFILESIZE=2000

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

# If set, the pattern "**" used in a pathname expansion context will
# match all files and zero or more directories and subdirectories.
#shopt -s globstar

# make less more friendly for non-text input files, see lesspipe(1)
#[ -x /usr/bin/lesspipe ] && eval "$(SHELL=/bin/sh lesspipe)"

# set variable identifying the chroot you work in (used in the prompt below)
if [ -z "${debian_chroot:-}" ] && [ -r /etc/debian_chroot ]; then
    debian_chroot=$(cat /etc/debian_chroot)
fi

# set a fancy prompt (non-color, unless we know we "want" color)
case "$TERM" in
    xterm-color) color_prompt=yes;;
esac

# Alias definitions.
# You may want to put all your additions into a separate file like
# ~/.bash_aliases, instead of adding them here directly.
# See /usr/share/doc/bash-doc/examples in the bash-doc package.

if [ -f ~/.bash_aliases ]; then
    . ~/.bash_aliases
fi

# 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 ! shopt -oq posix; then
  if [ -f /usr/share/bash-completion/bash_completion ]; then
    . /usr/share/bash-completion/bash_completion
  elif [ -f /etc/bash_completion ]; then
    . /etc/bash_completion
  fi
fi

function __func_head() { echo "Usage:" ${FUNCNAME[1]} $1; }
function __func_help() { echo "       ${FUNCNAME[1]} $1"; }

function __download {
    which curl 2>&1 > /dev/null
    if [ $? -eq 0 ]; then
        curl -s "$1"
        return
    fi

    which wget 2>&1 > /dev/null
    if [ $? -eq 0 ]; then
        wget -qO- "$1"
        return
    fi
}

function rbash_cache {
    local CACHE_FILE
    MLINK=$( echo "$1" | md5sum | cut -d' ' -f1 )

    CACHE_FILE="$RBASH_SOURCES/$MLINK"

    if [ ! -f "$CACHE_FILE" ]; then
        __download "$1" > "$CACHE_FILE"
    fi

    chmod 700 "$CACHE_FILE"
    echo -n "$CACHE_FILE"
}

_RBASH_LOADED=
mkdir -p "$RBASH_SOURCES"

if [ ! -f "$RBASH_SOURCES/merged" ]; then
	_RBASH_MERGE_START=1
fi

function rbash_load {
    local f path

	if [ -z "$_RBASH_MERGE_START" ]; then
		if [ -n "$_RBASH_LOADED" ]; then
			return 0
		fi

		if [ -f "$RBASH_MERGED" ]; then
			source $RBASH_MERGED
			if [ $? -eq 0 ]; then
				_RBASH_LOADED=1
				return 0
			fi
			rm "$RBASH_MERGED"
		fi
	fi

    echo "Load: $1"
    f=`rbash_cache "$RBASH_REMOTE/$RBASH_REPO/raw/branch/master/$RBASH_PATH/$1"`
    source "$f"
	if [ $? -eq 0 ]; then
		echo "echo \"Load: $1\"" >> "$RBASH_MERGED"
		cat "$f" >> "$RBASH_SOURCES/merged"
	else
		echo "    ..failed"
		echo "echo \"Skipped: $1\". Use rbash-reload to try load this again." >> "$RBASH_MERGED"
	fi

	chmod 700 "$RBASH_MERGED"
}

function rbash-reload {
	_RBASH_LOADED=
	_RBASH_MERGE_START=1
	rm -f "$RBASH_SOURCES/merged"
	source "$RBASH_CONFIG"
	_RBASH_MERGE_START=
}

function rbash-upgrade {
    echo "Updating $__SCRIPT"

    local TMPFILE _P_HOSTNAME _P_DOMAIN _P_COLOR

    TMPFILE=$( mktemp )
    __download "$RBASH_REMOTE/$RBASH_REPO/raw/branch/master/$RBASH_PATH/rbashrc" > $TMPFILE

    if [ -n "$1" ]; then
        RHOSTNAME=$1
        RDOMAIN=$2
    fi

    _P_HOSTNAME="<HOSTNAME"">"
    _P_DOMAIN="<DOMAIN"">"
    _P_COLOR_CODE="<RCOLOR"">"

    sed -i \
        -e "s/$_P_HOSTNAME/$RHOSTNAME/g" \
        -e "s/$_P_DOMAIN/$RDOMAIN/g" \
        -e "s/$_P_COLOR_CODE/$RCOLOR/g" \
         $TMPFILE

    mv $TMPFILE $__SCRIPT

    # Clean up the old dir
    if [ $? -eq 0 ] && [ -d "$RBASH_SOURCES" ]; then
        rm -r "$RBASH_SOURCES"
    fi

    source $__SCRIPT
}

function rbash-run {
    local f
    echo "Getting: $1"

    f=`rbash_cache "$RBASH_REMOTE/$RBASH_REPO/raw/branch/master/$1"`
    shift

    "$f" "$@"
}

function _require_jq {
    which jq 2>&1 > /dev/null
    if [ $? -ne 0 ]; then
        echo This command requires jq.
        return 1
    fi
}

declare -A _RBASH_DICT
function rbash-list {
    _require_jq || return 1

    echo "Available sources:"
	local _k
    for i in `__download "$RBASH_REMOTE/api/v1/repos/$RBASH_REPO/contents/$RBASH_PATH/sources/" | jq -r '.[].name'`; do
		_k=`echo -n $i | grep -o "^[0-9]\+"`
		_RBASH_DICT["s$_k"]=$i
        grep -q -e "^rbash_load \"sources/$i\"\$" "$RBASH_CONFIG"
        if [ $? -eq 0 ]; then
            echo "  * $i"
        else
            echo "    $i"
        fi
    done
}

function r2ensource {
	rbash-list || return 1

	local _id _sel _tmp _en
	_en=()
	read -p "Enter the script id XX, delimited by space: " _ids
	for _id in $_ids; do
		_sel=${_RBASH_DICT["s$_id"]}
		if [ -z "$_sel" ]; then
			echo "No such id $_id"
			return 1
		fi

		grep -q -e "^rbash_load \"sources/$_sel\"\$" "$RBASH_CONFIG"
		if [ $? -eq 0 ]; then
			echo "\"$_sel\" is already enabled."
			continue
		fi

		_en+=("$_sel")
	done

	_tmp=$( mktemp )
	cp "$RBASH_CONFIG" "$_tmp"
	for _id in "${_en[@]}"; do
		echo "rbash_load \"sources/$_id\"" >> $_tmp
		echo "Enabling $_id"
	done
	sort $_tmp > "$RBASH_CONFIG"
	rm -r "$_tmp" "$RBASH_SOURCES"
}

function r2dissource {
	echo "Currently enabled sources:"
	grep -e "^rbash_load \"sources/" "$RBASH_CONFIG" | cut -c 21- | tr -d '"' | sed "s/^/    $1/g"

	local _id _dis _opts
	_dis=()
	read -p "Enter the script id XX, delimited by space: " _ids
	for _id in $_ids; do
		_dis+=("$( grep -e "^rbash_load \"sources/${_id}_" "$RBASH_CONFIG" )")
		if [ $? -ne 0 ]; then
			echo "No such id: $_id"
			return 1
		fi
	done

	_opts=""
	for _id in "${_dis[@]}"; do
		_opts="-e \"^$( echo -n $_id | sed 's/"/\\"/g' )$\" $_opts"
		echo "Disabling "$( echo -n $_id | sed 's/rbash_load \"sources\/\([^"]\+\)"/\1/g' )
	done

	_dis=$( mktemp )
	sh -c "grep -v $_opts \"$RBASH_CONFIG\"" > $_dis
	sort $_dis > $RBASH_CONFIG
	rm -r "$_dis" "$RBASH_SOURCES"
}

export PS1="This is ${RHOSTNAME}\e[1;3${RCOLOR}m${RDOMAIN}\e[0m: \w\n\$ "
export EDITOR=vim

# Source for custom bash env, allows overriding
if [ -f "$RBASH_ENV" ]; then
    source "$RBASH_ENV"
fi

# Create default source config
if [ ! -f "$RBASH_CONFIG" ]; then
    cat <<___DEFAULT___ > "$RBASH_CONFIG"
rbash_load "sources/10_aliases"
rbash_load "sources/12_shortcuts"
rbash_load "sources/20_fast-greps"
rbash_load "sources/40_go-command"
rbash_load "sources/41_pivot-command"
___DEFAULT___
    chmod 600 "$RBASH_CONFIG"
fi

source "$RBASH_CONFIG"
unset _RBASH_MERGE_START