auto_snapshot.bash

#!/bin/bash
#test
### Enable Debug
#set -xv

### Capture logs
logdate=$(date +%Y-%m-%d@%H:%M)

# Debug log
#exec > >(tee -i "$logdate".log)

# Primary Log
if [ -d "/var/log/doctl-remote-snapshot" ]
	then
		:
	else
		sudo mkdir /var/log/doctl-remote-snapshot
fi
exec > >(tee -i /var/log/doctl-remote-snapshot/"$logdate".log)


### Check for root or sudo execution
#if [ "$EUID" -ne 0 ]
#  then echo "Please run as root using sudo bash dev-config.bash"
#  exit 1
#fi


### Spinner
# http://fitnr.com/showing-a-bash-spinner.html
#spinner()
#{
#    local pid=$1
#    local delay=0.3
#    local spinstr='|/-\'
#    while [ "$(ps a | awk '{print $1}' | grep $pid)" ]; do
#        local temp=${spinstr#?}
#        printf " [%c]  " "$spinstr"
#        local spinstr=$temp${spinstr%"$temp"}
#        sleep $delay
#        printf "\b\b\b\b\b\b"
#    done
#    printf "    \b\b\b\b"
#}
#& spinner $!

### Call dodroplet.config
scriptdir=$(dirname "$0")
source $scriptdir/dodroplet.config

### Constants
date=$(date '+%Y-%m-%d-%a@%H:%M-%Z')"$snap_name_append"
dropletname=$(sudo /snap/bin/doctl compute droplet list | grep -e $dropletid | awk '{print$2}')
name="$dropletname""_""$date"
host=$(hostname)
ipadd=$(hostname -I | awk '{print $1}')
today=$(date '+%A, %B %d, %Y at %I:%M%p %Z')
snap_ip=$(sudo /snap/bin/doctl compute droplet list | grep $dropletid | awk '{print$3}')


# Log Archive
if [ -d "$log_archive_dir" ]
	then
		:
	else
		sudo mkdir $log_archive_dir
fi
exec > >(tee -i $log_archive_dir/"$logdate".log)
log_file=$log_archive_dir/"$logdate".log

## $numretain validation
if [ "$numretain " -lt 1 ]
	then 
		echo "Please assign a value greater than "0" to \$numretain";
		exit 1;
fi


### Create email template
tmpdir=$(sudo mktemp -d -t doctl-remote-snapshot-$(date +%Y-%m-%d@%H:%M)-XXXXXXXXXX)
email_notification="$tmpdir"/email_notification.txt

mkdir $tmpdir
touch $email_notification
echo "To: $recipient_email"$'\r' >> $email_notification
echo "From: $host <$host@$ipadd>"$'\r' >> $email_notification
echo "Subject: Snapshot on "$dropletname" Completed"$'\r' >> $email_notification
echo $'\r' >> $email_notification
echo "A remote snapshot job has been initated. Below is a brief log of that job."$'\r' >> $email_notification
echo $'\r' >> $email_notification

echo "Starting script"
sleep 3

### Shutdown droplet
if [ "$1" == "-p" ] || [ "$1" == "p" ] || [ "$2" == "-p" ] || [ "$3" == "p" ] 
	then
		echo "Droplet was not powered off because power-off flag was used." 
		echo "The droplet was not powered off because power-off flag was used." $'\r' >> $email_notification
	else
		echo "Shutting down droplet"
		sudo /snap/bin/doctl compute droplet-action shutdown $dropletid --wait
fi

### Create new snapshot using $name
echo "Creating snapshot titled \"$name\""
echo "Please wait this may take awhile. About 1 minute per GB."
sudo /snap/bin/doctl compute droplet-action snapshot --snapshot-name "$name" $dropletid --wait
new_snap=$(sudo /snap/bin/doctl compute snapshot list | grep $dropletid | tail -n 1 | awk '{print$2}')

