#!/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: build_tsk.sh [-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 && \
# export CPPFLAGS="-I/usr/local/opt/libpq/include" && \
./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