#!/bin/bash

#################################################################
# Property of Caton Inc.                                        #
# Copyright 2022 - All rights reserved.                         #
#                                                               #
# For details, contact www.catontechnology.com/en/contact       #
#################################################################
set +e
#AGENT_ID
#AGENT_ID=${AGENT_ID}
#PASSWORD
#PASSWORD=${PASSWORD}

updateonly=""
minlogopen=""
configpath=/etc
configname=netScopeCfg
configfullpath=${configpath}/${configname}
echo -e "\033[32mStart installing NetScope ..........."
if [ -f "${configfullpath}" ]; then
    echo -e "\033[31mNetScope config file existed :\033[33m ${configfullpath}"
    while true; do
        echo -e "\033[33mDo you want to overwrite the existing file?"
        read -p "Please answer yes or no : " yn </dev/tty
        case $yn in
        [Yy]*)
            echo -e "\033[32mContinue installing..........."
            echo -e "\033[0m"
            break
            ;;
        [Nn]*)
            echo -e "\033[32mContinue updating..........."
            echo -e "\033[0m"
            updateonly="True"
            break
            ;;
        *) echo -e "\033[31mPlease answer yes or no." ;;
        esac
    done
fi
echo -e "\033[0m"

if [[ "${updateonly}" == "" ]]; then
    if [[ "${AGENT_ID}" == "" ]]; then
        echo -e "\033[31mRrror AGENT_ID : null"
        echo -e "\033[0m"
        exit 1
    fi

    if [[ "${PASSWORD}" == "" ]]; then
        echo -e "\033[31mError PASSWORD : null"
        echo -e "\033[0m"
        exit 1
    fi
fi

if [[ "${MINLOG}" != "" ]]; then
    minlogopen="${MINLOG}"
fi

if [ "$(whoami)" != "root" ]; then
    echo -e "\033[31mMust run as root. If you are using sudo, you should"
    echo -e "\033[31mrun 'sudo su -' before to open a root shell."
    echo -e "\033[0m"
    exit 1
fi

SYSTEMINFOTYPE=""
function getsysteminfo() {
    if [[ -e /etc/os-release ]]; then
        source /etc/os-release
        SYSTEMINFOTYPE=${ID}" "${VERSION_ID}
    elif [[ -e /etc/issue ]]; then
        SYSTEMINFOTYPE=$(cat /etc/issue)
    fi
    echo ${SYSTEMINFOTYPE}
}
getsysteminfo
gzdownloadurlfix="https://install.netscope.caton.cloud"
gzdownloadurl=""

case "${SYSTEMINFOTYPE}" in
# Ubuntu 18.*
[Uu]buntu\ 18.*)
    gzdownloadurl="" #todo
    ;;
# Ubuntu 20.*
[Uu]buntu\ 20.*)
    gzdownloadurl="/Ubuntu20/agent.netscope.tar.gz"
    ;;
# Ubuntu 22.*
[Uu]buntu\ 22.*)
    gzdownloadurl="/Ubuntu22/agent.netscope.tar.gz"
    ;;
# CentOS 7
[Cc]ent[Oo][Ss]\ *7*)
    gzdownloadurl="/Centos7/agent.netscope.tar.gz"
    ;;
# CentOS 8
[Cc]ent[Oo][Ss]\ *8*)
    gzdownloadurl="/Centos7/agent.netscope.tar.gz"
    ;;
# Debian 12*
[Dd]ebian\ *12*)
    gzdownloadurl="/Ubuntu22/agent.netscope.tar.gz"
    ;;
# Debian 11*
[Dd]ebian\ *11*)
    gzdownloadurl="/Ubuntu20/agent.netscope.tar.gz"
    ;;
# Debian 10*
[Dd]ebian\ *10*)
    gzdownloadurl="" #todo
    ;;
# Amazon centos 2*
amzn\ *2*)
    gzdownloadurl="/Centos7/agent.netscope.tar.gz"
    ;;
*)
    echo -e "\033[31mUnsupported system type: ${SYSTEMINFOTYPE}"
    echo -e "\033[0m"
    exit 1
    ;;
esac

if [[ "${gzdownloadurl}" == "" ]]; then
    echo -e "\033[31mUnsupported system type: ${SYSTEMINFOTYPE}, unable to download"
    echo -e "\033[0m"
    exit 1
fi
#==============
pakagename=agent.netscope.tar.gz
pakageuntar=-zxvf
downloadmethod='/usr/bin/curl'

installpath=./netscope

#config
SaaSServerIp=neconn.caton.cloud
SaasServerPort=28081
SaasServerPath=/netscope
AgentID=${AGENT_ID}
AgentPassWord=${PASSWORD}
publicIpApi=https://caton.cloud/netscope/api/v1/ip
#==============
if ! type ${downloadmethod} >/dev/null 2>&1; then
    echo -e "\033[31m${downloadmethod} is not installed. Please install ${downloadmethod} first \033[0m"
    echo -e "\033[0m"
    exit 1
else
    echo -e "\033[32m${downloadmethod} detection passed\033[0m"
    echo -e "\033[0m"
fi

#==============
if [[ "${updateonly}" == "" ]]; then
    touch ${configfullpath}
    chmod 777 ${configfullpath}
    echo "{" >${configfullpath}
    echo "\"SaaSServerIp\":\"${SaaSServerIp}\", " >>${configfullpath}
    echo "\"SaasServerPort\":\"${SaasServerPort}\", " >>${configfullpath}
    echo "\"SaasServerPath\":\"${SaasServerPath}\", " >>${configfullpath}
    echo "\"AgentID\":\"${AgentID}\"," >>${configfullpath}
    echo "\"AgentPassWord\": \"${AgentPassWord}\"," >>${configfullpath}
    echo "\"publicIpApi\": \"${publicIpApi}\"," >>${configfullpath}
    echo "\"MtrLogEnable\": 0," >>${configfullpath}
    echo "\"zlogEnable\": 0," >>${configfullpath}
    echo "\"MtrThreadNumber\": 60," >>${configfullpath}
    echo "\"cacheNumber\": 60000," >>${configfullpath}
    echo "\"writeInterval\": 100000" >>${configfullpath}
    echo "}" >>${configfullpath}
fi


#==============
cd /tmp/
mkdir --m=777 ${installpath} >/dev/null 2>&1
cd ${installpath}
#==============
${downloadmethod} -o ./${pakagename} ${gzdownloadurlfix}${gzdownloadurl}
tar ${pakageuntar} ${pakagename} -C .
rm -f ${pakagename}
chmod 777 ./install.sh

if [[ "${minlogopen}" == "" ]]; then
    echo -e "\033[0mstart install"
    bash ./install.sh
else
    echo -e "\033[0mOPEN minLog"
    bash ./install.sh minLog
fi
#==============
echo -e "\033[32minstallation is complete"
echo -e "\033[0m"