"""Backup script for pre-installed instance"""

import shutil
import utils
import os

#TODO: have version of each modoboa componenent saved into the config file to restore the same version  

class Backup():


    def __init__(self, config):
        self.config = config
        self.destinationPath = ""
        self.BACKUPDIRECTORY = ["mails", "custom", "databases"]


    def preparePath(self):
        for dir in self.BACKUPDIRECTORY:
            os.mkdir(self.destinationPath + dir)


    def validatePath(self, path):
        """Check basic condition for backup directory"""

        if os.path.isfile(path):
            print("Error, you provided a file instead of a directory!")
            return False

        if not os.path.exists(path):
            createDir = input(f"\"{path}\" doesn't exists, would you like to create it ? [Y/n]\n").lower()

            if createDir == "y" or createDir == "yes":
                os.mkdir(path)
            else:
                return False

        if len(os.listdir(path)) != 0:
            delDir = input("Warning : backup folder is not empty, it will be purged if you continue... [Y/n]").lower()
            if delDir == "y" or delDir == "yes":
                shutil.rmtree(path)
            else:
                return False

        self.destinationPath = path

        if self.destinationPath[-1] != "/":
            self.destinationPath += "/"

        self.preparePath()
        return True


    def setPath(self):
        """Setup backup directory"""
        user_value = None
        while (user_value != '' and not self.validatePath(user_value)):
            print("Enter backup path, please provide an empty folder.")
            user_value = utils.user_input("-> ")
    

    def backupConfigFile(self):
        utils.copy_file("installer.cfg", self.destinationPath)


    def backupMails(self):

        utils.printcolor("Backing up mails", utils.MAGENTA)

        home_path = self.config.get("dovecot", "home_dir")

        if not os.path.exists(home_path) or os.path.isfile(home_path):
            utils.printcolor("Error backing up Email, provided path "
            f" ({home_path}) seems not right...", utils.RED)
        
        else:
            shutil.copytree(home_path, self.destinationPath+"mails/")
            utils.printcolor("Mail backup complete!", utils.GREEN)


    def backupCustomConfig(self):
        """Custom config :
        - Amavis : /etc/amavis/conf.d/99-custom
        - Postscreen : /etc/postfix/custom_whitelist.cidr
        Feel free to suggest to add others!"""
        utils.printcolor("Backing up some custom configuration...", utils.MAGENTA)

        custom_path = self.destinationPath+"custom/"

        """AMAVIS"""
        amavis_custom = "/etc/amavis/conf.d/99-custom"
        if os.path.isfile(amavis_custom):
            utils.copy_file(amavis_custom, custom_path)
            utils.printcolor("Amavis custom configuration saved!", utils.GREEN)

        """POSTSCREEN"""
        postscreen_custom = "/etc/postfix/custom_whitelist.cidr"
        if os.path.isfile(postscreen_custom):
            utils.copy_file(postscreen_custom, custom_path)
            utils.printcolor("Postscreen whitelist custom configuration saved!", utils.GREEN)


    def backupDBs(self):
        """Backing up databases"""


    def run(self):
        self.setPath()
        self.backupConfigFile()
        self.backupMails()
        self.backupCustomConfig()
        self.backupDBs()