#!/usr/bin/python3.6

#import subprocess
#os functions
import os
import sys
import datetime
from datetime import datetime, timedelta

#text manipulation
import glob
import string

#emails
import smtplib
import email

#basic config - please do not change
HOSTNAME = os.uname()[1]
LOCKDIR = "/tmp/rn-innobackup.lock" #LOCKDIR
LOGFILE = "/var/log/rn-mysqlbackup.log" #LOGFILE
INNOBACKUPEX = "/usr/bin/innobackupex"
ALERTMAIL = "mmaros@reflected.net" #to use multiple recepients, this needs to be turned into a list ["email1@bla.com", "email2@bla.com", "email3@bla.com", ...]
#ALERTMAIL = "support@reflected.net" #to use multiple recepients, this needs to be turned into a list ["email1@bla.com", "email2@bla.com", "email3@bla.com", ...]


#backup configuration
#to use one binary over other, please put it at the start of the list, if it doesnt exist it will get skipped
COMPRESS = ["/usr/bin/pbzip2","/usr/local/sbin/pbzip2","/usr/bin/pigz","/usr/local/sbin/pigz"]
BACKUPDIR = '/home/backups/mysql/'
RETENTION  =int(7)
STREAMING = False
PARALEL = int(4)
USEMEMORY = "1G"
CUSTOMOPT = "" #separate them with a space between, use syntax same as on the cli "--option=value --option2=value
DATADIR = '/home/mysql/data'

#
#check which compression to use
#
#initialize the variable -making changes here will get overwritten
compression = ""
for i in range(0, len(COMPRESS)):
    if os.path.isfile(COMPRESS[i]):
        compression = COMPRESS[i]
        break
#exit if compression is not set up
if compression == "":
    print ("none of compressions are available, please check which ones are available and which ones are provided")
    sys.exit()

#exit if datadir doesnt exist
if os.path.isdir(DATADIR) is False:
    print (DATADIR + "does not exit")
    sys.exit()

#compression being used:
if ("pbzip2" in compression):
    backupSuffix = ".tar.bz2"
elif ("pigz" in compression):
    backupSuffix = ".tar.gz"
else:
    print ("wrong  compression")
    sys.exit()
curDate = ""

#check total freeDisk disk space on partition containting BACKUPDIR, in bytes
def freeSpace(BACKUPDIR):
    st = os.statvfs(BACKUPDIR)
    freeDisk = int(st.f_bavail * st.f_frsize)
    return freeDisk
#print ("Free disk space on" , BACKUPDIR + ":" , freeDisk)


def lastBackup(BACKUPDIR, backupSuffix, oldDate):
    if STREAMING:
        justTest = os.path.join(BACKUPDIR + 'mysql_backups_' + HOSTNAME + '_' + '*-streaming' + backupSuffix)
    else:
        justTest = os.path.join(BACKUPDIR + 'mysql_backups_' + HOSTNAME + '_' + '*' + backupSuffix)
    print (justTest)
    lastSize = os.path.getsize(max(glob.iglob(justTest) , key = os.path.getctime))
    print (lastSize)
    # lastSize = os.path.getsize(max(glob.iglob(os.path.join(BACKUPDIR , '*' + backupSuffix)) , key = os.path.getctime))
    return lastSize
#print ("Size of last backup is:" , lastSize)

#get current date in "20180409"/"YYYYMMDD" format, minus N days
def curDate(oldAge):
    curDate = datetime.today() - timedelta(days=oldAge)
    curDate = curDate.strftime("%Y%m%d")
    return curDate;

def dirSize(path):
    total_size = 0
    for dirpath, dirnames, filenames in os.walk(path):
        for f in filenames:
            fp = os.path.join(dirpath, f)
            total_size += os.path.getsize(fp)
    return total_size


def isRunnable():
    if STREAMING:
        oldAge = 1
        oldDate = curDate(oldAge)
        lastSize = lastBackup(BACKUPDIR , backupSuffix, oldDate)
        freeDisk = freeSpace(BACKUPDIR)
        print("Size of last backup")
        print(lastSize)
        print("Free disk size")
        print(freeDisk)
        if ((freeDisk - lastSize) * 1.05) < lastSize:
            msgBody = "Cant run backups as there is not enough free disk space"
            sendMail(msgBody)
        print ("All gucci")
    else:
        oldAge = 1
        oldDate = curDate(oldAge)
        lastSize = lastBackup(BACKUPDIR , backupSuffix, oldDate)
        freeDisk = freeSpace(BACKUPDIR)
        datadirSize = dirSize(DATADIR)
        print("Size of last backup")
        print(lastSize)
        print("Free disk size")
        print(freeDisk)
        print("Datadir size")
        print(datadirSize)
        if ((freeDisk - lastSize - datadirSize) * 1.05) < (lastSize + datadirSize):
            msgBody = "Cant run backups as there is not enough free disk space"
            sendMail(msgBody)
        print ("All gucci")

def sendMail(msgBody):
    sender = 'root@' + HOSTNAME + '.ded.reflected.net'
    receivers = ALERTMAIL
    msgBody = msgBody

    #print ("this works")
    #message = """From: """ + sender + """\nTo: """ + receivers + """\nSubject: backup failed on """ + HOSTNAME + """\n\n""" + msgBody + """\n"""

    #its still ugly, but ugly in mutliple lines
    message = ("From: " + sender + "\n"
    "To:" + receivers + "\n"
    "Subject: Backup failed on " + HOSTNAME + "\n\n"
    + msgBody + "\n")

    #smtpObj = smtplib.SMTP('smtp-out.reflected.net')
    #smtpObj.sendmail(sender, receivers, message)
    print ("Successfully sent email")
    sys.exit()


isRunnable()