#!/bin/bash

# This script will install Docker, Grafana, InfluxDB, Graphite & CollectD. It will also add systemd service files to ensure auto startup each boot.
# This script was created by reddit user /u/tyler_hammer. This is a combination of several guides with edits to make it as easy as possible for new people to start.

# Checking for Root Permissions # Thanks to Github User "codygarver" for the recommendation
check_your_privilege () {
    if [[ "$(id -u)" != 0 ]]; then
        echo -e "\e[91mError: This setup script requires root permissions. Please run the script as root.\e[0m" > /dev/stderr
        exit 1
    fi
}
check_your_privilege

# Docker Installation for Ubuntu 16.04

clear

# Grab IP for Link
myip=$(ip route | head -n 1 | egrep 'dev (.*)' -o | cut -d' ' -f2 | xargs ip a s $1 | egrep '([0-9]{1,3}\.){3}[0-9]{1,3}' -o | awk '{ if (NR == 1) print $1}')

# Update Package Database
while true; do
    echo -n -e "\e[7mDo you wish to run system updates? [y/n]:\e[0m "
    read yn
    case $yn in
        [yY] | [yY][Ee][Ss] ) echo -ne "\e[36mUpdating System - This may take awhile!\e[0m";  sudo apt-get -y update >/dev/null 2>>install.log && sudo apt-get -y upgrade >/dev/null 2>>install.log;clear;echo -e "\r\033[K\e[36mUpdating System ----- Complete\e[0m"; break;; #(Run both in one line)
        [nN] | [n|N][O|o] ) echo -e "\e[36mSkipping Updates\e[0m"; break;;  #Boring people don't update
        * ) echo -e "\e[7mPlease answer y or n.\e[0m ";;  #Error handling to get the right answer
    esac
done

