#!/usr/bin/env bash

clear

AUTOMATIX=true
PACK='update'
#USE_MODULES=true
COLLECTD_INSTALL=true
GEL_NAME=

echo "NEXTGEN DP INSTALLER"
echo "<tomas.hetmer@cdn77.com>"
echo "--------------------"
echo ""
curl https://nextgen-10gc.datapacket.com/logo
echo ""
echo "-> Now loading modules"

echo "--> 00_color_logger"
# Some credit to https://github.com/maxtsepkov/bash_colors/blob/master/bash_colors.sh
#
# Constants and functions for terminal colors. Not using tput
# Author: Steve Wyckoff

COLOR_SCRIPT=color-logger.bash
COLOR_VERSION=0.9.0

COLOR_ESC="\x1B["

COLOR_RESET=0             # reset all attributes to their defaults
COLOR_RESET_UNDERLINE=24  # underline off
COLOR_RESET_REVERSE=27    # reverse off
COLOR_DEFAULT=39          # set underscore off, set default foreground color
COLOR_DEFAULTB=49         # set default background color

COLOR_BOLD=1              # set bold
COLOR_BRIGHT=2            # set half-bright (simulated with color on a color display)
COLOR_UNDERSCORE=4        # set underscore (simulated with color on a color display)
COLOR_REVERSE=7           # set reverse video

COLOR_BLACK=30            # set black foreground
COLOR_RED=31              # set red foreground
COLOR_GREEN=32            # set green foreground
COLOR_BROWN=33            # set brown foreground
COLOR_BLUE=34             # set blue foreground
COLOR_MAGENTA=35          # set magenta foreground
COLOR_CYAN=36             # set cyan foreground
COLOR_WHITE=37            # set white foreground

COLOR_BLACK_BG=40           # set black background
COLOR_RED_BG=41             # set red background
COLOR_GREEN_BG=42           # set green background
COLOR_BROWN_BG=43           # set brown background
COLOR_BLUE_BG=44            # set blue background
COLOR_MAGENTA_BG=45         # set magenta background
COLOR_CYAN_BG=46            # set cyan background
COLOR_WHITE_BG=47           # set white background

COLOR_DEBUG=$COLOR_BLUE
COLOR_INFO=$COLOR_MAGENTA
COLOR_HIGHLIGHT=$COLOR_CYAN
COLOR_WARN=$COLOR_BROWN
COLOR_ERROR=$COLOR_RED

COLOR_SUCCESS=$COLOR_GREEN

logger_wrap_escape() {
  local paint="$1"
  local message="$2"

  if ! [ $paint -ge 0 -a $paint -le 47 ] 2>/dev/null; then
    echo "escape: argument for \"$paint\" is out of range" >&2 && return 1
  fi

  if [ -z "$message" ]; then
    echo "No message passed in"

    exit 1
  fi

  message="${COLOR_ESC}${paint}m${message}${COLOR_ESC}${COLOR_RESET}m"

  echo -ne "$message"
}

logger_color(){
  local paint="$1"
  shift

  for message in "$@";
  do
    logger_wrap_escape "$paint" "$message"
  done

  echo
}

# PUBLIC API
debug(){
  logger_color "$COLOR_DEBUG" "$@"
}

info(){
  logger_color "$COLOR_INFO" "$@"
}

warn(){
  logger_color "$COLOR_WARN" "$@"
}

error(){
  logger_color "$COLOR_ERROR" "$@"
}

highlight(){
  logger_wrap_escape $COLOR_HIGHLIGHT "$1"
}

success(){
  logger_color "$COLOR_SUCCESS" "$@"
}
# END PUBLIC API

show_help() {
  cat << EOF
NAME
  $COLOR_SCRIPT - Add color logging to bash scripts.

SYNOPSIS
  color-logger.bash [-hcv] logger_function

DESCRIPTION
  Ideal use is to "source $COLOR_SCRIPT" and then use a logger_function directly in script.

  logger_functions: debug, info, warn, error, success, highlight

  e.g. 'debug "debug message"'

  highlight is a special case. It can be used in conjunection with other logging functions.

  e.g. 'error "epic" "\$(highlight "fail")" "as usual"'

OPTIONS
  -h          Show help
  -c          Show some pretty colors
  -v          Show version
EOF
}

