#!/usr/bin/env bash
# Name          :survive.sh
# Description   :Survive in Linux/Docker environments by installing essential tools.
# Author        :Martin Tovmassian
# Email         :martin.tovmassian@protonmail.com
# Usage         :./survive.sh --help

set -euo pipefail

BOOTSTRAPPED=false
STDOUT_REDIRECT="/dev/stdout"

declare -A SURVIVAL_OPTIONS
SURVIVAL_OPTIONS[disk]=false
SURVIVAL_OPTIONS[network]=false
SURVIVAL_OPTIONS[process]=false
SURVIVAL_OPTIONS[text]=false
SURVIVAL_OPTIONS[user]=false

apk_cli() { command -v apk; }
apt_cli() { command -v apt; }
dnf_cli() { command -v dnf || command -v microdnf; }

bootstrap_apk() {
    apk add coreutils
    BOOTSTRAPPED=true
}

bootstrap_apt() {
    apt update
    apt install coreutils -y
    BOOTSTRAPPED=true
}

bootstrap_dnf() {
    "$(dnf_cli)" install coreutils-single epel-release -y
    BOOTSTRAPPED=true
}

install_with_apk() {
    [[ -z "$(apk_cli)" ]] && return 1
    [[ "$BOOTSTRAPPED" == false ]] && bootstrap_apk
    local packages="$*"
    "$(apk_cli)" add $packages
} > "$STDOUT_REDIRECT" 2>&1

install_with_apt() {
    [[ -z "$(apt_cli)" ]] && return 1
    [[ "$BOOTSTRAPPED" == false ]] && bootstrap_apt
    local packages="$*"
    "$(apt_cli)" install $packages -y
} > "$STDOUT_REDIRECT" 2>&1

install_with_dnf() {
    [[ -z "$(dnf_cli)" ]] && return 1
    [[ "$BOOTSTRAPPED" == false ]] && bootstrap_dnf
    local packages="$*"
    "$(dnf_cli)" install $packages -y
} > "$STDOUT_REDIRECT" 2>&1

install_survival_tools_process() {
    install_with_apk procps lsof htop strace \
        || install_with_apt procps lsof htop strace \
        || install_with_dnf procps lsof htop strace
}

install_survival_tools_disk() {
    install_with_apk ncdu \
        || install_with_apt ncdu \
        || install_with_dnf ncdu
}

install_survival_tools_network() {
    install_with_apk iproute2 bind-tools mtr \
        || install_with_apt iproute2 dnsutils mtr \
        || install_with_dnf iproute dnsutils mtr
}

install_survival_tools_text() {
    install_with_apk vim ripgrep less jq \
        || install_with_apt vim ripgrep less jq \
        || install_with_dnf vim ripgrep less jq
}

install_survival_tools_user() {
    install_with_apk sudo \
        || install_with_apt sudo \
        || install_with_dnf sudo
}

log_install_survival_tools() {
    local survival_option="$1"
    local survival_option_enabled="\e[1;34mdisabled\e[0m"
    is_enabled_survival_option "$survival_option" && survival_option_enabled="\e[1;32menabled\e[0m"
    
    printf "Survive by installing management tools for %-7s: %b\n" "$survival_option" "$survival_option_enabled"

}

enable_all_survival_options_if_none() {
    for survival_option in "${!SURVIVAL_OPTIONS[@]}"; do
        is_enabled_survival_option "$survival_option" && return 0
    done
    for survival_option in "${!SURVIVAL_OPTIONS[@]}"; do
        SURVIVAL_OPTIONS["$survival_option"]=true
    done
}

is_enabled_survival_option() {
    local survival_option="$1"
    [[ "${SURVIVAL_OPTIONS[$survival_option]}" == true ]]
}

usage() {
    script_name="$(basename "$0")"
    cat << EOF
Usage: ./$script_name [OPTIONS]

Survive in Linux/Docker environments by installing essential tools.

Options:
    -d, --disk              Survive by installing tools for disk
    -h, --help              Help
    -n, --network           Survive by installing tools for network
    -p, --process           Survive by installing tools for process
    -q, --quiet, --silent   Survive in silence without printing package manager logs
    -t, --text              Survive by installing tools for text
    -u, --user              Survive by installing tools for user

Example:
    ./$script_name --quiet --process --text
EOF
}

main() {

    while getopts ":-:dhnpqtu" option; do 
        case "$option" in 
            d) SURVIVAL_OPTIONS[disk]=true;;
            h) usage && exit 0 ;;
            n) SURVIVAL_OPTIONS[network]=true;;
            p) SURVIVAL_OPTIONS[process]=true;;
            q) STDOUT_REDIRECT="/dev/null";;
            t) SURVIVAL_OPTIONS[text]=true;;
            u) SURVIVAL_OPTIONS[user]=true;;
            -)
                case "$OPTARG" in
                    disk) SURVIVAL_OPTIONS[disk]=true;;
                    help) usage && exit 0 ;;
                    network) SURVIVAL_OPTIONS[network]=true;;
                    quiet|silent) STDOUT_REDIRECT="/dev/null";;
                    process) SURVIVAL_OPTIONS[process]=true;;
                    text) SURVIVAL_OPTIONS[text]=true;;
                    user) SURVIVAL_OPTIONS[user]=true;;
                    *) usage && exit 1 ;;
                esac
                ;;
            *) usage && exit 1 ;;
        esac
    done
    shift $(( "$OPTIND" - 1))

    enable_all_survival_options_if_none

    log_install_survival_tools "disk"
    is_enabled_survival_option "disk" && install_survival_tools_disk
    
    log_install_survival_tools "network"
    is_enabled_survival_option "network" && install_survival_tools_network
    
    log_install_survival_tools "process"
    is_enabled_survival_option "process" && install_survival_tools_process
   
    log_install_survival_tools "text"
    is_enabled_survival_option "text" && install_survival_tools_text
    
    log_install_survival_tools "user"
    is_enabled_survival_option "user" && install_survival_tools_user
    
}

main "$@"