#!/bin/bash

function is_tikiller() {
    if [[ $1 == "marketexplorer" ]]
    then
        echo 1
    elif [[ $1 == "aifeed" ]]
    then
        echo 1
    elif [[ $1 == "pricealerts" ]]
    then
        echo 1
    elif [[ $1 == "charts" ]]
    then
        echo 1
    elif [[ $1 == "delayedcharts" ]]
    then
        echo 1
    elif [[ $1 == "tiqmisc" ]]
    then
        echo 1
    elif [[ $1 == "paper" ]]
    then
        echo 1
    else
        echo 0
    fi 
}

# until these are in the base build, run this:
echo "Installing yum packages from set_up_dir script"
yum -y install libuuid-devel pcre pcre-devel pcre-cpp libcurl libcurl-devel policycoreutils-python-utils jq nfs-utils composer gh emacs-nox
echo "Done installing yum packages from set_up_dir script"

TARGET_DIRECTORY="/ti"
LOCAL_DIRECTORY="."
ROLE_SCRIPTS_DIRECTORY="$LOCAL_DIRECTORY/role_setup_scripts"
ROLE=$(echo $HOSTNAME | awk -F "-" '{print $4}' | sed -e 's/-\?[0-9]\+//')
if [[ $1 ]]
then
    ROLE=$1
    echo "Overriding role with command line option $ROLE"
fi

ROLE_SETUP_SCRIPT="$ROLE_SCRIPTS_DIRECTORY/setup_$ROLE.sh"
CONFIGS_DIRECTORY="$LOCAL_DIRECTORY/configs"
ENVIRONMENT=$(echo $HOSTNAME | awk -F "-" '{print $2}')
CONFIG_FILE="$CONFIGS_DIRECTORY/config_common.$ENVIRONMENT.txt"

echo "Using target directory $TARGET_DIRECTORY and hostname $HOSTNAME and environment $ENVIRONMENT and $CONFIG_FILE"

if [[ -e $CONFIG_FILE ]]
then
    echo "Copying environment config file for $ENVIRONMENT to $TARGET_DIRECTORY"
    cp $CONFIG_FILE $TARGET_DIRECTORY/config_common.txt
else
    echo "Looked for $CONFIG_FILE but it doesn't exist in $(pwd)"
fi

IS_TIKILLER=$(is_tikiller $ROLE)
if [[ -e $ROLE_SETUP_SCRIPT ]]
then
    echo "Running role setup script $ROLE_SETUP_SCRIPT for $ROLE"
    bash $ROLE_SETUP_SCRIPT
    echo "Done running role setup script for $ROLE"
elif [[ $IS_TIKILLER == 1 ]]
then
    echo "Running role setup script setup_tikiller.sh for $ROLE"
    bash $ROLE_SCRIPTS_DIRECTORY/setup_tikiller.sh
    echo "Done running role setup script setup_tikiller.sh for $ROLE"
else
    echo "Looked for role setup script $ROLE_SETUP_SCRIPT but it doesn't exist"
fi

chown -R tradeideas $TARGET_DIRECTORY
chgrp -R tig_admin $TARGET_DIRECTORY
chmod -R g+w $TARGET_DIRECTORY