install.sh

name=$(hostname)
branch="main"
verbosity="-q"

install() {
    echo "[*] Installing monitor..."
    install_monitor
    echo "[+] Installed monitor!"
    echo "[*] Installing monitor service..."
    install_service "/usr/local/bin/srvcheck"
    echo "[+] Installed monitor service!"
}

print_help () {
    echo "Usage: install [options...]
     --active-set <active_set_number> number of the validators in the active set [number of active validators provided by default]
     --admin <@username> the Telegram username that is tagged once new governance proposals are live
 -b  --block-time <time> expected block time [default is 60 seconds]
     --branch <name> name of the branch to use for the installation [default is main]
     --endpoint <url:port> node local rpc address
     --git <git_repo> git repo to query the latest realease version that is installed
     --gov enable checks on new governance proposals
     --mount <mount_point> mount point where the node saves data
 -n  --name <name> monitor name [default is the server hostname]
     --rel <version> release version installed (required if git_repo is specified)
     --signed-blocks <max_misses> <blocks_window> max number of blocks not signed in a specified blocks window [default is 5 blocks missed out of the latest 100 blocks]
 -s  --service <name> service name of the node to monitor [required]
 -t  --telegram <chat_id> <token> telegram chat options (id and token) where the alerts will be sent [required]
 -v  --verbose enable verbose installation
 -e  --exporter <port> enable prometheus exporter on <port> (port optional, default 9001)"
}

install_monitor () {
    config_file="/etc/srvcheck.conf"
    apt -qq update
    apt -qq install git python3-pip -y
    systemctl stop node-monitor.service
    rm -rf /etc/srvcheck.conf
    rm -rf /etc/systemd/system/node-monitor.service
    pip3 $verbosity install git+https://github.com/openbitlab/celestia-srvcheck.git@$branch#egg=celestia-srvcheck --exists-action w --ignore-installed 
    wget $verbosity https://raw.githubusercontent.com/openbitlab/celestia-srvcheck/$branch/conf/srvcheck.conf -O $config_file ## TODO add args to change service name
    sed -i -e "s/^apiToken =.*/apiToken = \"$api_token\"/" $config_file
    sed -i -e "s/^chatIds =.*/chatIds = [\"$chat_id\"]/" $config_file
    sed -i -e "s/^tagVersion =.*/tagVersion = $(curl https://api.github.com/repos/openbitlab/celestia-srvcheck/git/refs/tags -s | jq .[-1] | jq .ref | grep -oP '(?<=v).*?(?=\")')/" $config_file
    sed -i -e "s/^commit =.*/commit = $(curl https://api.github.com/repos/openbitlab/celestia-srvcheck/git/refs/heads/main -s | jq .object.sha -r | head -c 7)/" $config_file
    sed -i -e "s/^name =.*/name = $name/" $config_file
    sed -i -e "s/^service =.*/service = $service/" $config_file
    if [ ! -z "$block_time" ]
    then
        sed -i -e "s/^blockTime =.*/blockTime = $block_time/" $config_file
    fi
    if [ ! -z "$active_set" ]
    then
        sed -i -e "s/^activeSet =.*/activeSet = $active_set/" $config_file
    fi
    if [ ! -z "$admin" ]
    then
        sed -i -e "s/^govAdmin =.*/govAdmin = $admin/" $config_file
    fi
    if [ ! -z "$git_api" ]
    then
        sed -i -e "s,^ghRepository =.*,ghRepository = $git_api,g" $config_file
    fi
    if [ ! -z "$local_version" ]
    then
        sed -i -e "s/^localVersion =.*/localVersion = $local_version/" $config_file
    fi
    if [ ! -z "$mountPoint" ]
    then
        sed -i -e "s,^mountPoint =.*,mountPoint = $mountPoint,g" $config_file
    fi
    if [ ! -z "$endpoint" ]
    then
        sed -i -e "s,^endpoint =.*,endpoint = $endpoint,g" $config_file
    fi
    if [[ ! -z "$threshold_notsigned" && ! -z "$block_window" ]]
    then
        sed -i -e "s/^thresholdNotsigned =.*/thresholdNotsigned = $threshold_notsigned/" $config_file
        sed -i -e "s/^blockWindow =.*/blockWindow = $block_window/" $config_file
    fi
    if [ "$enable_gov" = true ]
    then
        sed -i -r 's/(.TaskTendermintNewProposal?;|.TaskTendermintNewProposal?;?$)//' $config_file #enable checks on tendermint governance module
    fi
    if [ "$enable_exporter" = true ]
    then
        sed -i -r 's/(.TaskExporter?;|.TaskExporter?;?$)//' $config_file #enable exporter
        sed -i -e "s/^exporterPort =.*/exporterPort = $exporter_port/" $config_file
    fi
}