### Reboot droplet.
if [ "$1" == "-p" ] || [ "$1" == "p" ] || [ "$2" == "-p" ] || [ "$2" == "p" ]
	then
		echo "Skipping Reboot"
	else
		echo "Powering on droplet. Please wait."
		sudo /snap/bin/doctl compute droplet-action power-on $dropletid --wait
		sleep 5
			# While loop to test if server IP is reachable by ping, if not, powers on until live.
			while ! ping -c 1 "$snap_ip" &> /dev/null
				do
					echo "Droplet and Server are NOT Live. Waiting 10 seconds to test again"
					sleep 10
					if ping -c 1 "$snap_ip" &> /dev/null
						then
							echo "Droplet and Server is Live"
							break
						else 
							echo "Still not live. Attempting to power on again"
							sudo /snap/bin/doctl compute droplet-action power-on $dropletid --wait
					fi
			done
fi

snapshots()
{
        snapshots=$(sudo /snap/bin/doctl compute snapshot list | grep $dropletid | wc -l)
}

#snapshots=$(sudo /snap/bin/doctl compute snapshot list | grep $dropletid | wc -l)
snapshots
### Retention
a=$(($snapshots - $numretain))
if [ "$1" == "-r" ] || [ "$1" == "r" ] || [ "$2" == "-r" ] || [ "$2" == "r" ]
        then
                echo "Nothing will be deleted."
                echo "Nothing has been deleted." $'\r' >> $email_notification
        else
                # List snapshots and get oldest snapshots after $numretain
                #snapshots=$(sudo /snap/bin/doctl compute snapshot list | grep $dropletid | wc -l)
                snapshots
                if [ "$snapshots" -lt "$numretain" ]
                        then
                                echo "Skipping retention because \$numretain is greater than \$snapshot."
                                echo "Skipped retention because \$numretain is greater than \$snapshot."$'\r' >> $email_notification
                        elif [ "$a" > 0 ]
                                then
                                        a=$(($snapshots - $numretain))
                                                echo "Deleting the last "$a" snapshot(s)"

                                                # Deleting all snapshots beyond $numretain
                                                while [[ "$snapshots" -gt "$numretain" ]]
                                                        do
                                                                oldest=$(sudo /snap/bin/doctl compute snapshot list | grep -e "$dropletid" | awk '{print$1}' | head -n 1)
                                                                oldest_name=$(sudo /snap/bin/doctl compute snapshot list --format "ID,Name,ResourceId" | grep $dropletid | awk '{print$2}' | head -n 1)
                                                                echo "Deleted "$oldest_name""$'\r' >> $email_notification
                                                                echo "Deleting "$oldest_name""$'\r'
                                                                sudo /snap/bin/doctl compute image delete $oldest --force
                                                                snapshots
                                                done
                        fi
                sleep 1
fi

### Send email end program
echo "Sending completion email to "$recipient_email""
echo >> $email_notification
echo "Snapshot of "$dropletname" titled "\"$new_snap\"" was created on $today"$'\r' >> $email_notification
echo "Server was confirmed to be live. "$'\r' >> $email_notification
echo "Log file can be found here: $log_file"$'\r' >> $email_notification
droplet_heading=$(sudo /snap/bin/doctl compute droplet snapshots $dropletid | grep name)
#remaining_snapshots=$(sudo /snap/bin/doctl compute droplet snapshots $dropletid)
remaining_snapshots=$(sudo /snap/bin/doctl compute droplet snapshots $dropletid | tail -n +2 | wc | awk {'print$1'})
echo -e "Remaining Snapshots for this Droplet:\n" $remaining_snapshots >> $email_notification
echo "$remaining_snapshots" | wc -l >> $email_notification
#echo $droplet_heading >> $email_notification
#echo $remaining_snapshots >> $email_notification

sudo sendmail -f "$recipient_email" $recipient_email < $email_notification

### Clean up work
sudo rm -r $email_notification $tmpdir

echo "Snapshot Process Completed"
exit 0