# Add GPG Key for Docker Repo
echo -ne "\e[36mAdding GPG Key for Docker Repo\e[0m"
apt-key adv --keyserver hkp://p80.pool.sks-keyservers.net:80 --recv-keys 58118E89F3A912897C070ADBF76221572C52609D >>/dev/null 2>>install.log
echo "deb https://apt.dockerproject.org/repo ubuntu-xenial main" | sudo tee /etc/apt/sources.list.d/docker.list >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mAdding GPG Key for Docker Repo ----- Complete\e[0m"

# Update Database
echo -ne "\e[36mUpdating Database\e[0m"
apt-get update >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mUpdating Database ----- Complete\e[0m"

# Verify Repo
echo -ne "\e[36mVerifying Repo\e[0m"
apt-cache policy docker-engine >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mVerifying Repo ----- Complete\e[0m"

# Install Docker
echo -ne "\e[36mInstalling Docker\e[0m"
apt-get install -y docker-engine >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mInstalling Docker ----- Complete\e[0m"

# Grafana Install - Docker - Ubuntu 16.04

echo

clear

echo -e "\e[7mPlease specify an admin password for Grafana\e[0m"
read -p "> " -s GADMINPW

echo
echo

echo -e "\e[7mPlease re-enter the password\e[0m"
read -p "> " -s GADMINPW2

echo
echo

while [ "${GADMINPW}" != "${GADMINPW2}" ];
do
    echo
    echo -e "\e[41mPasswords do not match, please try again!\e[0m"
    echo
    echo -e "\e[7mPlease specify an admin password for Grafana\e[0m"
    read -p "> " -s GADMINPW
    echo
    echo
    echo -e "\e[7mPlease re-enter the password\e[0m"
    read -p "> " -s GADMINPW2
    echo
done

# Create Grafana Docker
clear
echo -e "\r\033[K\e[36mUpdating System ----- Complete\e[0m"
echo -e "\r\033[K\e[36mAdding GPG Key for Docker Repo ----- Complete\e[0m"
echo -e "\r\033[K\e[36mUpdating Database ----- Complete\e[0m"
echo -e "\r\033[K\e[36mVerifying Repo ----- Complete\e[0m"
echo -e "\r\033[K\e[36mInstalling Docker ----- Complete\e[0m"
echo -e "\r\033[K\e[36mCreating persistent storage for Grafana ----- Complete\e[0m"

echo -ne "\e[36mCreating Grafana docker container - This make take awhile!\e[0m"
sudo docker create \
--name=grafana \
-e PUID=1000 -e PGID=1000 \
-p 3000:3000 \
--restart always \
-v /docker/containers/grafana/sqlite3/:/var/lib/grafana \
-v /docker/containers/grafana/config/:/etc/grafana \
-e "GF_SECURITY_ADMIN_PASSWORD=${GADMINPW}" \
grafana/grafana >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mCreating Grafana docker container ----- Complete\e[0m"

# Start Grafana Docker
echo -ne "\e[36mStarting Grafana\e[0m"
docker start grafana >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mStarting Grafana ----- Complete\e[0m"

# Make Bin folder for update scripts
echo -ne "\e[36mCreating Update Folder\e[0m"
mkdir ~/updates >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mCreating Update Folder ----- Complete\e[0m"

# Download Grafana Update Script
echo -ne "\e[36mDownloading Grafana update script\e[0m"
wget https://raw.githubusercontent.com/tylerhammer/grafana/master/Update%20Scripts/grafanaupdate.sh -O ~/updates/updategrafana.sh >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mDownloading Grafana update script ----- Complete\e[0m"

# InfluxDB - Docker - Ubuntu 16.04

# Create Local Storage
echo -ne "\e[36mCreating local storage for InfluxDB\e[0m"
mkdir -p /docker/containers/influxdb/conf/ >>/dev/null 2>>install.log
mkdir -p /docker/containers/influxdb/db/ >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mCreating local storage for InfluxDB ----- Complete\e[0m"

# Check Ownership
echo -ne "\e[36mVerifying ownership\e[0m"
chown ${USER:=$(/usr/bin/id -run)}:$USER -R /docker >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mVerifying ownership ----- Complete\e[0m"

# Generate Default Config
echo -ne "\e[36mGenerating default config file for InfluxDB\e[0m"
docker run --rm influxdb influxd config > /docker/containers/influxdb/conf/influxdb.conf 2>>install.log
echo -e "\r\033[K\e[36mGenerating default config file for InfluxDB ----- Complete\e[0m"

# Create InfluxDB Container
echo -ne "\e[36mCreating InfluxDB docker container - This make take awhile!\e[0m"
docker create \
--name influxdb \
-e PUID=1000 -e PGID=1000 \
-p 8083:8083 -p 8086:8086 \
-v /docker/containers/influxdb/conf/influxdb.conf:/etc/influxdb/influxdb.conf:ro \
-v /docker/containers/influxdb/db:/var/lib/influxdb \
influxdb -config /etc/influxdb/influxdb.conf >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mCreating InfluxDB docker container ----- Complete\e[0m"

# Start InfluxDB
echo -ne "\e[36mStarting InfluxDB\e[0m"
docker start influxdb >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mStarting InfluxDB ----- Complete\e[0m"

# Create Influx Update Script
echo -ne "\e[36mDownloading InfluxDB update script\e[0m"
wget https://raw.githubusercontent.com/tylerhammer/grafana/master/Update%20Scripts/influxdbupdate.sh -O ~/updates/influxupdate.sh >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mDownloading InfluxDB update script ----- Complete\e[0m"

# Setup Auto Start
echo -ne "\e[36mCreating InfluxDB SystemD file\e[0m"
sudo bash -c "cat >/lib/systemd/system/influxdb.service" << EOF  2>>install.log
[Unit]
 Description=influxdb container
 Requires=docker.service
 After=docker.service

[Service]
 User=root
 Restart=on-failure
 RestartSec=45
 ExecStart=/usr/bin/docker start -a influxdb
 ExecStop=/usr/bin/docker stop -t 2 influxdb

[Install]
 WantedBy=multi-user.target
EOF
echo -e "\r\033[K\e[36mCreating InfluxDB SystemD file ----- Complete\e[0m"

# Enable Service
echo -ne "\e[36mEnabling InfluxDB Service\e[0m"
systemctl enable influxdb.service >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mEnabling InfluxDB Service ----- Complete\e[0m"

# Graphite Install - Docker - Ubuntu 16.04

# Create Graphite Container
echo -ne "\e[36mCreating Graphite docker container - This may take awhile!\e[0m"
docker run -d\
 --name graphite\
 --restart=always\
 -p 80:80\
 -p 2003-2004:2003-2004\
 -p 2023-2024:2023-2024\
 -p 8125:8125/udp\
 -p 8126:8126\
 hopsoft/graphite-statsd >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mCreating Graphite docker container ----- Complete\e[0m"

# Auto Start
echo -ne "\e[36mCreating Graphite SystemD file\e[0m"
sudo bash -c "cat >/lib/systemd/system/graphite.service" << EOF
[Unit]
 Description=Graphite container
 Requires=docker.service
 After=docker.service

[Service]
 User=root
 Restart=on-failure
 RestartSec=45
 ExecStart=/usr/bin/docker start -a graphite
 ExecStop=/usr/bin/docker stop -t 2 graphite

[Install]
 WantedBy=multi-user.target
EOF
echo -e "\r\033[K\e[36mCreating Graphite SystemD file ----- Complete\e[0m"

# Enable Service
echo -ne "\e[36mEnabling Graphite Service\e[0m"
systemctl enable graphite.service >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mEnabling Graphite Service ----- Complete\e[0m"

# Enable InfluxDB WebUI
echo -ne "\e[36mEnabling InfluxDB WebUI\e[0m"
sed -i '40s/.*/  enabled = true/' /docker/containers/influxdb/conf/influxdb.conf >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mEnabling InfluxDB WebUI ----- Complete\e[0m"

# Install other dependencies
echo -ne "\e[36mInstalling SSHPASS and SNMP dependencies - This may take awhile!\e[0m"
apt-get install -y sshpass >>/dev/null 2>>install.log
apt-get install -y snmp snmp-mibs-downloader >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mInstalling SSHPASS and SNMP dependencies ----- Complete\e[0m"

# Remove the need to user Sudo before docker. This generally requires you to log out and log back in, which is why we restart at the end of the script.
echo -ne "\e[36mRemoving "Sudo" requirement from docker command\e[0m"
sudo usermod -aG docker $(logname) >>/dev/null 2>>install.log
echo -e "\r\033[K\e[36mRemoving "Sudo" requirement from docker command ----- Complete\e[0m"


# Restart Announcment for previous command
echo -e "\e[7mThe VM needs to be restarted in order to apply changes and finalize the installation.\e[0m"

echo -e "\e[7mAfter the restart, Grafana can be accessed via http://${myip}:3000 & the InfluxDB AdminUI can be accessed via http://${myip}:8083\e[0m"

echo -n "Press any key to restart"
read -rsn1

# Restart
reboot