#!/bin/bash
#
# Script to stop the client and uninstall database, but retain the config and log files.
set -e
# set -x

RED='\033[0;31m'
GREEN='\033[1;32m'
NC='\033[0m'
verMode=edge

installDir="/usr/local/taos"
clientName="taos"
uninstallScript="rmtaos"

clientName2="taos"
productName2="TDengine"

benchmarkName2="${clientName2}Benchmark"
demoName2="${clientName2}demo"
dumpName2="${clientName2}dump"
inspect_name="${clientName2}inspect"
taosgen_name="${PREFIX}gen"
uninstallScript2="rm${clientName2}"

installDir="/usr/local/${clientName2}"

#install main path
install_main_dir=${installDir}

log_link_dir=${installDir}/log
cfg_link_dir=${installDir}/cfg
bin_link_dir="/usr/bin"
lib_link_dir="/usr/lib"
lib64_link_dir="/usr/lib64"
inc_link_dir="/usr/include"
log_dir="/var/log/${clientName2}"
cfg_dir="/etc/${clientName2}"

csudo=""
if command -v sudo >/dev/null; then
    csudo="sudo "
fi

function kill_client() {
    pid=$(ps -C ${clientName2} | grep -w ${clientName2} | grep -v $uninstallScript2 | awk '{print $1}')
    if [ -n "$pid" ]; then
        ${csudo}kill -9 $pid || :
    fi
}

function clean_bin() {
    # Remove links
    ${csudo}rm -f ${bin_link_dir}/${clientName2} || :
    ${csudo}rm -f ${bin_link_dir}/${demoName2} || :
    ${csudo}rm -f ${bin_link_dir}/${benchmarkName2} || :
    ${csudo}rm -f ${bin_link_dir}/${dumpName2} || :
    ${csudo}rm -f ${bin_link_dir}/${uninstallScript2} || :
    ${csudo}rm -f ${bin_link_dir}/set_core || :
    [ -L ${bin_link_dir}/${inspect_name} ] && ${csudo}rm -f ${bin_link_dir}/${inspect_name} || :
    [ -L ${bin_link_dir}/${taosgen_name} ] && ${csudo}unlink ${bin_link_dir}/${taosgen_name} || :

    if [ "$verMode" == "cluster" ] && [ "$clientName" != "$clientName2" ]; then
        ${csudo}rm -f ${bin_link_dir}/${clientName2} || :
        ${csudo}rm -f ${bin_link_dir}/${demoName2} || :
        ${csudo}rm -f ${bin_link_dir}/${benchmarkName2} || :
        ${csudo}rm -f ${bin_link_dir}/${dumpName2} || :
        ${csudo}rm -f ${bin_link_dir}/${uninstallScript2} || :
        [ -L ${bin_link_dir}/${inspect_name} ] && ${csudo}rm -f ${bin_link_dir}/${inspect_name} || :
        [ -L ${bin_link_dir}/${taosgen_name} ] && ${csudo}unlink ${bin_link_dir}/${taosgen_name} || :
    fi
}

function clean_lib() {
    # Remove links in lib and lib64 directories
    for dir in "${lib_link_dir}" "${lib64_link_dir}"; do
        if [ -d "$dir" ]; then
            for pattern in "libtaos.*" "libtaosnative.*" "libtaosws.*"; do
                ${csudo}find "$dir" -name "$pattern" -exec ${csudo}rm -f {} \; || :
            done
        fi
    done
}

function clean_header() {
    # Remove link
    ${csudo}rm -f ${inc_link_dir}/taos.h || :
    ${csudo}rm -f ${inc_link_dir}/taosdef.h || :
    ${csudo}rm -f ${inc_link_dir}/taoserror.h || :
    ${csudo}rm -f ${inc_link_dir}/tdef.h || :
    ${csudo}rm -f ${inc_link_dir}/taosudf.h || :
    ${csudo}rm -f ${inc_link_dir}/taosws.h || :
}

function clean_config() {
    # Remove link
    ${csudo}rm -f ${cfg_link_dir}/* || :
}

function clean_log() {
    # Remove link
    ${csudo}rm -rf ${log_link_dir} || :
}

function clean_config_and_log_dir() {
    # Remove link
    echo "Do you want to remove all the log and configuration files? [y/n]"
    read answer
    if [ X$answer == X"y" ] || [ X$answer == X"Y" ]; then
        confirmMsg="I confirm that I would like to delete all log and configuration files"
        echo "Please enter '${confirmMsg}' to continue"
        read answer
        if [ X"$answer" == X"${confirmMsg}" ]; then
            # Remove dir
            rm -rf ${cfg_dir} || :
            rm -rf ${log_dir} || :
        else
            echo "answer doesn't match, skip this step"
        fi
    fi
}

# Stop client.
kill_client
# Remove binary file and links
clean_bin
# Remove header file.
clean_header
# Remove lib file
clean_lib
# Remove link log directory
clean_log
# Remove link configuration file
clean_config
# Remove dir
clean_config_and_log_dir

${csudo}rm -rf ${install_main_dir}

echo -e "${GREEN}${productName2} client is removed successfully!${NC}"
echo
