#!/bin/bash
#
# Script to stop the service and uninstall TSDB

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

data_dir="/var/lib/taos"
log_dir="/var/log/taos"

#install main path
install_main_dir="/usr/local/taos"

data_link_dir="/usr/local/taos/data"
log_link_dir="/usr/local/taos/log"

cfg_link_dir="/usr/local/taos/cfg"
bin_link_dir="/usr/bin"
lib_link_dir="/usr/lib"
inc_link_dir="/usr/include"

header_dir="/usr/local/include/taos"
cfg_dir="/etc/taos"
bin_dir="/usr/local/bin/taos"
lib_dir="/usr/local/lib/taos"
link_dir="/usr/bin"
service_config_dir="/etc/systemd/system"
taos_service_name="taosd"
nginx_service_name="tdnginx"

function is_using_systemd() {
    if pidof systemd &> /dev/null; then
        return 0
    else
        return 1
    fi
}

if ! is_using_systemd; then
    service_config_dir="/etc/init.d"
fi

function clean_bin() {
    # Remove link
    sudo rm -f ${bin_link_dir}/taos      || :
    sudo rm -f ${bin_link_dir}/taosd     || :
    sudo rm -f ${bin_link_dir}/taosdump  || :
    sudo rm -f ${bin_link_dir}/rmtaos    || :

    # Remove binary files
    #sudo rm -rf ${bin_dir}        || :
}
function clean_lib() {
    # Remove link
    sudo rm -f ${lib_link_dir}/libtaos.*      || :
    
    #sudo rm -f /usr/lib/libtaos.so || :
    #sudo rm -rf ${lib_dir}         || :
}

function clean_header() {
    # Remove link
    sudo rm -f ${inc_link_dir}/taos.h       || :
    
    #sudo rm -rf ${header_dir}
}

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

function clean_log() {
    if grep -e '^\s*logDir.*$' ${cfg_dir}/taos.cfg &> /dev/null; then
        config_log_dir=$(cut -d ' ' -f2 <<< $(grep -e '^\s*logDir.*$' ${cfg_dir}/taos.cfg))
        # echo "Removing log dir ${config_log_dir}......"
        sudo rm -rf ${config_log_dir}    || : 
    fi
    
    # Remove link
    sudo rm -rf ${log_link_dir}    || :     
    sudo rm -rf ${log_dir}         || :
}

function clean_service_on_systemd() {
    taosd_service_config="${service_config_dir}/${taos_service_name}.service"

    if systemctl is-active --quiet ${taos_service_name}; then
        echo "TDengine taosd is running, stopping it..."
        sudo systemctl stop ${taos_service_name} &> /dev/null || echo &> /dev/null
    fi
    sudo systemctl disable ${taos_service_name} &> /dev/null || echo &> /dev/null

    sudo rm -f ${taosd_service_config}
}

function clean_service_on_sysvinit() {
    restart_config_str="taos:2345:respawn:${service_config_dir}/taosd start"

    if pidof taosd &> /dev/null; then
        echo "TDengine taosd is running, stopping it..."
        sudo service taosd stop || :
    fi

    sudo sed -i "\|${restart_config_str}|d" /etc/inittab || :
    sudo rm -f ${service_config_dir}/taosd || :
    sudo update-rc.d -f taosd remove || :
    sudo init q || :
}

function clean_service() {
    if is_using_systemd; then
        clean_service_on_systemd
    else
        clean_service_on_sysvinit
    fi
}

isAll="true"
if ! type taosd &> /dev/null; then
    isAll="false"
fi

config_data_dir=''
if grep -e '^\s*dataDir.*$' ${cfg_dir}/taos.cfg &> /dev/null; then
    config_data_dir=$(cut -d ' ' -f2 <<< $(grep -e '^\s*dataDir.*$' ${cfg_dir}/taos.cfg))
fi

# Stop service and disable booting start.
clean_service
# Remove binary file and links
clean_bin
# Remove header file.
clean_header
# Remove lib file
clean_lib
# Remove log directory
clean_log
# Remove configuration file
clean_config
# Remove data directory
sudo rm -rf ${data_link_dir}    || : 

[ "$isAll" = "false" ] && exit 0 || :
echo -e -n "${RED}Do you want to delete data stored in TDengine? [y/N]: ${NC}" 
read is_delete
while true; do
    if [ "${is_delete}" = "y" ] || [ "${is_delete}" = "Y" ]; then
        sudo rm -rf ${data_dir}
        # echo "Removing data file ${config_data_dir}..."
        [ -n ${config_data_dir} ] && sudo rm -rf ${config_data_dir}
        break
    elif [ "${is_delete}" = "n" ] || [ "${is_delete}" = "N" ]; then
        break
    else
        read -p "Please enter 'y' or 'n': " is_delete
    fi
done

sudo rm -rf ${install_main_dir}

osinfo=$(awk -F= '/^NAME/{print $2}' /etc/os-release)
if echo $osinfo | grep -qwi "ubuntu" ; then
#  echo "this is ubuntu system"
   sudo rm -f /var/lib/dpkg/info/tdengine* || :
elif  echo $osinfo | grep -qwi "centos" ; then
  echo "this is centos system"
  sudo rpm -e --noscripts tdengine || :
fi

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