list_colors(){
  local T='gYw'

  echo -e "\n                 40m     41m     42m     43m     44m     45m     46m     47m";

  for FGs in '   0m' '   1m' '  30m' '1;30m' '  31m' '1;31m' \
    '  32m' '1;32m' '  33m' '1;33m' '  34m' '1;34m' \
    '  35m' '1;35m' '  36m' '1;36m' '  37m' '1;37m';
  do
    FG=${FGs// /}
    echo -en " $FGs \033[$FG  $T  "

    for BG in 40m 41m 42m 43m 44m 45m 46m 47m; do
      echo -en " \033[$FG\033[$BG  $T  \033[0m";
    done

    echo;
  done

  echo;
  echo "For what maps the rows and columns to this script you just need to examine $COLOR_SCRIPT"
}

show_version(){
  echo $COLOR_VERSION
}

OPTIND=1
while getopts "hcv" opt; do
  case "$opt" in
    h)
      show_help
      exit 0
      ;;
    c)
      list_colors
      exit 0
      ;;
    v)
      show_version
      exit 0
      ;;
    '?')
      show_help >&2
      exit 1
      ;;
  esac
done

shift "$((OPTIND-1))" # Shift off the options and optional --.

echo "--> 00_functions"
# Run CMD and check status
runCMD () {
  COMMAND=${1}
  printf ">> ${COMMAND} "
    ${COMMAND}
    EC=$?
    echo ""
  if [ ${EC} -ne 0 ]; then
    error "ec: ${EC}"
    echo ""
    exit 255
  fi
}
echo "--> 00_os_detect"
if [ -f "/etc/arch-release" ]; then
  OS_NAME=2;
elif [ -f "/etc/centos-release" ]; then
  OS_NAME=3;
  CENTOS_VERSION=$(rpm -qa \*-release | grep -Ei "oracle|redhat|centos" | cut -d"-" -f3 | head -c 1)
  if [ $CENTOS_VERSION -eq 6 ]; then
    HISTORICAL_OS=true
  fi
elif [ -f "/etc/lsb-release" ]; then
  OS_NAME=1;
elif [ -f "/etc/debian_version" ]; then
  OS_NAME=0;
fi

echo "--> 1003_collectd_mc_types_db"
if [ "$COLLECTD_INSTALL" == "true" ]; then

  mkdir -p /opt/dp/collectd
  $(curl -s https://nextgen-10gc.datapacket.com/types.db.custom -o /opt/dp/collectd/types.db.custom -z /opt/dp/collectd/types.db.custom)

fi

echo "--> 1500_collectd_set_conf"
if [ "$COLLECTD_INSTALL" == "true" ]; then

	generateCollectdCFG () {
	debug "Generate CollectD config"
		COLLECTD_CFG_PATH=${1}
		$(curl -s https://nextgen-10gc.datapacket.com/collectd.conf -o "${COLLECTD_CFG_PATH}" -z "${COLLECTD_CFG_PATH}")
	}

	case "$OS_NAME" in
		0)  debug "OS detected as Debian"
			generateCollectdCFG "/etc/collectd/collectd.conf"
			runCMD "/usr/sbin/service collectd restart"
			#runCMD "service collectd status"
		;;
		1)  debug "OS detected as Ubuntu"
			generateCollectdCFG "/etc/collectd/collectd.conf"
			runCMD "/usr/sbin/service collectd restart"
			#runCMD "service collectd status"
		;;
		2)  debug "OS detected as Arch"
			generateCollectdCFG "/etc/collectd.conf"
			runCMD "systemctl restart collectd"
			#runCMD "systemctl status collectd"
		;;
		3)  debug "OS detected as Centos"
			CENTOS_VERSION=$(rpm -qa \*-release | grep -Ei "oracle|redhat|centos" | cut -d"-" -f3 | head -c 1)
			case "${CENTOS_VERSION}" in
			[7-8])  generateCollectdCFG "/etc/collectd.conf"
				runCMD "systemctl restart collectd"
			;;
			6)  generateCollectdCFG "/etc/collectd.conf"
				runCMD "service collectd restart"
			;;
			esac
			#runCMD "service collectd status"
		;;
		*)  error "OS not detected"
			exit 255
		;;
	esac

fi