#!/bin/bash
#  _   _
# | |_| |__   ___  ___ __ _
# | __| '_ \ / _ \/ __/ _` |
# | |_| | | |  __/ (_| (_| |
#  \__|_| |_|\___|\___\__,_|
#
# licensed under the MIT license <http://opensource.org/licenses/MIT>
#
# build.sh - v0.9.0
#   a simple build script because i forget things

INSTALL_DIR="/usr/local/bin"
MAN_DIR="/usr/local/share/man/man1"
BASH_COMPLETE_DIR="/usr/local/etc/bash_completion.d"
ZSH_COMPLETE_DIR="/usr/local/share/zsh/site-functions"

# sh functions
p() {
	echo $"# $1"
}

err() {
	p "ERROR: $1" >&2
	exit 1
}

ok() {
	if [ $? != 0 ]; then
		err "$1"
	fi
}

set_version_var() {
    if [[ $( uname -s ) = "Darwin" ]]; then
        local sed_opts="-En"
    else
        local sed_opts="-rn"
    fi
    local CARGO_VER=$( grep "version" Cargo.toml | sed $sed_opts 's/.*"(.*)"/\1/p' )
    local COMMIT=$( git rev-parse --short HEAD )
    local BUILD_DATE=$( date +%d-%m-%Y )
    export THECA_BUILD_VER="$CARGO_VER ($COMMIT, built $BUILD_DATE)"
}

# check subcommand
case "$1" in
    # building the binary (just pass through to cargo)
    build)
        if command -v cargo >/dev/null 2>&1; then
            BUILD_CMD="cargo build"
            shift
            for arg in "$@"; do
                BUILD_CMD="$BUILD_CMD $arg"
            done
            set_version_var
            $BUILD_CMD
            ok "$BUILD_CMD failed"
            if [[ ! -d "bin" ]]; then
                mkdir bin
            fi
            if [[ "$@" =~ "--release" ]]; then
                bin_loc="target/release/theca"
            else
                bin_loc="target/debug/theca"
            fi
            p "built $bin_loc"
            cp "$bin_loc" bin/theca
            p "copied $bin_loc -> bin/theca"
        else
            err "cargo could not be found"
        fi
        ;;

    # build the man page, i can never remember the name of this thing
    build-man)
        if command -v md2man-roff >/dev/null 2>&1; then
            md2man-roff docs/THECA.1.md > docs/THECA.1
            p "built THECA.1 man page"
        else
            err "md2man-roff could not be found"
        fi
        ;;

    # install the binary from bin/theca so we don't have to bother about --dev/--release
    # binaries
    install)
        if [ -e "bin/theca" ]; then
            cp bin/theca $INSTALL_DIR/
            if [ $? -ne 0 ]; then
                p "theca could not be copied to $INSTALL_DIR"
                err "please rerun the script with sudo if you do not have write permissions in $INSTALL_DIR"
            fi
            p "copied bin/theca -> $INSTALL_DIR/theca"
        else
            err "bin/theca doesn't exist (did you run ./build.sh build [--release]?)"
        fi
        if [[ "$@" =~ "--bash-complete" ]]; then
            cp completion/bash_complete.sh $BASH_COMPLETE_DIR/theca
            p "copied completion/bash_complete.sh -> $BASH_COMPLETE_DIR/theca"
        fi
        if [[ "$@" =~ "--zsh-complete" ]]; then
            cp completion/_theca $ZSH_COMPLETE_DIR/_theca
            p "copied completion/_theca -> $ZSH_COMPLETE_DIR/theca"
        fi
        if [[ "$@" =~  "--man" ]]; then
            if [ -e "docs/THECA.1" ]; then
                cp docs/THECA.1 $MAN_DIR/
                p "copied docs/THECA.1 -> $MAN_DIR/THECA.1"
            else
                err "docs/THECA.1 doesn't exist"
            fi
        fi
        p "have fun :>"
        ;;

    # run all the tests in one place
    test)
        if [[ "$@" =~  "--travis" ]]; then
			travis_build=1
			cargo_cmd="travis-cargo -q"
            python="python3"
        else
			cargo_cmd="cargo"
            python="python3"
        fi

        # run the rust tests
        if ! $cargo_cmd test; then
            err "rust tests did't pass!"
        fi


        python_cmd="$python tools/theca_test_harness.py --condensed -tc"
        if [[ "$@" =~ "--release" ]]; then
            build_profile="release"
            python_cmd="$python_cmd target/release/theca"
			if [[ -z "$travis_build" ]]; then
				cargo_cmd="$cargo_cmd build --release"
			else
				cargo_cmd="$cargo_cmd build"
			fi
        else
            build_profile="dev"
            python_cmd="$python_cmd target/debug/theca"
            cargo_cmd="$cargo_cmd build"
        fi

        # build the dev binary
        if ! $cargo_cmd; then
            err "couldn't build the binary!"
        fi

        # run the python tests
        p "running python harness tests"
        if ! eval "$python_cmd -pt"; then
            err "[$build_profile] profile tests did not pass!"
        fi
        if ! eval "$python_cmd -jt"; then
            err "[$build_profile] json output tests did not pass!"
        fi
        if ! eval "$python_cmd -tt"; then
            err "[$build_profile] text output tests did not pass!"
        fi

        p "it seems like everything is ok..."
        ;;

    # delete the target/ dir, the binary in . and the man page (if --man is used)
    clean)
        if [ -d "target" ]; then
            rm -r target
            p "deleted ./target/"
        fi
        if [ d "bin" ]; then
            rm -r bin
            p "deleted ./bin/"
        fi
        if [ -e "docs/THECA.1" ] && [[ "$@" =~  "--man" ]]; then
            rm docs/THECA.1
            p "deleted ./docs/THECA.1"
        fi
        ;;

    # print the help
    *)
        err "Usage: $0 {build|build-man|test|install|clean}"
        ;;
esac