#!/bin/bash

############################################################################
#   You can also pass some arguments to script to set some these options:
#   
#       --config: change the user and password to grafana and specify the mikrotik IP address
#       --stop: stop docker containers
#   
#   For example:
#       bash run.sh --config
#   
############################################################################

set -e

REPO=Grafana-Mikrotik
ENV_FILE=${ENV_FILE:-.env}
ENV_FILE_GF=${ENV_FILE_GF:-.grafana}
ENV_FILE_PROMETHEUS=${ENV_FILE_PROMETHEU:-.prometheus}

#? Colors
RED='\033[31m'
GREEN='\033[32m'
YELLOW='\033[33m'
BLUE='\033[34m'
BOLD='\033[1m'
RESET='\033[m' #? No Color

BOLD='\033[1m'

ask() {
    local prompt default reply

    if [[ ${2} = 'Y' ]]; then
        prompt='Y/n'
        default='Y'
    elif [[ ${2} = 'N' ]]; then
        prompt='y/N'
        default='N'
    else
        prompt='y/n'
        default=''
    fi

    while true; do

        #? Ask the question (not using "read -p" as it uses stderr not stdout)
        echo -n "$1 [${prompt}] "

        #? Read the answer (use /dev/tty in case stdin is redirected from somewhere else)
        read -r reply </dev/tty

        #? Default?
        if [[ -z ${reply} ]]; then
            reply=${default}
        fi

        #? Check if the reply is valid
        case "${reply}" in
        Y* | y*) return 0 ;;
        N* | n*) return 1 ;;
        esac
    done
}

command_exists() {
    command -v "$@" >/dev/null 2>&1 || {
        fmt_error "$* is not installed. Please install $* first."
        exit 1
    }
}

fmt_error() {
    echo -e "\n${BOLD}${RED}Error: $*${RESET}\n" >&2
}

help() {
    if [ "${HELP}" = yes ]; then
        sed -n -e 's/^#   //' -e '3,12p;' "$0"
        exit 1
        return
    fi
}

clone_git() {

    echo -e "${BLUE}Git cloning ${REPO}...${RESET}"
    git clone --depth=1 https://github.com/IgorKha/${REPO}.git ||
        {
            fmt_error "git clone of ${REPO} repo failed"
            exit 1
        }

    echo
}

router_ip() {
    if [[ "${CONFIG}" = yes ]]; then
        IP=$(grep -R 'MIKROTIK_IP' "${ENV_FILE_PROMETHEUS}" 2>&1 | cut -d= -f2)
        echo -e "\n${BLUE}===================================="
        echo -e "\n${BOLD}Prometheus${RESET}\n"
        if ask "Change target mikrotik IP address ? (current ${IP})" Y; then
            read -rp 'Enter target mikrotik IP address: ' IP
            if [ -d "./${REPO}" ]; then
                sed -ri -e '/mikrotik_ip/s/(- ).*( #.*)/\1'"${IP}"'\2/g' \
                    ${REPO}/prometheus/prometheus.yml
                sed -ri -e 's/^(MIKROTIK_IP=)(.*)$/\1'"$IP"'/g' "${ENV_FILE_PROMETHEUS}"
                echo -e "\n${GREEN}... Prometheus target IP changed to ${IP}"
            else
                sed -ri -e '/mikrotik_ip/s/(- ).*( #.*)/\1'"${IP}"'\2/g' \
                    ./prometheus/prometheus.yml
                sed -ri -e 's/^(MIKROTIK_IP=)(.*)$/\1'"${IP}"'/g' "${ENV_FILE_PROMETHEUS}"
                echo -e "\n${GREEN}... Prometheus target IP changed to ${IP}"
            fi
            return
        fi
        echo -e "\n${BLUE}...Skipped step"
        return
    fi
}

# snmp_on() {
#     if [ "$CONFIG" = yes ]; then
#         echo -e "\n${BLUE}===================================="
#         echo -e "${BOLD}Mikrotik SNMP ACTIVATION${RESET}"
#         if ask "Activate snmp mikrotik using ssh?" N; then
#             read -rp 'Enter login: ' MK_LOGIN
#             # read -rsp 'Enter password: ' MK_PASSWD

#             COMM="
#             snmp
#             "
#             ssh ${MK_LOGIN}@$IP "${COMM}"

#         else
#             echo "skipped"
#         fi
#         return
#     fi
# }

grafana_credentials() {
    if [[ "${CONFIG}" = yes ]]; then
        echo -e "\n${YELLOW}===================================="
        echo -e "\n${BOLD}Grafana${RESET}\n"
        if ask "Change default credentials Grafana ?" N; then
            read -rp 'Enter grafana Username: ' GF_USER
            read -rsp 'Enter grafana Password: ' GF_PASSWD

            sed -ri -e 's/^(GF_SECURITY_ADMIN_USER=)(.*)$/\1'"${GF_USER}"'/g' "${ENV_FILE_GF}"
            sed -ri -e 's/^(GF_SECURITY_ADMIN_PASSWORD=)(.*)$/\1'"${GF_PASSWD}"'/g' "${ENV_FILE_GF}"
        else
            echo -e "Default Grafana:
            User: ${YELLOW}admin${RESET}
            Password: ${YELLOW}mikrotik${RESET}"
        fi
        return
    fi
}

docker-cmd() {
    if [[ "${STOP}" = yes ]]; then
        if [ -d "./${REPO}" ]; then
            cd ${REPO} && docker compose down
        else
            docker compose down
        fi
    else
        if [[ -d "./${REPO}" ]]; then
            cd ${REPO} && docker compose up -d
            print_success
        else
            docker compose up -d
            print_success
        fi
    fi
}

print_success() {
    echo "============================================="
    echo -e "${GREEN}Grafana http://localhost:3000"
    echo -e "Prometheus http://localhost:9090/targets${RESET}"
}

main() {

    #? Parse arguments
    while [[ $# -gt 0 ]]; do
        case $1 in
        --help) HELP=yes ;;
        --config) CONFIG=yes ;;
        --stop) STOP=yes ;;
        esac
        shift
    done

    help

    command_exists git
    command_exists docker

    #? init
    if [[ -d "./${REPO}" ]]; then
        ENV_FILE=${REPO}/.env
        ENV_FILE_GF=${REPO}/.grafana
        ENV_FILE_PROMETHEUS=${REPO}/.prometheus
    elif [[ ! -e ${ENV_FILE} ]]; then
        clone_git
        ENV_FILE=${REPO}/.env
        ENV_FILE_GF=${REPO}/.grafana
        ENV_FILE_PROMETHEUS=${REPO}/.prometheus
    fi

    router_ip
    # snmp_on
    grafana_credentials

    # Change UID:GID prometheus container to current user
    sed -ri -e 's/^(CURRENT_USER=)(.*)$/\1'"$(id -u)\:$(id -g)"'/g' "${ENV_FILE}"

    docker-cmd
}

main "$@"