#!/bin/bash
# Clones sleuthkit repo from github (if necessary) and installs
# this script does require sudo privileges
# called like: build_tsk.sh -r <repo path to be created or existing> -b <tsk branch to checkout>

usage() {
    echo "Usage: install_tsk_from_src [-r repo_path] [-b tsk_branch]" 1>&2
}

while getopts "r:b:" o; do
    case "${o}" in
    r)
        SLEUTHKIT_SRC_DIR=${OPTARG}
        ;;
    b)
        TSK_BRANCH=${OPTARG}
        ;;
    *)
        usage
        exit 1
        ;;
    esac
done

if [[ -z "${SLEUTHKIT_SRC_DIR}" ]] || [[ -z "${TSK_BRANCH}" ]]; then
    usage
    exit 1
fi

if [[ ! -d $SLEUTHKIT_SRC_DIR ]]; then
    TSK_REPO_PATH=$(dirname "$SLEUTHKIT_SRC_DIR")
    echo "Cloning Sleuthkit to $TSK_REPO_PATH..."
    mkdir -p $TSK_REPO_PATH &&
        pushd $TSK_REPO_PATH &&
        git clone https://github.com/sleuthkit/sleuthkit.git &&
        popd
    if [[ ! -d $SLEUTHKIT_SRC_DIR ]]; then
        echo "Unable to successfully clone Sleuthkit" >>/dev/stderr
        exit 1
    fi
fi

echo "Getting latest of Sleuthkit branch: $TSK_BRANCH..."
pushd $SLEUTHKIT_SRC_DIR &&
    git reset --hard &&
    git checkout $TSK_BRANCH &&
    git pull &&
    popd
if [[ $? -ne 0 ]]; then
    echo "Unable to reset Sleuthkit repo and pull latest on $TSK_BRANCH" >>/dev/stderr
    exit 1
fi

echo "Installing Sleuthkit..."
pushd $SLEUTHKIT_SRC_DIR &&
    ./bootstrap &&
    ./configure &&
    make &&
    sudo make install &&
    popd
if [[ $? -ne 0 ]]; then
    echo "Unable to build Sleuthkit." >>/dev/stderr
    exit 1
fi

JAVA_INSTALLS=/usr/local/share/java
echo "Sleuthkit in $JAVA_INSTALLS:"
ls $JAVA_INSTALLS | grep sleuthkit