import os
import shutil
from os.path import isdir

from config import config, savePaths
from distutils.dir_util import copy_tree

from file_util import forcedir, copyfile, recursive_overwrite, copytree

__author__ = "Hsuan Han Lai (Edward Lai)"
__date__ = "2019-04-02"

workDir = os.path.dirname(os.path.realpath(__file__))


class OP1Backup:
    def getLatestProjectName(self, path):
        """
        Search Project Names in a given folder path
        And return next eligible folder name starting with "Project_"
        :param path:
        :return: New Folder Name
        """
        projects = os.listdir(path)
        print(projects)
        maxProjectNum = 0
        for i in projects:
            if "Project_" in i:
                temp = i.split("_")
                try:
                    temp = int(temp[-1])
                    if temp > maxProjectNum:
                        maxProjectNum = temp
                except IndexError:
                    pass
        return "Project_" + str(maxProjectNum + 1)

    def getTime(self):
        import datetime
        currentDT = datetime.datetime.now()
        return str(currentDT.year) + str(currentDT.month) + str(currentDT.day) + "_" + str(
            currentDT.hour) + str(currentDT.minute) + str(currentDT.second)

    def copyToLocal(self):
        op1TapePath, op1AlbumPath = config["OP_1_Mounted_Dir"] + "/tape", config["OP_1_Mounted_Dir"] + "/album"
        # currentTimeFolder = savePaths["Local_Projects"] + "/" + self.getTime()
        currentTimeFolder = savePaths["Local_Projects"] + "/" + self.getLatestProjectName(savePaths["Local_Projects"])
        forcedir(currentTimeFolder + "/album")
        forcedir(currentTimeFolder + "/tape")
        try:
            copy_tree(op1AlbumPath, currentTimeFolder + "/album")
            copy_tree(op1TapePath, currentTimeFolder + "/tape")
        except:
            # Connection broken, remove created folder
            shutil.rmtree(currentTimeFolder)

    def copyOnlyTapesToLocal(self):
        op1TapePath, op1AlbumPath = config["OP_1_Mounted_Dir"] + "/tape", config["OP_1_Mounted_Dir"] + "/album"
        # currentTimeFolder = savePaths["Local_Projects"] + "/" + self.getTime()
        currentTimeFolder = savePaths["Local_Projects"] + "/" + self.getLatestProjectName(savePaths["Local_Projects"])
        forcedir(currentTimeFolder + "/tape")
        try:
            copy_tree(op1TapePath, currentTimeFolder + "/tape")
        except:
            shutil.rmtree(currentTimeFolder)

    def loadProjectToOP1(self, localProjectPath):
        shutil.rmtree(config["OP_1_Mounted_Dir"] + "/tape")
        forcedir(config["OP_1_Mounted_Dir"] + "/tape")
        op1TapePath = config["OP_1_Mounted_Dir"] + "/tape"
        copy_tree(localProjectPath + "/tape", op1TapePath)

        if isdir(localProjectPath + "/album"):
            shutil.rmtree(config["OP_1_Mounted_Dir"] + "/album")
            forcedir(config["OP_1_Mounted_Dir"] + "/album")
            op1AlbumPath = config["OP_1_Mounted_Dir"] + "/album"
            copy_tree(localProjectPath + "/album", op1AlbumPath)

    def backupOP1Patches(self):
        """
        OP-1 ------PATCHES------> LOCAL STORAGE
        Backing up all the patch files on the OP-1, and copy the file if the folder have already exist on the local drive.
        Overwrite when redundant file occurs
        """

        try:
            recursive_overwrite(config["OP_1_Mounted_Dir"] + "/synth", savePaths["Local_Synth"])
            recursive_overwrite(config["OP_1_Mounted_Dir"] + "/drum", savePaths["Local_Drum"])
        except IOError:
            print ("Copy Error")

    def backupOPZState(self):
        if os.listdir(config["OP_Z_Mounted_Dir"]):
            OPZStatePath = config["OP_Z_Mounted_Dir"]
            # currentTimeFolder = savePaths["OP_Z_Local_Backup_States_Path"] + "/" + self.getTime()
            currentTimeFolder = savePaths["OP_Z_Local_Backup_States_Path"] + "/" + self.getLatestProjectName(
                savePaths["OP_Z_Local_Backup_States_Path"])
            print(OPZStatePath)
            print(currentTimeFolder)
            forcedir(currentTimeFolder)
            try:
                copy_tree(OPZStatePath, currentTimeFolder)
                print("Finished CPY")
            except:
                shutil.rmtree(currentTimeFolder)

    def loadStateToOPZ(self, localPath):
        OPZStatePath = config["OP_Z_Mounted_Dir"]
        print("Load local to OPZ")
        print(localPath)
        print(OPZStatePath)
        for root, dirs, files in os.walk(OPZStatePath):
            for f in files:
                print("remove", f)
                os.unlink(os.path.join(root, f))
            for d in dirs:
                print("remove Dir", d)
                shutil.rmtree(os.path.join(root, d))
        print("OPZ Dir Cleared")
        print("Copying to OPZ")
        copytree(localPath, OPZStatePath)
        print("Finished Copying to OPZ")