install_service () {
    wget -q https://raw.githubusercontent.com/openbitlab/celestia-srvcheck/$branch/conf/node-monitor.service -O /etc/systemd/system/node-monitor.service ## TODO add args to change service name
    sed -i -e "s,^ExecStart=.*,ExecStart=$1,g" /etc/systemd/system/node-monitor.service
    systemctl daemon-reload 
    systemctl start node-monitor
    systemctl enable node-monitor
}

POSITIONAL_ARGS=()

enable_gov=false
enable_exporter=false
while [[ $# -gt 0 ]]; do
case $1 in
    -n|--name)
        if [[ -z $2 ]]
        then
            print_help
            exit 1
        else
            name="$2"
        fi
        shift # past argument
        shift # past value
    ;;
    --admin)
        if [[ -z $2 ]]
        then
            print_help
            exit 1
        else
            admin="$2"
        fi
        shift # past argument
        shift # past value
    ;;
    -b|--block-time)
        if [[ -z $2 ]]
        then
            print_help
            exit 1
        else
            block_time="$2"
        fi
        shift # past argument
        shift # past value
    ;;
    --mount)
        if [[ -z $2 ]]
        then
            print_help
            exit 1
        else
            mountPoint="$2"
        fi
        shift # past argument
        shift # past value
    ;;
    -v|--verbose)
        verbosity=""
        shift # past argument
    ;;
    --branch)
        if [[ -z $2 ]]
        then
            print_help
            exit 1
        else
            branch="$2"
        fi
        shift # past argument
        shift # past value
    ;;
    -t|--telegram)
        if [[ -z $2 || -z $3 ]]
        then
            print_help
            exit 1
        else
	        chat_id="$2"
            api_token="$3"
        fi
        shift # past argument
        shift # past value
        shift # past value
    ;;
    --git)
        if [[ -z $2 ]]
        then
            print_help
            exit 1
        else
	        git_api="$2"
        fi
        shift # past argument
        shift # past value
    ;;
    --rel)
        if [[ -z $2 ]]
        then
            print_help
            exit 1
        else
	        local_version="$2"
        fi
        shift # past argument
        shift # past value
    ;;
    --signed-blocks)
        threshold_notsigned="$2"
        block_window="$3"
        shift # past argument
        shift # past value
        shift # past value
    ;;
    --active-set)
        if [[ -z $2 ]]
        then
            print_help
            exit 1
        else
            active_set="$2"
        fi
        shift # past argument
        shift # past value
    ;;
    -s|--service)
        if [[ -z $2 ]]
        then
            print_help
            exit 1
        else
            service="$2"
        fi
        shift # past argument
        shift # past value
    ;;
    --endpoint)
        if [[ -z $2 ]]
        then
            print_help
            exit 1
        else
	        endpoint="$2"
        fi
        shift # past argument
        shift # past value
    ;;
    --exporter)
        enable_exporter=true
	      exporter_port="$2"
        shift # past argument
        shift # past value
    ;;
    --gov)
        enable_gov=true
        shift # past argument
    ;;
    -*|--*)
        echo "Unknown option $1"
        print_help
        exit 1
    ;;
    *)
        POSITIONAL_ARGS+=("$1") # save positional arg
        shift # past argument
    ;;
esac
done

set -- "${POSITIONAL_ARGS[@]}" # restore positional parameters

if [[ -z $chat_id || -z $api_token || -z $service ]]
then
    print_help
    exit 1
fi

install